generateFlavCfromFlavB Namespace Reference

Functions
def	generate_flav_c
def	main

Function Documentation

def generateFlavCfromFlavB.generate_flav_c

(

loaded_data

)

Definition at line 10 of file generateFlavCfromFlavB.py.

References python.multivaluedict.dict, alcazmumu_cfi.filter, and list().

Referenced by main().

def generate_flav_c(loaded_data):
    flav_b_data = filter(
        lambda e: e.params.jetFlavor == 0,
        loaded_data.entries
    )
    flav_b_data = sorted(flav_b_data, key=lambda e: e.params.operatingPoint)
    flav_b_data = sorted(flav_b_data, key=lambda e: e.params.measurementType)
    flav_b_data = sorted(flav_b_data, key=lambda e: e.params.etaMin)
    flav_b_data = sorted(flav_b_data, key=lambda e: e.params.ptMin)
    flav_b_data = sorted(flav_b_data, key=lambda e: e.params.discrMin)
    sys_groups = itertools.groupby(
        flav_b_data,
        key=lambda e: '%d, %s, %.02f, %.02f, %.02f' % (
            e.params.operatingPoint,
            e.params.measurementType,
            e.params.etaMin,
            e.params.ptMin,
            e.params.discrMin
        )
    )

    def gen_entry_dict(groups):
        for _, grp in groups:
            grp = list(grp)
            entries_by_sys = dict((e.params.sysType, e) for e in grp)
            assert len(grp) == len(entries_by_sys)  # every sysType is unique
            yield entries_by_sys
    sys_dicts = gen_entry_dict(sys_groups)

    def gen_flavb_csv_line(dicts):
        for d in dicts:
            central = d.pop('central')
            central.params.jetFlavor = 1
            yield central.makeCSVLine()
            for e in d.itervalues():
                e.params.jetFlavor = 1
                e.formula = '2*(%s)-(%s)' % (e.formula, central.formula)
                yield e.makeCSVLine()
    csv_lines = gen_flavb_csv_line(sys_dicts)

    return list(l for l in csv_lines)