NuclearInteractionFinder.h

Go to the documentation of this file.

#ifndef CD_NuclearInteractionFinder_H_
#define CD_NuclearInteractionFinder_H_

//----------------------------------------------------------------------------
//-----------------------------------------------------------------------------

#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"
#include "TrackingTools/PatternTools/interface/TrajectoryStateUpdator.h"
#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"
#include "TrackingTools/PatternTools/interface/TrajectoryMeasurement.h"
#include "TrackingTools/DetLayers/interface/MeasurementEstimator.h"
#include "TrackingTools/MeasurementDet/interface/LayerMeasurements.h"

#include "RecoTracker/MeasurementDet/interface/MeasurementTracker.h"
#include "RecoTracker/MeasurementDet/interface/MeasurementTrackerEvent.h"
#include "RecoTracker/TkDetLayers/interface/GeometricSearchTracker.h"

#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"

#include "RecoTracker/NuclearSeedGenerator/interface/NuclearTester.h"
#include "RecoTracker/NuclearSeedGenerator/interface/SeedFromNuclearInteraction.h"
#include "RecoTracker/NuclearSeedGenerator/interface/TangentHelix.h"

#include "TrackingTools/DetLayers/interface/NavigationSchool.h"

class NuclearInteractionFinder {
private:
  typedef TrajectoryStateOnSurface TSOS;
  typedef FreeTrajectoryState FTS;
  typedef TrajectoryMeasurement TM;
  typedef std::vector<Trajectory> TrajectoryContainer;
  typedef TrajectoryMeasurement::ConstRecHitPointer ConstRecHitPointer;

  void fillSeeds(const std::pair<TrajectoryMeasurement, std::vector<TrajectoryMeasurement> >& tmPairs);

  std::vector<TrajectoryMeasurement> findCompatibleMeasurements(const TM& lastMeas,
                                                                double rescaleFactor,
                                                                const LayerMeasurements& layerMeasurements) const;

  std::vector<TrajectoryMeasurement> findMeasurementsFromTSOS(const TSOS& currentState,
                                                              DetId detid,
                                                              const LayerMeasurements& layerMeasurements) const;

  void definePrimaryHelix(std::vector<TrajectoryMeasurement>::const_iterator it_meas);

public:
  struct Config {
    double rescaleErrorFactor;
    double ptMin;  //passed to SeedFromNuclearInteraction
    unsigned int maxHits;
    bool checkCompletedTrack; 
  };

  NuclearInteractionFinder(const Config& iConfig,
                           const TrackerGeometry* theTrckerGeom,
                           const Propagator* thePropagator,
                           const MeasurementEstimator* theEstimator,
                           const MeasurementTracker* theMeasurementTracker,
                           const GeometricSearchTracker* theGeomSearchTracker,
                           const NavigationSchool* theNavigationSchool);

  bool run(const Trajectory& traj, const MeasurementTrackerEvent& event);

  void improveSeeds(const MeasurementTrackerEvent& event);

  std::unique_ptr<TrajectorySeedCollection> getPersistentSeeds();

  TrajectoryStateOnSurface rescaleError(float rescale, const TSOS& state) const;

  const NavigationSchool* nav() const { return theNavigationSchool; }

private:
  const Propagator* thePropagator;
  const MeasurementEstimator* theEstimator;
  const MeasurementTracker* theMeasurementTracker;
  const GeometricSearchTracker* theGeomSearchTracker;
  const NavigationSchool* theNavigationSchool;

  std::unique_ptr<NuclearTester> nuclTester;
  std::unique_ptr<SeedFromNuclearInteraction> currentSeed;
  std::vector<SeedFromNuclearInteraction> allSeeds;
  std::unique_ptr<TangentHelix> thePrimaryHelix;

  // parameters
  unsigned int maxHits;
  double rescaleErrorFactor;
  bool checkCompletedTrack; 
};
#endif