PythiaID.py

Go to the documentation of this file.

def properties(pid, sgn=1):
    return {
        "name": pid.name(sgn),
        "threeCharge": pid.chargeType(sgn),
        "hasAnti": pid.hasAnti(),
        "isLepton": pid.isLepton(),
        "isQuark": pid.isQuark(),
        "isDiQuark": pid.isDiquark(),
        "isHadron": pid.isHadron(),
        "isMeson": pid.isMeson(),
        "isBaryon": pid.isBaryon(),
        "jSpin": pid.spinType(),
    }
 
 
# Generate the PID dictionary and optionally archive.
def generate(dat=None):
    import pythia8
 
    pythia = pythia8.Pythia("", False)
 
    # Loop over the particles.
    pid, pids = 1, {}
    while pid != 0:
        pids[pid] = properties(pythia.particleData.particleDataEntryPtr(pid))
        if pids[pid]["hasAnti"]:
            pids[-pid] = properties(pythia.particleData.particleDataEntryPtr(pid), -1)
        pid = pythia.particleData.nextId(abs(pid))
 
    # Archive if requested.
    if dat:
        with open(dat, "w") as dat:
            dat.write(repr(pids))
    return pids
 
 
# Load the Pythia dictionary.
try:
    pids = eval(open("PythiaID.dat").read())
except FileNotFoundError:
    pids = generate("PythiaID.dat")