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