DuplicateTrackMerger.h

Go to the documentation of this file.

#ifndef RecoTracker_DuplicateTrackMerger_h
#define RecoTracker_DuplicateTrackMerger_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Provenance/interface/ParameterSetID.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
#include "MagneticField/Engine/interface/MagneticField.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidate.h"
#include "RecoTracker/TrackProducer/interface/TrackMerger.h"
#include <vector>
#include <algorithm>
#include <string>
#include <iostream>
#include <map>

#include "CondFormats/EgammaObjects/interface/GBRForest.h"

namespace reco { namespace modules {
    class DuplicateTrackMerger : public edm::stream::EDProducer<> {
       public:
         explicit DuplicateTrackMerger(const edm::ParameterSet& iPara);
     virtual ~DuplicateTrackMerger();

     typedef std::vector<std::pair<TrackCandidate,std::pair<reco::TrackRef,reco::TrackRef> > > CandidateToDuplicate;

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

       private:
     GBRForest* forest_;

     float tmva_ddsz_;
     float tmva_ddxy_;
     float tmva_dphi_;
     float tmva_dlambda_;
     float tmva_dqoverp_;
     float tmva_d3dr_;
     float tmva_d3dz_;
     float tmva_outer_nMissingInner_;
     float tmva_inner_nMissingOuter_;

     float* gbrVals_;

     edm::EDGetTokenT<reco::TrackCollection> trackSource_;
     std::string dbFileName_;
     bool useForestFromDB_;
     std::string forestLabel_;
     double minDeltaR3d_;
     double minBDTG_;
     double minpT_;
     double minP_;
     double maxDCA_;
     double maxDPhi_;
     double maxDLambda_;
     double maxDdxy_;
     double maxDdsz_;
     double maxDQoP_;

     edm::ESHandle<MagneticField> magfield_;

     TrackMerger merger_;
     };
  }
}
#endif