#if !defined(ROOTDATAREADER)
#define ROOTDATAREADER

#include "IUAmpTools/DataReader.h"
#include "IUAmpTools/Kinematics.h"

#include "TString.h"
#include "TFile.h"
#include "TTree.h"

#include <string>

using namespace std;

class ROOTDataReader : public DataReader
{
	
public:
  
  /**
   * Constructor for ROOTDataReader
   * \param[in] inFileName ROOT file name to be read.
   * \param[in] inTreeName name of tree from which to read events.
   * \param[in] useWeight (optional) enables reading in event weights
   */
  ROOTDataReader( const string& inFileName, 
		  const string& inTreeName, bool useWeight=false);
  
  virtual Kinematics* getEvent();
  virtual void resetSource();

  /**
   * This function returns a true if the file was open
   * with weight-reading enabled and had this tree branch,
   * false, if these criteria are not met.
   */
  virtual bool hasWeight(){ return m_useWeight; };
  virtual unsigned int numEvents() const;
  
private:
	
  TFile* m_inFile;
  TTree* m_inTree;
  unsigned int m_eventCounter;
  bool m_useWeight;
  
  int m_nPart;
  float m_e[Kinematics::kMaxParticles];
  float m_px[Kinematics::kMaxParticles];
  float m_py[Kinematics::kMaxParticles];
  float m_pz[Kinematics::kMaxParticles];
  float m_eBeam;
  float m_pxBeam;
  float m_pyBeam;
  float m_pzBeam;
  float m_eRecoil;
  float m_pxRecoil;
  float m_pyRecoil;
  float m_pzRecoil;
  float m_weight;
};

#endif