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

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_2024v12_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_2024v12_cff::eta1, HLT_2024v12_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_