#ifndef QATOOLS_WRITERS_ANALYSISTREE2_H
#define QATOOLS_WRITERS_ANALYSISTREE2_H

#include <Rtypes.h>
#include <TString.h>
#include <TFile.h>
#include <TTree.h>

#include <qaEvent.h>
#include <qaParticle.h>
#include <Utility.h>

// AnalysisTree headers
#include <AnalysisTree/Configuration.hpp>
#include <AnalysisTree/DataHeader.hpp>
#include <AnalysisTree/EventHeader.hpp>
#include <AnalysisTree/Detector.hpp>
#include <AnalysisTree/Matching.hpp>

class qaWriter_analysistree2
{
private:
  /* data */
  TFile *fFile;
  TTree *fTree;
  AnalysisTree::DataHeader *fDataHeader;
  AnalysisTree::Configuration *fConfig;
  AnalysisTree::EventHeader *fEvent;
  AnalysisTree::Particles *fParticles;

  int iB, iPhiRp, icharge;

  bool isInit;
public:
  virtual ~qaWriter_analysistree2();

  virtual void Init(std::string filename, std::string treename);
  virtual void WriteEvent(qaEvent* event);
  virtual void WriteParticle(qaParticle* particle);
  virtual void WriteTree();
  virtual void FillTree();

  ClassDef(qaWriter_analysistree2,0);
};

#endif