ChamberSegmentUtility.h

Go to the documentation of this file.

#ifndef EstimatorTool_RecSegment_h
#define EstimatorTool_RecSegment_h

#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"
#include "DataFormats/GeometrySurface/interface/LocalError.h"
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"
#include "Geometry/CSCGeometry/interface/CSCGeometry.h"
#include "Geometry/Records/interface/GlobalTrackingGeometryRecord.h"
#include "Geometry/DTGeometry/interface/DTChamber.h"
#include "DataFormats/DTRecHit/interface/DTRecSegment4D.h"
#include "DataFormats/DTRecHit/interface/DTRecSegment4DCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCSegment.h"
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"


class ChamberSegmentUtility {

 public:

  ChamberSegmentUtility(const edm::Event&, const edm::EventSetup&);

  // Get the 4D segments in a CSC chamber
  std::vector<CSCSegment> getCSCSegmentsInChamber(CSCDetId);

  // Get the 4D segments in a DT chamber
  std::vector<DTRecSegment4D> getDTSegmentsInChamber(DTChamberId);

  // Get the list of DT chambers with segments
  const std::map<int, std::vector<DTRecSegment4D> >& getDTlist() const { return dtsegMap; };

  // Get the list of CSC chambers with segments
  const std::map<int, std::vector<CSCSegment> >& getCSClist() const { return cscsegMap; };

  // Get the map association between segments4d and rechits
  std::vector<DTRecHit1D> getDTRHmap(const DTRecSegment4D&);

  // Get the map association between segments4d and rechits 
  std::vector<CSCRecHit2D> getCSCRHmap(const CSCSegment&);

  
 private:

  edm::ESHandle<CSCGeometry> cscGeometry;
  edm::Handle<CSCSegmentCollection> CSCSegments;
  edm::ESHandle<DTGeometry> dtGeom;
  edm::Handle<DTRecSegment4DCollection> all4DSegments;

  std::vector<DTRecSegment4D> dtseg;
  std::vector<CSCSegment> cscseg;
  std::map<int, std::vector<DTRecSegment4D> > dtsegMap;
  std::map<int, std::vector<CSCSegment> > cscsegMap;
  DTChamberId selectedDT;
  CSCDetId selectedCSC;
  std::vector<DTRecHit1D> phiSegRH;
  std::vector<DTRecHit1D> zSegRH;  
};

#endif