// Author: Pierre-Luc Drouin <http://www.pldrouin.net>
// Copyright Carleton University

#ifndef _QSIGEXCLEANDATA_
#define _QSIGEXCLEANDATA_

#include "QSigExDirHandler.h"
#include "TDirectory.h"
#include "TTree.h"
#include "TList.h"
#include "TNamed.h"
#include "TFormula.h"
#include "TCut.h"
#include "TCollection.h"
#include "TFile.h"
#include "TBranch.h"
#include "TLeaf.h"
#include "TEventList.h"
#include "TKey.h"
#include "QFormulaUtils.h"

//#define DEBUG
//#define DEBUG2

#include "debugger.h"


//////////////////////////////////////////////////////////////////////////
//                                                                      //
// QSigExCleanData                                                      //
//                                                                      //
// This class applies the cuts defined in the TDirectory structure and  //
// applies them on raw data events to produce a TTree containing clean  //
// data. The new TTree is placed in "Event Info" TDirectory             //
//                                                                      //
//////////////////////////////////////////////////////////////////////////

class QSigExCleanData: public QSigExDirHandler{
 public:
  QSigExCleanData():QSigExDirHandler(),fEInfoDir(NULL),fDataFName(),fDataOName(){

    PRINTF2(this,"\tQSigExCleanData::QSigExCleanData()\n")
  }

  QSigExCleanData(const QSigExCleanData& rhs):QSigExDirHandler(*this),fEInfoDir(rhs.fEInfoDir),fDataFName(rhs.fDataFName),fDataOName(rhs.fDataOName){
    PRINTF2(this,"\tQSigExCleanData::QSigExCleanData(const QSigExCleanData& rhs)\n")
  }

  QSigExCleanData(TDirectory* dir,const Char_t* cardfilename=NULL):QSigExDirHandler(),fEInfoDir(NULL),fDataFName(),fDataOName()
    {
      PRINTF6(this,"\tQSigExCleanData::QSigExCleanData(TDirectory* dir<",
	      dir,">,const Char_t* cardfilename<",cardfilename,">)\n")
      try{
	SetDir(dir);
	LoadCardFile(cardfilename);
      }catch(int e){
	cout << "Exception handled by QSigExCleanData::QSigExCleanData\n";
	throw e;
      }
    }

  virtual ~QSigExCleanData(){}

  void LoadCardFile(const Char_t* cardfilename=NULL);

  void CleanDir();
  void ClearCardBuf();

  Int_t Get();
  const Char_t* GetDataFName() {return fDataFName;}
  const Char_t* GetDataOName() {return fDataOName;}

  void SetDataFName(const Char_t* datafname) {fDataFName=datafname;}
  void SetDataOName(const Char_t* dataoname) {fDataOName=dataoname;}

 protected:
  void FormatDir();

  void SetDataFilename(const Char_t* fname);
  void SetDataObjectName(const Char_t* fname);

  void CheckIfDataFilename() const;
  void CheckIfDataObjectName() const;

  void CheckCuts() const;
 private:
  TDirectory* fEInfoDir; //Pointer to the TDirectory that contains event information
  TString fDataFName; //String that contains the raw data filename
  TString fDataOName; //String that contains the raw data object name

  ClassDef (QSigExCleanData,1)  //Applies cuts on raw data events and fills a TTree with clean data
};


#include "debugger.h"
#endif