DTRecSegment4D.h

Go to the documentation of this file.

#ifndef DTRecHit_DTRecSegment4D_h
#define DTRecHit_DTRecSegment4D_h

/* Base Class Headers */
#include "DataFormats/TrackingRecHit/interface/RecSegment.h"

/* Collaborating Class Declarations */
#include "DataFormats/DTRecHit/interface/DTSLRecSegment2D.h"
#include "DataFormats/DTRecHit/interface/DTChamberRecSegment2D.h"

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

class DTRecSegment4D : public RecSegment {
public:
  friend class DTSegmentUpdator;
  DTRecSegment4D() : theProjection(none), theDimension(0) {}

  DTRecSegment4D(const DTChamberRecSegment2D& phiSeg,
                 const DTSLRecSegment2D& zedSeg,
                 const LocalPoint& posZInCh,
                 const LocalVector& dirZInCh);

  DTRecSegment4D(const DTChamberRecSegment2D& phiSeg);

  DTRecSegment4D(const DTSLRecSegment2D& zedSeg, const LocalPoint& posZInCh, const LocalVector& dirZInCh);

  ~DTRecSegment4D() override;

  //--- Base class interface

  DTRecSegment4D* clone() const override { return new DTRecSegment4D(*this); }

  AlgebraicVector parameters() const override;

  AlgebraicSymMatrix parametersError() const override;

  AlgebraicMatrix projectionMatrix() const override;

  LocalPoint localPosition() const override { return thePosition; }

  LocalError localPositionError() const override;

  LocalVector localDirection() const override { return theDirection; }

  LocalError localDirectionError() const override;

  // Chi2 of the segment fit
  double chi2() const override;

  // Degrees of freedom of the segment fit
  int degreesOfFreedom() const override;

  // Dimension (in parameter space)
  int dimension() const override { return theDimension; }

  // Access to component RecHits (if any)
  std::vector<const TrackingRecHit*> recHits() const override;

  // Non-const access to component RecHits (if any)
  std::vector<TrackingRecHit*> recHits() override;

  //--- Extension of the interface

  bool hasPhi() const { return (theProjection == full || theProjection == phi); }

  bool hasZed() const { return (theProjection == full || theProjection == Z); }

  const DTChamberRecSegment2D* phiSegment() const { return hasPhi() ? &thePhiSeg : nullptr; }

  const DTSLRecSegment2D* zSegment() const { return hasZed() ? &theZedSeg : nullptr; }

  void setPosition(LocalPoint pos) { thePosition = pos; }

  void setDirection(LocalVector dir) { theDirection = dir; }

  void setCovMatrix(const AlgebraicSymMatrix& mat) { theCovMatrix = mat; }

  virtual DTChamberId chamberId() const;

private:
  enum Projection { full, phi, Z, none };
  Projection theProjection;

  DTChamberRecSegment2D* phiSegment() { return &thePhiSeg; }

  DTSLRecSegment2D* zSegment() { return &theZedSeg; }

  LocalPoint thePosition;    // in chamber frame
  LocalVector theDirection;  // in chamber frame

  void setCovMatrixForZed(const LocalPoint& posZInCh);

  // the covariance matrix, has the following meaning
  // mat[0][0]=sigma (dx/dz)
  // mat[1][1]=sigma (dy/dz)
  // mat[2][2]=sigma (x)
  // mat[3][3]=sigma (y)
  // mat[0][2]=cov(dx/dz,x)
  // mat[1][3]=cov(dy/dz,y)
  AlgebraicSymMatrix theCovMatrix;

  DTChamberRecSegment2D thePhiSeg;
  DTSLRecSegment2D theZedSeg;

  int theDimension;  // the dimension of this rechit
};

std::ostream& operator<<(std::ostream& os, const DTRecSegment4D& seg);

#endif  // DTRecHit_DTRecSegment4D_h