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