format_google_benchmark_data.py

import json
import os
import sys
import datetime
import re

def normalize_name(name):
    # Remove prefix if exists (e.g., "BenchmarkModel/")
    if "/" in name:
        name = name.split("/", 1)[1]
    
    # Clean up name similar to pycasbin
    parts = name.split("_")
    new_parts = []
    for p in parts:
        if p.lower() in ["rbac", "abac", "acl", "api", "rest"]:
            new_parts.append(p.upper())
        else:
            new_parts.append(p.capitalize())
    return "".join(new_parts)


def main():
    if len(sys.argv) < 3:
        print("Usage: python format_google_benchmark_data.py input.json output.json")
        sys.exit(1)

    input_path = sys.argv[1]
    output_path = sys.argv[2]

    try:
        with open(input_path, "r", encoding="utf-8") as f:
            data = json.load(f)
    except Exception as e:
        print(f"Error loading {input_path}: {e}")
        sys.exit(1)

    # Get commit info from environment variables
    commit_info = {
        "author": {
            "email": os.environ.get("COMMIT_AUTHOR_EMAIL", ""),
            "name": os.environ.get("COMMIT_AUTHOR_NAME", ""),
            "username": os.environ.get("COMMIT_AUTHOR_USERNAME", ""),
        },
        "committer": {
            "email": os.environ.get("COMMIT_COMMITTER_EMAIL", ""),
            "name": os.environ.get("COMMIT_COMMITTER_NAME", ""),
            "username": os.environ.get("COMMIT_COMMITTER_USERNAME", ""),
        },
        "distinct": True,
        "id": os.environ.get("COMMIT_ID", ""),
        "message": os.environ.get("COMMIT_MESSAGE", ""),
        "timestamp": os.environ.get("COMMIT_TIMESTAMP", ""),
        "tree_id": os.environ.get("COMMIT_TREE_ID", ""),
        "url": os.environ.get("COMMIT_URL", ""),
    }

    # Get CPU count
    cpu_count = data.get("context", {}).get("num_cpus")
    if not cpu_count:
        cpu_count = os.cpu_count() or 1

    benches = []
    for bench in data.get("benchmarks", []):
        # Skip aggregate items (mean, median, stddev) if any
        if "run_type" in bench and bench["run_type"] == "aggregate":
            continue

        name = bench["name"]
        
        # Google Benchmark outputs time in the unit specified by time_unit
        # We want to standardize on ns/op
        val = bench["real_time"]
        unit = bench.get("time_unit", "ns")
        
        if unit == "ms":
            val *= 1e6
        elif unit == "us":
            val *= 1e3
        elif unit == "s":
            val *= 1e9
        
        # Extra info
        iterations = bench.get("iterations", 0)
        extra = f"{iterations} times"

        benches.append(
            {"name": normalize_name(name), "value": round(val, 2), "unit": "ns/op", "extra": extra}
        )

    output_data = {
        "commit": commit_info,
        "date": int(datetime.datetime.now().timestamp() * 1000),
        "tool": "cpp",
        "procs": cpu_count,
        "benches": benches,
    }

    with open(output_path, "w", encoding="utf-8") as f:
        json.dump(output_data, f, indent=2)
    
    print(f"Successfully formatted benchmark data to {output_path}")


if __name__ == "__main__":
    main()