RootChamberWriter.h

Go to the documentation of this file.

#ifndef SimMuon_Neutron_RootChamberWriter_h
#define SimMuon_Neutron_RootChamberWriter_h

#include <TTree.h>
#include <TClonesArray.h>
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"

class RootChamberWriter
{
public:
  RootChamberWriter() : theTree(nullptr), theHits(nullptr) {}
  RootChamberWriter(const std::string & treeName);

  ~RootChamberWriter();

  void write(const edm::PSimHitContainer & hits);

  TTree * tree() {return theTree;}

private:
  TTree * theTree;
  TClonesArray * theHits;
};

#endif