CosmicTrackFinder.cc

Go to the documentation of this file.

// Package:    RecoTracker/SingleTrackPattern
// Class:      CosmicTrackFinder
// Original Author:  Michele Pioppi-INFN perugia
#include <memory>
#include <string>

#include "RecoTracker/SingleTrackPattern/interface/CosmicTrackFinder.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "TrackingTools/PatternTools/interface/TSCPBuilderNoMaterial.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"

#include "RecoTracker/TransientTrackingRecHit/interface/Traj2TrackHits.h"


namespace cms
{

  CosmicTrackFinder::CosmicTrackFinder(edm::ParameterSet const& conf) : 
    cosmicTrajectoryBuilder_(conf) ,
    crackTrajectoryBuilder_(conf)
  {
    geometry=conf.getUntrackedParameter<std::string>("GeometricStructure","STANDARD");
    useHitsSplitting_=conf.getParameter<bool>("useHitsSplitting");
    matchedrecHitsToken_ = consumes<SiStripMatchedRecHit2DCollection>(
        conf.getParameter<edm::InputTag>("matchedRecHits"));
    rphirecHitsToken_ = consumes<SiStripRecHit2DCollection>(
        conf.getParameter<edm::InputTag>("rphirecHits"));
    stereorecHitsToken_ = consumes<SiStripRecHit2DCollection>(
        conf.getParameter<edm::InputTag>("stereorecHits"));
    pixelRecHitsToken_ = consumes<SiPixelRecHitCollection>(
        conf.getParameter<edm::InputTag>("pixelRecHits"));
    seedToken_ = consumes<TrajectorySeedCollection>(
        conf.getParameter<edm::InputTag>("cosmicSeeds"));

    produces<TrackCandidateCollection>();
  }


  // Virtual destructor needed.
  CosmicTrackFinder::~CosmicTrackFinder() { }  

  // Functions that gets called by framework every event
  void CosmicTrackFinder::produce(edm::Event& e, const edm::EventSetup& es)
  {
    using namespace std;

    // retrieve seeds
    edm::Handle<TrajectorySeedCollection> seed;
    e.getByToken(seedToken_, seed);

    //retrieve PixelRecHits
    static const SiPixelRecHitCollection s_empty;
    const SiPixelRecHitCollection *pixelHitCollection = &s_empty;
    edm::Handle<SiPixelRecHitCollection> pixelHits;
    if (geometry!="MTCC" && (geometry!="CRACK" )) {
      if( e.getByToken(pixelRecHitsToken_, pixelHits)) {
    pixelHitCollection = pixelHits.product();
      } else {
        Labels l;
        labelsForToken(pixelRecHitsToken_, l);
    edm::LogWarning("CosmicTrackFinder")
            << "Collection SiPixelRecHitCollection with InputTag "
            << l.module
            << " cannot be found, using empty collection of same type.";
      }
    }
    
   


 //retrieve StripRecHits
    edm::Handle<SiStripMatchedRecHit2DCollection> matchedrecHits;
    e.getByToken(matchedrecHitsToken_, matchedrecHits);
    edm::Handle<SiStripRecHit2DCollection> rphirecHits;
    e.getByToken(rphirecHitsToken_, rphirecHits);
    edm::Handle<SiStripRecHit2DCollection> stereorecHits;
    e.getByToken(stereorecHitsToken_, stereorecHits);

    // Step B: create empty output collection
    auto output = std::make_unique<TrackCandidateCollection>();

    edm::ESHandle<TrackerGeometry> tracker;
    es.get<TrackerDigiGeometryRecord>().get(tracker);
    edm::LogVerbatim("CosmicTrackFinder") << "========== Cosmic Track Finder Info ==========";
    edm::LogVerbatim("CosmicTrackFinder") << " Numbers of Seeds " << (*seed).size();
    if(!(*seed).empty()){

      std::vector<Trajectory> trajoutput;
      

      const TransientTrackingRecHitBuilder * hitBuilder = nullptr;
      if(geometry!="CRACK" ) {
        cosmicTrajectoryBuilder_.run(*seed,
                                   *stereorecHits,
                                   *rphirecHits,
                                   *matchedrecHits,
                   *pixelHitCollection,
                                   es,
                                   e,
                                   trajoutput);
        hitBuilder = cosmicTrajectoryBuilder_.hitBuilder();
      } else {
        crackTrajectoryBuilder_.run(*seed,
                                   *stereorecHits,
                                   *rphirecHits,
                                   *matchedrecHits,
                   *pixelHitCollection,
                                   es,
                                   e,
                                   trajoutput);
    hitBuilder = crackTrajectoryBuilder_.hitBuilder();
      }
      assert(hitBuilder);
      Traj2TrackHits t2t(hitBuilder,true);

      edm::LogVerbatim("CosmicTrackFinder") << " Numbers of Temp Trajectories " << trajoutput.size();
      edm::LogVerbatim("CosmicTrackFinder") << "========== END Info ==========";
      if(!trajoutput.empty()){

        // crazyness...
    std::vector<Trajectory*> tmpTraj;
    std::vector<Trajectory>::iterator itr;
    for (itr=trajoutput.begin();itr!=trajoutput.end();itr++)tmpTraj.push_back(&(*itr));

    //The best track is selected
    //FOR MTCC the criteria are:
    //1)# of layers,2) # of Hits,3)Chi2
    if (geometry=="MTCC")  stable_sort(tmpTraj.begin(),tmpTraj.end(),CompareTrajLay());
    else  stable_sort(tmpTraj.begin(),tmpTraj.end(),CompareTrajChi());


    const Trajectory  theTraj = *(*tmpTraj.begin());
    bool seedplus=(theTraj.seed().direction()==alongMomentum);

        // std::cout << "CosmicTrackFinder " <<"Reconstruction " <<  (seedplus ? "along" : "opposite to") << " momentum" << std::endl;
        LogDebug("CosmicTrackFinder")<<"Reconstruction " <<  (seedplus ? "along" : "opposite to") << " momentum";

    /*
    // === the convention is to save always final tracks with hits sorted *along* momentum
        // --- this is NOT what was necessaraly happening before and not consistent with what done in standard CKF (this is a candidate not a track)    
    */
         edm::OwnVector<TrackingRecHit> recHits;
         if(theTraj.direction() == alongMomentum) std::cout << "cosmic: along momentum... " << std::endl;
         t2t(theTraj,recHits,useHitsSplitting_);
         recHits.reverse();  // according to original code

        /*
    Trajectory::RecHitContainer thits;
    //it->recHitsV(thits);
        theTraj.recHitsV(thits,useHitsSplitting_);
    edm::OwnVector<TrackingRecHit> recHits;
    recHits.reserve(thits.size());
    // reverse hit order
    for (Trajectory::RecHitContainer::const_iterator hitIt = thits.end()-1;
         hitIt >= thits.begin(); hitIt--) {
      recHits.push_back( (**hitIt).hit()->clone());
    }
       */

    TSOS firstState;
    unsigned int firstId;

        // assume not along momentum....
    firstState=theTraj.lastMeasurement().updatedState();
        firstId = theTraj.lastMeasurement().recHitR().rawId();
    //firstId = recHits.front().rawId();

    /*
    cout << "firstState y, z: " << firstState.globalPosition().y() 
         << " , " << firstState.globalPosition().z() <<  endl;

    */

    AlgebraicSymMatrix55 C = AlgebraicMatrixID();
    TSOS startingState( firstState.localParameters(), LocalTrajectoryError(C),
                firstState.surface(),
                firstState.magneticField() );

    // protection againt invalid initial states
    if (! firstState.isValid()) {
      edm::LogWarning("CosmicTrackFinder") << "invalid innerState, will not make TrackCandidate";
      edm::OrphanHandle<TrackCandidateCollection> rTrackCand = e.put(std::move(output));
      return;
    }

        // FIXME in case of slitting this can happen see CkfTrackCandidateMakerBase
    if(firstId != recHits.front().rawId()){
      edm::LogWarning("CosmicTrackFinder") <<"Mismatch in DetID of first hit: firstID= " <<firstId
                           << "   DetId= " << recHits.front().geographicalId().rawId();
      edm::OrphanHandle<TrackCandidateCollection> rTrackCand = e.put(std::move(output));
      return;
    }

    PTrajectoryStateOnDet const &  state = trajectoryStateTransform::persistentState( startingState, firstId);
    
    
    output->push_back(TrackCandidate(recHits,theTraj.seed(),state,theTraj.seedRef() ) );
    
      }

    }
    edm::OrphanHandle<TrackCandidateCollection> rTrackCand = e.put(std::move(output));
  }
}