#include <Alignment/OfflineValidation/plugins/EopTreeWriter.cc>

Inheritance diagram for EopTreeWriter:

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

	~EopTreeWriter () override=default

Public Member Functions inherited from edm::one::EDAnalyzer< edm::one::SharedResources >
	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)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)

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

double	getDistInCM (double eta1, double phi1, double eta2, double phi2)

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

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

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

Definition at line 62 of file EopTreeWriter.cc.

Constructor & Destructor Documentation

◆ EopTreeWriter()

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

explicit

Definition at line 94 of file EopTreeWriter.cc.

References edm::EDConsumerBase::consumesCollector(), ecalRecHitTokenAlCaToken_, ecalRecHitTokenEBRecoToken_, ecalRecHitTokenEERecoToken_, fs_, edm::ParameterSet::getParameter(), ProducerED_cfi::InputTag, isoPixelTkToken_, TFileService::kSharedResource, TrackAssociatorParameters::loadParameters(), TFileService::make(), parameters_, src_, theTrackCollectionToken_, tree_, and treeMemPtr_.

     : src_(iConfig.getParameter<edm::InputTag>("src")), geometryToken_(esConsumes()) {
   usesResource(TFileService::kSharedResource);
   //now do what ever initialization is needed
 
   // TrackAssociator parameters
   edm::ParameterSet parameters = iConfig.getParameter<edm::ParameterSet>("TrackAssociatorParameters");
   edm::ConsumesCollector iC = consumesCollector();
   parameters_.loadParameters(parameters, iC);
 
   tree_ = fs_->make<TTree>("EopTree", "EopTree");
   treeMemPtr_ = new EopVariables;
   tree_->Branch("EopVariables", &treeMemPtr_);  // address of pointer!
 
   // do the consumes
   theTrackCollectionToken_ = consumes<reco::TrackCollection>(src_);
   isoPixelTkToken_ =
       consumes<reco::IsolatedPixelTrackCandidateCollection>(edm::InputTag("IsoProd", "HcalIsolatedTrackCollection"));
 
   ecalRecHitTokenAlCaToken_ = consumes<EcalRecHitCollection>(edm::InputTag("IsoProd", "IsoTrackEcalRecHitCollection"));
   ecalRecHitTokenEBRecoToken_ = consumes<EcalRecHitCollection>(edm::InputTag("ecalRecHit", "EcalRecHitsEB"));
   ecalRecHitTokenEERecoToken_ = consumes<EcalRecHitCollection>(edm::InputTag("ecalRecHit", "EcalRecHitsEE"));
 }

◆ ~EopTreeWriter()

EopTreeWriter::~EopTreeWriter ( )

overridedefault

Member Function Documentation

◆ analyze()

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 123 of file EopTreeWriter.cc.

References TrackDetectorAssociator::associate(), TrackAssociatorParameters::bFieldToken, TrackDetMatchInfo::EcalRecHits, ecalRecHitTokenAlCaToken_, ecalRecHitTokenEBRecoToken_, ecalRecHitTokenEERecoToken_, PV3DBase< T, PVType, FrameType >::eta(), Exception, EopVariables::fillVariables(), geometryToken_, edm::EventSetup::getData(), getDistInCM(), TrackDetectorAssociator::getFreeTrajectoryState(), CaloGeometry::getPosition(), TrackDetMatchInfo::HcalRecHits, mps_fire::i, iEvent, info(), isoPixelTkToken_, edm::HandleBase::isValid(), parameters_, PV3DBase< T, PVType, FrameType >::phi(), rpcPointValidation_cfi::recHit, theTrackCollectionToken_, HLT_2024v14_cff::track, trackAssociator_, DiMuonV_cfg::tracks, tree_, treeMemPtr_, TrackDetMatchInfo::trkGlobPosAtEcal, TrackDetectorAssociator::useDefaultPropagator(), and TrackAssociatorParameters::useMuon.

                                                                                {
   using namespace edm;
 
   // get geometry
   const CaloGeometry* geo = &iSetup.getData(geometryToken_);
 
   // temporary collection of EB+EE recHits
   std::unique_ptr<EcalRecHitCollection> tmpEcalRecHitCollection(new EcalRecHitCollection);
   bool ecalInAlca = iEvent.getHandle(ecalRecHitTokenAlCaToken_).isValid();
   bool ecalInReco =
       iEvent.getHandle(ecalRecHitTokenEBRecoToken_) && iEvent.getHandle(ecalRecHitTokenEERecoToken_).isValid();
 
   std::vector<edm::EDGetTokenT<EcalRecHitCollection> > ecalTokens_;
   if (ecalInAlca)
     ecalTokens_.push_back(ecalRecHitTokenAlCaToken_);
   else if (ecalInReco) {
     ecalTokens_.push_back(ecalRecHitTokenEBRecoToken_);
     ecalTokens_.push_back(ecalRecHitTokenEERecoToken_);
   } else {
     throw cms::Exception("MissingProduct", "can not find EcalRecHits");
   }
 
   for (const auto& i : ecalTokens_) {
     edm::Handle<EcalRecHitCollection> ec = iEvent.getHandle(i);
     for (EcalRecHitCollection::const_iterator recHit = (*ec).begin(); recHit != (*ec).end(); ++recHit) {
       tmpEcalRecHitCollection->push_back(*recHit);
     }
   }
 
   const auto& tracks = iEvent.get(theTrackCollectionToken_);
 
   edm::Handle<reco::IsolatedPixelTrackCandidateCollection> isoPixelTracks = iEvent.getHandle(isoPixelTkToken_);
   bool pixelInAlca = isoPixelTracks.isValid();
 
   double trackemc1;
   double trackemc3;
   double trackemc5;
   double trackhac1;
   double trackhac3;
   double trackhac5;
   double maxPNearby;
   double dist;
   double EnergyIn;
   double EnergyOut;
 
   parameters_.useMuon = false;
 
   if (pixelInAlca)
     if (isoPixelTracks->empty())
       return;
 
   for (const auto& track : tracks) {
     bool noChargedTracks = true;
 
     if (track.p() < 9.)
       continue;
 
     trackAssociator_.useDefaultPropagator();
     TrackDetMatchInfo info = trackAssociator_.associate(
         iEvent,
         iSetup,
         trackAssociator_.getFreeTrajectoryState(&iSetup.getData(parameters_.bFieldToken), track),
         parameters_);
 
     trackemc1 = info.nXnEnergy(TrackDetMatchInfo::EcalRecHits, 0);
     trackemc3 = info.nXnEnergy(TrackDetMatchInfo::EcalRecHits, 1);
     trackemc5 = info.nXnEnergy(TrackDetMatchInfo::EcalRecHits, 2);
     trackhac1 = info.nXnEnergy(TrackDetMatchInfo::HcalRecHits, 0);
     trackhac3 = info.nXnEnergy(TrackDetMatchInfo::HcalRecHits, 1);
     trackhac5 = info.nXnEnergy(TrackDetMatchInfo::HcalRecHits, 2);
 
     if (trackhac3 < 5.)
       continue;
 
     double etaecal = info.trkGlobPosAtEcal.eta();
     double phiecal = info.trkGlobPosAtEcal.phi();
 
     maxPNearby = -10;
     dist = 50;
     for (const auto& track1 : tracks) {
       if (&track == &track1) {
         continue;
       }
       TrackDetMatchInfo info1 = trackAssociator_.associate(iEvent, iSetup, track1, parameters_);
       double etaecal1 = info1.trkGlobPosAtEcal.eta();
       double phiecal1 = info1.trkGlobPosAtEcal.phi();
 
       if (etaecal1 == 0 && phiecal1 == 0)
         continue;
 
       double ecDist = getDistInCM(etaecal, phiecal, etaecal1, phiecal1);
 
       if (ecDist < 40.) {
         //calculate maximum P and sum P near seed track
         if (track1.p() > maxPNearby) {
           maxPNearby = track1.p();
           dist = ecDist;
         }
 
         //apply loose isolation criteria
         if (track1.p() > 5.) {
           noChargedTracks = false;
           break;
         }
       }
     }
     EnergyIn = 0;
     EnergyOut = 0;
     if (noChargedTracks) {
       for (std::vector<EcalRecHit>::const_iterator ehit = tmpEcalRecHitCollection->begin();
            ehit != tmpEcalRecHitCollection->end();
            ehit++) {
         // R-scheme of ECAL CLUSTERIZATION
         const GlobalPoint& posH = geo->getPosition((*ehit).detid());
         double phihit = posH.phi();
         double etahit = posH.eta();
 
         double dHitCM = getDistInCM(etaecal, phiecal, etahit, phihit);
 
         if (dHitCM < 9.0) {
           EnergyIn += ehit->energy();
         }
         if (dHitCM > 15.0 && dHitCM < 35.0) {
           EnergyOut += ehit->energy();
         }
       }
 
       treeMemPtr_->fillVariables(track.charge(),
                                  track.innerOk(),
                                  track.outerRadius(),
                                  track.numberOfValidHits(),
                                  track.numberOfLostHits(),
                                  track.chi2(),
                                  track.normalizedChi2(),
                                  track.p(),
                                  track.pt(),
                                  track.ptError(),
                                  track.theta(),
                                  track.eta(),
                                  track.phi(),
                                  trackemc1,
                                  trackemc3,
                                  trackemc5,
                                  trackhac1,
                                  trackhac3,
                                  trackhac5,
                                  maxPNearby,
                                  dist,
                                  EnergyIn,
                                  EnergyOut);
 
       tree_->Fill();
     }
   }
 }

◆ endJob()

void EopTreeWriter::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 281 of file EopTreeWriter.cc.

References treeMemPtr_.

Referenced by o2olib.O2ORunMgr::executeJob().

                            {
   delete treeMemPtr_;
   treeMemPtr_ = nullptr;
 }

◆ fillDescriptions()

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

static

Definition at line 328 of file EopTreeWriter.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, TrackAssociatorParameters::fillPSetDescription(), and ProducerED_cfi::InputTag.

 {
   edm::ParameterSetDescription desc;
   desc.setComment("Generate tree for Tracker Alignment E/p validation");
   desc.add<edm::InputTag>("src", edm::InputTag("generalTracks"));
 
   // track association (take defaults)
   edm::ParameterSetDescription psd0;
   TrackAssociatorParameters::fillPSetDescription(psd0);
   desc.add<edm::ParameterSetDescription>("TrackAssociatorParameters", psd0);
 
   descriptions.addWithDefaultLabel(desc);
 }

◆ getDistInCM()

double EopTreeWriter::getDistInCM	(	double	eta1,
		double	phi1,
		double	eta2,
		double	phi2
	)

private

Definition at line 287 of file EopTreeWriter.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), funct::cos(), SiPixelRawToDigiRegional_cfi::deltaPhi, HGC3DClusterGenMatchSelector_cfi::dR, HLT_2024v14_cff::eta1, HLT_2024v14_cff::eta2, JetChargeProducer_cfi::exp, M_PI, mathSSE::sqrt(), and funct::tan().

Referenced by analyze().

                                                                                     {
   //*************************************************************
 
   static constexpr float EEBoundary = 1.479;  // psedo-rapidity
   static constexpr float EBRadius = 129;      // in cm
   static constexpr float EEIPdis = 317;       // in cm
 
   double deltaPhi = phi1 - phi2;
   while (deltaPhi > M_PI)
     deltaPhi -= 2 * M_PI;
   while (deltaPhi <= -M_PI)
     deltaPhi += 2 * M_PI;
   double dR;
   // double Rec;
   double theta1 = 2 * atan(exp(-eta1));
   double theta2 = 2 * atan(exp(-eta2));
   double cotantheta1;
   if (cos(theta1) == 0)
     cotantheta1 = 0;
   else
     cotantheta1 = 1 / tan(theta1);
   double cotantheta2;
   if (cos(theta2) == 0)
     cotantheta2 = 0;
   else
     cotantheta2 = 1 / tan(theta2);
   // if (fabs(eta1)<1.479) Rec=129; //radius of ECAL barrel
   // else Rec=317; //distance from IP to ECAL endcap
   //|vect| times tg of acos(scalar product)
   // dR=fabs((Rec/sin(theta1))*tan(acos(sin(theta1)*sin(theta2)*(sin(phi1)*sin(phi2)+cos(phi1)*cos(phi2))+cos(theta1)*cos(theta2))));
 
   if (fabs(eta1) < EEBoundary) {
     dR = EBRadius * sqrt((cotantheta1 - cotantheta2) * (cotantheta1 - cotantheta2) + deltaPhi * deltaPhi);
   } else {
     dR = EEIPdis *
          sqrt(tan(theta1) * tan(theta1) + tan(theta2) * tan(theta2) - 2 * tan(theta1) * tan(theta2) * cos(deltaPhi));
   }
   return dR;
 }

Member Data Documentation

◆ ecalRecHitTokenAlCaToken_

edm::EDGetTokenT<EcalRecHitCollection> EopTreeWriter::ecalRecHitTokenAlCaToken_

private

Definition at line 86 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ ecalRecHitTokenEBRecoToken_

edm::EDGetTokenT<EcalRecHitCollection> EopTreeWriter::ecalRecHitTokenEBRecoToken_

private

Definition at line 87 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ ecalRecHitTokenEERecoToken_

edm::EDGetTokenT<EcalRecHitCollection> EopTreeWriter::ecalRecHitTokenEERecoToken_

private

Definition at line 88 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ fs_

edm::Service<TFileService> EopTreeWriter::fs_

private

Definition at line 78 of file EopTreeWriter.cc.

Referenced by EopTreeWriter().

◆ geometryToken_

edm::ESGetToken<CaloGeometry, CaloGeometryRecord> EopTreeWriter::geometryToken_

private

Definition at line 76 of file EopTreeWriter.cc.

Referenced by analyze().

◆ isoPixelTkToken_

edm::EDGetTokenT<reco::IsolatedPixelTrackCandidateCollection> EopTreeWriter::isoPixelTkToken_

private

Definition at line 85 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ parameters_

TrackAssociatorParameters EopTreeWriter::parameters_

private

Definition at line 82 of file EopTreeWriter.cc.

Referenced by analyze(), Mixins._TypedParameterizable::clone(), Types.PSet::clone(), Types.EDAlias::clone(), Mixins._TypedParameterizable::copy(), and EopTreeWriter().

◆ src_

edm::InputTag EopTreeWriter::src_

private

Definition at line 75 of file EopTreeWriter.cc.

Referenced by EopTreeWriter().

◆ theTrackCollectionToken_

edm::EDGetTokenT<reco::TrackCollection> EopTreeWriter::theTrackCollectionToken_

private

Definition at line 84 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ trackAssociator_

TrackDetectorAssociator EopTreeWriter::trackAssociator_

private

Definition at line 81 of file EopTreeWriter.cc.

Referenced by analyze().

◆ tree_

TTree* EopTreeWriter::tree_

private

Definition at line 79 of file EopTreeWriter.cc.

Referenced by analyze(), and EopTreeWriter().

◆ treeMemPtr_

EopVariables* EopTreeWriter::treeMemPtr_

private

Definition at line 80 of file EopTreeWriter.cc.

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

Private Attributes
edm::EDGetTokenT< EcalRecHitCollection >	ecalRecHitTokenAlCaToken_

edm::EDGetTokenT< EcalRecHitCollection >	ecalRecHitTokenEBRecoToken_

edm::EDGetTokenT< EcalRecHitCollection >	ecalRecHitTokenEERecoToken_

edm::Service< TFileService >	fs_

edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geometryToken_

edm::EDGetTokenT< reco::IsolatedPixelTrackCandidateCollection >	isoPixelTkToken_

TrackAssociatorParameters	parameters_

edm::InputTag	src_

edm::EDGetTokenT< reco::TrackCollection >	theTrackCollectionToken_

TrackDetectorAssociator	trackAssociator_

TTree *	tree_

EopVariables *	treeMemPtr_

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ EopTreeWriter()

◆ ~EopTreeWriter()

Member Function Documentation

◆ analyze()

◆ endJob()

◆ fillDescriptions()

◆ getDistInCM()

Member Data Documentation

◆ ecalRecHitTokenAlCaToken_

◆ ecalRecHitTokenEBRecoToken_

◆ ecalRecHitTokenEERecoToken_

◆ fs_

◆ geometryToken_

◆ isoPixelTkToken_

◆ parameters_

◆ src_

◆ theTrackCollectionToken_

◆ trackAssociator_

◆ tree_

◆ treeMemPtr_