Functions
None	processFile (str fin, bool noCharm=True)

None	run ()

None	run1GeV ()

None	run4Charm ()

None	run4Charm1GeV ()

None	run4beauty ()

None	finalResult ()

Variables
str	path = "/eos/experiment/ship/data/Mbias/background-prod-2018/"

list	charmExtern = [4332, 4232, 4132, 4232, 4122, 431, 411, 421, 15]

list	neutrinos = [12, 14, 16]

float	weightMbias = 768.75

float	weightMbias1GeV = 27778.0

float	weightCharm = 326.0

float	weightCharm1GeV = 4895.24

float	weightBeauty = 9.37

dict	h = {}

ROOT	PDG = ROOT.TDatabasePDG.Instance()

idnu	idw = idnu

int	idhnu = 1000 + idw

ROOT	name = PDG.GetParticle(idw).GetName()

ROOT	title = name + " momentum (GeV)"

int	key = idhnu

Function Documentation

◆ finalResult()

None extractNeutrinosAndUpdateWeight.finalResult ( )

Definition at line 134 of file extractNeutrinosAndUpdateWeight.py.

def finalResult() -> None:
    h10 = {}
    ut.readHists(h10, "pythia8_Geant4_10.0_c0-67000_nu.root")
    h1 = {}
    ut.readHists(h1, "pythia8_Geant4_1.0_c0-19000_nu.root")
    c10 = {}
    ut.readHists(c10, "pythia8_Geant4_charm_10.0_nu.root")
    c1 = {}
    ut.readHists(c1, "pythia8_Geant4_charm_1.0_nu.root")
    cmd = "hadd pythia8_Geant4_10.0_withCharm_nu.root pythia8_Geant4_10.0_c0-67000_nu.root pythia8_Geant4_charm_10.0_nu.root"
    os.system(cmd)
    cmd = (
        "hadd pythia8_Geant4_1.0_withCharm_nu.root pythia8_Geant4_1.0_c0-19000_nu.root pythia8_Geant4_charm_1.0_nu.root"
    )
    os.system(cmd)

◆ processFile()

None extractNeutrinosAndUpdateWeight.processFile	(	str	fin,
		bool	noCharm = `True`
	)

Definition at line 54 of file extractNeutrinosAndUpdateWeight.py.

def processFile(fin: str, noCharm: bool = True) -> None:
    f = ROOT.TFile.Open(os.environ["EOSSHIP"] + path + fin)
    print("opened file ", fin)
    sTree = f.Get("cbmsim")
    for n in range(sTree.GetEntries()):
        sTree.GetEntry(n)
        for v in sTree.vetoPoint:
            nu = v.GetTrackID()
            t = sTree.MCTrack[nu]
            pdgCode = t.GetPdgCode()
            if abs(pdgCode) not in neutrinos:
                continue
            moID = abs(sTree.MCTrack[t.GetMotherId()].GetPdgCode())
            if moID in charmExtern and noCharm:
                continue  # take heavy flavours from separate production
            idhnu = abs(pdgCode) + 1000
            if pdgCode < 0:
                idhnu += 1000
            l10ptot = ROOT.TMath.Min(ROOT.TMath.Max(ROOT.TMath.Log10(t.GetP()), -0.3), 1.69999)
            l10pt = ROOT.TMath.Min(ROOT.TMath.Max(ROOT.TMath.Log10(t.GetPt()), -2.0), 0.4999)
            key = idhnu
            h[key].Fill(t.GetP(), weight)
            key = idhnu + 100
            h[key].Fill(l10ptot, l10pt, weight)
            key = idhnu + 200
            h[key].Fill(l10ptot, l10pt, weight)
 
 

◆ run()

None extractNeutrinosAndUpdateWeight.run ( )

Definition at line 82 of file extractNeutrinosAndUpdateWeight.py.

def run() -> None:
    tmp = "pythia8_Geant4_10.0_cXX.root"
    global weight
    weight = weightMbias
    for run in range(0, 67000, 1000):
        processFile(tmp.replace("XX", str(run)))
    ut.writeHists(h, "pythia8_Geant4_10.0_c0-67000_nu.root")
 
 

◆ run1GeV()

None extractNeutrinosAndUpdateWeight.run1GeV ( )

Definition at line 91 of file extractNeutrinosAndUpdateWeight.py.

def run1GeV() -> None:
    tmp = "pythia8_Geant4_1.0_cXX.root"
    global weight
    weight = weightMbias1GeV
    for run in range(0, 19000, 1000):
        processFile(tmp.replace("XX", str(run)))
    ut.writeHists(h, "pythia8_Geant4_1.0_c0-19000_nu.root")
 
 

◆ run4beauty()

None extractNeutrinosAndUpdateWeight.run4beauty ( )

Definition at line 120 of file extractNeutrinosAndUpdateWeight.py.

def run4beauty() -> None:
    global weight
    weight = weightBeauty
    fname = "pythia8_Geant4_beauty_5336B_10.0.root"
    rc = processFile(fname, False)
    if rc == 0:
        fmu = fname.replace(".root", "_mu.root")
        rc = os.system("xrdcp " + fmu + " $EOSSHIP/eos/experiment/ship/data/Mbias/background-prod-2018/" + fmu)
        if rc != 0:
            print("copy to EOS failed, stop", fmu)
        else:
            rc = os.system("rm " + fmu)
 
 

◆ run4Charm()

None extractNeutrinosAndUpdateWeight.run4Charm ( )

Definition at line 100 of file extractNeutrinosAndUpdateWeight.py.

def run4Charm() -> None:
    tmp = "pythia8_Geant4_charm_XX-YY_10.0.root"
    global weight
    weight = weightCharm
    for cycle in [0, 1, 2]:
        for run in range(0, 100, 20):
            crun = run + cycle * 1000
            fname = tmp.replace("XX", str(crun)).replace("YY", str(crun + 19))
            processFile(fname, False)
    ut.writeHists(h, "pythia8_Geant4_charm_10.0_nu.root")
 
 

◆ run4Charm1GeV()

None extractNeutrinosAndUpdateWeight.run4Charm1GeV ( )

Definition at line 112 of file extractNeutrinosAndUpdateWeight.py.

def run4Charm1GeV() -> None:
    fname = "pythia8_Geant4_charm_0-19_1.0.root"  # renamed pythia8_Geant4_charm_XX-YY_10.0.root
    global weight
    weight = weightCharm1GeV
    processFile(fname, False)
    ut.writeHists(h, "pythia8_Geant4_charm_1.0_nu.root")
 
 