CbmMCDataArray.h

Go to the documentation of this file.

 
#ifndef CBMMCDATAARRAY_H
#define CBMMCDATAARRAY_H
 
#include <Rtypes.h>      // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>  // for Int_t, Long64_t
#include <TString.h>     // for TString
 
#include <list>    // for list
#include <map>     // for map
#include <vector>  // for vector
 
#include "CbmLink.h"  // for CbmLink
 
class TChain;
class TClonesArray;
class TObject;
 
class CbmMCDataArray {
  friend class CbmMCDataManager;
 
public:
  virtual ~CbmMCDataArray() {}
 
 
  TObject* Get(const CbmLink* lnk) {
    return Get(lnk->GetFile(), lnk->GetEntry(), lnk->GetIndex());
  }
 
 
  TObject* Get(const CbmLink& lnk) {
    return Get(lnk.GetFile(), lnk.GetEntry(), lnk.GetIndex());
  }
 
 
  TObject* Get(Int_t fileNumber, Int_t eventNumber, Int_t index);
 
  Int_t Size(Int_t fileNumber, Int_t eventNumber);
 
  Int_t Size(const CbmLink& lnk) { return Size(lnk.GetFile(), lnk.GetEntry()); }
 
  Int_t Size(const CbmLink* lnk) {
    return Size(lnk->GetFile(), lnk->GetEntry());
  }
 
private:
  CbmMCDataArray()
    : fLegacy(0)
    , fLegacyArray(nullptr)
    , fBranchName()
    , fSize(0)
    , fChains()
    , fTArr()
    , fN(0)
    , fArrays() {}
 
  CbmMCDataArray(const char* branchName,
                 const std::vector<std::list<TString>>& fileList);
 
  CbmMCDataArray(const char* branchName);
 
 
  CbmMCDataArray(const CbmMCDataArray&);
 
 
  void Done();
 
 
  void FinishEvent();
 
  void AddFriend(Int_t chainNum1, Int_t chainNum2);
 
 
  TObject* LegacyGet(Int_t fileNumber, Int_t eventNumber, Int_t index);
 
 
  //----   Private data members -------
  Int_t fLegacy;                 
  TClonesArray* fLegacyArray;    
  TString fBranchName;           
  Int_t fSize;                   
  std::vector<TChain*> fChains;  
  std::vector<TClonesArray*>
    fTArr;                   
  std::vector<Long64_t> fN;  
 
  std::vector<std::map<Int_t, TClonesArray*>> fArrays;  
 
  CbmMCDataArray& operator=(const CbmMCDataArray&);
 
  ClassDef(CbmMCDataArray, 1);
};
 
#endif