CbmDigiVector.h

Go to the documentation of this file.

 
#ifndef CBMDIGIVECTOR_H
#define CBMDIGIVECTOR_H 1
 
 
#include <FairLogger.h>
#include <vector>
 
#include "CbmDigiContainer.h"
#include "CbmMatch.h"
 
template<class Digi>
class CbmDigiVector : public CbmDigiContainer {
 
public:
  CbmDigiVector(Bool_t hasMatches = kFALSE) : fHasMatches(hasMatches) {
    TString name = Digi::GetClassName();
    name += "_vector";
    SetName(name.Data());
  }
 
 
  virtual ~CbmDigiVector() {}
 
 
  virtual void AddDigi(boost::any digi, const CbmMatch* match = nullptr) {
    if (fIsLocked) {
      LOG(fatal) << GetName() << "::AddDigi: Vector is locked.";
      return;
    }
    const Digi* thisDigi = boost::any_cast<const Digi*>(digi);
    if (!thisDigi) {
      LOG(fatal) << GetName() << "::AddDigi: Wrong argument type"
                 << " (should be " << Digi::GetClassName() << "*) !";
      return;
    }  //? Any_cast successful
    fDigis.push_back(*thisDigi);
    if (fHasMatches) {
      if (match == nullptr) {
        LOG(fatal) << GetName() << "::AddDigi: Valid match object required!";
        return;
      }
      fMatches.push_back(*match);
    }  //? Has matches
  }
 
 
  virtual Bool_t ConnectToTree() {
    LOG(fatal) << "Digi vector cannot be connected to a TTree!";
    return false;
  }
 
 
  virtual boost::any GetDigi(UInt_t index) {
    fIsLocked          = kTRUE;
    const Digi* result = nullptr;
    if (index < fDigis.size()) result = &(fDigis[index]);
    return boost::any(result);
  }
 
 
  virtual const CbmMatch* GetDigiMatch(UInt_t index) {
    fIsLocked              = kTRUE;
    const CbmMatch* result = nullptr;
    if (fHasMatches && index < fMatches.size()) result = &(fMatches[index]);
    return result;
  }
 
 
  virtual Bool_t HasMatches() const { return fHasMatches; }
 
 
  virtual ULong64_t GetNofDigis() const {
    if (fHasMatches) assert(fMatches.size() == fDigis.size());
    return static_cast<ULong64_t>(fDigis.size());
  }
 
 
  virtual std::string ToString() const {
    std::stringstream ss;
    ss << GetName() << ", size " << fDigis.size();
    if (HasMatches()) ss << ", matches present";
    return ss.str();
  }
 
 
private:
  std::vector<Digi> fDigis {};
  std::vector<CbmMatch> fMatches {};
  Bool_t fHasMatches = kFALSE;
  Bool_t fIsLocked   = kFALSE;
 
 
  ClassDef(CbmDigiVector, 1);
};
 
#endif /* CBMDIGIVECTOR_H */