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