#include <RecoTracker/TkSeedGenerator/src/ConversionSeedFilter.cc>

Inheritance diagram for ConversionSeedFilter:

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

	~ConversionSeedFilter () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	getKine (const TrajectoryStateOnSurface &tsos, double *vars)

TrajectoryStateOnSurface	getTSOS (const TrajectorySeed &ts)

TrajectoryStateOnSurface	getTSOS (const reco::Track &tk)

TrajectoryStateOnSurface	getTSOS (const Trajectory &tj, const TrajectorySeed &ts)

bool	isCompatible (double vars1, double vars2)

void	produce (edm::Event &, const edm::EventSetup &) override

void	SearchAmongSeeds (const TrajectorySeedCollection pInPos, const TrajectorySeedCollection pInNeg, TrajectorySeedCollection &selectedColl, std::vector< bool > &idxPosColl1, std::vector< bool > &idxPosColl2)

void	SearchAmongTracks (const TrajectorySeedCollection pInSeed, const reco::TrackCollection pInTk, TrajectorySeedCollection &selectedColl, std::vector< bool > &idxPosColl)

void	SearchAmongTrajectories (const TrajectorySeedCollection pInSeed, const Trajectory InTj, TrajectorySeedCollection &selectedColl, std::vector< bool > &idxPosColl)

Private Attributes
double	deltaCotThetaCut

double	deltaPhiCut

double	deltaRCut

double	deltaZCut

edm::EDGetTokenT< TrajectorySeedCollection >	inputCollSeedNeg

edm::EDGetTokenT< TrajectorySeedCollection >	inputCollSeedPos

edm::EDGetTokenT< TrajTrackAssociationCollection >	inputTrajectory

uint32_t	maxInputSeeds

bool	takeAll

edm::ESHandle< TrackerGeometry >	theG

edm::ESHandle< MagneticField >	theMF

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

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

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

Definition at line 41 of file ConversionSeedFilter.cc.

Constructor & Destructor Documentation

ConversionSeedFilter::ConversionSeedFilter ( const edm::ParameterSet & cfg )

explicit

Definition at line 69 of file ConversionSeedFilter.cc.

                                                                     : 
   //inputCollTkPos(cfg.getParameter<edm::InputTag>("tkCollectionPos")),
   inputCollSeedPos(consumes<TrajectorySeedCollection>(cfg.getParameter<edm::InputTag>("seedCollectionPos"))),
   //inputCollTkNeg(cfg.getParameter<edm::InputTag>("tkCollectionNeg")),
   inputCollSeedNeg(consumes<TrajectorySeedCollection>(cfg.getParameter<edm::InputTag>("seedCollectionNeg"))),
   inputTrajectory(consumes<TrajTrackAssociationCollection>(cfg.getParameter<edm::InputTag>("inputTrajectory"))),
   deltaPhiCut(cfg.getParameter<double>("deltaPhiCut")),
   deltaCotThetaCut(cfg.getParameter<double>("deltaCotThetaCut")),
   deltaRCut(cfg.getParameter<double>("deltaRCut")),
   deltaZCut(cfg.getParameter<double>("deltaZCut")),
   maxInputSeeds(cfg.getParameter<uint32_t>("maxInputSeeds")),
   takeAll(cfg.getParameter<bool>("takeAll"))
 {
   produces<TrajectorySeedCollection>();
 }

ConversionSeedFilter::~ConversionSeedFilter ( )

override

Definition at line 86 of file ConversionSeedFilter.cc.

86 {}

Member Function Documentation

void ConversionSeedFilter::getKine	(	const TrajectoryStateOnSurface &	tsos,
		double *	vars
	)

private

Definition at line 270 of file ConversionSeedFilter.cc.

References DEFINE_FWK_MODULE, TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), PV3DBase< T, PVType, FrameType >::perp(), PV3DBase< T, PVType, FrameType >::phi(), funct::tan(), PV3DBase< T, PVType, FrameType >::theta(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by isCompatible(), SearchAmongSeeds(), SearchAmongTracks(), and SearchAmongTrajectories().

                                                            {
 
   vars[0] = tsos.globalMomentum().phi();            //phi
   vars[1] = 2/tan(tsos.globalMomentum().theta()); //cotTheta
   vars[2] = tsos.globalPosition().perp();         //R
   vars[3] = tsos.globalPosition().z();            //Z
 }

TrajectoryStateOnSurface ConversionSeedFilter::getTSOS ( const TrajectorySeed & ts )

private

Definition at line 229 of file ConversionSeedFilter.cc.

References PTrajectoryStateOnDet::detId(), TrackerGeometry::idToDet(), edm::ESHandle< T >::product(), TrajectorySeed::startingState(), GeomDet::surface(), theG, theMF, and trajectoryStateTransform::transientState().

Referenced by getTSOS(), SearchAmongSeeds(), SearchAmongTracks(), and SearchAmongTrajectories().

                                  {
 
   PTrajectoryStateOnDet state = ts.startingState();
   DetId detId(state.detId());
   return trajectoryStateTransform::transientState( state, &(theG->idToDet(detId)->surface()), theMF.product());
 }

TrajectoryStateOnSurface ConversionSeedFilter::getTSOS ( const reco::Track & tk )

private

Definition at line 237 of file ConversionSeedFilter.cc.

References getTSOS(), trajectoryStateTransform::innerStateOnSurface(), edm::ESHandle< T >::product(), theG, and theMF.

                             {
   return trajectoryStateTransform::innerStateOnSurface( tk, *theG.product(), theMF.product());
 }

TrajectoryStateOnSurface ConversionSeedFilter::getTSOS	(	const Trajectory &	tj,
		const TrajectorySeed &	ts
	)

private

Definition at line 242 of file ConversionSeedFilter.cc.

References Trajectory::closestMeasurement(), PTrajectoryStateOnDet::detId(), TrackerGeometry::idToDet(), isCompatible(), AlCaHLTBitMon_ParallelJobs::p, PTrajectoryStateOnDet::parameters(), LocalTrajectoryParameters::position(), TrajectorySeed::startingState(), GeomDet::surface(), theG, Surface::toGlobal(), and TrajectoryMeasurement::updatedState().

                                                        {
 
   PTrajectoryStateOnDet state = ts.startingState();
   DetId detId(state.detId());
   GlobalPoint p = theG->idToDet(detId)->surface().toGlobal(state.parameters().position());
   return tj.closestMeasurement(p).updatedState();
 }

bool ConversionSeedFilter::isCompatible	(	double *	vars1,
		double *	vars2
	)

private

Definition at line 251 of file ConversionSeedFilter.cc.

References deltaCotThetaCut, reco::deltaPhi(), hiPixelPairStep_cff::deltaPhi, deltaPhiCut, boostedElectronIsolation_cff::deltaR, deltaRCut, deltaZCut, and getKine().

Referenced by getTSOS(), SearchAmongSeeds(), SearchAmongTracks(), and SearchAmongTrajectories().

                                           {
 
   double deltaPhi      = fabs(reco::deltaPhi(vars1[0],vars2[0])  );
   double deltaCotTheta = fabs(               vars1[1]-vars2[1]   );
   double deltaR        = fabs(               vars1[2]-vars2[2]   );
   double deltaZ        = fabs(               vars1[3]-vars2[3]   );
   /*
   std::cout << "\nnewCheck" << std::endl;
   if(deltaPhi>deltaPhiCut)             std::cout << "\nko deltaphi" << deltaPhi; 
   if(deltaCotTheta>deltaCotThetaCut )  std::cout << "\nko deltaCt " << deltaCotTheta; 
   if(deltaR>deltaRCut           )  std::cout << "\nko deltaR  " << deltaR; 
   if(deltaZ>deltaZCut               )  std::cout << "\nko deltaZ  " << deltaZ; 
   if(deltaPhi<deltaPhiCut && deltaCotTheta<deltaCotThetaCut && deltaR<deltaRCut && deltaZ<deltaZCut) std::cout << "\nok :)\n";
   */
   return deltaPhi<deltaPhiCut && deltaCotTheta<deltaCotThetaCut && deltaR<deltaRCut && deltaZ<deltaZCut;
     
 }

void ConversionSeedFilter::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 89 of file ConversionSeedFilter.cc.

References TrackValidation_cff::association, edm::AssociationMap< Tag >::begin(), reco::TrackBase::charge(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, edm::AssociationMap< Tag >::end(), edm::EventSetup::get(), edm::Event::getByToken(), inputCollSeedNeg, inputCollSeedPos, inputTrajectory, edm::HandleBase::isValid(), maxInputSeeds, eostools::move(), edm::Handle< T >::product(), edm::Event::put(), mps_fire::result, SearchAmongSeeds(), SearchAmongTrajectories(), edm::AssociationMap< Tag >::size(), takeAll, theG, theMF, and HiIsolationCommonParameters_cff::track.

                                                                                {
    using namespace edm;
    using namespace std;
    Handle<TrajectorySeedCollection> pInPos; iEvent.getByToken(inputCollSeedPos,pInPos);
    Handle<TrajectorySeedCollection> pInNeg; iEvent.getByToken(inputCollSeedNeg,pInNeg);
 
    // Handle<reco::TrackCollection> pInTkPos;  iEvent.getByLabel(inputCollTkPos,pInTkPos);
    // Handle<reco::TrackCollection> pInTkNeg;  iEvent.getByLabel(inputCollTkNeg,pInTkNeg);
 
    edm::Handle<TrajTrackAssociationCollection> trajTrackAssociations; iEvent.getByToken(inputTrajectory,trajTrackAssociations);
 
    iSetup.get<TrackerDigiGeometryRecord>().get(theG);
    iSetup.get<IdealMagneticFieldRecord>().get(theMF);  
 
    auto result = std::make_unique<TrajectorySeedCollection>();
 
    TrajectorySeedCollection selectedColl;
    
    if(takeAll){
      result->insert(result->end(),pInPos->begin(),pInPos->end());
      result->insert(result->end(),pInNeg->begin(),pInNeg->end());
    }else{
      edm::LogInfo("ConversionSeedFilter") << "takeAll " << takeAll ;
      if (pInPos->size()<maxInputSeeds && pInNeg->size()<maxInputSeeds) {
 
        std::vector<bool> idxPosColl1(pInPos->size(),false);  
        std::vector<bool> idxPosColl2(pInNeg->size(),false);  
        selectedColl.reserve(pInPos->size());
        
        edm::LogInfo("ConversionSeedFilter") << "New Event \t Pos " << pInPos->size() << " \t Neg " << pInNeg->size() << std::endl;
        
        SearchAmongSeeds(pInPos.product(),pInNeg.product(),selectedColl,idxPosColl1,idxPosColl2);
        //SearchAmongTracks(pInPos.product(),pInTkNeg.product(),selectedColl,idxPosColl1);
        //SearchAmongTracks(pInNeg.product(),pInTkPos.product(),selectedColl,idxPosColl2);
        
        if(trajTrackAssociations.isValid()){
      edm::LogInfo("ConversionSeedFilter") << "Reconstructed tracks " << trajTrackAssociations->size() << std::endl;
      for( TrajTrackAssociationCollection::const_iterator association = trajTrackAssociations->begin(); 
           association != trajTrackAssociations->end(); association++) {
        const Trajectory*  traj  = association->key.get();
        const reco::Track* track = association->val.get();
        
        //edm::LogInfo("ConversionSeedFilter") << "Traj charge " << track->charge() << std::endl;
        
        if(track->charge()<0){SearchAmongTrajectories(pInPos.product(),traj,selectedColl,idxPosColl1);}
        else                 {SearchAmongTrajectories(pInNeg.product(),traj,selectedColl,idxPosColl2);}
        
      } 
        }
      }
      result->insert(result->end(),selectedColl.begin(),selectedColl.end());
    }
 
       
    edm::LogInfo("ConversionSeedFilter") << "\nNew Event : result size " << result->size()<< std::endl;
 
    iEvent.put(std::move(result));
    
 }

void ConversionSeedFilter::SearchAmongSeeds	(	const TrajectorySeedCollection *	pInPos,
		const TrajectorySeedCollection *	pInNeg,
		TrajectorySeedCollection &	selectedColl,
		std::vector< bool > &	idxPosColl1,
		std::vector< bool > &	idxPosColl2
	)

private

Definition at line 151 of file ConversionSeedFilter.cc.

References getKine(), getTSOS(), isCompatible(), and SearchAmongTracks().

Referenced by produce().

                                                                                                                                                                                                    {
 
   for (TrajectorySeedCollection::const_iterator iS1=pInPos->begin(); iS1!=pInPos->end(); ++iS1){
     bool pushed1 = false;
         
     double vars1[4];
     getKine(getTSOS(*iS1),vars1);
     
     for (TrajectorySeedCollection::const_iterator iS2=pInNeg->begin(); iS2!=pInNeg->end(); ++iS2){
       
       double vars2[4];
       getKine(getTSOS(*iS2),vars2);
       
       if(isCompatible(vars1,vars2)){
     edm::LogInfo("ConversionSeedFilter") << "[SearchAmongSeeds] match in pos " << iS1-pInPos->begin() << " " << iS2-pInNeg->begin() << std::endl;
     if (!pushed1) {
       idxPosColl1[iS1-pInPos->begin()]=true;
       selectedColl.push_back(*iS1);
       pushed1 = true;
     }
     if(!idxPosColl2[iS2-pInNeg->begin()]){
       selectedColl.push_back(*iS2);
       idxPosColl2[iS2-pInNeg->begin()]=true;
     }
       }
     }
   }
 }

void ConversionSeedFilter::SearchAmongTracks	(	const TrajectorySeedCollection *	pInSeed,
		const reco::TrackCollection *	pInTk,
		TrajectorySeedCollection &	selectedColl,
		std::vector< bool > &	idxPosColl
	)

private

Definition at line 181 of file ConversionSeedFilter.cc.

References getKine(), getTSOS(), isCompatible(), and SearchAmongTrajectories().

Referenced by SearchAmongSeeds().

                                                                                                                                                                 {
 
   for (TrajectorySeedCollection::const_iterator iS1=pInSeed->begin(); iS1!=pInSeed->end(); ++iS1){
     
     if(idxPosColl[iS1-pInSeed->begin()])
       continue;
     
     double vars1[4];
     getKine(getTSOS(*iS1),vars1);
     
     for (reco::TrackCollection::const_iterator iS2=pInTk->begin(); iS2!=pInTk->end(); ++iS2){
       
       double vars2[4];
       getKine(getTSOS(*iS2),vars2);
       
       if(isCompatible(vars1,vars2)){
     edm::LogInfo("ConversionSeedFilter") << "[SearchAmongTracks] match in pos " << iS1-pInSeed->begin() << std::endl;
     idxPosColl[iS1-pInSeed->begin()]=true;
     selectedColl.push_back(*iS1);
     continue;
       }
     }
   }
 }

void ConversionSeedFilter::SearchAmongTrajectories	(	const TrajectorySeedCollection *	pInSeed,
		const Trajectory *	InTj,
		TrajectorySeedCollection &	selectedColl,
		std::vector< bool > &	idxPosColl
	)

private

Definition at line 207 of file ConversionSeedFilter.cc.

References getKine(), getTSOS(), and isCompatible().

Referenced by produce(), and SearchAmongTracks().

                                                                                                                                                             {
   
   for (TrajectorySeedCollection::const_iterator iS1=pInSeed->begin(); iS1!=pInSeed->end(); ++iS1){
     
     if(idxPosColl[iS1-pInSeed->begin()])
       continue;
 
     double vars1[4];
     getKine(getTSOS(*iS1),vars1);
     
     double vars2[4];
     getKine(getTSOS(*InTj,*iS1),vars2);
     
     if(isCompatible(vars1,vars2)){
       edm::LogInfo("ConversionSeedFilter") << "[SearchAmongTrajectories] match seed in pos " << iS1-pInSeed->begin() << " of " << pInSeed->size() <<  " seed charge " <<  iS1->startingState().parameters().charge()<<std::endl;
       idxPosColl[iS1-pInSeed->begin()]=true;
       selectedColl.push_back(*iS1);
     }
   }
 }

Member Data Documentation

double ConversionSeedFilter::deltaCotThetaCut

private

Definition at line 61 of file ConversionSeedFilter.cc.

Referenced by isCompatible().

double ConversionSeedFilter::deltaPhiCut

private

Definition at line 61 of file ConversionSeedFilter.cc.

Referenced by isCompatible().

double ConversionSeedFilter::deltaRCut

private

Definition at line 61 of file ConversionSeedFilter.cc.

Referenced by isCompatible().

double ConversionSeedFilter::deltaZCut

private

Definition at line 61 of file ConversionSeedFilter.cc.

Referenced by isCompatible().

edm::EDGetTokenT<TrajectorySeedCollection> ConversionSeedFilter::inputCollSeedNeg

private

Definition at line 59 of file ConversionSeedFilter.cc.

Referenced by produce().

edm::EDGetTokenT<TrajectorySeedCollection> ConversionSeedFilter::inputCollSeedPos

private

Definition at line 58 of file ConversionSeedFilter.cc.

Referenced by produce().

edm::EDGetTokenT<TrajTrackAssociationCollection> ConversionSeedFilter::inputTrajectory

private

Definition at line 60 of file ConversionSeedFilter.cc.

Referenced by produce().

uint32_t ConversionSeedFilter::maxInputSeeds

private

Definition at line 65 of file ConversionSeedFilter.cc.

Referenced by produce().

bool ConversionSeedFilter::takeAll

private

Definition at line 66 of file ConversionSeedFilter.cc.

Referenced by produce().

edm::ESHandle<TrackerGeometry> ConversionSeedFilter::theG

private

Definition at line 63 of file ConversionSeedFilter.cc.

Referenced by getTSOS(), and produce().

edm::ESHandle<MagneticField> ConversionSeedFilter::theMF

private

Definition at line 64 of file ConversionSeedFilter.cc.

Referenced by getTSOS(), and produce().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation