Inheritance diagram for CTPPSProtonReconstructionSimulationValidator:

Classes
struct	DoubleArmPlotGroup

struct	PlotGroup

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

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

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

std::vector< ESProxyIndex > 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

bool	registeredToConsumeMany (TypeID const &, 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::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

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

void	endJob () override

void	fillPlots (unsigned int meth_idx, unsigned int idx, const reco::ForwardProton &rec_pr, const HepMC::FourVector &vtx, const HepMC::FourVector &mom, const LHCInfo &lhcInfo)

Private Attributes
DoubleArmPlotGroup	double_arm_plots_

edm::ESGetToken< LHCInfo, LHCInfoRcd >	lhcInfoESToken_

std::string	outputFile_

std::map< unsigned int, std::map< unsigned int, PlotGroup > >	plots_

edm::EDGetTokenT< edm::HepMCProduct >	tokenHepMCAfterSmearing_

edm::EDGetTokenT< edm::HepMCProduct >	tokenHepMCBeforeSmearing_

edm::EDGetTokenT< reco::ForwardProtonCollection >	tokenRecoProtonsMultiRP_

edm::EDGetTokenT< reco::ForwardProtonCollection >	tokenRecoProtonsSingleRP_

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 ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

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

Definition at line 37 of file CTPPSProtonReconstructionSimulationValidator.cc.

Constructor & Destructor Documentation

◆ CTPPSProtonReconstructionSimulationValidator()

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

explicit

Definition at line 194 of file CTPPSProtonReconstructionSimulationValidator.cc.

     : tokenHepMCBeforeSmearing_(
           consumes<edm::HepMCProduct>(iConfig.getParameter<edm::InputTag>("tagHepMCBeforeSmearing"))),
       tokenHepMCAfterSmearing_(
           consumes<edm::HepMCProduct>(iConfig.getParameter<edm::InputTag>("tagHepMCAfterSmearing"))),
       tokenRecoProtonsSingleRP_(
           consumes<reco::ForwardProtonCollection>(iConfig.getParameter<InputTag>("tagRecoProtonsSingleRP"))),
       tokenRecoProtonsMultiRP_(
           consumes<reco::ForwardProtonCollection>(iConfig.getParameter<InputTag>("tagRecoProtonsMultiRP"))),
       lhcInfoESToken_(esConsumes(ESInputTag("", iConfig.getParameter<std::string>("lhcInfoLabel")))),
       outputFile_(iConfig.getParameter<string>("outputFile")) {}

Member Function Documentation

◆ analyze()

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 209 of file CTPPSProtonReconstructionSimulationValidator.cc.

References double_arm_plots_, CTPPSProtonReconstructionSimulationValidator::DoubleArmPlotGroup::fill(), fillPlots(), edm::EventSetup::getData(), edm::HepMCProduct::GetEvent(), patZpeak::handle, heavyIonCSV_trainingSettings::idx, iEvent, lhcInfoESToken_, reco::ForwardProton::multiRP, reco::ForwardProton::sector45, reco::ForwardProton::sector56, reco::ForwardProton::singleRP, tokenHepMCAfterSmearing_, tokenHepMCBeforeSmearing_, tokenRecoProtonsMultiRP_, tokenRecoProtonsSingleRP_, and extraflags_cff::vtx.

                                                                                                               {
   // get conditions
   const auto &lhcInfo = iSetup.getData(lhcInfoESToken_);
 
   // get input
   edm::Handle<edm::HepMCProduct> hHepMCBeforeSmearing;
   iEvent.getByToken(tokenHepMCBeforeSmearing_, hHepMCBeforeSmearing);
   HepMC::GenEvent *hepMCEventBeforeSmearing = (HepMC::GenEvent *)hHepMCBeforeSmearing->GetEvent();
 
   edm::Handle<edm::HepMCProduct> hHepMCAfterSmearing;
   iEvent.getByToken(tokenHepMCAfterSmearing_, hHepMCAfterSmearing);
   HepMC::GenEvent *hepMCEventAfterSmearing = (HepMC::GenEvent *)hHepMCAfterSmearing->GetEvent();
 
   Handle<reco::ForwardProtonCollection> hRecoProtonsSingleRP;
   iEvent.getByToken(tokenRecoProtonsSingleRP_, hRecoProtonsSingleRP);
 
   Handle<reco::ForwardProtonCollection> hRecoProtonsMultiRP;
   iEvent.getByToken(tokenRecoProtonsMultiRP_, hRecoProtonsMultiRP);
 
   // extract vertex position
   bool vertex_set = false;
   FourVector vtx;
   for (auto it = hepMCEventAfterSmearing->vertices_begin(); it != hepMCEventAfterSmearing->vertices_end(); ++it) {
     if (vertex_set) {
       LogError("CTPPSProtonReconstructionSimulationValidator") << "Multiple vertices found.";
       return;
     }
 
     vertex_set = true;
     vtx = (*it)->position();
   }
 
   // extract forward protons
   bool proton_45_set = false;
   bool proton_56_set = false;
   FourVector mom_45, mom_56;
 
   for (auto it = hepMCEventBeforeSmearing->particles_begin(); it != hepMCEventBeforeSmearing->particles_end(); ++it) {
     const auto &part = *it;
 
     // accept only stable non-beam protons
     if (part->pdg_id() != 2212)
       continue;
 
     if (part->status() != 1)
       continue;
 
     if (part->is_beam())
       continue;
 
     const auto &mom = part->momentum();
 
     if (mom.e() < 4500.)
       continue;
 
     if (mom.z() > 0) {
       // 45
       if (proton_45_set) {
         LogError("CTPPSProtonReconstructionSimulationValidator") << "Found multiple protons in sector 45.";
         return;
       }
 
       proton_45_set = true;
       mom_45 = mom;
     } else {
       // 56
       if (proton_56_set) {
         LogError("CTPPSProtonReconstructionSimulationValidator") << "Found multiple protons in sector 56.";
         return;
       }
 
       proton_56_set = true;
       mom_56 = mom;
     }
   }
 
   // single-arm comparison
   for (const auto &handle : {hRecoProtonsSingleRP, hRecoProtonsMultiRP}) {
     for (const auto &rec_pr : *handle) {
       if (!rec_pr.validFit())
         continue;
 
       unsigned int idx = 12345;
 
       bool mom_set = false;
       FourVector mom;
 
       if (rec_pr.lhcSector() == reco::ForwardProton::LHCSector::sector45) {
         idx = 0;
         mom_set = proton_45_set;
         mom = mom_45;
       }
       if (rec_pr.lhcSector() == reco::ForwardProton::LHCSector::sector56) {
         idx = 1;
         mom_set = proton_56_set;
         mom = mom_56;
       }
 
       if (!mom_set)
         continue;
 
       unsigned int meth_idx = 1234;
 
       if (rec_pr.method() == reco::ForwardProton::ReconstructionMethod::singleRP) {
         meth_idx = 0;
 
         CTPPSDetId rpId((*rec_pr.contributingLocalTracks().begin())->rpId());
         idx = 100 * rpId.arm() + 10 * rpId.station() + rpId.rp();
       }
 
       if (rec_pr.method() == reco::ForwardProton::ReconstructionMethod::multiRP)
         meth_idx = 1;
 
       fillPlots(meth_idx, idx, rec_pr, vtx, mom, lhcInfo);
     }
   }
 
   // double-arm comparison
   bool time_set_45 = false, time_set_56 = false;
   double time_45 = 0., time_56 = 0.;
   for (const auto &rec_pr : *hRecoProtonsMultiRP) {
     if (!rec_pr.validFit())
       continue;
 
     // skip protons with no timing information
     if (rec_pr.timeError() < 1E-3)
       continue;
 
     if (rec_pr.lhcSector() == reco::ForwardProton::LHCSector::sector45) {
       time_set_45 = true;
       time_45 = rec_pr.time();
     }
     if (rec_pr.lhcSector() == reco::ForwardProton::LHCSector::sector56) {
       time_set_56 = true;
       time_56 = rec_pr.time();
     }
   }
 
   if (time_set_45 && time_set_56)
     double_arm_plots_.fill(time_45, time_56, vtx.z(), vtx.t());
 }

◆ endJob()

void CTPPSProtonReconstructionSimulationValidator::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 396 of file CTPPSProtonReconstructionSimulationValidator.cc.

References double_arm_plots_, AlcaSiPixelAliHarvester0T_cff::method, outputFile_, plots_, and CTPPSProtonReconstructionSimulationValidator::DoubleArmPlotGroup::write().

                                                           {
   auto f_out = std::make_unique<TFile>(outputFile_.c_str(), "recreate");
 
   for (const auto &mit : plots_) {
     const char *method = (mit.first == 0) ? "single rp" : "multi rp";
     TDirectory *d_method = f_out->mkdir(method);
 
     for (const auto &eit : mit.second) {
       gDirectory = d_method->mkdir(Form("%i", eit.first));
       eit.second.write();
     }
   }
 
   gDirectory = f_out->mkdir("double arm");
   double_arm_plots_.write();
 }

◆ fillPlots()

void CTPPSProtonReconstructionSimulationValidator::fillPlots	(	unsigned int	meth_idx,
		unsigned int	idx,
		const reco::ForwardProton &	rec_pr,
		const HepMC::FourVector &	vtx,
		const HepMC::FourVector &	mom,
		const LHCInfo &	lhcInfo
	)

private

Definition at line 353 of file CTPPSProtonReconstructionSimulationValidator.cc.

References reco::ForwardProton::calculateT(), LHCInfo::energy(), heavyIonCSV_trainingSettings::idx, plots_, mathSSE::sqrt(), reco::ForwardProton::t(), reco::ForwardProton::thetaX(), reco::ForwardProton::thetaY(), reco::ForwardProton::vertex(), extraflags_cff::vtx, and reco::ForwardProton::xi().

Referenced by analyze().

                                                                                      {
   const double p_nom = lhcInfo.energy();
   const double xi_simu = (p_nom - mom.rho()) / p_nom;
   const double th_x_simu = mom.x() / mom.rho();
   const double th_y_simu = mom.y() / mom.rho();
   const double vtx_y_simu = vtx.y();
   const double th_simu = sqrt(th_x_simu * th_x_simu + th_y_simu * th_y_simu);
   const double t_simu = -reco::ForwardProton::calculateT(p_nom, mom.rho(), th_simu);
 
   const double xi_reco = rec_pr.xi();
   const double th_x_reco = rec_pr.thetaX();
   const double th_y_reco = rec_pr.thetaY();
   const double vtx_y_reco = rec_pr.vertex().y() * 10.;  // conversion: cm --> mm
   const double t_reco = -rec_pr.t();
 
   auto &plt = plots_[meth_idx][idx];
 
   plt.h_xi_reco_vs_xi_simu->Fill(xi_simu, xi_reco);
   plt.h_de_xi->Fill(xi_reco - xi_simu);
   plt.p_de_xi_vs_xi_simu->Fill(xi_simu, xi_reco - xi_simu);
 
   plt.h_de_th_x->Fill(th_x_reco - th_x_simu);
   plt.p_de_th_x_vs_xi_simu->Fill(xi_simu, th_x_reco - th_x_simu);
 
   plt.h_de_th_y->Fill(th_y_reco - th_y_simu);
   plt.p_de_th_y_vs_xi_simu->Fill(xi_simu, th_y_reco - th_y_simu);
 
   plt.h_de_vtx_y->Fill(vtx_y_reco - vtx_y_simu);
   plt.p_de_vtx_y_vs_xi_simu->Fill(xi_simu, vtx_y_reco - vtx_y_simu);
 
   plt.h_de_t->Fill(t_reco - t_simu);
   plt.p_de_t_vs_xi_simu->Fill(xi_simu, t_reco - t_simu);
   plt.p_de_t_vs_t_simu->Fill(t_simu, t_reco - t_simu);
   plt.h2_de_t_vs_t_simu->Fill(t_simu, t_reco - t_simu);
 }

Member Data Documentation

◆ double_arm_plots_

DoubleArmPlotGroup CTPPSProtonReconstructionSimulationValidator::double_arm_plots_

private

Definition at line 183 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze(), and endJob().

◆ lhcInfoESToken_

edm::ESGetToken<LHCInfo, LHCInfoRcd> CTPPSProtonReconstructionSimulationValidator::lhcInfoESToken_

private

Definition at line 58 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

◆ outputFile_

std::string CTPPSProtonReconstructionSimulationValidator::outputFile_

private

Definition at line 60 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by endJob().

◆ plots_

std::map<unsigned int, std::map<unsigned int, PlotGroup> > CTPPSProtonReconstructionSimulationValidator::plots_

private

Definition at line 151 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by endJob(), and fillPlots().

◆ tokenHepMCAfterSmearing_

edm::EDGetTokenT<edm::HepMCProduct> CTPPSProtonReconstructionSimulationValidator::tokenHepMCAfterSmearing_

private

Definition at line 53 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

◆ tokenHepMCBeforeSmearing_

edm::EDGetTokenT<edm::HepMCProduct> CTPPSProtonReconstructionSimulationValidator::tokenHepMCBeforeSmearing_

private

Definition at line 52 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

◆ tokenRecoProtonsMultiRP_

edm::EDGetTokenT<reco::ForwardProtonCollection> CTPPSProtonReconstructionSimulationValidator::tokenRecoProtonsMultiRP_

private

Definition at line 56 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

◆ tokenRecoProtonsSingleRP_

edm::EDGetTokenT<reco::ForwardProtonCollection> CTPPSProtonReconstructionSimulationValidator::tokenRecoProtonsSingleRP_

private

Definition at line 55 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

Classes

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ CTPPSProtonReconstructionSimulationValidator()

Member Function Documentation

◆ analyze()

◆ endJob()

◆ fillPlots()

Member Data Documentation

◆ double_arm_plots_

◆ lhcInfoESToken_

◆ outputFile_

◆ plots_

◆ tokenHepMCAfterSmearing_

◆ tokenHepMCBeforeSmearing_

◆ tokenRecoProtonsMultiRP_

◆ tokenRecoProtonsSingleRP_