/scratch0/jsnuveri/BDSIM/BDSIMgit/bdsim/include/BDSOutputROOT.hh

#ifndef BDSOutputROOT_h
#define BDSOutputROOT_h 

#ifdef USE_ROOT

#include "BDSOutputBase.hh"
#include "BDSTrajectory.hh"

#include "TROOT.h"
#include "TH1F.h"
#include "TFile.h"
#include "TTree.h"

class BDSOutputROOT : public BDSOutputBase {

public: 

  BDSOutputROOT(); // default constructor
  //  BDSOutput(BDSOutputFormat format);
  virtual ~BDSOutputROOT();

  virtual void WriteHits(BDSSamplerHitsCollection*);
  virtual void WriteEnergyLoss(BDSEnergyCounterHitsCollection*);
  virtual void WritePrimaryLoss(BDSEnergyCounterHit*);
  virtual void WritePrimaryHit(BDSEnergyCounterHit*);
  virtual void WriteTrajectory(std::vector<BDSTrajectory*> &TrajVec);
  virtual void WritePrimary(G4String samplerName, 
                            G4double E,
                            G4double x0,
                            G4double y0,
                            G4double z0,
                            G4double xp,
                            G4double yp,
                            G4double zp,
                            G4double t,
                            G4double weight,
                            G4int PDGType, 
                            G4int nEvent, 
                            G4int TurnsTaken);

  virtual void WriteHistogram(BDSHistogram1D* histogramIn);
  virtual void Commit();  
  virtual void Write();   

private:

  void Init(); 

  void BuildSamplerTree(G4String name);
  TFile* theRootOutputFile;
  //  TTree *theLWCalorimeterTree;
  
  TTree* PrecisionRegionEnergyLossTree;
  TTree* EnergyLossTree;
  TTree* PrimaryLossTree;
  TTree* PrimaryHitsTree;
  
  float x0,xp0,y0,yp0,z0,zp0,E0,t0;
  float x_prod,xp_prod,y_prod,yp_prod,z_prod,zp_prod,E_prod,t_prod;
  float x_lastScat,xp_lastScat,y_lastScat,yp_lastScat,z_lastScat,zp_lastScat,E_lastScat,t_lastScat;
  float x,xp,y,yp,z,zp,E,t; //Edep;
  float X,Xp,Y,Yp,Z,Zp,s,weight; //,EWeightZ;
  int part,nev, pID, track_id, turnnumber;
  float S_el,E_el;
  float X_pl,Y_pl,Z_pl,S_pl,x_pl,y_pl,z_pl,E_pl,weight_pl;
  int part_pl, turnnumber_pl, eventno_pl;
  float X_ph,Y_ph,Z_ph,S_ph,x_ph,y_ph,z_ph,E_ph,weight_ph;
  int part_ph, turnnumber_ph, eventno_ph;
  float X_el_p,Y_el_p,Z_el_p,S_el_p,x_el_p,y_el_p,z_el_p,E_el_p;
  int part_el_p, weight_el_p, turnnumber_el_p, eventno_el_p;
  char volumeName_el_p[100];

  void WriteRootHit(G4String Name, 
                    G4double InitMom, 
                    G4double InitX, 
                    G4double InitY, 
                    G4double InitZ, 
                    G4double InitXPrime, 
                    G4double InitYPrime, 
                    G4double InitZPrime, 
                    G4double InitT, 
                    G4double ProdMom, 
                    G4double ProdX, 
                    G4double ProdY, 
                    G4double ProdZ, 
                    G4double ProdXPrime, 
                    G4double ProdYPrime, 
                    G4double ProdZPrime, 
                    G4double ProdT, 
                    G4double LastScatMom, 
                    G4double LastScatX, 
                    G4double LastScatY, 
                    G4double LastScatZ, 
                    G4double LastScatXPrime, 
                    G4double LastScatYPrime, 
                    G4double LastScatZPrime, 
                    G4double LastScatT, 
                    G4double Mom, 
                    G4double X, 
                    G4double Y, 
                    G4double Z, 
                    G4double XPrime, 
                    G4double YPrime, 
                    G4double ZPrime, 
                    G4double T, 
                    G4double GlobalX, 
                    G4double GlobalY, 
                    G4double GlobalZ, 
                    G4double GlobalXPrime, 
                    G4double GlobalYPrime, 
                    G4double GlobalZPrime, 
                    G4double S, 
                    G4double Weight, 
                    G4int    PDGtype, 
                    G4int    EventNo, 
                    G4int    ParentID,
                    G4int    TrackID, 
                    G4int    TurnsTaken);
  
};

extern BDSOutputBase* bdsOutput;

#endif // USE_ROOT
#endif

---