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);
  virtual ~RootNeutronWriter();

  virtual void initialize(int detType);

  RootChamberWriter & chamberWriter(int chamberType);

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

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

#endif