Inheritance diagram for CTPPSPixelLocalTrackProducer:

Public Member Functions
	CTPPSPixelLocalTrackProducer (const edm::ParameterSet &parameterSet)

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

	~CTPPSPixelLocalTrackProducer () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

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

Private Member Functions
void	run (const edm::DetSetVector< CTPPSPixelRecHit > &input, edm::DetSetVector< CTPPSPixelLocalTrack > &output)

Private Attributes
edm::ESWatcher< VeryForwardRealGeometryRecord >	geometryWatcher_

edm::InputTag	inputTag_

std::vector< uint32_t >	listOfAllPlanes_

int	maxHitPerPlane_

int	maxHitPerRomanPot_

int	maxTrackPerPattern_

int	maxTrackPerRomanPot_

uint32_t	numberOfPlanesPerPot_

std::unique_ptr< RPixDetPatternFinder >	patternFinder_

edm::EDGetTokenT< edm::DetSetVector< CTPPSPixelRecHit > >	tokenCTPPSPixelRecHit_

std::unique_ptr< RPixDetTrackFinder >	trackFinder_

int	verbosity_

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

Definition at line 40 of file CTPPSPixelLocalTrackProducer.cc.

Constructor & Destructor Documentation

CTPPSPixelLocalTrackProducer::CTPPSPixelLocalTrackProducer ( const edm::ParameterSet & parameterSet )

explicit

Definition at line 72 of file CTPPSPixelLocalTrackProducer.cc.

References Exception, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), mps_fire::i, inputTag_, listOfAllPlanes_, maxHitPerPlane_, maxHitPerRomanPot_, maxTrackPerPattern_, maxTrackPerRomanPot_, numberOfPlanesPerPot_, patternFinder_, AlCaHLTBitMon_QueryRunRegistry::string, tokenCTPPSPixelRecHit_, trackFinder_, and verbosity_.

 {
   inputTag_             = parameterSet.getParameter<std::string>  ("label");
   verbosity_            = parameterSet.getUntrackedParameter<int> ("verbosity");
   maxHitPerRomanPot_    = parameterSet.getParameter<int> ("maxHitPerRomanPot");
   maxHitPerPlane_       = parameterSet.getParameter<int> ("maxHitPerPlane");
   maxTrackPerRomanPot_  = parameterSet.getParameter<int> ("maxTrackPerRomanPot");
   maxTrackPerPattern_   = parameterSet.getParameter<int> ("maxTrackPerPattern");
   numberOfPlanesPerPot_ = parameterSet.getParameter<int> ("numberOfPlanesPerPot");
 
   std::string patternFinderAlgorithm = parameterSet.getParameter<std::string>("patternFinderAlgorithm");
   std::string trackFitterAlgorithm  = parameterSet.getParameter<std::string>("trackFinderAlgorithm" );
 
   // pattern algorithm selector
   if(patternFinderAlgorithm == "RPixRoadFinder"){
     patternFinder_ = std::unique_ptr<RPixRoadFinder>(new RPixRoadFinder(parameterSet));
   }
   else{
     throw cms::Exception("CTPPSPixelLocalTrackProducer") << "Pattern finder algorithm" << patternFinderAlgorithm << " does not exist";
   }
   
   listOfAllPlanes_.reserve(6);
   for(uint32_t i=0; i<numberOfPlanesPerPot_; ++i){
     listOfAllPlanes_.push_back(i);
   }
 
   //tracking algorithm selector
   if(trackFitterAlgorithm == "RPixPlaneCombinatoryTracking"){
     trackFinder_ = std::unique_ptr<RPixPlaneCombinatoryTracking>(new RPixPlaneCombinatoryTracking(parameterSet));
   }
   else{
     throw cms::Exception("CTPPSPixelLocalTrackProducer") << "Tracking fitter algorithm" << trackFitterAlgorithm << " does not exist";
   }
   trackFinder_->setListOfPlanes(listOfAllPlanes_);
   trackFinder_->initialize();
   
   tokenCTPPSPixelRecHit_ = consumes<edm::DetSetVector<CTPPSPixelRecHit> >(edm::InputTag(inputTag_));
  
   produces<edm::DetSetVector<CTPPSPixelLocalTrack> > ();
 
 }

CTPPSPixelLocalTrackProducer::~CTPPSPixelLocalTrackProducer ( )

override

Definition at line 116 of file CTPPSPixelLocalTrackProducer.cc.

116 {

117 }

Member Function Documentation

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

static

Definition at line 121 of file CTPPSPixelLocalTrackProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addUntracked(), AlCaHLTBitMon_QueryRunRegistry::string, and parallelization::uint().

                                                                                               {
   edm::ParameterSetDescription desc;
 
   desc.add<std::string>    ("label"                            , "ctppsPixelRecHits"           )
     ->setComment( "label of the RecHits input for the tracking algorithm"                       );
   desc.add<std::string>    ("patternFinderAlgorithm"           , "RPixRoadFinder"              )
     ->setComment( "algorithm type for pattern finder"                                           );
   desc.add<std::string>    ("trackFinderAlgorithm"             , "RPixPlaneCombinatoryTracking")
     ->setComment( "algorithm type for track finder"                                             );
   desc.add<uint>           ("trackMinNumberOfPoints"           , 3                             )
     ->setComment( "minimum number of planes to produce a track"                                 );
   desc.addUntracked<int>   ("verbosity"                        , 0                             )
     ->setComment( "verbosity for track producer"                                                );
   desc.add<double>         ("maximumChi2OverNDF"               , 5.                            )
     ->setComment( "maximum Chi2OverNDF for accepting the track"                                 );
   desc.add<double>         ("maximumXLocalDistanceFromTrack"   , 0.2                           )
     ->setComment( "maximum x distance in mm to associate a point not used for fit to the track" );
   desc.add<double>         ("maximumYLocalDistanceFromTrack"   , 0.3                           )
     ->setComment( "maximum y distance in mm to associate a point not used for fit to the track" );
   desc.add<int>            ("maxHitPerPlane"                   , 20                            )
     ->setComment( "maximum hits per plane, events with higher number will not be fitted"        );
   desc.add<int>            ("maxHitPerRomanPot"                , 60                            )
     ->setComment( "maximum hits per roman pot, events with higher number will not be fitted"    );
   desc.add<int>            ("maxTrackPerRomanPot"              , 10                            )
     ->setComment( "maximum tracks per roman pot, events with higher track will not be saved"    );
   desc.add<int>            ("maxTrackPerPattern"               , 5                             )
     ->setComment( "maximum tracks per pattern, events with higher track will not be saved"      );
   desc.add<int>            ("numberOfPlanesPerPot"             , 6                             )
     ->setComment( "number of planes per pot"                                                    );
   desc.add<double>         ("roadRadius"                       , 1.0                           )
     ->setComment( "radius of pattern search window"                                             );
   desc.add<int>            ("minRoadSize"                      , 3                             )
     ->setComment( "minimum number of points in a pattern"                                       );
   desc.add<int>            ("maxRoadSize"                      , 20                            )
     ->setComment( "maximum number of points in a pattern"                                       );
 
   descriptions.add("ctppsPixelLocalTracks", desc);
 }

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

override

Definition at line 162 of file CTPPSPixelLocalTrackProducer.cc.

References edm::ESWatcher< T >::check(), edm::DetSet< T >::clear(), DEFINE_FWK_MODULE, Exception, spr::find(), edm::DetSetVector< T >::find_or_insert(), geometryWatcher_, edm::EventSetup::get(), edm::Event::getByToken(), CTPPSDetId::getRPId(), CTPPSGeometry::getRPTranslation(), edm::Event::getRun(), edm::RunBase::id(), maxHitPerPlane_, maxHitPerRomanPot_, maxTrackPerPattern_, maxTrackPerRomanPot_, listBenchmarks::pattern, patternFinder_, edm::DetSet< T >::push_back(), edm::Event::put(), edm::RunID::run(), tokenCTPPSPixelRecHit_, HiIsolationCommonParameters_cff::track, trackFinder_, and verbosity_.

                                                                                        {
   // Step A: get inputs
   
   edm::Handle<edm::DetSetVector<CTPPSPixelRecHit> > recHits;
   iEvent.getByToken(tokenCTPPSPixelRecHit_, recHits);
   edm::DetSetVector<CTPPSPixelRecHit> recHitVector(*recHits);
 
   // get geometry
   edm::ESHandle<CTPPSGeometry> geometryHandler;
   iSetup.get<VeryForwardRealGeometryRecord>().get(geometryHandler);
   const CTPPSGeometry &geometry = *geometryHandler;
   geometryWatcher_.check(iSetup);
 
   std::vector<CTPPSPixelDetId> listOfPotWithHighOccupancyPlanes; 
   std::map<CTPPSPixelDetId, uint32_t> mapHitPerPot;
 
   for(const auto & recHitSet : recHitVector){
     
     if(verbosity_>2) edm::LogInfo("CTPPSPixelLocalTrackProducer")<<"Hits found in plane = "<<recHitSet.detId()<< " number = " << recHitSet.size() ;
     CTPPSPixelDetId tmpRomanPotId = CTPPSPixelDetId(recHitSet.detId()).getRPId();
     uint32_t hitOnPlane = recHitSet.size();
 
     //Get the number of hits per pot
     if(mapHitPerPot.find(tmpRomanPotId)==mapHitPerPot.end()){
       mapHitPerPot[tmpRomanPotId] = hitOnPlane;
     }
     else mapHitPerPot[tmpRomanPotId]+=hitOnPlane;
 
     //check is the plane occupancy is too high and save the corresponding pot
     if(maxHitPerPlane_>=0 && hitOnPlane>(uint32_t)maxHitPerPlane_){
       if(verbosity_>2) edm::LogInfo("CTPPSPixelLocalTrackProducer")<<" ---> To many hits in the plane, pot will be excluded from tracking cleared";
       listOfPotWithHighOccupancyPlanes.push_back(tmpRomanPotId);
     }
   }
 
   //remove rechit for pot with too many hits or containing planes with too many hits
   for(const auto & recHitSet : recHitVector){
     const auto tmpDetectorId = CTPPSPixelDetId(recHitSet.detId());
     const auto tmpRomanPotId = tmpDetectorId.getRPId();
 
     if((maxHitPerRomanPot_>=0 && mapHitPerPot[tmpRomanPotId]>(uint32_t)maxHitPerRomanPot_) || 
       find (listOfPotWithHighOccupancyPlanes.begin(), listOfPotWithHighOccupancyPlanes.end(), tmpRomanPotId) != listOfPotWithHighOccupancyPlanes.end() ){
       edm::DetSet<CTPPSPixelRecHit> &tmpDetSet = recHitVector[tmpDetectorId];
       tmpDetSet.clear();
     }
 
   }
 
   edm::DetSetVector<CTPPSPixelLocalTrack> foundTracks;
 
   // Pattern finder
 
   patternFinder_->clear();
   patternFinder_->setHits(&recHitVector);
   patternFinder_->setGeometry(&geometry);
   patternFinder_->findPattern();
   std::vector<RPixDetPatternFinder::Road> patternVector = patternFinder_->getPatterns();
 
   //loop on all the patterns
   int numberOfTracks = 0;
 
   for(const auto & pattern : patternVector){
     CTPPSPixelDetId firstHitDetId = CTPPSPixelDetId(pattern.at(0).detId);
     CTPPSPixelDetId romanPotId = firstHitDetId.getRPId();
     
     std::map<CTPPSPixelDetId, std::vector<RPixDetPatternFinder::PointInPlane> > hitOnPlaneMap; //hit of the pattern organized by plane
 
     //loop on all the hits of the pattern
     for(const auto & hit : pattern){
       CTPPSPixelDetId hitDetId = CTPPSPixelDetId(hit.detId);
       CTPPSPixelDetId tmpRomanPotId = hitDetId.getRPId();
 
       if(tmpRomanPotId!=romanPotId){ //check that the hits belong to the same tracking station
         throw cms::Exception("CTPPSPixelLocalTrackProducer") << "Hits in the pattern must belong to the same tracking station";
       }
 
       if(hitOnPlaneMap.find(hitDetId)==hitOnPlaneMap.end()){ //add the plane key in case it is the first hit of that plane
         std::vector<RPixDetPatternFinder::PointInPlane> hitOnPlane;
         hitOnPlane.push_back(hit);
         hitOnPlaneMap[hitDetId] = hitOnPlane;
       }
       else hitOnPlaneMap[hitDetId].push_back(hit); //add the hit to an existing plane key
     
     }
 
     trackFinder_->clear();
     trackFinder_->setRomanPotId(romanPotId);
     trackFinder_->setHits(&hitOnPlaneMap);
     trackFinder_->setGeometry(&geometry);
     trackFinder_->setZ0(geometry.getRPTranslation(romanPotId).z());
     trackFinder_->findTracks(iEvent.getRun().id().run());
     std::vector<CTPPSPixelLocalTrack> tmpTracksVector = trackFinder_->getLocalTracks();
 
     if(verbosity_>2)edm::LogInfo("CTPPSPixelLocalTrackProducer")<<"tmpTracksVector = "<<tmpTracksVector.size();
     if(maxTrackPerPattern_>=0 && tmpTracksVector.size()>(uint32_t)maxTrackPerPattern_){
       if(verbosity_>2)edm::LogInfo("CTPPSPixelLocalTrackProducer")<<" ---> To many tracks in the pattern, cleared";
       continue;
     }
 
     for(const auto & track : tmpTracksVector){
       ++numberOfTracks;
       edm::DetSet<CTPPSPixelLocalTrack> &tmpDetSet = foundTracks.find_or_insert(romanPotId);
         tmpDetSet.push_back(track);
     }
 
 
   }
 
   if(verbosity_>1) edm::LogInfo("CTPPSPixelLocalTrackProducer")<<"Number of tracks will be saved = "<<numberOfTracks;
 
   for(const auto & track : foundTracks){
     if(verbosity_>1) edm::LogInfo("CTPPSPixelLocalTrackProducer")<<"Track found in detId = "<<track.detId()<< " number = " << track.size() ;
     if(maxTrackPerRomanPot_>=0 && track.size()>(uint32_t)maxTrackPerRomanPot_){
       if(verbosity_>1) edm::LogInfo("CTPPSPixelLocalTrackProducer")<<" ---> Too many tracks in the pot, cleared";
       CTPPSPixelDetId tmpRomanPotId = CTPPSPixelDetId(track.detId());
       edm::DetSet<CTPPSPixelLocalTrack> &tmpDetSet = foundTracks[tmpRomanPotId];
       tmpDetSet.clear();
     }
   }
 
   iEvent.put(std::make_unique<edm::DetSetVector<CTPPSPixelLocalTrack> >(foundTracks));
 
   return;
 
 }