DTSegmentUpdator.h

Go to the documentation of this file.

#ifndef DTSegment_DTSegmentUpdator_h
#define DTSegment_DTSegmentUpdator_h

/* C++ Headers */
#include <vector>

#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "Geometry/DTGeometry/interface/DTGeometry.h"

/* ====================================================================== */

/* Collaborating Class Declarations */
class DTSegmentCand;
class DTRecSegment2D;
class DTRecSegment4D;
class DTLinearFit;
class DTRecHitBaseAlgo;
class DTChamberRecSegment2D;
class DTGeometry;
class MuonGeometryRecord;

namespace edm {
  class EventSetup;
  class ParameterSet;
}  // namespace edm

/* Class DTSegmentUpdator Interface */
class DTSegmentUpdator {
public:
  DTSegmentUpdator(const edm::ParameterSet& config, edm::ConsumesCollector);

  ~DTSegmentUpdator();

  /* Operations */

  bool fit(DTSegmentCand* seg, bool allow3par, const bool fitdebug) const;

  void fit(DTRecSegment2D* seg, bool allow3par, bool block3par) const;

  void fit(DTRecSegment4D* seg, bool allow3par) const;

  void update(DTRecSegment4D* seg, const bool calcT0, bool allow3par) const;

  void update(DTRecSegment2D* seg, bool allow3par) const;

  void calculateT0corr(DTRecSegment2D* seg) const;
  void calculateT0corr(DTRecSegment4D* seg) const;

  void setES(const edm::EventSetup& setup);

protected:
private:
  std::unique_ptr<DTLinearFit> theFitter;     // the linear fitter
  std::unique_ptr<DTRecHitBaseAlgo> theAlgo;  // the algo for hit reconstruction
  edm::ESHandle<DTGeometry> theGeom;          // the geometry
  const edm::ESGetToken<DTGeometry, MuonGeometryRecord> theGeomToken;

  void updateHits(DTRecSegment2D* seg, GlobalPoint& gpos, GlobalVector& gdir, const int step = 2) const;

  //rejects bad hits (due to deltas) for phi segment
  void rejectBadHits(DTChamberRecSegment2D*) const;

  void fit(const std::vector<float>& x,
           const std::vector<float>& y,
           const std::vector<int>& lfit,
           const std::vector<double>& dist,
           const std::vector<float>& sigy,
           LocalPoint& pos,
           LocalVector& dir,
           float& cminf,
           float& vminf,
           AlgebraicSymMatrix& covMat,
           double& chi2,
           const bool allow3par = false,
           const bool block3par = false) const;

  double intime_cut;
  bool vdrift_4parfit;
  double T0_hit_resolution;
  bool perform_delta_rejecting;
  bool debug;
};

#endif  // DTSegment_DTSegmentUpdator_h