SubsystemNeutronWriter.h

Go to the documentation of this file.

#ifndef SubsystemNeutronWriter_h
#define SubsystemNeutronWriter_h
 
#include <vector>
#include <map>
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
namespace CLHEP {
  class HepRandomEngine;
}
 
class NeutronWriter;
 
class SubsystemNeutronWriter : public edm::stream::EDProducer<> {
public:
  explicit SubsystemNeutronWriter(edm::ParameterSet const& pset);
 
  ~SubsystemNeutronWriter() override;
 
  void printStats();
 
  void produce(edm::Event& e, edm::EventSetup const& c) override;
 
  virtual int localDetId(int globalDetId) const = 0;
 
  virtual int chamberType(int globalDetId) const = 0;
 
  virtual int chamberId(int globalDetId) const = 0;
 
  virtual bool accept(const edm::PSimHitContainer& cluster) const = 0;
 
  void initialize(int chamberType);
 
protected:
  virtual void writeHits(int chamberType, edm::PSimHitContainer& chamberHits, CLHEP::HepRandomEngine*);
 
  void writeCluster(int chamberType, const edm::PSimHitContainer& cluster);
 
  void adjust(PSimHit& h, float timeOffset, float smearing);
 
  void updateCount(int chamberType);
 
private:
  NeutronWriter* theHitWriter;
  bool useRandFlat;
  edm::InputTag theInputTag;
  double theNeutronTimeCut;
  double theTimeWindow;
  double theT0;
  int theNEvents;
  bool initialized;
  // true means to translate DetId into just layer number, e.g., 1-6 in CSC
  bool useLocalDetId_;
  std::map<int, int> theCountPerChamberType;
};
 
#endif