lepMass, ) print " Helicity angles names:\n helcosthetaK = %s \n helcosthetaL = %s \n helphi = %s" % ( obsDict["helcosthetaK"][0], obsDict["helcosthetaL"][0], obsDict["helphi"][0], ) addHelicityAnglesToTree( intermediateTree, daughterPartNames["posHad"], daughterPartNames["negHad"], daughterPartNames["posLep"], daughterPartNames["negLep"], posHadMass, negHadMass, lepMass, lepMass, obsDict["helcosthetaK"][0], obsDict["helcosthetaL"][0], obsDict["helphi"][0], ) # close intermediate files intermediateTree.Write() intermediateFile.Close() del intermediateFile print "P2VV - INFO: Wrote intermediate tree to file: %s" % intermediateFileName print "P2VV - INFO: Finished refining tree. Continuing to create RooDataSet"
if daughterPartMasses['negLep'] == 'muon': lepMass = Mmu else: assert False, 'P2VV - ERROR: Cannot assign mass to negative hadron.' # add helicity angles to tree print ' P2VV - INFO: The following associations will be made:\n '\ ' Positive hadron name: %s, mass=%s \n Negative hadron name: %s, mass=%s \n '\ ' Positive lepton name: %s, mass=%s \n Negative lepton name: %s, mass=%s \n'\ ' Units MUST be in MeV. Check!!!'\ %(daughterPartNames['posHad'], posHadMass, daughterPartNames['negHad'], negHadMass,\ daughterPartNames['posLep'], lepMass, daughterPartNames['negLep'], lepMass) print ' Helicity angles names:\n helcosthetaK = %s \n helcosthetaL = %s \n helphi = %s'\ %( angleNames['helcosthetaK'], angleNames['helcosthetaL'], angleNames['helphi'] ) addHelicityAnglesToTree(tree, daughterPartNames['posHad'], daughterPartNames['negHad'], daughterPartNames['posLep'], daughterPartNames['negLep'], posHadMass, negHadMass, lepMass, lepMass, angleNames['helcosthetaK'], angleNames['helcosthetaL'], angleNames['helphi'], sufixes) # add all the possible configurations as well if options.addAllConfigrtns: print 'P2VV - INFO: Fliping hadrons and recalculating helicity angles' addHelicityAnglesToTree(tree, daughterPartNames['negHad'], daughterPartNames['posHad'], daughterPartNames['posLep'], daughterPartNames['negLep'], negHadMass, posHadMass, lepMass, lepMass, 'helcosthetaK__pim_Kpl_mup_mum','helcosthetaL__pim_Kpl_mup_mum','helphi__pim_Kpl_mup_mum', sufixes) print 'P2VV - INFO: Fliping leptons and recalculating helicity angles' addHelicityAnglesToTree(tree,