#!/bin/bash

# Get llvm-profdata from `apt install llvm-20`.
# Must match rust's llvm version or it will crash.

# Build instrumented wheel.
cargo clean
uv cache clean
uv pip install maturin
rm -rf dist/
UV_NO_BUILD_CACHE=1 uv run --no-sync maturin build --compatibility pypi --out dist --verbose -- "-Cprofile-generate=${PWD}/scripts/pgo-profiles/pgo.profraw"

# Install instrumented wheel.
uv pip install $(find dist/ -name '*.whl')[pydantic] --group test --group bench --reinstall

# Clear existing profiles.
rm -rf ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles; mkdir ./scripts/pgo-profiles/pgo.profraw

# Run reference workload to generate profile.
# Name each profile file per-binary (%m), per-process (%p)
# so that they don't overwrite each other's results.
# Unfortunately, per-thread (%t) is not always available.
LLVM_PROFILE_FILE="${PWD}/scripts/pgo-profiles/pgo.profraw/%m-%p.profraw" uv run --no-sync ./scripts/profile_workload.py

# Merge profiles.
/usr/lib/llvm-21/bin/llvm-profdata merge -o scripts/pgo-profiles/pgo.profdata $(find scripts/pgo-profiles/pgo.profraw -name '*.profraw')