MuonTrajectoryUpdator.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackingTools_MuonTrajectoryUpdator_H
#define RecoMuon_TrackingTools_MuonTrajectoryUpdator_H
 
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
#include "TrackingTools/DetLayers/interface/NavigationDirection.h"
 
#include <functional>
 
class Propagator;
class MeasurementEstimator;
class TrajectoryMeasurement;
class Trajectory;
class TrajectoryStateOnSurface;
class TrajectoryStateUpdator;
class DetLayer;
 
namespace edm {
  class ParameterSet;
}
 
class MuonTrajectoryUpdator {
public:
  MuonTrajectoryUpdator(const edm::ParameterSet &par, NavigationDirection fitDirection);
 
  MuonTrajectoryUpdator(NavigationDirection fitDirection, double chi2, int granularity);
 
  virtual ~MuonTrajectoryUpdator();
 
  // Operations
 
  virtual std::pair<bool, TrajectoryStateOnSurface> update(const TrajectoryMeasurement *measurement,
                                                           Trajectory &trajectory,
                                                           const Propagator *propagator);
 
  const MeasurementEstimator *estimator() const { return theEstimator; }
  const TrajectoryStateUpdator *measurementUpdator() const { return theUpdator; }
 
  double maxChi2() const { return theMaxChi2; }
 
  NavigationDirection fitDirection() { return theFitDirection; }
 
  void setMaxChi2(double chi2) { theMaxChi2 = chi2; }
 
  void setFitDirection(NavigationDirection fitDirection) { theFitDirection = fitDirection; }
 
  void makeFirstTime();
 
protected:
private:
  TrajectoryStateOnSurface propagateState(const TrajectoryStateOnSurface &state,
                                          const TrajectoryMeasurement *measurement,
                                          const TransientTrackingRecHit::ConstRecHitPointer &current,
                                          const Propagator *propagator) const;
 
  double theMaxChi2;
 
  int theGranularity;
  // FIXME: ask Tim if the CSC segments can be used, since in ORCA they wasn't.
 
  struct RadiusComparatorInOut {
    bool operator()(const TransientTrackingRecHit::ConstRecHitPointer &a,
                    const TransientTrackingRecHit::ConstRecHitPointer &b) const {
      return a->det()->surface().position().perp() < b->det()->surface().position().perp();
    }
  };
 
  struct RadiusComparatorOutIn {
    bool operator()(const TransientTrackingRecHit::ConstRecHitPointer &a,
                    const TransientTrackingRecHit::ConstRecHitPointer &b) const {
      return a->det()->surface().position().perp() > b->det()->surface().position().perp();
    }
  };
 
  struct ZedComparatorInOut {
    bool operator()(const TransientTrackingRecHit::ConstRecHitPointer &a,
                    const TransientTrackingRecHit::ConstRecHitPointer &b) const {
      return fabs(a->globalPosition().z()) < fabs(b->globalPosition().z());
    }
  };
 
  struct ZedComparatorOutIn {
    bool operator()(const TransientTrackingRecHit::ConstRecHitPointer &a,
                    const TransientTrackingRecHit::ConstRecHitPointer &b) const {
      return fabs(a->globalPosition().z()) > fabs(b->globalPosition().z());
    }
  };
 
  void sort(TransientTrackingRecHit::ConstRecHitContainer &, const DetLayer *);
 
  TrajectoryMeasurement updateMeasurement(const TrajectoryStateOnSurface &propagatedTSOS,
                                          const TrajectoryStateOnSurface &lastUpdatedTSOS,
                                          const TransientTrackingRecHit::ConstRecHitPointer &recHit,
                                          const double &chi2,
                                          const DetLayer *detLayer,
                                          const TrajectoryMeasurement *initialMeasurement);
 
  // FIXME: change in a ESHandle
  MeasurementEstimator *theEstimator;
  TrajectoryStateUpdator *theUpdator;
 
  // The fit direction.This is the global fit direction and it could be (LOCALLY!) different w.r.t. the
  // propagation direction embeeded in the propagator (i.e. when it is used in the "anyDirection" mode)
  // This data member is not set via parameter set since it must be consistent with the RefitterParameter.
  NavigationDirection theFitDirection;
 
  // Parameters for the error rescaling
  bool theFirstTSOSFlag;
  bool theRescaleErrorFlag;
  double theRescaleFactor;
 
  // parameter to use the "bad hits"
  bool useInvalidHits;
  // exclude RPC from the fit (but allows to keep it in the muon signal confirmation)
  bool theRPCExFlag;
};
#endif