CbmMCDataArray.h

Go to the documentation of this file.

/* Copyright (C) 2015-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Mikhail Prokudin, Volker Friese [committer], Florian Uhlig */
 
#ifndef CBMMCDATAARRAY_H
#define CBMMCDATAARRAY_H
 
#include "CbmLink.h"  // for CbmLink
 
#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
 
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