RecoLocalMuon/DTSegment/src/DTMeantimerPatternReco4D.h - master - githubmirror/cmssw

RecoLocalMuon/DTSegment/src/DTMeantimerPatternReco4D.h
97 строк3 KB
Christopher Jones
Added esConsumes to code in RecoLocalMuon/DTSegment
08 сен 2021, 01:03
08 сен 2021, 01:0309f7c0b
#ifndef DTSegment_DTMeantimerPatternReco4D_h
#define DTSegment_DTMeantimerPatternReco4D_h

/** \class DTMeantimerPatternReco4D
 *
 * Algo for reconstructing 4d segment in DT using a Meantimer approach
 *  
 * \author Stefano Lacaprara - INFN Legnaro <stefano.lacaprara@pd.infn.it>
 * \author Riccardo Bellan - INFN TO <riccardo.bellan@cern.ch>
 *
 */

// Base Class Headers
#include "RecoLocalMuon/DTSegment/src/DTRecSegment4DBaseAlgo.h"

class DTRecSegment2DBaseAlgo;

// Collaborating Class Declarations
namespace edm {
  class ParameterSet;
  class EventSetup;
}  // namespace edm
class DTSegmentUpdator;
class MuonGeometryRecord;
//class DTSegmentCleaner;

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

#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "FWCore/Framework/interface/FrameworkfwdMostUsed.h"

// ======================================================================
class DTSegmentCand;
class DTMeantimerPatternReco;
class DTHitPairForFit;

// Class DTMeantimerPatternReco4D Interface

class DTMeantimerPatternReco4D : public DTRecSegment4DBaseAlgo {
public:
  /// Constructor
  DTMeantimerPatternReco4D(const edm::ParameterSet& pset, edm::ConsumesCollector);

  /// Destructor
  ~DTMeantimerPatternReco4D() override;

  /// Operations
  edm::OwnVector<DTRecSegment4D> reconstruct() override;

  std::string algoName() const override { return theAlgoName; }

  void setES(const edm::EventSetup& setup) override;
  void setDTRecHit1DContainer(edm::Handle<DTRecHitCollection> all1DHits) override;
  void setDTRecSegment2DContainer(edm::Handle<DTRecSegment2DCollection> all2DSegments) override;
  void setChamber(const DTChamberId& chId) override;
  bool wants2DSegments() override { return !allDTRecHits; }

protected:
private:
  std::vector<DTSegmentCand*> buildPhiSuperSegmentsCandidates(
      std::vector<std::shared_ptr<DTHitPairForFit>>& pairPhiOwned);
  DTRecSegment4D* segmentSpecialZed(DTRecSegment4D* seg);

  std::string theAlgoName;

  bool debug;
  // DTSegmentUpdator* theUpdator; // the updator and fitter
  // DTSegmentCleaner* theCleaner; // the cleaner

  edm::ESHandle<DTGeometry> theDTGeometry;  // the DT geometry
  const edm::ESGetToken<DTGeometry, MuonGeometryRecord> theDTGeometryToken;

  // The reconstruction 2D algorithm
  // For the 2D reco I use thei reconstructor!
  DTMeantimerPatternReco* the2DAlgo;

  // the updator
  DTSegmentUpdator* theUpdator;

  const DTChamber* theChamber;

  //the input type
  bool allDTRecHits;
  bool applyT0corr;
  bool computeT0corr;

  //  std::vector<DTRecHit1D> the1DPhiHits;
  std::vector<DTSLRecSegment2D> theSegments2DTheta;
  std::vector<DTRecHit1DPair> theHitsFromPhi1;
  std::vector<DTRecHit1DPair> theHitsFromTheta;
  std::vector<DTRecHit1DPair> theHitsFromPhi2;
};
#endif