#!/usr/bin/env python3

import json, argparse

p = argparse.ArgumentParser()

p.add_argument('-i', '--input', action="store",
               help="input JSON file name", required = True)

p.add_argument('-o', '--output', action="store",
               help="output C file name", required = True)

p.add_argument('-m', '--model', action="append",
               help="CPU model in format: model[,stepping0]",
               required = True)

r = p.parse_args()

with open(r.input, 'r') as fp:
    objects = json.load(fp)

c = open(r.output, 'w')

c.write ("""
#include <perfmon/perfmon_intel.h>

static perfmon_intel_pmc_cpu_model_t cpu_model_table[] = {
""")

for v in r.model:
    if "," in v:
        (m, s)  = v.split(",")
        m = int(m, 0)
        s = int(s, 0)
        c.write ("  {}0x{:02X}, 0x{:02X}, 1{},\n".format("{", m, s, "}"))
    else:
        m = int(v, 0)
        c.write ("  {}0x{:02X}, 0x00, 0{},\n".format("{", m, "}"))
c.write ("""
};

static perfmon_intel_pmc_event_t event_table[] = {
""")

for obj in objects:
    MSRIndex = obj["MSRIndex"]
    if MSRIndex != "0":
      continue

    EventCode = obj["EventCode"]
    UMask = obj["UMask"]
    EventName = obj["EventName"].lower()
    if "," in EventCode:
        continue

    c.write ("  {\n")
    c.write ("   .event_code = {}{}{},\n".format("{", EventCode, "}"))
    c.write ("   .umask = {},\n".format(UMask))
    c.write ("   .event_name = \"{}\",\n".format(EventName))
    c.write ("   },\n")


c.write ("""  {
   .event_name = 0,
   },
};

PERFMON_REGISTER_INTEL_PMC (cpu_model_table, event_table);

""")

c.close()