/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/RecoTracker/FinalTrackSelectors/interface/DuplicateTrackMerger.h

Go to the documentation of this file.

#ifndef RecoTracker_DuplicateTrackMerger_h
#define RecoTracker_DuplicateTrackMerger_h

#include "FWCore/Framework/interface/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::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::InputTag 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