PERF_DATA="perf.data"

BENCH_ARGS="${PERF_BENCH_ARGS:---no-render}"
# Results directory configuration
RESULTS_ROOT="${PERF_RESULTS_ROOT:-perf}"
RUN_LABEL="${PERF_RUN_LABEL:-$(date +%H%M%S)}"
DATE_DIR="${RESULTS_ROOT}/$(date +%Y-%m-%d)"
REPORT_DIR="${DATE_DIR}/report"
FLAME_DIR="${DATE_DIR}/flamegraph"
PERF_DATA="${DATE_DIR}/perf-${RUN_LABEL}.data"
PERF_REPORT="${REPORT_DIR}/perf-${RUN_LABEL}.log"
PERF_FOLDED="${FLAME_DIR}/perf-${RUN_LABEL}.folded"
FLAMEGRAPH_SVG="${FLAME_DIR}/flamegraph-${RUN_LABEL}.svg"
# Mode: perf | c2c | both
PERF_MODE="${PERF_MODE:-perf}"
C2C_DIR="${DATE_DIR}/c2c"
C2C_DATA="${C2C_DIR}/c2c-${RUN_LABEL}.data"
C2C_REPORT="${C2C_DIR}/c2c-${RUN_LABEL}.log"

echo "Results: ${DATE_DIR}"

# 3. Run perf record
echo "Running perf record with $ITERATIONS iterations..."
echo "This may require sudo permissions for kernel-level profiling."

# 3. Prepare output directories
mkdir -p "$REPORT_DIR" "$FLAME_DIR"

perf record -g --call-graph dwarf "$BENCH_EXE" "$ITERATIONS"

# 3. Run perf record (mode: perf/both)
if [ "$PERF_MODE" = "perf" ] || [ "$PERF_MODE" = "both" ]; then
    echo "Running perf record with $ITERATIONS iterations..."
    echo "This may require sudo permissions for kernel-level profiling."

if [ $? -eq 0 ]; then
    echo "=== Profile Complete ==="
    echo "To view the results, run:"
    echo "  perf report"
    echo ""
    echo "To generate a flamegraph (if tools are installed):"
    echo "  perf script | stackcollapse-perf.pl | flamegraph.pl > flamegraph.svg"
else
    echo "Error: perf record failed. You might need to run: sudo sysctl -w kernel.perf_event_paranoid=-1"

    perf record -g --call-graph dwarf -o "$PERF_DATA" "$BENCH_EXE" "$ITERATIONS" $BENCH_ARGS
    if [ $? -eq 0 ]; then
        echo "=== perf record complete ==="
        echo "Writing report to ${PERF_REPORT}"
        perf report -i "$PERF_DATA" --stdio > "$PERF_REPORT"
        if command -v stackcollapse-perf.pl &> /dev/null && command -v flamegraph.pl &> /dev/null; then
            echo "Generating flamegraph..."
            perf script -i "$PERF_DATA" | stackcollapse-perf.pl > "$PERF_FOLDED"
            flamegraph.pl "$PERF_FOLDED" > "$FLAMEGRAPH_SVG"
        else
            echo "Flamegraph tools not found. Skipping flamegraph generation."
        fi
    else
        echo "Error: perf record failed. You might need to run: sudo sysctl -w kernel.perf_event_paranoid=-1"
        exit 1
    fi

# 4. Run perf c2c record/report (mode: c2c/both)
if [ "$PERF_MODE" = "c2c" ] || [ "$PERF_MODE" = "both" ]; then
    if perf c2c record --help &> /dev/null; then
        echo "Running perf c2c record with $ITERATIONS iterations..."
        mkdir -p "$C2C_DIR"
        perf c2c record -o "$C2C_DATA" -- "$BENCH_EXE" "$ITERATIONS" $BENCH_ARGS
        if [ $? -eq 0 ]; then
            echo "Writing c2c report to ${C2C_REPORT}"
            perf c2c report -i "$C2C_DATA" --stdio > "$C2C_REPORT"
        else
            echo "Warning: perf c2c record failed."
            exit 1
        fi
    else
        echo "Warning: perf c2c is not available in this perf build."
        exit 1
    fi
fi

/-- Simulate a heavy editing session -/
def runBenchmark (iterations : Nat) : IO Unit := do

/-- Simulate a heavy editing session. -/
def runBenchmark (iterations : Nat) (render : Bool := true) : IO Unit := do

  -- We don't actually print to TTY to avoid I/O bottlenecks in perf
  let _ ← ViE.UI.render s

  if render then
    let _ ← ViE.UI.render s

  ViE.Benchmark.runBenchmark iter

  let noRender := args.any (fun a => a == "--no-render")
  let render := !noRender
  -- Run benchmark; rendering can be disabled with --no-render.
  ViE.Benchmark.runBenchmark iter render

.lake

.lake
perf
perf.data*
perf-*.log
flamegraph*.svg

perf/
perf.data*
perf-*.log
flamegraph*.svg