CTPPSProtonReconstructionSimulationValidator Class Reference

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
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)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

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 double energy)

Private Attributes
DoubleArmPlotGroup	double_arm_plots_
const edm::ESGetToken< LHCInfoPerFill, LHCInfoPerFillRcd >	lhcInfoPerFillToken_
const edm::ESGetToken< LHCInfoPerLS, LHCInfoPerLSRcd >	lhcInfoPerLSToken_
const edm::ESGetToken< LHCInfo, LHCInfoRcd >	lhcInfoToken_
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_
const bool	useNewLHCInfo_

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
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

Definition at line 36 of file CTPPSProtonReconstructionSimulationValidator.cc.

Constructor & Destructor Documentation

CTPPSProtonReconstructionSimulationValidator()

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

explicit

Definition at line 198 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"))),
      lhcInfoToken_(esConsumes(ESInputTag("", iConfig.getParameter<std::string>("lhcInfoLabel")))),
      lhcInfoPerLSToken_(esConsumes(ESInputTag("", iConfig.getParameter<std::string>("lhcInfoPerLSLabel")))),
      lhcInfoPerFillToken_(esConsumes(ESInputTag("", iConfig.getParameter<std::string>("lhcInfoPerFillLabel")))),
      useNewLHCInfo_(iConfig.getParameter<bool>("useNewLHCInfo")),
      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 233 of file CTPPSProtonReconstructionSimulationValidator.cc.

References double_arm_plots_, LHCInfoCombined::energy, CTPPSProtonReconstructionSimulationValidator::DoubleArmPlotGroup::fill(), fillPlots(), edm::HepMCProduct::GetEvent(), patZpeak::handle, heavyIonCSV_trainingSettings::idx, iEvent, lhcInfoPerFillToken_, lhcInfoPerLSToken_, lhcInfoToken_, reco::ForwardProton::multiRP, reco::ForwardProton::sector45, reco::ForwardProton::sector56, reco::ForwardProton::singleRP, tokenHepMCAfterSmearing_, tokenHepMCBeforeSmearing_, tokenRecoProtonsMultiRP_, tokenRecoProtonsSingleRP_, useNewLHCInfo_, and L1BJetProducer_cff::vtx.

                                                                                                              {
  // get conditions
  const LHCInfoCombined lhcInfoCombined(
      iSetup, lhcInfoPerLSToken_, lhcInfoPerFillToken_, lhcInfoToken_, useNewLHCInfo_);

  // 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, lhcInfoCombined.energy);
    }
  }

  // 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 421 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();
}

fillDescriptions()

void CTPPSProtonReconstructionSimulationValidator::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 216 of file CTPPSProtonReconstructionSimulationValidator.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                              {
  edm::ParameterSetDescription desc;

  desc.add<std::string>("lhcInfoLabel", "")->setComment("label of the LHCInfo record");
  desc.add<std::string>("lhcInfoPerLSLabel", "")->setComment("label of the LHCInfoPerLS record");
  desc.add<std::string>("lhcInfoPerFillLabel", "")->setComment("label of the LHCInfoPerFill record");
  desc.add<bool>("useNewLHCInfo", false)->setComment("flag whether to use new LHCInfoPer* records or old LHCInfo");

  desc.add<std::string>("outputFile", "output.root")->setComment("output file name");

  desc.addUntracked<unsigned int>("verbosity", 0)->setComment("verbosity level");

  descriptions.add("ctppsProtonReconstructionSimulationValidatorDefault", desc);
}

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 double  energy  )

private

Definition at line 378 of file CTPPSProtonReconstructionSimulationValidator.cc.

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

Referenced by analyze().

                                                                                  {
  const double p_nom = 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 187 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze(), and endJob().

lhcInfoPerFillToken_

const edm::ESGetToken<LHCInfoPerFill, LHCInfoPerFillRcd> CTPPSProtonReconstructionSimulationValidator::lhcInfoPerFillToken_

private

Definition at line 61 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

lhcInfoPerLSToken_

const edm::ESGetToken<LHCInfoPerLS, LHCInfoPerLSRcd> CTPPSProtonReconstructionSimulationValidator::lhcInfoPerLSToken_

private

Definition at line 60 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

lhcInfoToken_

const edm::ESGetToken<LHCInfo, LHCInfoRcd> CTPPSProtonReconstructionSimulationValidator::lhcInfoToken_

private

Definition at line 59 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

outputFile_

std::string CTPPSProtonReconstructionSimulationValidator::outputFile_

private

Definition at line 64 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by endJob().

plots_

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

private

Definition at line 155 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by endJob(), and fillPlots().

tokenHepMCAfterSmearing_

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

private

Definition at line 54 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

tokenHepMCBeforeSmearing_

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

private

Definition at line 53 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

tokenRecoProtonsMultiRP_

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

private

Definition at line 57 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

tokenRecoProtonsSingleRP_

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

private

Definition at line 56 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().

useNewLHCInfo_

const bool CTPPSProtonReconstructionSimulationValidator::useNewLHCInfo_

private

Definition at line 62 of file CTPPSProtonReconstructionSimulationValidator.cc.

Referenced by analyze().