#include <MuonAnalysis/MomentumScaleCalibration/plugins/TestCorrection.cc>

Inheritance diagram for TestCorrection:

Public Member Functions
	TestCorrection (const edm::ParameterSet &)

	~TestCorrection () override

Public Member Functions inherited from edm::one::EDAnalyzer<>
	EDAnalyzer ()=default

	EDAnalyzer (const EDAnalyzer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

const EDAnalyzer &	operator= (const EDAnalyzer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESResolverIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESResolverIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const *> const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override

lorentzVector	correctMuon (const lorentzVector &muon)

void	endJob () override

template<typename T >
std::vector< MuScleFitMuon >	fillMuonCollection (const std::vector< T > &tracks)

virtual void	initialize (const edm::EventSetup &)

Private Member Functions inherited from MuScleFitBase
	MuScleFitBase (const edm::ParameterSet &iConfig)

virtual	~MuScleFitBase () noexcept(false)

void	clearHistoMap ()
	Clean the histograms map. More...

void	fillHistoMap (TFile *outputFile, unsigned int iLoop)
	Create the histograms map. More...

void	readProbabilityDistributionsFromFile ()
	Read probability distributions from a local root file. More...

void	writeHistoMap (const unsigned int iLoop)
	Save the histograms map to file. More...

Private Attributes
std::unique_ptr< BackgroundFunction >	background_

TH1F *	correctedPt_

TProfile *	correctedPtVsEta_

std::unique_ptr< MomentumScaleCorrector >	corrector_

int	eventCounter_

edm::EDGetTokenT< reco::MuonCollection >	glbMuonsToken_

std::unique_ptr< ResolutionFunction >	resolution_

edm::EDGetTokenT< reco::TrackCollection >	saMuonsToken_

edm::EDGetTokenT< reco::TrackCollection >	tracksToken_

TH1F *	uncorrectedPt_

TProfile *	uncorrectedPtVsEta_

Private Attributes inherited from MuScleFitBase
int	debug_

std::vector< GenMuonPair >	genMuonPairs_
	Stores the genMuon pairs and the motherId prior to the creation of the internal tree. More...

std::map< std::string, Histograms * >	mapHisto_
	The map of histograms. More...

std::vector< MuonPair >	muonPairs_
	Used to store the muon pairs plus run and event number prior to the creation of the internal tree. More...

std::string	probabilitiesFile_

std::string	probabilitiesFileInPath_

int	theCompressionSettings_

std::vector< TFile * >	theFiles_
	The files were the histograms are saved. More...

std::string	theGenInfoRootFileName_

edm::InputTag	theMuonLabel_

int	theMuonType_

std::string	theRootFileName_

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 60 of file TestCorrection.cc.

Constructor & Destructor Documentation

◆ TestCorrection()

TestCorrection::TestCorrection ( const edm::ParameterSet & iConfig )

explicit

Definition at line 133 of file TestCorrection.cc.

References background_, correctedPt_, correctedPtVsEta_, corrector_, eventCounter_, MuScleFitBase::fillHistoMap(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), download_sqlite_cfg::outputFile, MuScleFitUtils::parResol, MuScleFitUtils::resfind, resolution_, MuScleFitUtils::resolutionFunction, MuScleFitUtils::resolutionFunctionForVec, resolutionFunctionVecService(), AlCaHLTBitMon_QueryRunRegistry::string, MuScleFitBase::theFiles_, MuScleFitBase::theRootFileName_, uncorrectedPt_, and uncorrectedPtVsEta_.

     : MuScleFitBase(iConfig),
       glbMuonsToken_(mayConsume<reco::MuonCollection>(theMuonLabel_)),
       saMuonsToken_(mayConsume<reco::TrackCollection>(theMuonLabel_)),
       tracksToken_(mayConsume<reco::TrackCollection>(theMuonLabel_)) {
   //now do what ever initialization is needed
   TFile* outputFile = new TFile(theRootFileName_.c_str(), "RECREATE");
   theFiles_.push_back(outputFile);
   // outputFile_ = new TFile(theRootFileName_.c_str(), "RECREATE");
   // outputFile_->cd();
   outputFile->cd();
   MuScleFitUtils::resfind = iConfig.getParameter<std::vector<int> >("resfind");
   fillHistoMap(outputFile, 0);
   uncorrectedPt_ = new TH1F("uncorrectedPt", "uncorrected pt", 1000, 0, 100);
   uncorrectedPtVsEta_ = new TProfile("uncorrectedPtVsEta", "uncorrected pt vs eta", 1000, 0, 100, -3., 3.);
   correctedPt_ = new TH1F("correctedPt", "corrected pt", 1000, 0, 100);
   correctedPtVsEta_ = new TProfile("correctedPtVsEta", "corrected pt vs eta", 1000, 0, 100, -3., 3.);
   eventCounter_ = 0;
   // Create the corrector and set the parameters
   corrector_ =
       std::make_unique<MomentumScaleCorrector>(iConfig.getUntrackedParameter<std::string>("CorrectionsIdentifier"));
   edm::LogPrint("TestCorrection") << "corrector_ = " << &*corrector_ << std::endl;
   resolution_ =
       std::make_unique<ResolutionFunction>(iConfig.getUntrackedParameter<std::string>("ResolutionsIdentifier"));
   edm::LogPrint("TestCorrection") << "resolution_ = " << &*resolution_ << std::endl;
   background_ =
       std::make_unique<BackgroundFunction>(iConfig.getUntrackedParameter<std::string>("BackgroundIdentifier"));
 
   // Initialize the parameters of MuScleFitUtils from those saved in the functions.
   // MuScleFitUtils::parScale = corrector_.getFunction(0)->parameters();
   MuScleFitUtils::resolutionFunction = resolution_->function(0);
   MuScleFitUtils::resolutionFunctionForVec = resolutionFunctionVecService(resolution_->identifiers()[0]);
 
   MuScleFitUtils::parResol = resolution_->parameters();
 }

◆ ~TestCorrection()

TestCorrection::~TestCorrection ( )

override

Definition at line 169 of file TestCorrection.cc.

References svgfig::canvas(), correctedPt_, correctedPtVsEta_, eventCounter_, compareTotals::legend, MuScleFitBase::theFiles_, uncorrectedPt_, uncorrectedPtVsEta_, and MuScleFitBase::writeHistoMap().

                                 {
   theFiles_[0]->cd();
   TCanvas canvas("ptComparison", "pt comparison", 1000, 800);
   canvas.cd();
   uncorrectedPt_->GetXaxis()->SetTitle("Pt(GeV)");
   correctedPt_->SetLineColor(kRed);
   TLegend* legend = new TLegend(0.7, 0.71, 0.98, 1.);
   legend->SetTextSize(0.02);
   legend->SetFillColor(0);  // Have a white background
   legend->AddEntry(uncorrectedPt_, "original pt");
   legend->AddEntry(correctedPt_, "corrected pt");
   uncorrectedPt_->Draw();
   correctedPt_->Draw("same");
   legend->Draw("same");
 
   canvas.Write();
   uncorrectedPt_->Write();
   uncorrectedPtVsEta_->Write();
   correctedPt_->Write();
   correctedPtVsEta_->Write();
 
   writeHistoMap(0);
   theFiles_[0]->Close();
 
   edm::LogPrint("TestCorrection") << "Total analyzed events = " << eventCounter_ << std::endl;
 }

Member Function Documentation

◆ analyze()

void TestCorrection::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 201 of file TestCorrection.cc.

References funct::abs(), correctedPt_, correctedPtVsEta_, correctMuon(), eventCounter_, fillMuonCollection(), MuScleFitUtils::findBestRecoRes(), glbMuonsToken_, iEvent, initialize(), MuScleFitBase::mapHisto_, PDWG_BPHSkim_cff::muons, MuScleFitUtils::ResFound, saMuonsToken_, MuScleFitUtils::SavedPair, MuScleFitBase::theMuonType_, pwdgSkimBPark_cfi::tracks, tracksToken_, uncorrectedPt_, and uncorrectedPtVsEta_.

                                                                                 {
   using namespace edm;
 
   initialize(iSetup);
 
   ++eventCounter_;
   if (eventCounter_ % 100 == 0) {
     edm::LogPrint("TestCorrection") << "Event number " << eventCounter_ << std::endl;
   }
 
   // Take the reco-muons, depending on the type selected in the cfg
   // --------------------------------------------------------------
 
   std::vector<MuScleFitMuon> muons;
 
   if (theMuonType_ == 1) {  // GlobalMuons
     Handle<reco::MuonCollection> glbMuons;
     iEvent.getByToken(glbMuonsToken_, glbMuons);
     muons = fillMuonCollection(*glbMuons);
   } else if (theMuonType_ == 2) {  // StandaloneMuons
     Handle<reco::TrackCollection> saMuons;
     iEvent.getByToken(saMuonsToken_, saMuons);
     muons = fillMuonCollection(*saMuons);
   } else if (theMuonType_ == 3) {  // Tracker tracks
     Handle<reco::TrackCollection> tracks;
     iEvent.getByToken(tracksToken_, tracks);
     muons = fillMuonCollection(*tracks);
   }
 
   // Find the two muons from the resonance, and set ResFound bool
   // ------------------------------------------------------------
   std::pair<MuScleFitMuon, MuScleFitMuon> recMuFromBestRes = MuScleFitUtils::findBestRecoRes(muons);
   if (MuScleFitUtils::ResFound) {
     MuScleFitUtils::SavedPair.push_back(std::make_pair(recMuFromBestRes.first.p4(), recMuFromBestRes.second.p4()));
   } else {
     MuScleFitUtils::SavedPair.push_back(std::make_pair(lorentzVector(0., 0., 0., 0.), lorentzVector(0., 0., 0., 0.)));
   }
 
   // If resonance found, do the hard work
   // ------------------------------------
   if (MuScleFitUtils::ResFound) {
     // Find weight and reference mass for this muon pair
     // -------------------------------------------------
     // double weight = MuScleFitUtils::computeWeight ((recMu1+recMu2).mass());
 
     // Use the correction function to correct the pt scale of the muons. Note that this takes into
     // account the corrections from all iterations.
     lorentzVector recMu1;
     recMu1 = correctMuon(recMu1);
     lorentzVector recMu2;
     recMu2 = correctMuon(recMu2);
 
     reco::Particle::LorentzVector bestRecRes(recMu1 + recMu2);
 
     //Fill histograms
     //------------------
     mapHisto_["hRecBestMu"]->Fill(recMu1);
     if ((std::abs(recMu1.eta()) < 2.5) && (recMu1.pt() > 2.5)) {
       mapHisto_["hRecBestMu_Acc"]->Fill(recMu1);
     }
     mapHisto_["hRecBestMu"]->Fill(recMu2);
     if ((std::abs(recMu2.eta()) < 2.5) && (recMu2.pt() > 2.5)) {
       mapHisto_["hRecBestMu_Acc"]->Fill(recMu2);
     }
     mapHisto_["hDeltaRecBestMu"]->Fill(recMu1, recMu2);
 
     mapHisto_["hRecBestRes"]->Fill(bestRecRes);
     if ((std::abs(recMu1.eta()) < 2.5) && (recMu1.pt() > 2.5) && (std::abs(recMu2.eta()) < 2.5) &&
         (recMu2.pt() > 2.5)) {
       mapHisto_["hRecBestRes_Acc"]->Fill(bestRecRes);
       // Fill histogram of Res mass vs muon variable
       mapHisto_["hRecBestResVSMu"]->Fill(recMu1, bestRecRes, -1);
       mapHisto_["hRecBestResVSMu"]->Fill(recMu2, bestRecRes, +1);
     }
   }
 
   // Loop on the recMuons
   std::vector<MuScleFitMuon>::const_iterator recMuon = muons.begin();
   int muonCount = 0;
   for (; recMuon != muons.end(); ++recMuon, ++muonCount) {
     // Fill the histogram with uncorrected pt values
     uncorrectedPt_->Fill(recMuon->pt());
     uncorrectedPtVsEta_->Fill(recMuon->pt(), recMuon->eta());
 
     // Fill the histogram with corrected pt values
     edm::LogPrint("TestCorrection") << "correcting muon[" << muonCount << "] with pt = " << recMuon->pt() << std::endl;
     double corrPt = (*corrector_)(*recMuon);
     edm::LogPrint("TestCorrection") << "to pt = " << corrPt << std::endl;
     correctedPt_->Fill(corrPt);
     correctedPtVsEta_->Fill(corrPt, recMuon->eta());
     // correctedPt_->Fill(recMuon->pt());
   }
 }

◆ correctMuon()

lorentzVector TestCorrection::correctMuon ( const lorentzVector & muon )

private

Definition at line 295 of file TestCorrection.cc.

References corrector_, and MuScleFitUtils::fromPtEtaPhiToPxPyPz().

Referenced by analyze().

                                                                    {
   double corrPt = corrector_->correct(muon);
   double ptEtaPhiE[4] = {corrPt, muon.Eta(), muon.Phi(), muon.E()};
   return MuScleFitUtils::fromPtEtaPhiToPxPyPz(ptEtaPhiE);
 }

◆ endJob()

void TestCorrection::endJob ( void )

inlineoverrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 68 of file TestCorrection.cc.

68 {};

◆ fillMuonCollection()

template<typename T >

std::vector<MuScleFitMuon> TestCorrection::fillMuonCollection ( const std::vector< T > & tracks )

inlineprivate

Definition at line 70 of file TestCorrection.cc.

References MuScleFitBase::debug_, amptDefaultParameters_cff::mu, HLT_2023v12_cff::muon, PDWG_BPHSkim_cff::muons, mathSSE::sqrt(), HLT_2023v12_cff::track, and pwdgSkimBPark_cfi::tracks.

Referenced by analyze().

                                                                           {
     std::vector<MuScleFitMuon> muons;
     typename std::vector<T>::const_iterator track;
     for (track = tracks.begin(); track != tracks.end(); ++track) {
       reco::Particle::LorentzVector mu;
       mu = reco::Particle::LorentzVector(
           track->px(), track->py(), track->pz(), sqrt(track->p() * track->p() + +0.011163612));
 
       Double_t hitsTk(0), hitsMuon(0), ptError(0);
       if (const reco::Muon* myMu = dynamic_cast<const reco::Muon*>(&(*track))) {
         hitsTk = myMu->innerTrack()->hitPattern().numberOfValidTrackerHits();
         hitsMuon = myMu->innerTrack()->hitPattern().numberOfValidMuonHits();
         ptError = myMu->innerTrack()->ptError();
       } else if (const pat::Muon* myMu = dynamic_cast<const pat::Muon*>(&(*track))) {
         hitsTk = myMu->innerTrack()->hitPattern().numberOfValidTrackerHits();
         hitsMuon = myMu->innerTrack()->hitPattern().numberOfValidMuonHits();
         ptError = myMu->innerTrack()->ptError();
       } else if (const reco::Track* myMu = dynamic_cast<const reco::Track*>(&(*track))) {
         hitsTk = myMu->hitPattern().numberOfValidTrackerHits();
         hitsMuon = myMu->hitPattern().numberOfValidMuonHits();
         ptError = myMu->ptError();
       }
 
       MuScleFitMuon muon(mu, track->charge(), ptError, hitsTk, hitsMuon);
 
       if (debug_ > 0) {
         edm::LogPrint("TestCorrection") << "[TestCorrection::fillMuonCollection] after MuScleFitMuon initialization"
                                         << std::endl;
         edm::LogPrint("TestCorrection") << "  muon = " << muon << std::endl;
       }
 
       muons.push_back(muon);
     }
     return muons;
   }

◆ initialize()

void TestCorrection::initialize ( const edm::EventSetup & )

privatevirtual

Definition at line 302 of file TestCorrection.cc.

References MuScleFitBase::readProbabilityDistributionsFromFile().

Referenced by analyze().

                                                     {
   // Read the pdf from root file. They are used by massProb when finding the muon pair, needed
   // for the mass histograms.
   readProbabilityDistributionsFromFile();
 }

Member Data Documentation

◆ background_

std::unique_ptr<BackgroundFunction> TestCorrection::background_

private

Definition at line 121 of file TestCorrection.cc.

Referenced by TestCorrection().

◆ correctedPt_

TH1F* TestCorrection::correctedPt_

private

Definition at line 114 of file TestCorrection.cc.

Referenced by analyze(), TestCorrection(), and ~TestCorrection().

◆ correctedPtVsEta_

TProfile* TestCorrection::correctedPtVsEta_

private

Definition at line 115 of file TestCorrection.cc.

Referenced by analyze(), TestCorrection(), and ~TestCorrection().

◆ corrector_

std::unique_ptr<MomentumScaleCorrector> TestCorrection::corrector_

private

Definition at line 119 of file TestCorrection.cc.

Referenced by correctMuon(), and TestCorrection().

◆ eventCounter_

int TestCorrection::eventCounter_

private

Definition at line 117 of file TestCorrection.cc.

Referenced by analyze(), TestCorrection(), and ~TestCorrection().

◆ glbMuonsToken_

edm::EDGetTokenT<reco::MuonCollection> TestCorrection::glbMuonsToken_

private

Definition at line 123 of file TestCorrection.cc.

Referenced by analyze().

◆ resolution_

std::unique_ptr<ResolutionFunction> TestCorrection::resolution_

private

Definition at line 120 of file TestCorrection.cc.

Referenced by TestCorrection().

◆ saMuonsToken_

edm::EDGetTokenT<reco::TrackCollection> TestCorrection::saMuonsToken_

private

Definition at line 124 of file TestCorrection.cc.

Referenced by analyze().

◆ tracksToken_

edm::EDGetTokenT<reco::TrackCollection> TestCorrection::tracksToken_

private

Definition at line 125 of file TestCorrection.cc.

Referenced by analyze().

◆ uncorrectedPt_

TH1F* TestCorrection::uncorrectedPt_

private

Definition at line 112 of file TestCorrection.cc.

Referenced by analyze(), TestCorrection(), and ~TestCorrection().

◆ uncorrectedPtVsEta_

TProfile* TestCorrection::uncorrectedPtVsEta_

private

Definition at line 113 of file TestCorrection.cc.

Referenced by analyze(), TestCorrection(), and ~TestCorrection().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ TestCorrection()

◆ ~TestCorrection()

Member Function Documentation

◆ analyze()

◆ correctMuon()

◆ endJob()

◆ fillMuonCollection()

◆ initialize()

Member Data Documentation

◆ background_

◆ correctedPt_

◆ correctedPtVsEta_

◆ corrector_

◆ eventCounter_

◆ glbMuonsToken_

◆ resolution_

◆ saMuonsToken_

◆ tracksToken_

◆ uncorrectedPt_

◆ uncorrectedPtVsEta_