GEMSegmentBuilder.h

Go to the documentation of this file.

#ifndef GEMRecHit_GEMSegmentBuilder_h
#define GEMRecHit_GEMSegmentBuilder_h

#include "DataFormats/GEMRecHit/interface/GEMRecHitCollection.h"
#include "DataFormats/GEMRecHit/interface/GEMSegmentCollection.h"
#include "Geometry/GEMGeometry/interface/GEMGeometry.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

class GEMSegmentAlgorithmBase;

class GEMSegmentBuilder {
public:
  explicit GEMSegmentBuilder(const edm::ParameterSet&);
  ~GEMSegmentBuilder();

  void build(const GEMRecHitCollection* rechits, GEMSegmentCollection& oc);

  void setGeometry(const GEMGeometry* g);

  static void fillDescription(edm::ParameterSetDescription& descriptions);

private:
  bool enableGE0;
  bool enableGE12;
  std::string segAlgoName;
  std::string ge0AlgoName;
  edm::ParameterSet segAlgoPSet;
  edm::ParameterSet ge0AlgoPSet;
  std::unique_ptr<GEMSegmentAlgorithmBase> segAlgo;
  std::unique_ptr<GEMSegmentAlgorithmBase> ge0Algo;
  const GEMGeometry* geom_;
};

#endif