#include <AlignmentTwoBodyDecayTrackSelector.h>

Public Types
typedef std::vector< const reco::Track * >	Tracks

Public Member Functions
	AlignmentTwoBodyDecayTrackSelector (const edm::ParameterSet &cfg, edm::ConsumesCollector &iC)
	constructor More...

Tracks	select (const Tracks &tracks, const edm::Event &iEvent, const edm::EventSetup &iSetup)
	select tracks More...

bool	useThisFilter ()
	returns if any of the Filters is used. More...

	~AlignmentTwoBodyDecayTrackSelector ()
	destructor More...

Private Member Functions
bool	checkAcoplanarity (const reco::Track trk1, const reco::Track trk2) const
	checks if the [cands] are acoplanar (returns empty set if not) More...

bool	checkCharge (const reco::Track trk1, const reco::Track trk2=nullptr) const
	checks if the mother has charge = [theCharge] More...

Tracks	checkMass (const Tracks &cands) const
	checks if the mass of the mother is in the mass region More...

bool	checkMETAcoplanarity (const reco::Track trk, const reco::CaloMET met) const
	checks if [cands] contains a acoplanar track w.r.t missing ET (returns empty set if not) More...

Tracks	checkMETMass (const Tracks &cands, const edm::Event &iEvent) const
	checks if the mass of the mother is in the mass region adding missing E_T More...

void	printTracks (const Tracks &col) const
	print Information on Track-Collection More...

Private Attributes
bool	secThrBool

double	theAcoplanarDistance

bool	theAcoplanarityFilterSwitch

unsigned int	theCandNumber

int	theCharge

bool	theChargeSwitch

double	theDaughterMass

bool	theMassrangeSwitch
	private data members More...

double	theMaxMass

double	theMinMass

bool	theMissingETSwitch

edm::EDGetTokenT< reco::CaloMETCollection >	theMissingETToken

double	thesecThr

bool	theUnsignedSwitch

Detailed Description

Definition at line 20 of file AlignmentTwoBodyDecayTrackSelector.h.

Member Typedef Documentation

◆ Tracks

typedef std::vector<const reco::Track*> AlignmentTwoBodyDecayTrackSelector::Tracks

Definition at line 22 of file AlignmentTwoBodyDecayTrackSelector.h.

Constructor & Destructor Documentation

◆ AlignmentTwoBodyDecayTrackSelector()

AlignmentTwoBodyDecayTrackSelector::AlignmentTwoBodyDecayTrackSelector	(	const edm::ParameterSet &	cfg,
		edm::ConsumesCollector &	iC
	)

constructor

Definition at line 24 of file AlignmentTwoBoyDecayTrackSelector.cc.

References funct::abs(), looper::cfg, edm::ConsumesCollector::consumes(), LogDebug, and AlignmentTrackSelector_cfi::theCharge.

                                                                                                  {
   LogDebug("Alignment") << "> applying two body decay Trackfilter ...";
   theMassrangeSwitch = cfg.getParameter<bool>("applyMassrangeFilter");
   if (theMassrangeSwitch) {
     theMinMass = cfg.getParameter<double>("minXMass");
     theMaxMass = cfg.getParameter<double>("maxXMass");
     theDaughterMass = cfg.getParameter<double>("daughterMass");
     theCandNumber = cfg.getParameter<unsigned int>("numberOfCandidates");  //Number of candidates to keep
     secThrBool = cfg.getParameter<bool>("applySecThreshold");
     thesecThr = cfg.getParameter<double>("secondThreshold");
     LogDebug("Alignment") << ">  Massrange min,max         :   " << theMinMass << "," << theMaxMass
                           << "\n>  Mass of daughter Particle :   " << theDaughterMass;
 
   } else {
     theMinMass = 0;
     theMaxMass = 0;
     theDaughterMass = 0;
   }
   theChargeSwitch = cfg.getParameter<bool>("applyChargeFilter");
   if (theChargeSwitch) {
     theCharge = cfg.getParameter<int>("charge");
     theUnsignedSwitch = cfg.getParameter<bool>("useUnsignedCharge");
     if (theUnsignedSwitch)
       theCharge = std::abs(theCharge);
     LogDebug("Alignment") << ">  Desired Charge, unsigned: " << theCharge << " , " << theUnsignedSwitch;
   } else {
     theCharge = 0;
     theUnsignedSwitch = true;
   }
   theMissingETSwitch = cfg.getParameter<bool>("applyMissingETFilter");
   if (theMissingETSwitch) {
     edm::InputTag theMissingETSource = cfg.getParameter<InputTag>("missingETSource");
     theMissingETToken = iC.consumes<reco::CaloMETCollection>(theMissingETSource);
     LogDebug("Alignment") << ">  missing Et Source: " << theMissingETSource;
   }
   theAcoplanarityFilterSwitch = cfg.getParameter<bool>("applyAcoplanarityFilter");
   if (theAcoplanarityFilterSwitch) {
     theAcoplanarDistance = cfg.getParameter<double>("acoplanarDistance");
     LogDebug("Alignment") << ">  Acoplanar Distance: " << theAcoplanarDistance;
   }
 }

◆ ~AlignmentTwoBodyDecayTrackSelector()

AlignmentTwoBodyDecayTrackSelector::~AlignmentTwoBodyDecayTrackSelector ( )

destructor

Definition at line 69 of file AlignmentTwoBoyDecayTrackSelector.cc.

69 {}

Member Function Documentation

◆ checkAcoplanarity()

bool AlignmentTwoBodyDecayTrackSelector::checkAcoplanarity	(	const reco::Track *	trk1,
		const reco::Track *	trk2
	)		const

private

checks if the [cands] are acoplanar (returns empty set if not)

Definition at line 252 of file AlignmentTwoBoyDecayTrackSelector.cc.

References SiPixelRawToDigiRegional_cfi::deltaPhi, M_PI, and reco::TrackBase::phi().

                                                                                                              {
   if (fabs(deltaPhi(trk1->phi(), trk2->phi() - M_PI)) < theAcoplanarDistance)
     return true;
   return false;
 }

◆ checkCharge()

bool AlignmentTwoBodyDecayTrackSelector::checkCharge	(	const reco::Track *	trk1,
		const reco::Track *	trk2 = `nullptr`
	)		const

private

checks if the mother has charge = [theCharge]

Definition at line 240 of file AlignmentTwoBoyDecayTrackSelector.cc.

References funct::abs(), reco::TrackBase::charge(), and AlignmentTrackSelector_cfi::theCharge.

                                                                                                        {
   int sumCharge = trk1->charge();
   if (trk2)
     sumCharge += trk2->charge();
   if (theUnsignedSwitch)
     sumCharge = std::abs(sumCharge);
   if (sumCharge == theCharge)
     return true;
   return false;
 }

◆ checkMass()

AlignmentTwoBodyDecayTrackSelector::Tracks AlignmentTwoBodyDecayTrackSelector::checkMass ( const Tracks & cands ) const

private

checks if the mass of the mother is in the mass region

checks if the mass of the X is in the mass region

Definition at line 95 of file AlignmentTwoBoyDecayTrackSelector.cc.

References a, b, HLT_2022v12_cff::cands, PA_ZEESkim_cff::checkCharge, mps_fire::i, LogDebug, mps_fire::result, jetUpdater_cfi::sort, and mathSSE::sqrt().

                                                                                                                 {
   Tracks result;
 
   LogDebug("Alignment") << ">  cands size : " << cands.size();
 
   if (cands.size() < 2)
     return result;
 
   TLorentzVector track0;
   TLorentzVector track1;
   TLorentzVector mother;
   typedef pair<const reco::Track*, const reco::Track*> constTrackPair;
   typedef pair<double, constTrackPair> candCollectionItem;
   vector<candCollectionItem> candCollection;
 
   for (unsigned int iCand = 0; iCand < cands.size(); iCand++) {
     track0.SetXYZT(cands.at(iCand)->px(),
                    cands.at(iCand)->py(),
                    cands.at(iCand)->pz(),
                    sqrt(cands.at(iCand)->p() * cands.at(iCand)->p() + theDaughterMass * theDaughterMass));
 
     for (unsigned int jCand = iCand + 1; jCand < cands.size(); jCand++) {
       track1.SetXYZT(cands.at(jCand)->px(),
                      cands.at(jCand)->py(),
                      cands.at(jCand)->pz(),
                      sqrt(cands.at(jCand)->p() * cands.at(jCand)->p() + theDaughterMass * theDaughterMass));
       if (secThrBool == true && track1.Pt() < thesecThr && track0.Pt() < thesecThr)
         continue;
       mother = track0 + track1;
 
       const reco::Track* trk1 = cands.at(iCand);
       const reco::Track* trk2 = cands.at(jCand);
 
       bool correctCharge = true;
       if (theChargeSwitch)
         correctCharge = this->checkCharge(trk1, trk2);
 
       bool acoplanarTracks = true;
       if (theAcoplanarityFilterSwitch)
         acoplanarTracks = this->checkAcoplanarity(trk1, trk2);
 
       if (mother.M() > theMinMass && mother.M() < theMaxMass && correctCharge && acoplanarTracks) {
         candCollection.push_back(candCollectionItem(mother.Pt(), constTrackPair(trk1, trk2)));
       }
     }
   }
 
   if (candCollection.empty())
     return result;
 
   sort(candCollection.begin(), candCollection.end(), [](auto& a, auto& b) { return a.first > b.first; });
 
   std::map<const reco::Track*, unsigned int> uniqueTrackIndex;
   std::map<const reco::Track*, unsigned int>::iterator it;
   for (unsigned int i = 0; i < candCollection.size() && i < theCandNumber; i++) {
     constTrackPair& trackPair = candCollection[i].second;
 
     it = uniqueTrackIndex.find(trackPair.first);
     if (it == uniqueTrackIndex.end()) {
       result.push_back(trackPair.first);
       uniqueTrackIndex[trackPair.first] = i;
     }
 
     it = uniqueTrackIndex.find(trackPair.second);
     if (it == uniqueTrackIndex.end()) {
       result.push_back(trackPair.second);
       uniqueTrackIndex[trackPair.second] = i;
     }
   }
 
   return result;
 }

◆ checkMETAcoplanarity()

bool AlignmentTwoBodyDecayTrackSelector::checkMETAcoplanarity	(	const reco::Track *	trk,
		const reco::CaloMET *	met
	)		const

private

checks if [cands] contains a acoplanar track w.r.t missing ET (returns empty set if not)

checks if the [cands] are acoplanar (returns empty set if not)

Definition at line 259 of file AlignmentTwoBoyDecayTrackSelector.cc.

References SiPixelRawToDigiRegional_cfi::deltaPhi, M_PI, BTaggingMonitor_cfi::met, and reco::TrackBase::phi().

                                                                                                                  {
   if (fabs(deltaPhi(trk1->phi(), met->phi() - M_PI)) < theAcoplanarDistance)
     return true;
   return false;
 }

◆ checkMETMass()

AlignmentTwoBodyDecayTrackSelector::Tracks AlignmentTwoBodyDecayTrackSelector::checkMETMass	(	const Tracks &	cands,
		const edm::Event &	iEvent
	)		const

private

checks if the mass of the mother is in the mass region adding missing E_T

checks if the mass of the X is in the mass region adding missing E_T

Definition at line 169 of file AlignmentTwoBoyDecayTrackSelector.cc.

References a, b, HLT_2022v12_cff::cands, PA_ZEESkim_cff::checkCharge, mps_fire::i, iEvent, edm::HandleBase::isValid(), LogDebug, BTaggingMonitor_cfi::met, mps_fire::result, edm::second(), jetUpdater_cfi::sort, mathSSE::sqrt(), and HLT_2022v12_cff::track.

                                                        {
   Tracks result;
 
   LogDebug("Alignment") << ">  cands size : " << cands.size();
 
   if (cands.empty())
     return result;
 
   TLorentzVector track;
   TLorentzVector met4;
   TLorentzVector mother;
 
   Handle<reco::CaloMETCollection> missingET;
   iEvent.getByToken(theMissingETToken, missingET);
   if (!missingET.isValid()) {
     LogError("Alignment") << "@SUB=AlignmentTwoBodyDecayTrackSelector::checkMETMass"
                           << ">  could not optain missingET Collection!";
     return result;
   }
 
   typedef pair<double, const reco::Track*> candCollectionItem;
   vector<candCollectionItem> candCollection;
 
   for (reco::CaloMETCollection::const_iterator itMET = missingET->begin(); itMET != missingET->end(); ++itMET) {
     met4.SetXYZT((*itMET).px(), (*itMET).py(), (*itMET).pz(), (*itMET).p());
 
     for (unsigned int iCand = 0; iCand < cands.size(); iCand++) {
       track.SetXYZT(cands.at(iCand)->px(),
                     cands.at(iCand)->py(),
                     cands.at(iCand)->pz(),
                     sqrt(cands.at(iCand)->p() * cands.at(iCand)->p() + theDaughterMass * theDaughterMass));
 
       mother = track + met4;
 
       const reco::Track* trk = cands.at(iCand);
       const reco::CaloMET* met = &(*itMET);
 
       bool correctCharge = true;
       if (theChargeSwitch)
         correctCharge = this->checkCharge(trk);
 
       bool acoplanarTracks = true;
       if (theAcoplanarityFilterSwitch)
         acoplanarTracks = this->checkMETAcoplanarity(trk, met);
 
       if (mother.M() > theMinMass && mother.M() < theMaxMass && correctCharge && acoplanarTracks) {
         candCollection.push_back(candCollectionItem(mother.Pt(), trk));
       }
     }
   }
 
   if (candCollection.empty())
     return result;
 
   sort(candCollection.begin(), candCollection.end(), [](auto& a, auto& b) { return a.first > b.first; });
 
   std::map<const reco::Track*, unsigned int> uniqueTrackIndex;
   std::map<const reco::Track*, unsigned int>::iterator it;
   for (unsigned int i = 0; i < candCollection.size() && i < theCandNumber; i++) {
     it = uniqueTrackIndex.find(candCollection[i].second);
     if (it == uniqueTrackIndex.end()) {
       result.push_back(candCollection[i].second);
       uniqueTrackIndex[candCollection[i].second] = i;
     }
   }
 
   return result;
 }

◆ printTracks()

void AlignmentTwoBodyDecayTrackSelector::printTracks ( const Tracks & col ) const

private

print Information on Track-Collection

Definition at line 268 of file AlignmentTwoBoyDecayTrackSelector.cc.

References cuy::col, submitPVResolutionJobs::count, and LogDebug.

                                                                             {
   int count = 0;
   LogDebug("Alignment") << ">......................................";
   for (Tracks::const_iterator it = col.begin(); it < col.end(); ++it, ++count) {
     LogDebug("Alignment") << ">  Track No. " << count << ": p = (" << (*it)->px() << "," << (*it)->py() << ","
                           << (*it)->pz() << ")\n"
                           << ">                        pT = " << (*it)->pt() << " eta = " << (*it)->eta()
                           << " charge = " << (*it)->charge();
   }
   LogDebug("Alignment") << ">......................................";
 }

◆ select()

AlignmentTwoBodyDecayTrackSelector::Tracks AlignmentTwoBodyDecayTrackSelector::select	(	const Tracks &	tracks,
		const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

select tracks

Definition at line 78 of file AlignmentTwoBoyDecayTrackSelector.cc.

References iEvent, LogDebug, mps_fire::result, and tracks.

Referenced by TrackConfigSelector::select().

                                                                                                                    {
   Tracks result = tracks;
 
   if (theMassrangeSwitch) {
     if (theMissingETSwitch)
       result = checkMETMass(result, iEvent);
     else
       result = checkMass(result);
   }
 
   LogDebug("Alignment") << ">  TwoBodyDecay tracks all,kept: " << tracks.size() << "," << result.size();
   return result;
 }

◆ useThisFilter()

bool AlignmentTwoBodyDecayTrackSelector::useThisFilter ( )

returns if any of the Filters is used.

Definition at line 72 of file AlignmentTwoBoyDecayTrackSelector.cc.

Referenced by TrackConfigSelector::TrackConfigSelector().

                                                        {
   return theMassrangeSwitch || theChargeSwitch || theAcoplanarityFilterSwitch;
 }

Member Data Documentation

◆ secThrBool

bool AlignmentTwoBodyDecayTrackSelector::secThrBool

private

Definition at line 59 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theAcoplanarDistance

double AlignmentTwoBodyDecayTrackSelector::theAcoplanarDistance

private

Definition at line 67 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theAcoplanarityFilterSwitch

bool AlignmentTwoBodyDecayTrackSelector::theAcoplanarityFilterSwitch

private

Definition at line 53 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theCandNumber

unsigned int AlignmentTwoBodyDecayTrackSelector::theCandNumber

private

Definition at line 58 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theCharge

int AlignmentTwoBodyDecayTrackSelector::theCharge

private

Definition at line 62 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theChargeSwitch

bool AlignmentTwoBodyDecayTrackSelector::theChargeSwitch

private

Definition at line 51 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theDaughterMass

double AlignmentTwoBodyDecayTrackSelector::theDaughterMass

private

Definition at line 57 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theMassrangeSwitch

bool AlignmentTwoBodyDecayTrackSelector::theMassrangeSwitch

private

private data members

Definition at line 50 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theMaxMass

double AlignmentTwoBodyDecayTrackSelector::theMaxMass

private

Definition at line 56 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theMinMass

double AlignmentTwoBodyDecayTrackSelector::theMinMass

private

Definition at line 55 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theMissingETSwitch

bool AlignmentTwoBodyDecayTrackSelector::theMissingETSwitch

private

Definition at line 52 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theMissingETToken

edm::EDGetTokenT<reco::CaloMETCollection> AlignmentTwoBodyDecayTrackSelector::theMissingETToken

private

Definition at line 65 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ thesecThr

double AlignmentTwoBodyDecayTrackSelector::thesecThr

private

Definition at line 60 of file AlignmentTwoBodyDecayTrackSelector.h.

◆ theUnsignedSwitch

bool AlignmentTwoBodyDecayTrackSelector::theUnsignedSwitch

private

Definition at line 63 of file AlignmentTwoBodyDecayTrackSelector.h.

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ Tracks

Constructor & Destructor Documentation

◆ AlignmentTwoBodyDecayTrackSelector()

◆ ~AlignmentTwoBodyDecayTrackSelector()

Member Function Documentation

◆ checkAcoplanarity()

◆ checkCharge()

◆ checkMass()

◆ checkMETAcoplanarity()

◆ checkMETMass()

◆ printTracks()

◆ select()

◆ useThisFilter()

Member Data Documentation

◆ secThrBool

◆ theAcoplanarDistance

◆ theAcoplanarityFilterSwitch

◆ theCandNumber

◆ theCharge

◆ theChargeSwitch

◆ theDaughterMass

◆ theMassrangeSwitch

◆ theMaxMass

◆ theMinMass

◆ theMissingETSwitch

◆ theMissingETToken

◆ thesecThr

◆ theUnsignedSwitch