RootNeutronWriter.h

Go to the documentation of this file.

#ifndef SimMuon_Neutron_RootNeutronWriter_h
#define SimMuon_Neutron_RootNeutronWriter_h

#include "SimMuon/Neutron/src/NeutronWriter.h"
#include "SimMuon/Neutron/src/RootChamberWriter.h"
#include <TFile.h>
#include <vector>
class RootNeutronWriter : public NeutronWriter {
public:
  RootNeutronWriter(const std::string& fileName);
  ~RootNeutronWriter() override;

  void initialize(int detType) override;

  RootChamberWriter& chamberWriter(int chamberType);

  void writeCluster(int chamberType, const edm::PSimHitContainer& hits) override;

private:
  std::map<int, RootChamberWriter> theChamberWriters;
  TFile* theFile;
};

#endif