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

Inheritance diagram for EopTreeWriter:

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

	~EopTreeWriter () override

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

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	beginJob () override

void	endJob () override

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

Private Attributes
edm::Service< TFileService >	fs_

edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geometryToken_

TrackAssociatorParameters	parameters_

edm::InputTag	src_

TrackDetectorAssociator	trackAssociator_

TTree *	tree_

EopVariables *	treeMemPtr_

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

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag) noexcept

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 67 of file EopTreeWriter.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 101 of file EopTreeWriter.cc.

References edm::EDConsumerBase::consumesCollector(), fs_, edm::ParameterSet::getParameter(), TFileService::kSharedResource, TrackAssociatorParameters::loadParameters(), TFileService::make(), parameters_, 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!
 }

EopTreeWriter::~EopTreeWriter ( )

override

Definition at line 116 of file EopTreeWriter.cc.

                               {
   // do anything here that needs to be done at destruction time
   // (e.g. close files, deallocate resources etc.)
 }

Member Function Documentation

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 126 of file EopTreeWriter.cc.

References TrackDetectorAssociator::associate(), TrackAssociatorParameters::bFieldToken, TrackDetMatchInfo::EcalRecHits, PV3DBase< T, PVType, FrameType >::eta(), Exception, EopVariables::fillVariables(), geometryToken_, edm::Event::getByLabel(), edm::EventSetup::getData(), getDistInCM(), TrackDetectorAssociator::getFreeTrajectoryState(), TrackDetMatchInfo::HcalRecHits, mps_fire::i, info(), TrackDetMatchInfo::nXnEnergy(), parameters_, PV3DBase< T, PVType, FrameType >::phi(), src_, HLT_FULL_cff::track, trackAssociator_, tracks, tree_, treeMemPtr_, TrackDetMatchInfo::trkGlobPosAtEcal, TrackDetectorAssociator::useDefaultPropagator(), and TrackAssociatorParameters::useMuon.

                                                                                {
   using namespace edm;
 
   // get geometry
   const CaloGeometry* geo = &iSetup.getData(geometryToken_);
   //    const CaloSubdetectorGeometry* towerGeometry =
   //      geo->getSubdetectorGeometry(DetId::Calo, CaloTowerDetId::SubdetId);
 
   // temporary collection of EB+EE recHits
   std::unique_ptr<EcalRecHitCollection> tmpEcalRecHitCollection(new EcalRecHitCollection);
   std::vector<edm::InputTag> ecalLabels_;
 
   edm::Handle<EcalRecHitCollection> tmpEc;
   bool ecalInAlca = iEvent.getByLabel(edm::InputTag("IsoProd", "IsoTrackEcalRecHitCollection"), tmpEc);
   bool ecalInReco = iEvent.getByLabel(edm::InputTag("ecalRecHit", "EcalRecHitsEB"), tmpEc) &&
                     iEvent.getByLabel(edm::InputTag("ecalRecHit", "EcalRecHitsEE"), tmpEc);
   if (ecalInAlca)
     ecalLabels_.push_back(edm::InputTag("IsoProd", "IsoTrackEcalRecHitCollection"));
   else if (ecalInReco) {
     ecalLabels_.push_back(edm::InputTag("ecalRecHit", "EcalRecHitsEB"));
     ecalLabels_.push_back(edm::InputTag("ecalRecHit", "EcalRecHitsEE"));
   } else
     throw cms::Exception("MissingProduct", "can not find EcalRecHits");
 
   std::vector<edm::InputTag>::const_iterator i;
   for (i = ecalLabels_.begin(); i != ecalLabels_.end(); i++) {
     edm::Handle<EcalRecHitCollection> ec;
     iEvent.getByLabel(*i, ec);
     for (EcalRecHitCollection::const_iterator recHit = (*ec).begin(); recHit != (*ec).end(); ++recHit) {
       tmpEcalRecHitCollection->push_back(*recHit);
     }
   }
 
   edm::Handle<reco::TrackCollection> tracks;
   iEvent.getByLabel(src_, tracks);
 
   edm::Handle<reco::IsolatedPixelTrackCandidateCollection> isoPixelTracks;
   edm::Handle<reco::IsolatedPixelTrackCandidateCollection> tmpPix;
   bool pixelInAlca = iEvent.getByLabel(edm::InputTag("IsoProd", "HcalIsolatedTrackCollection"), tmpPix);
   if (pixelInAlca)
     iEvent.getByLabel(edm::InputTag("IsoProd", "HcalIsolatedTrackCollection"), isoPixelTracks);
 
   Double_t trackemc1;
   Double_t trackemc3;
   Double_t trackemc5;
   Double_t trackhac1;
   Double_t trackhac3;
   Double_t trackhac5;
   Double_t maxPNearby;
   Double_t dist;
   Double_t EnergyIn;
   Double_t EnergyOut;
 
   parameters_.useMuon = false;
 
   if (pixelInAlca)
     if (isoPixelTracks->empty())
       return;
 
   for (reco::TrackCollection::const_iterator track = tracks->begin(); track != tracks->end(); ++track) {
     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 (reco::TrackCollection::const_iterator track1 = tracks->begin(); track1 != tracks->end(); track1++) {
       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();
     }
   }
 }

void EopTreeWriter::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 291 of file EopTreeWriter.cc.

291 {}

void EopTreeWriter::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 294 of file EopTreeWriter.cc.

References treeMemPtr_.

Referenced by o2olib.O2ORunMgr::executeJob().

                            {
   delete treeMemPtr_;
   treeMemPtr_ = nullptr;
 }

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

private

Definition at line 299 of file EopTreeWriter.cc.

References funct::cos(), srCondWrite_cfg::deltaPhi, funct::exp(), Pi, mathSSE::sqrt(), and funct::tan().

Referenced by analyze().

                                                                                     {
   double deltaPhi = phi1 - phi2;
   while (deltaPhi > TMath::Pi())
     deltaPhi -= 2 * TMath::Pi();
   while (deltaPhi <= -TMath::Pi())
     deltaPhi += 2 * TMath::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) < 1.479)
     dR = 129 * sqrt((cotantheta1 - cotantheta2) * (cotantheta1 - cotantheta2) + deltaPhi * deltaPhi);
   else
     dR = 317 *
          sqrt(tan(theta1) * tan(theta1) + tan(theta2) * tan(theta2) - 2 * tan(theta1) * tan(theta2) * cos(deltaPhi));
   return dR;
 }