#if !defined(ROOTDATAREADER)
#define ROOTDATAREADER

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

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

#include <string>

using namespace std;

class ROOTDataReader : public UserDataReader< ROOTDataReader >
{
	
public:
  
  /**
   * Default constructor for ROOTDataReader
   */
  ROOTDataReader() : UserDataReader< ROOTDataReader >(), m_inFile( NULL ) { }
  
  ~ROOTDataReader();
  
  /**
   * Constructor for ROOTDataReader
   * \param[in] args vector of string arguments
   */
  ROOTDataReader( const vector< string >& args );
  
  string name() const { return "ROOTDataReader"; }
  
  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