StEStructMCReader.h

/*********************************************************************
 *
 *  $Id: StEStructMCReader.h,v 1.6 2012/11/16 21:19:07 prindle Exp $
 *
 *  Author: Chunhui Han
 *
 *********************************************************************
 *
 *  Discription:
 *    Read STAR standard MC rootuples, and apply track and
 *    event cuts on it. The output is StEStructEvent.
 *
 *    Modified from the automatically generated files from ROOT.
 *
 ********************************************************************
 *
 *  $Log: StEStructMCReader.h,v $
 *  Revision 1.6  2012/11/16 21:19:07  prindle
 *  Moved EventCuts, TrackCuts to EventReader. Affects most readers.
 *  Added support to write and read EStructEvents.
 *  Cuts: 3D histo support, switch to control filling of histogram for reading EStructEvents
 *  EventCuts: A few new cuts
 *  MuDstReader: Add 2D to some histograms, treat ToFCut, PrimaryCuts, VertexRadius histograms like other cut histograms.
 *  QAHists: Add refMult
 *  TrackCuts: Add some hijing cuts.
 *
 *  Revision 1.5  2006/02/22 22:03:21  prindle
 *  Removed all references to multRef
 *
 *  Revision 1.4  2004/03/18 18:35:17  chunhuih
 *
 *  use const char * instead of char * for the constructor argument filelistfile.
 *
 *  Revision 1.3  2004/03/03 23:17:10  chunhuih
 *
 *  added mNentries to store the total entries in the rootuple, to reduce
 *  redundant code.
 *
 *  Revision 1.2  2004/03/02 21:35:10  chunhuih
 *
 *  added impact parameter information to the StEStructEvent
 *
 *  Revision 1.1  2004/02/26 20:05:33  chunhuih
 *
 *  initial import
 *
 *
 ********************************************************************/

//   This class has been automatically generated 
//     (Thu Feb 12 19:19:06 2004 by ROOT version3.10/01)
//   from TTree h999/HEPEVNT
//   found on file: evgen.1.root

#ifndef STESTRUCTMCREADER_H
#define STESTRUCTMCREADER_H

#include "StEStructPool/AnalysisMaker/StEStructEventReader.h"

#include <TROOT.h>
#include <TChain.h>
#include <TFile.h>
#include <TH1F.h>
class StEStructEventCuts;
class StEStructTrackCuts;

class StEStructMCReader : public StEStructEventReader {
 private:
  TTree          *fChain;   
  Int_t           fCurrent; 
  //Declaration of leaves types
  Int_t           itrac;
  Int_t           istat;
  Int_t           ipdg;
  Int_t           moth1;
  Int_t           moth2;
  Int_t           idau1;
  Int_t           idau2;
  Float_t         Pxyz[3];
  Float_t         ener;
  Float_t         mass;
  Float_t         Vxyz[3];
  Float_t         Vtime;

//List of branches
  TBranch        *b_itrac;   
  TBranch        *b_istat;   
  TBranch        *b_ipdg;   
  TBranch        *b_moth1;   
  TBranch        *b_moth2;   
  TBranch        *b_idau1;   
  TBranch        *b_idau2;   
  TBranch        *b_Pxyz;   
  TBranch        *b_ener;   
  TBranch        *b_mass;   
  TBranch        *b_Vxyz;   
  TBranch        *b_Vtime;   

  int meventsToDo;
  int meventCount;
  int mloopIndex;
  bool mAmDone;
  int mnumTracks;
  int mNentries;

  void fillTracks(StEStructEvent *estructEvent);
  int getTotalEventCount();
  int getCharge(int);

  const int mIPMAX;

 public:
  StEStructMCReader(TTree *tree=0);
  // I moved EventCuts and TrackCuts handling to StEStructEventReader which this inherits from.
  // Think they should be removed from arguments to contructors here. Need to check xml code generating macro where these are used.
  StEStructMCReader(int nevents, TTree *tree = 0, StEStructEventCuts *ecuts = 0, StEStructTrackCuts *tcuts = 0);
  StEStructMCReader(int nevents, const char *fileListFile, StEStructEventCuts *ecuts = 0, StEStructTrackCuts *tcuts = 0);
  ~StEStructMCReader();

  bool hasTree();
  bool measureable(int pid);
  float *globalDCA(float *p, float *v);

  virtual StEStructEvent* next();
  virtual bool            done();

  Int_t  Cut(Int_t entry);
  Int_t  GetEntry(Int_t entry);
  Int_t  LoadTree(Int_t entry);
  void   Init(TTree *tree);
  void   Loop();
  Bool_t Notify();
  void   Show(Int_t entry = -1);

  ClassDef(StEStructMCReader,1)
};

inline bool StEStructMCReader::hasTree() {
  return (fChain == 0);
}

inline bool StEStructMCReader::done() { return mAmDone; }

#endif