CMSSW Reference Manual

00001 #ifndef RecoAlgos_AnalyticalTrackSelector_h
00002 #define RecoAlgos_AnalyticalTrackSelector_h
00003 
00015 #include <utility>
00016 #include <vector>
00017 #include <memory>
00018 #include <algorithm>
00019 #include <map>
00020 #include "FWCore/Framework/interface/EDProducer.h"
00021 #include "FWCore/Framework/interface/EventPrincipal.h" 
00022 #include "FWCore/Framework/interface/Event.h"
00023 #include "FWCore/ParameterSet/interface/ParameterSet.h"
00024 #include "FWCore/ParameterSet/interface/InputTag.h"
00025 
00026 #include "DataFormats/TrackReco/interface/TrackFwd.h"
00027 #include "DataFormats/TrackReco/interface/Track.h"
00028 #include "DataFormats/TrackReco/interface/TrackExtra.h"
00029 #include "DataFormats/VertexReco/interface/VertexFwd.h"
00030 #include "DataFormats/VertexReco/interface/Vertex.h"
00031 #include "DataFormats/BeamSpot/interface/BeamSpot.h"
00032 #include "TrackingTools/PatternTools/interface/Trajectory.h"
00033 #include "TrackingTools/PatternTools/interface/TrajTrackAssociation.h"
00034 
00035 
00036 namespace reco { namespace modules {
00037 
00038     class AnalyticalTrackSelector : public edm::EDProducer {
00039         private:
00040         public:
00042             explicit AnalyticalTrackSelector( const edm::ParameterSet & cfg ) ;
00044             virtual ~AnalyticalTrackSelector() ;
00045 
00046         private:
00047             typedef math::XYZPoint Point;
00049             void produce( edm::Event& evt, const edm::EventSetup& es ) ;
00051             bool select (const reco::BeamSpot &vertexBeamSpot, const reco::Track &tk, const std::vector<Point> &points);
00052             void selectVertices ( const reco::VertexCollection &vtxs, std::vector<Point> &points);
00054             edm::InputTag src_;
00055             edm::InputTag beamspot_;
00056             edm::InputTag vertices_;
00058             bool copyExtras_;
00060             bool copyTrajectories_;
00061 
00063             bool keepAllTracks_;
00065             bool setQualityBit_;
00066             TrackBase::TrackQuality qualityToSet_;
00067 
00069             int32_t vtxNumber_;
00070             size_t  vtxTracks_;
00071             double  vtxChi2Prob_;
00073                         std::vector<double> res_par_;
00074             double  chi2n_par_;
00075                         std::vector<double> d0_par1_;
00076                         std::vector<double> dz_par1_;
00077                         std::vector<double> d0_par2_;
00078                         std::vector<double> dz_par2_;
00079 
00081                         uint32_t min_layers_;
00082 
00084             std::auto_ptr<reco::TrackCollection> selTracks_;
00085             std::auto_ptr<reco::TrackExtraCollection> selTrackExtras_;
00086             std::auto_ptr< TrackingRecHitCollection>  selHits_;
00087             std::auto_ptr< std::vector<Trajectory> > selTrajs_;
00088             std::auto_ptr< TrajTrackAssociationCollection >  selTTAss_;
00089             reco::TrackRefProd rTracks_;
00090             reco::TrackExtraRefProd rTrackExtras_;
00091             TrackingRecHitRefProd rHits_;
00092             edm::RefProd< std::vector<Trajectory> > rTrajectories_;
00093             std::vector<reco::TrackRef> trackRefs_;
00094 
00095     };
00096 
00097 } }
00098 
00099 #endif
AnalyticalTrackSelector.h
