Public Member Functions
def	__init__ (self, isMC, isReReco, isSync=False)

def	correct (self, mu, run)

def	correct_all (self, mus, run)

def	corrected_p4 (self, mu, run)

Public Attributes
	corr

	corrABC

	corrD

	isMC

	isSync

Detailed Description

Definition at line 4 of file MuScleFitCorrector.py.

Constructor & Destructor Documentation

def MuScleFitCorrector.MuScleFitCorr.__init__	(	self,
		isMC,
		isReReco,
		isSync = `False`
	)

Definition at line 5 of file MuScleFitCorrector.py.

     def __init__(self,isMC,isReReco,isSync=False):
         #colin need to import muscle fit inputs, if tool still in use
         path = "%s/src/CMGTools/RootTools/data/musclefit/" % os.environ['CMSSW_BASE'];
         self.isMC = isMC
         if self.isMC:
             self.corr = heppy.MuScleFitCorrector(path+"MuScleFit_2012_MC_53X_smear%s.txt" % ("ReReco" if isReReco else "Prompt"))
             self.isSync = isSync
         else:
             self.corrABC = heppy.MuScleFitCorrector(path+"MuScleFit_2012ABC_DATA%s_53X.txt" % ("_ReReco" if isReReco else ""))
             self.corrD   = heppy.MuScleFitCorrector(path+"MuScleFit_2012D_DATA%s_53X.txt"   % ("_ReReco" if isReReco else ""))

Member Function Documentation

def MuScleFitCorrector.MuScleFitCorr.correct	(	self,
		mu,
		run
	)

Definition at line 26 of file MuScleFitCorrector.py.

References MuScleFitCorrector.MuScleFitCorr.corrected_p4().

     def correct(self, mu, run):
         mu.setP4( self.corrected_p4(mu, run) )
 

def MuScleFitCorrector.MuScleFitCorr.correct_all	(	self,
		mus,
		run
	)

Definition at line 29 of file MuScleFitCorrector.py.

References MuScleFitCorrector.MuScleFitCorr.corrected_p4().

     def correct_all(self, mus, run):
         for mu in mus:
             mu.setP4( self.corrected_p4(mu, run) )
 

def MuScleFitCorrector.MuScleFitCorr.corrected_p4	(	self,
		mu,
		run
	)

Definition at line 15 of file MuScleFitCorrector.py.

References MuScleFitCorrector.MuScleFitCorr.corrABC, MuScleFitCorrector.MuScleFitCorr.corrD, KalmanMuonCorrector.KalmanMuonCorrector.isMC, MuScleFitCorrector.MuScleFitCorr.isMC, KalmanMuonCorrector.KalmanMuonCorrector.isSync, and MuScleFitCorrector.MuScleFitCorr.isSync.

Referenced by MuScleFitCorrector.MuScleFitCorr.correct(), and MuScleFitCorrector.MuScleFitCorr.correct_all().

     def corrected_p4(self, mu, run):
         p4 = TLorentzVector(mu.px(), mu.py(), mu.pz(), mu.energy())
         if self.isMC:
             self.corr.applyPtCorrection(p4, mu.charge())
             self.corr.applyPtSmearing(p4, mu.charge(), self.isSync)
         else:
             corr = self.corrD if run >= 203773 else self.corrABC
             corr.applyPtCorrection(p4, mu.charge())
         ## convert to the proper C++ class (but preserve the mass!)
         return ROOT.reco.Muon.PolarLorentzVector( p4.Pt(), p4.Eta(), p4.Phi(), mu.mass() )
 