StPicoDstReader.h

#ifndef StPicoDstReader_h
#define StPicoDstReader_h

// ROOT headers
#include "TChain.h"
#include "TTree.h"
#include "TFile.h"
#include "TString.h"
#include "TClonesArray.h"

// PicoDst headers
#include "StPicoDst.h"
#include "StPicoEvent.h"
#include "StPicoArrays.h"

//_________________
class StPicoDstReader : public TObject {

 public:

  StPicoDstReader(const Char_t* inFileName);
  ~StPicoDstReader();

  StPicoDst *picoDst()    { return mPicoDst; }
  TChain *chain()         { return mChain; }
  TTree *tree()           { return mTree; }

  void SetStatus(const Char_t* branchNameRegex, Int_t enable);

  void Init();
  Bool_t readPicoEvent(Long64_t iEvent);
  Bool_t ReadPicoEvent(Long64_t iEvent) { return readPicoEvent(iEvent); }
  void Finish();

 private:

  TString mInputFileName;

  void streamerOff();

  void createArrays();
  void clearArrays();
  void setBranchAddresses(TChain *chain);

  StPicoDst *mPicoDst;
  TChain *mChain;
  TTree *mTree;

  Int_t mEventCounter;

  TClonesArray *mPicoArrays[StPicoArrays::NAllPicoArrays];
  Char_t        mStatusArrays[StPicoArrays::NAllPicoArrays];

  ClassDef(StPicoDstReader, 0)
};

#endif