GoodSeedProducer.h

Go to the documentation of this file.

#ifndef GoodSeedProducer_H
#define GoodSeedProducer_H
// system include files
#include <memory>

// user include files

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PreIdFwd.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "TMVA/Reader.h"
#include "DataFormats/Math/interface/LorentzVector.h"

//namespace reco {
class PFResolutionMap;
// }

class PFTrackTransformer;
class TrajectoryFitter;
class TrajectorySmoother;
class TrackerGeometry;
class TrajectoryStateOnSurface;


class GoodSeedProducer : public edm::EDProducer {
  typedef TrajectoryStateOnSurface TSOS;
   public:
      explicit GoodSeedProducer(const edm::ParameterSet&);
      ~GoodSeedProducer();
  
   private:
      virtual void beginRun(const edm::Run & run,const edm::EventSetup&) override;
      virtual void produce(edm::Event&, const edm::EventSetup&) override;
      virtual void endRun(const edm::Run & run,const edm::EventSetup&) override;
 
      int getBin(float,float);
      int getBin(float);
      void PSforTMVA(const math::XYZTLorentzVector& mom,
             const math::XYZTLorentzVector& pos);
      bool IsIsolated(float  charge,float P,
                  math::XYZPointF, 
                      const reco::PFClusterCollection &ecalColl,
                      const reco::PFClusterCollection &hcalColl);

      void fillPreIdRefValueMap( edm::Handle<reco::TrackCollection> tkhandle,
                 const edm::OrphanHandle<reco::PreIdCollection>&,
                 edm::ValueMap<reco::PreIdRef>::Filler & filler);
      // ----------member data ---------------------------

      std::vector<reco::PFCluster> ps1Clus;
      std::vector<reco::PFCluster> ps2Clus;

      std::string preidckf_;

      std::string preidgsf_;

      std::string preidname_;

      edm::ESHandle<TrajectoryFitter> fitter_;

      edm::ESHandle<TrajectorySmoother> smoother_;

      PFTrackTransformer *pfTransformer_;

      int nHitsInSeed_;

      double minPt_;
      double maxPt_;
      double maxEta_;
      
      bool applyIsolation_;
      double HcalIsolWindow_;
      double EcalStripSumE_minClusEnergy_;
      double EcalStripSumE_deltaEta_;
      double EcalStripSumE_deltaPhiOverQ_minValue_;
      double EcalStripSumE_deltaPhiOverQ_maxValue_;
      double minEoverP_;
      double maxHoverP_;

      double clusThreshold_;

      double minEp_;
      double maxEp_;

      bool produceCkfseed_;

      bool disablePreId_;

      bool producePreId_;
      
      double PtThresholdSavePredId_;

      float thr[150];
      float thrPS[20];

      // ----------access to event data
      edm::ParameterSet conf_;
      edm::InputTag pfCLusTagPSLabel_;
      edm::InputTag pfCLusTagECLabel_;
      edm::InputTag pfCLusTagHCLabel_;
      std::vector<edm::InputTag> tracksContainers_;
      

      std::string fitterName_;
      std::string smootherName_;
      std::string propagatorName_;

      PFResolutionMap* resMapEtaECAL_;
      PFResolutionMap* resMapPhiECAL_;

      bool useQuality_;
      reco::TrackBase::TrackQuality trackQuality_;
    
      TMVA::Reader *reader;

      float eP,chi,eta,pt,nhit,dpt,chired,chiRatio;
      float ps1En,ps2En,ps1chi,ps2chi;
      bool useTmva_;

      std::string method_;

      math::XYZVector B_;

      bool usePreshower_;

      std::map<reco::TrackRef,unsigned> refMap_;
     
};
#endif