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