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;
  const edm::InputTag theInputTag;
  const double theNeutronTimeCut;
  const double theTimeWindow;
  const double theT0;
  const edm::EDGetTokenT<edm::PSimHitContainer> hitToken_;
  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