NuclearTrackCorrector.h

Go to the documentation of this file.

#ifndef CD_NuclearTrackCorrector_H_
#define CD_NuclearTrackCorrector_H_


// -*- C++ -*-
//
// Package:    NuclearTrackCorrector
// Class:      NuclearTrackCorrector
// 
//
// Original Author:  Loic QUERTENMONT, Vincent ROBERFROID
//         Created:  Tue Sep 18 14:22:48 CEST 2007
//
//


// system include files
#include <memory>
#include <string>
#include <stdio.h>

// user include files

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

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

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

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

#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidate.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/VertexReco/interface/NuclearInteraction.h"
#include "DataFormats/VertexReco/interface/NuclearInteractionFwd.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackBase.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"
#include "RecoTracker/NuclearSeedGenerator/interface/TrajectoryToSeedMap.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "RecoTracker/NuclearSeedGenerator/interface/TrackCandidateToTrajectoryMap.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerBase.h"
#include "RecoTracker/NuclearSeedGenerator/interface/NuclearTrackCorrector.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "RecoTracker/CkfPattern/interface/TransientInitialStateEstimator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateTransform.h"
#include "TrackingTools/Records/interface/TrackingComponentsRecord.h"

#include "RecoTracker/TrackProducer/interface/TrackProducerAlgorithm.h"





class TransientInitialStateEstimator;

//
// class decleration
//

class NuclearTrackCorrector :  public edm::EDProducer {

   public:
      typedef edm::RefVector<TrajectorySeedCollection> TrajectorySeedRefVector;
      typedef edm::Ref<TrajectoryCollection> TrajectoryRef;
      typedef edm::Ref<TrackCandidateCollection> TrackCandidateRef;
      typedef TransientTrackingRecHit::ConstRecHitContainer ConstRecHitContainer;
      typedef TrackProducerAlgorithm<reco::Track>::AlgoProductCollection AlgoProductCollection;

   public:

      explicit NuclearTrackCorrector(const edm::ParameterSet&);
      ~NuclearTrackCorrector();

   private:
      virtual void produce(edm::Event&, const edm::EventSetup&) override;
      virtual void endJob() ;

      bool newTrajNeeded(Trajectory& newtrajectory, const TrajectoryRef& trajRef, const reco::NuclearInteraction& ni);

      reco::TrackExtra getNewTrackExtra(const AlgoProductCollection& algoresults);

      bool getTrackFromTrajectory(const Trajectory& newTraj , const TrajectoryRef& initialTrajRef, AlgoProductCollection& algoResults);

      TrajectoryStateOnSurface getInitialState(const reco::Track * theT,
                                            TransientTrackingRecHit::RecHitContainer& hits,
                                            const TrackingGeometry * theG,
                                            const MagneticField * theMF);

      void  swap_map(const  edm::Handle< TrajectoryCollection >& trajColl , std::map< reco::TrackRef, edm::Ref<TrajectoryCollection> >& result);
      
      // ----------member data ---------------------------


      std::string str_Input_Trajectory;
      std::string str_Input_NuclearInteraction;
      int    int_Input_Hit_Distance;

      int    verbosity;
      int    KeepOnlyCorrectedTracks;

      std::vector< std::pair<unsigned int, unsigned int> > Indice_Map;

      
      edm::ESHandle<TrackerGeometry> theG;
      edm::ESHandle<MagneticField> theMF;
      edm::ESHandle<TrajectoryFitter> theFitter;
      edm::ESHandle<Propagator> thePropagator;
      edm::ParameterSet conf_;
      TransientInitialStateEstimator*  theInitialState;

      TrackProducerAlgorithm<reco::Track>* theAlgo;
      const TrajTrackAssociationCollection* m_TrajToTrackCollection;
};

#endif