#include "TROOT.h"
#include "IMidasFile.hxx"
#include "ICOMETRawEvent.hxx"
#include "TObjString.h"
#include "TTree.h"

namespace COMET{
  class IMidasFile;
};

class IDataTree {

public:

  void ProcessEvents();
  bool ParseArgs(int argc, char* argv[]);
  void InitialiseVariables();
  UInt_t HeaderTime();
  void ReadEvents();
  void InitialiseFilesAndTrees();
  void Usage();
  
  virtual ~IDataTree();

private:

  // Varoiables to be saved to flat tree for each selected hit
  Int_t fsubDet;
  Int_t fDetector;
  Int_t fnEvent;
  Int_t fnFile;
  Int_t fNumberFiles;
  Int_t fRMM;
  Int_t fTFB;
  Int_t fTripT;
  UShort_t fHighADC;
  UShort_t fLowADC;
  UInt_t fUnixTime;
  Long64_t fTimeRelTrig;
  // SMRD
  UInt_t fprmtv_top;
  UInt_t fprmtv_bottom;
  UInt_t fprmtv_south;
  UInt_t fprmtv_north;
  // POD
  UInt_t fprmtv_p0d;
  // DsECAL
  UInt_t fprmtv_dsecal;

  // Input Midas files
  COMET::IMidasFile* fInputFile;
  // Ouptput root file
  TFile* fRootFile;
  // Files read in from command line argument
  std::vector<std::string>* fFiles;
  // For naminng output ROOT file
  std::string fOutputRootFile;
  // Raw event
  COMET::ICOMETRawEvent* fRawEvent;
  // Output tree
  TTree* fOffsetTree;
};