/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Event/RootCnvSvc/RootCnvSvc-02-01-12/RootCnvSvc/RootInterface.h

Go to the documentation of this file.

#ifndef ROOTINTERFACE_H
#define ROOTINTERFACE_H
/*
 *Description:Interface for all Root I/O needed by the RootCnvSvc
 *Based on the RootInterface of Glast.
 */
#include "TObject.h"
#include <vector>
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/IService.h"

class TTree;
class TClonesArray;
class TFile;
class TList;


class RootInterface {

 public:

  RootInterface (MsgStream log);
  virtual  ~RootInterface();
  virtual   StatusCode finalize();

static  RootInterface* Instance(MsgStream log); 
 virtual bool checkEndOfTree();
 virtual bool getENDFILE(){ return m_ENDFILE;}
 virtual StatusCode addInput(const std::string& treename,const std::string& file);
  virtual StatusCode addOutput(const std::string& treename,const std::string& file,int splitx, int bufsize, int compression);
  virtual StatusCode createBranch(const std::string& tree,const std::string& branch ,const char *classname,void *addr,int & branchnr);
  
  virtual StatusCode setBranchAddress(const std::string treename, const std::string branchname, void *addr,int& nb);
  virtual StatusCode getBranchEntry( int nr, int entry,int& nb);
  
  virtual StatusCode getBranchEntry( int nr, int entry, void * addr, int& nb);
  
  
  virtual bool       getEOF() { return m_EOF;}
  virtual Int_t      getEntries() { return m_entries;}
  virtual StatusCode fillTrees();

  virtual std::string getCurrentFileName() { return m_currentFileName;}  

  virtual std::vector<int> getTotEvtNo();
  virtual std::string getJobOptions();
  virtual std::string getDecayOptions();
  virtual void printJobInfo(TFile* file, int level);

  virtual void setSelectFromTag(bool temp) {m_selectFromTag = temp;}
  virtual bool getSelectFromTag(){ return m_selectFromTag;}

  virtual int getCurrentFileNum() {return m_fileNum;}
  virtual int getTotalFileNum() {return m_fileNames.size();}

  virtual void setTagInputFile(std::vector<std::string> input); 
  virtual std::vector<std::string> getTagInputFile() {return m_tagInputFile;}

  //virtual void setTagFileFormat(int form) {m_tagFileFormat = form;}
  //virtual int getTagFileFormat() {return m_tagFileFormat;}


 private:
  virtual StatusCode getTreeNr(const std::string treename, unsigned int& treenr,bool doAdd=false);
  virtual TTree*  getTree(const std::string treename);
  virtual TTree*  getOtherTree(const std::string treename);
  virtual StatusCode createTree(unsigned int treenr, const std::string treename);
  
  

  std::vector<TTree *> m_otherTrees;
  std::vector<TFile *> inputFiles;
  std::vector<std::string> m_fileNames;
  std::string m_currentFileName;
  bool m_selectFromTag;
  std::vector<std::string> m_tagInputFile;
  //int m_tagFileFormat;

  int m_fileNum;
  bool m_ENDFILE;
  std::vector<std::string> m_treenames;  
  std::vector<std::string> m_inputFilenames;
  std::vector<std::string> m_outputFilenames;
  std::vector<TTree *> m_inputTrees;
  std::vector<TTree *> m_outputTrees;
  std::vector<TFile *> m_inputFiles;
  std::vector<TFile *> m_outputFiles;
  std::vector<int> m_splitModes;
  std::vector<int>  m_bufSizes;
  std::vector<int>  m_compressionLevels;


  TClonesArray *m_branches;

  TClonesArray *m_branchesRead;

  bool m_EOF;

  Int_t  m_entries;

  MsgStream log;

  static RootInterface *m_rootInterface;
  
  std::vector<std::string> m_jobOptions;
  std::string m_bossVer;
  std::string m_decayOptions;
  std::vector<int> m_totEvtNo;

  // tianhl add for besvis
 private:
  std::vector<std::string>      m_single_treenames;
  std::vector<std::string>      m_single_outputFileNames;
  std::vector<TFile *>          m_single_outputFiles;
  std::vector<TTree *>          m_single_outputTrees;
  std::vector<int>              m_single_splitModes;
  std::vector<int>              m_single_bufSizes;
  std::vector<int>              m_single_compressionLevels;

 public:
  virtual StatusCode f_addOutput(const std::string& treename,
                  const std::string& file,
                  int splitx = 1, int bufsize = 64000,
                  int compression = 1);
  virtual StatusCode f_createTree(unsigned int treenr, 
                  const std::string treename);
  virtual StatusCode f_createBranch(const std::string& treename,
                  const std::string& branchname,
                  const char *classname, 
                  void *addr,int & branchnr);
  virtual StatusCode f_getTreeNr(const std::string treename, 
                  unsigned int& treenr,bool doAdd=false);
  virtual StatusCode f_fillTrees();
  virtual StatusCode f_finalize();

};

#endif 

---