DataFormats/GEMRecHit/interface/GEMCSCSegment.h - master - githubmirror/cmssw

DataFormats/GEMRecHit/interface/GEMCSCSegment.h
96 строк4 KB
Matti Kortelainen
Move stored types in io_v1 namespace, reset class versions to 3
22 май 2026, 18:18
22 май 2026, 18:18909f324
#ifndef GEMRecHit_GEMCSCSegment_h
#define GEMRecHit_GEMCSCSegment_h

/** \class GEMCSCSegment
 *
 *  Based on CSCSegment class
 *  Describes a reconstructed track segment in the GEM + CSC chambers. 
 *  This is 4-dimensional since it has an origin (x,y) and a direction (x,y)
 *  in the local coordinate system of the (csc) chamber.
 *
 *  \author R. Radogna
 */

#include "DataFormats/TrackingRecHit/interface/RecSegment.h"

#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCRecHit2DCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
#include "DataFormats/MuonDetId/interface/CSCDetIdFwd.h"

#include <iosfwd>

namespace io_v1 {

  class GEMCSCSegment final : public RecSegment {
  public:
    /// Default constructor
    GEMCSCSegment() : theChi2(0.) {}

    /// Constructor
    GEMCSCSegment(const CSCSegment* csc_segment,
                  const std::vector<const GEMRecHit*> gem_rhs,
                  LocalPoint origin,
                  LocalVector direction,
                  AlgebraicSymMatrix errors,
                  double chi2);

    /// Destructor
    ~GEMCSCSegment() override;

    //--- Base class interface
    GEMCSCSegment* clone() const override { return new GEMCSCSegment(*this); }

    LocalPoint localPosition() const override { return theOrigin; }
    LocalError localPositionError() const override;

    LocalVector localDirection() const override { return theLocalDirection; }
    LocalError localDirectionError() const override;

    /// Parameters of the segment, for the track fit in the order (dx/dz, dy/dz, x, y )
    AlgebraicVector parameters() const override;

    /// Covariance matrix of parameters()
    AlgebraicSymMatrix parametersError() const override { return theCovMatrix; }

    /// The projection matrix relates the trajectory state parameters to the segment parameters().
    AlgebraicMatrix projectionMatrix() const override;

    double chi2() const override { return theChi2; };

    int dimension() const override { return 4; }

    int degreesOfFreedom() const override { return 2 * nRecHits() - 4; }

    int nRecHits() const { return (theGEMRecHits.size() + theCSCSegment.specificRecHits().size()); }

    //--- Return the constituents in different ways
    const CSCSegment cscSegment() const { return theCSCSegment; }
    const std::vector<GEMRecHit>& gemRecHits() const { return theGEMRecHits; }
    const std::vector<CSCRecHit2D>& cscRecHits() const { return theCSCSegment.specificRecHits(); }
    std::vector<const TrackingRecHit*> recHits() const override;
    std::vector<TrackingRecHit*> recHits() override;

    CSCDetId cscDetId() const { return geographicalId(); }

    void print() const;

  private:
    std::vector<GEMRecHit> theGEMRecHits;  // store GEM Rechits
    CSCSegment theCSCSegment;              // store CSC RecHits and store CSC Segment
                                           // eventually we have to disentangle if later on we decide
                                           // not to have a one-to-one relationship anymore
                                           // i.e. if we allow the GEMCSC segment to modify the
                                           // (selection of the rechits of the) CSC segment
    LocalPoint theOrigin;                  // in chamber frame - the GeomDet local coordinate system
    LocalVector theLocalDirection;         // in chamber frame - the GeomDet local coordinate system
    AlgebraicSymMatrix theCovMatrix;       // the covariance matrix
    double theChi2;
  };

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

}  // namespace io_v1
using GEMCSCSegment = io_v1::GEMCSCSegment;

#endif