CTPPSPixelLocalTrackProducer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/MakerMacros.h"

#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/Common/interface/DetSet.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
 
#include "FWCore/Framework/interface/ESWatcher.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "Geometry/Records/interface/VeryForwardRealGeometryRecord.h"

#include "DataFormats/CTPPSReco/interface/CTPPSPixelRecHit.h"
#include "DataFormats/CTPPSReco/interface/CTPPSPixelLocalTrack.h"
#include "DataFormats/DetId/interface/DetId.h"


#include "RecoCTPPS/PixelLocal/interface/RPixDetPatternFinder.h"
#include "RecoCTPPS/PixelLocal/interface/RPixDetTrackFinder.h"

#include <string>
#include <vector>

#include "FWCore/Framework/interface/Frameworkfwd.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/CTPPSDetId/interface/CTPPSPixelDetId.h"

#include "RecoCTPPS/PixelLocal/interface/RPixRoadFinder.h"
#include "RecoCTPPS/PixelLocal/interface/RPixPlaneCombinatoryTracking.h"


class CTPPSPixelLocalTrackProducer : public edm::stream::EDProducer<>
{
public:
  explicit CTPPSPixelLocalTrackProducer(const edm::ParameterSet& parameterSet);
 
  ~CTPPSPixelLocalTrackProducer() override;

  void produce(edm::Event&, const edm::EventSetup&) override;
  static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);

private:
  int verbosity_;
  int maxHitPerPlane_;
  int maxHitPerRomanPot_;
  int maxTrackPerRomanPot_;
  int maxTrackPerPattern_;
 
  edm::InputTag inputTag_;
  edm::EDGetTokenT<edm::DetSetVector<CTPPSPixelRecHit>> tokenCTPPSPixelRecHit_;
  edm::ESWatcher<VeryForwardRealGeometryRecord> geometryWatcher_;
  uint32_t numberOfPlanesPerPot_;
  std::vector<uint32_t> listOfAllPlanes_;

  std::unique_ptr<RPixDetPatternFinder> patternFinder_;
  std::unique_ptr<RPixDetTrackFinder>   trackFinder_  ;

  void run(const edm::DetSetVector<CTPPSPixelRecHit> &input, edm::DetSetVector<CTPPSPixelLocalTrack> &output);
  
};

//------------------------------------------------------------------------------------------------//

CTPPSPixelLocalTrackProducer::CTPPSPixelLocalTrackProducer(const edm::ParameterSet& parameterSet)
{
  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(){
}

//------------------------------------------------------------------------------------------------//

void CTPPSPixelLocalTrackProducer::fillDescriptions(edm::ConfigurationDescriptions & descriptions){
  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){
  // 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();
    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")<<" ---> To 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;

}

DEFINE_FWK_MODULE( CTPPSPixelLocalTrackProducer);