#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 noexcept final

bool	wantsGlobalRuns () const noexcept final

bool	wantsInputProcessBlocks () const noexcept final

bool	wantsProcessBlocks () const noexcept 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 noexcept

bool	wantsStreamRuns () const noexcept

	~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_, DiMuonV_cfg::muons, MuScleFitUtils::ResFound, saMuonsToken_, MuScleFitUtils::SavedPair, MuScleFitBase::theMuonType_, DiMuonV_cfg::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_2024v14_cff::muon, DiMuonV_cfg::muons, mathSSE::sqrt(), HLT_2024v14_cff::track, and DiMuonV_cfg::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_