MakeMiniprodValidationCuts.py

Go to the documentation of this file.

import os.path

DIR = "/nova/app/users/essmith/devsrepo/essmith/miniprod5_validation/events/lists/lists"
OUTFILE = "Miniprod5ValidationCuts.cxx"

CUT_TEMPLATE = """
  const Cut kIs_Prod4%(prod4_sel)s_Prod5%(prod5_sel)s_%(horn)s([](const caf::SRProxy * sr)
  {
    const std::unordered_set<std::size_t> hashes
    {
%(hash_list)s
    };

    return hashes.find(Hash(&sr->spill)) != hashes.end();
  });


"""

cut_text = ""
for horn_current in ("fhc", "rhc"):
        sels = ("unselected", "presel", "CVNeLow", "CVNeHigh")
        for prod4_sel in sels:
                for prod5_sel in sels:
                        f = os.path.join(DIR, "ND_%(horn)s_p4%(prod4_sel)s_p5%(prod5_sel)s.txt" % { "horn": horn_current, "prod4_sel": prod4_sel, "prod5_sel" : prod5_sel})
                        if not os.path.isfile(f):
                                continue
                        print f

                        hash_text = ""
                        with open(f) as infile:
                                for line in infile:
                                        numbers = (l.lstrip("0") if l != len(l) * "0" else "0" for l in line.split())
                                        hash_text += "      Hash(std::vector<std::size_t>{%s}),\n" % (", ".join(numbers))

                        subst = { "prod4_sel": prod4_sel, "prod5_sel": prod5_sel, "horn": horn_current.upper(), "hash_list": hash_text}
                        for sel in ("prod4_sel", "prod5_sel"):
                                subst[sel] = subst[sel].replace("elected", "el")
                                subst[sel] = subst[sel][:1].upper() + subst[sel][1:]
                        cut_text += CUT_TEMPLATE % subst

text = []
with open(OUTFILE, 'r') as infile:
        text = [line for line in infile]
with open(OUTFILE, 'w') as outfile:
        replacing = False
        for line in text:
                if not replacing and "AUTOFILL BELOW" in line:
                        replacing = True
                elif replacing and "END AUTOFILL" in line:
                        replacing = False
                        outfile.write(cut_text)
                elif replacing:
                        continue

                outfile.write(line)