StPicoDstMaker.h

#ifndef StPicoDstMaker_h
#define StPicoDstMaker_h

// StChain headers
#include "StChain/StMaker.h"

// PicoDst headers
#include "StPicoEvent/StPicoArrays.h"
#include "StPicoDstMaker/StPicoFmsFiller.h"

#if defined (__TFG__VERSION__)
#include "StMuDSTMaker/COMMON/StMuDst.h"
#endif /* __TFG__VERSION__ */

class TClonesArray;
class TChain;
class TFile;
class TTree;
class StMuDst;
class StMuTrack;
class StEmcCollection;
class StEmcPosition;
class StEmcGeom;
class StEmcRawHit;
class StPicoDst;
class StPicoEvent;

//_________________
class StPicoDstMaker : public StMaker {

 public:

  enum PicoIoMode {IoWrite=1, IoRead=2};
#if !defined (__TFG__VERSION__)
  enum PicoVtxMode {NotSet=0, Default=1, Vpd=2, VpdOrDefault=3, Mtd=4, FXT=5};
#endif /* ! __TFG__VERSION__ */
  enum PicoCovMtxMode {Skip=0, Write=1};
  enum PicoBEmcSmdMode {SmdSkip=0, SmdWrite=1};

  StPicoDstMaker(char const* name = "PicoDst");
  StPicoDstMaker(PicoIoMode ioMode, char const* fileName = "",
                 char const* name = "PicoDst");
  virtual ~StPicoDstMaker();

  virtual Int_t InitRun(Int_t const runnumber);
  virtual Int_t Init();
  virtual Int_t Make();
  virtual void  Clear(Option_t* option = "");
  virtual Int_t Finish();

  void printArrays();

  void SetStatus(char const* branchNameRegex, int enable);

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

  void setSplit(int = 99);
  void setBufferSize(int = 65536 * 4);
  void setCompression(int comp = 9);

#if defined (__TFG__VERSION__)
  PicoVtxMode vtxMode()                 { return StMuDst::instance()->vtxMode(); }
  void setVtxMode(const PicoVtxMode vtxMode)
  { StMuDst::instance()->setVtxMode(vtxMode); }
  void SetMaxTrackDca(Double_t cut = 50)
  { StMuDst::instance()->SetMaxTrackDca(cut); }
  void SetMaxVertexTransError(Double_t cut = 0.0050)
  { StMuDst::instance()->SetMaxVertexTransError(cut); }
  void SetVxXYrange(Double_t xmin = -0.3, Double_t xmax = 0.,
                    Double_t ymin = -0.27, Double_t ymax = -0.13)
  { StMuDst::instance()->SetVxXYrange(xmin,xmax,ymin,ymax) ;}
  void SetVxZrange(Double_t zmin = -70, Double_t zmax = 70.)
  { StMuDst::instance()->SetVxZrange(zmin, zmax); }
  void SetVxRmax(Double_t rmax = 2)     { StMuDst::instance()->SetVxRmax(rmax); }
  static StPicoDstMaker *instance()     { return fgPicoDstMaker; }
  TClonesArray** picoArrays()           { return mPicoArrays; }
#else /* ! __TFG__VERSION__ */

  void setVtxMode(const PicoVtxMode vtxMode)              { mVtxMode = vtxMode; }
#endif
  void setCovMtxMode(const PicoCovMtxMode covMtxMode)     { mCovMtxMode = covMtxMode; }
  void setBEmcSmdMode(const PicoBEmcSmdMode bemcSmdMode)  { mBEmcSmdMode = bemcSmdMode; }

 private:

  void streamerOff();

  void openWrite();
  void write();
  void closeWrite();
  Int_t openRead();
  void  read();
  void closeRead();
  void setBranchAddresses(TChain*);

  void buildEmcIndex(StEmcCollection*);
  void initEmc();
  void finishEmc();

  Bool_t initMtd(Int_t const runnumber);

  void clearArrays();
  void createArrays();

  Int_t MakeRead();
  Int_t MakeWrite();

  void fillEventHeader() const;

  void fillTracks();
  void fillEvent();
  void fillEmcTrigger();
  void fillMtdTrigger();
  void fillBTowHits();
  void fillBTofHits();
  void fillMtdHits();
  void fillEpdHits();
  void fillBbcHits();
  void fillETofHits();
  void fillMcVertices();
  void fillMcTracks();


  Bool_t getBEMC(const StMuTrack* t, int* id, int* adc, float* ene, float* d, int* nep, int* towid);
  Int_t  setVtxModeAttr();
  Int_t  setCovMtxModeAttr();
  Int_t  setBEmcSmdModeAttr();

#if defined (__TFG__VERSION__)
  Bool_t selectVertex()     { return StMuDst::instance()->selectVertex(); }
#else /* ! __TFG__VERSION__ */
  Bool_t selectVertex();
  Float_t   mTpcVpdVzDiffCut;
#endif

  StMuDst*  mMuDst;

  StPicoDst*  mPicoDst;

  StEmcCollection* mEmcCollection;
  StEmcPosition*   mEmcPosition;
  StEmcGeom*       mEmcGeom[4];
  StEmcRawHit*     mEmcIndex[4800];

  Float_t    mBField;

#if !defined (__TFG__VERSION__)
  PicoVtxMode mVtxMode;
#endif /* ! __TFG__VERSION__ */
  PicoCovMtxMode mCovMtxMode;
  PicoBEmcSmdMode mBEmcSmdMode;

  TString   mInputFileName;
  TString   mOutputFileName;
  TFile*    mOutputFile;

  TChain*   mChain;
  TTree*    mTTree;

  int mEventCounter;

  int mSplit;
  int mCompression;
  int mBufferSize;

  Int_t  mModuleToQT[30][5];
  Int_t  mModuleToQTPos[30][5];
  Int_t  mQTtoModule[8][8];
  Int_t  mQTSlewBinEdge[8][16][8];
  Int_t  mQTSlewCorr[8][16][8];

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

  StPicoFmsFiller  mFmsFiller;
#if defined (__TFG__VERSION__)
  static StPicoDstMaker *fgPicoDstMaker; 
#endif

  virtual const char *GetCVS() const {
    static const char cvs[]="Tag $Name:  $ $Id: StPicoDstMaker.h,v 1.30 2021/02/23 17:27:01 gnigmat Exp $ built " __DATE__ " " __TIME__ ;
    return cvs;
  }

  ClassDef(StPicoDstMaker, 0)
};

inline void StPicoDstMaker::setSplit(int split) { mSplit = split; }
inline void StPicoDstMaker::setCompression(int comp) { mCompression = comp; }
inline void StPicoDstMaker::setBufferSize(int buf) { mBufferSize = buf; }

#endif