CTPPSOpticsPlotter Class Reference

Inheritance diagram for CTPPSOpticsPlotter:

Classes
struct	ArmPlots
struct	RPPlots

Public Member Functions
	CTPPSOpticsPlotter (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 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
void	endJob () override

Private Attributes
std::map< unsigned int, ArmPlots >	arm_plots_
edm::ESGetToken< LHCInterpolatedOpticalFunctionsSetCollection, CTPPSInterpolatedOpticsRcd >	opticsESToken_
std::string	outputFile_
std::map< unsigned int, RPPlots >	rp_plots_
unsigned int	rpId_45_F_
unsigned int	rpId_45_N_
unsigned int	rpId_56_F_
unsigned int	rpId_56_N_

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

Definition at line 27 of file CTPPSOpticsPlotter.cc.

Constructor & Destructor Documentation

CTPPSOpticsPlotter()

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

explicit

Definition at line 108 of file CTPPSOpticsPlotter.cc.

References arm_plots_, rpId_45_F_, rpId_45_N_, rpId_56_F_, and rpId_56_N_.

    : opticsESToken_(esConsumes(ESInputTag("", iConfig.getParameter<std::string>("opticsLabel")))),

      rpId_45_N_(iConfig.getParameter<unsigned int>("rpId_45_N")),
      rpId_45_F_(iConfig.getParameter<unsigned int>("rpId_45_F")),
      rpId_56_N_(iConfig.getParameter<unsigned int>("rpId_56_N")),
      rpId_56_F_(iConfig.getParameter<unsigned int>("rpId_56_F")),

      outputFile_(iConfig.getParameter<string>("outputFile")) {
  arm_plots_[0].id_N = rpId_45_N_;
  arm_plots_[0].id_F = rpId_45_F_;

  arm_plots_[1].id_N = rpId_56_N_;
  arm_plots_[1].id_F = rpId_56_F_;
}

Member Function Documentation

analyze()

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 126 of file CTPPSOpticsPlotter.cc.

References CTPPSDetId::arm(), arm_plots_, edm::EventSetup::getData(), heavyIonCSV_trainingSettings::idx, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, profile_base_cff::opticalFunctions, opticsESToken_, CTPPSDetId::rp(), rp_plots_, CTPPSDetId::station(), LHCInterpolatedOpticalFunctionsSet::transport(), LHCInterpolatedOpticalFunctionsSet::Kinematics::x, protons_cff::xi, and LHCInterpolatedOpticalFunctionsSet::Kinematics::y.

                                                                                    {
  // stop if plots already made
  if (!rp_plots_.empty())
    return;

  // get conditions
  const auto& opticalFunctions = iSetup.getData(opticsESToken_);

  // stop if conditions invalid
  if (opticalFunctions.empty())
    return;

  // make per-RP plots
  for (const auto& it : opticalFunctions) {
    CTPPSDetId rpId(it.first);
    unsigned int rpDecId = rpId.arm() * 100 + rpId.station() * 10 + rpId.rp();

    auto& pl = rp_plots_[rpDecId];

    LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_beam = {0., 0., 0., 0., 0.};
    LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_beam;
    it.second.transport(k_in_beam, k_out_beam);

    const double vtx_ep = 1E-4;  // cm
    const double th_ep = 1E-6;   // rad

    for (double xi = 0.; xi < 0.30001; xi += 0.001) {
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi = {0., 0., 0., 0., xi};
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi;
      it.second.transport(k_in_xi, k_out_xi);

      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi_vtx_x = {vtx_ep, 0., 0., 0., xi};
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi_vtx_x;
      it.second.transport(k_in_xi_vtx_x, k_out_xi_vtx_x);

      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi_th_x = {0., th_ep, 0., 0., xi};
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi_th_x;
      it.second.transport(k_in_xi_th_x, k_out_xi_th_x);

      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi_vtx_y = {0., 0., vtx_ep, 0., xi};
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi_vtx_y;
      it.second.transport(k_in_xi_vtx_y, k_out_xi_vtx_y);

      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi_th_y = {0., 0., 0., th_ep, xi};
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi_th_y;
      it.second.transport(k_in_xi_th_y, k_out_xi_th_y);

      int idx = pl.g_v_x_vs_xi->GetN();

      pl.g_v_x_vs_xi->SetPoint(idx, xi, (k_out_xi_vtx_x.x - k_out_xi.x) / vtx_ep);
      pl.g_L_x_vs_xi->SetPoint(idx, xi, (k_out_xi_th_x.x - k_out_xi.x) / th_ep);
      pl.g_x_D_vs_xi->SetPoint(idx, xi, k_out_xi.x - k_out_beam.x);

      pl.g_v_y_vs_xi->SetPoint(idx, xi, (k_out_xi_vtx_y.y - k_out_xi.y) / vtx_ep);
      pl.g_L_y_vs_xi->SetPoint(idx, xi, (k_out_xi_th_y.y - k_out_xi.y) / th_ep);
      pl.g_y_D_vs_xi->SetPoint(idx, xi, k_out_xi.y - k_out_beam.y);

      pl.h_y_vs_x_disp->SetPoint(idx, k_out_xi.x - k_out_beam.x, k_out_xi.y - k_out_beam.y);
    }
  }

  // make per-arm plots
  for (const auto& ap : arm_plots_) {
    // find optics objects
    const LHCInterpolatedOpticalFunctionsSet *opt_N = nullptr, *opt_F = nullptr;

    for (const auto& it : opticalFunctions) {
      CTPPSDetId rpId(it.first);
      unsigned int rpDecId = rpId.arm() * 100 + rpId.station() * 10 + rpId.rp();

      if (rpDecId == ap.second.id_N)
        opt_N = &it.second;
      if (rpDecId == ap.second.id_F)
        opt_F = &it.second;
    }

    if (!opt_N || !opt_F) {
      edm::LogError("CTPPSOpticsPlotter::analyze") << "Cannot find optics objects for arm " << ap.first;
      continue;
    }

    LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_beam = {0., 0., 0., 0., 0.};

    LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_beam_N, k_out_beam_F;
    opt_N->transport(k_in_beam, k_out_beam_N);
    opt_F->transport(k_in_beam, k_out_beam_F);

    for (double xi = 0.; xi < 0.30001; xi += 0.001) {
      LHCInterpolatedOpticalFunctionsSet::Kinematics k_in_xi = {0., 0., 0., 0., xi};

      LHCInterpolatedOpticalFunctionsSet::Kinematics k_out_xi_N, k_out_xi_F;
      opt_N->transport(k_in_xi, k_out_xi_N);
      opt_F->transport(k_in_xi, k_out_xi_F);

      int idx = ap.second.g_de_x_vs_x_disp->GetN();

      ap.second.g_de_x_vs_x_disp->SetPoint(
          idx, k_out_xi_N.x - k_out_beam_N.x, (k_out_xi_F.x - k_out_beam_F.x) - (k_out_xi_N.x - k_out_beam_N.x));
      ap.second.g_de_y_vs_x_disp->SetPoint(
          idx, k_out_xi_N.x - k_out_beam_N.x, (k_out_xi_F.y - k_out_beam_F.y) - (k_out_xi_N.y - k_out_beam_N.y));
    }
  }
}

endJob()

void CTPPSOpticsPlotter::endJob ( void  )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 232 of file CTPPSOpticsPlotter.cc.

References arm_plots_, outputFile_, AlCaHLTBitMon_ParallelJobs::p, and rp_plots_.

                                {
  auto f_out = std::make_unique<TFile>(outputFile_.c_str(), "recreate");

  for (const auto& p : rp_plots_) {
    gDirectory = f_out->mkdir(Form("%u", p.first));
    p.second.write();
  }

  for (const auto& p : arm_plots_) {
    gDirectory = f_out->mkdir(Form("arm %u", p.first));
    p.second.write();
  }
}

Member Data Documentation

arm_plots_

std::map<unsigned int, ArmPlots> CTPPSOpticsPlotter::arm_plots_

private

Definition at line 98 of file CTPPSOpticsPlotter.cc.

Referenced by analyze(), CTPPSOpticsPlotter(), and endJob().

opticsESToken_

edm::ESGetToken<LHCInterpolatedOpticalFunctionsSetCollection, CTPPSInterpolatedOpticsRcd> CTPPSOpticsPlotter::opticsESToken_

private

Definition at line 35 of file CTPPSOpticsPlotter.cc.

Referenced by analyze().

outputFile_

std::string CTPPSOpticsPlotter::outputFile_

private

Definition at line 40 of file CTPPSOpticsPlotter.cc.

Referenced by endJob().

rp_plots_

std::map<unsigned int, RPPlots> CTPPSOpticsPlotter::rp_plots_

private

Definition at line 80 of file CTPPSOpticsPlotter.cc.

Referenced by analyze(), and endJob().

rpId_45_F_

unsigned int CTPPSOpticsPlotter::rpId_45_F_

private

Definition at line 37 of file CTPPSOpticsPlotter.cc.

Referenced by CTPPSOpticsPlotter().

rpId_45_N_

unsigned int CTPPSOpticsPlotter::rpId_45_N_

private

Definition at line 37 of file CTPPSOpticsPlotter.cc.

Referenced by CTPPSOpticsPlotter().

rpId_56_F_

unsigned int CTPPSOpticsPlotter::rpId_56_F_

private

Definition at line 38 of file CTPPSOpticsPlotter.cc.

Referenced by CTPPSOpticsPlotter().

rpId_56_N_

unsigned int CTPPSOpticsPlotter::rpId_56_N_

private

Definition at line 38 of file CTPPSOpticsPlotter.cc.

Referenced by CTPPSOpticsPlotter().