ME0DigiPreRecoModel.h

Go to the documentation of this file.

#ifndef SimMuon_GEMDigitizer_ME0DigiPreRecoModel_h
#define SimMuon_GEMDigitizer_ME0DigiPreRecoModel_h

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/GEMDigi/interface/ME0DigiPreRecoCollection.h"
#include "DataFormats/Common/interface/DetSet.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "SimDataFormats/CrossingFrame/interface/MixCollection.h"

#include <map>
#include <set>

namespace CLHEP {
  class HepRandomEngine;
}

class ME0EtaPartition;
class ME0Geometry;
class PSimHit;

class ME0DigiPreRecoModel {
public:
  virtual ~ME0DigiPreRecoModel() {}

  void setGeometry(const ME0Geometry* geom) { geometry_ = geom; }

  const ME0Geometry* getGeometry() { return geometry_; }

  virtual void simulateSignal(const ME0EtaPartition*, const edm::PSimHitContainer&, CLHEP::HepRandomEngine* engine) = 0;

  virtual void simulateNoise(const ME0EtaPartition*, CLHEP::HepRandomEngine* engine) = 0;

  //  virtual std::vector<std::pair<int,int> >
  //   simulateClustering(const ME0EtaPartition*, const PSimHit*, const int) = 0;

  void fillDigis(int rollDetId, ME0DigiPreRecoCollection&);

  virtual void setup() = 0;

protected:
  std::set<ME0DigiPreReco> digi_;

  ME0DigiPreRecoModel(const edm::ParameterSet&) {}

  const ME0Geometry* geometry_;

  //  DetectorHitMap detectorHitMap_;
};
#endif