CkfPattern/interface/CkfTrajectoryBuilder.h

0001 #ifndef CkfTrajectoryBuilder_H
0002 #define CkfTrajectoryBuilder_H
0003
0004 #include <vector>
0005
0006 class Propagator;
0007 class TrajectoryStateUpdator;
0008 class MeasurementEstimator;
0009 class TrajectorySeed;
0010 class TrajectoryStateOnSurface;
0011 class TrajectoryFilter;
0012
0013 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0014 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0015 #include "FWCore/Framework/interface/EventSetup.h"
0016 #include "FWCore/Framework/interface/Event.h"
0017
0018 #include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
0019 #include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
0020 #include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
0021
0022 #include "RecoTracker/CkfPattern/interface/BaseCkfTrajectoryBuilder.h"
0023
0024 #include "TrackingTools/PatternTools/interface/Trajectory.h"
0025 #include "TrackingTools/PatternTools/interface/TempTrajectory.h"
0026 #include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"
0027
0028 #include <map>
0029
0030 class TransientTrackingRecHitBuilder;
0031 class TrajectoryFilter;
0032
0033 class CkfTrajectoryBuilder : public BaseCkfTrajectoryBuilder {
0034 public:
0035   typedef std::vector<Trajectory> TrajectoryContainer;
0036   typedef std::vector<TempTrajectory> TempTrajectoryContainer;
0037
0038   CkfTrajectoryBuilder(const edm::ParameterSet& conf, edm::ConsumesCollector iC);
0039   CkfTrajectoryBuilder(const edm::ParameterSet& conf,
0040                        edm::ConsumesCollector iC,
0041                        std::unique_ptr<TrajectoryFilter> filter);
0042
0043   ~CkfTrajectoryBuilder() override {}
0044
0045   static void fillPSetDescription(edm::ParameterSetDescription& iDesc);
0046
0047   /// trajectories building starting from a seed
0048   TrajectoryContainer trajectories(const TrajectorySeed& seed) const override;
0049   /// trajectories building starting from a seed
0050   void trajectories(const TrajectorySeed& seed, TrajectoryContainer& ret) const override;
0051
0052   void buildTrajectories(const TrajectorySeed&,
0053                          TrajectoryContainer& ret,
0054                          unsigned int& nCandPerSeed,
0055                          const TrajectoryFilter*) const override;
0056
0057   void rebuildTrajectories(const TrajectorySeed&, TrajectoryContainer& result) const override {}
0058
0059   /// set Event for the internal MeasurementTracker data member
0060   //  virtual void setEvent(const edm::Event& event) const;
0061
0062 protected:
0063   int theMaxCand;               /**< Maximum number of trajectory candidates
0064                              to propagate to the next layer. */
0065   float theLostHitPenalty;      /**< Chi**2 Penalty for each lost hit. */
0066   float theFoundHitBonus;       /**< Chi**2 Bonus for each found hit. */
0067   int theMinHitForDoubleBonus;  /**< minimun number of hits to double the bonus above */
0068   bool theIntermediateCleaning; /**< Tells whether an intermediary cleaning stage
0069                                      should take place during TB. */
0070   bool theAlwaysUseInvalidHits;
0071
0072 protected:
0073   void setEvent_(const edm::Event& iEvent, const edm::EventSetup& iSetup) override;
0074
0075   virtual void findCompatibleMeasurements(const TrajectorySeed& seed,
0076                                           const TempTrajectory& traj,
0077                                           std::vector<TrajectoryMeasurement>& result) const;
0078
0079   unsigned int limitedCandidates(const TrajectorySeed& seed,
0080                                  TempTrajectory& startingTraj,
0081                                  TrajectoryContainer& result) const;
0082   unsigned int limitedCandidates(const std::shared_ptr<const TrajectorySeed>& sharedSeed,
0083                                  TempTrajectoryContainer& candidates,
0084                                  TrajectoryContainer& result) const;
0085
0086   void updateTrajectory(TempTrajectory& traj, TM&& tm) const;
0087
0088   /*
0089       //not mature for integration.
0090       bool theSharedSeedCheck;
0091       std::string theUniqueName;
0092       void rememberSeedAndTrajectories(const TrajectorySeed& seed,TrajectoryContainer &result) const;
0093       bool seedAlreadyUsed(const TrajectorySeed& seed,TempTrajectoryContainer &candidates) const;
0094       bool sharedSeed(const TrajectorySeed& seed1,const TrajectorySeed& seed2) const;
0095       //  mutable TempTrajectoryContainer theCachedTrajectories;
0096       typedef boost::unordered_multimap<uint32_t,TempTrajectory> SharedTrajectory;
0097       mutable SharedTrajectory theCachedTrajectories;
0098   */
0099 };
0100
0101 #endif