CbmDigiVector.h

Go to the documentation of this file.

/* Copyright (C) 2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */
 
#ifndef CBMDIGIVECTOR_H
#define CBMDIGIVECTOR_H 1
 
 
#include "CbmDigiContainer.h"
#include "CbmMatch.h"
 
#include <Logger.h>
 
#include <vector>
 
template<class Digi>
class CbmDigiVector : public CbmDigiContainer {
 
public:
  CbmDigiVector(bool hasMatches = false) : 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 ConnectToTree()
  {
    LOG(fatal) << "Digi vector cannot be connected to a TTree!";
    return false;
  }
 
 
  virtual boost::any GetDigi(uint32_t index)
  {
    fIsLocked          = true;
    const Digi* result = nullptr;
    if (index < fDigis.size()) result = &(fDigis[index]);
    return boost::any(result);
  }
 
 
  virtual const CbmMatch* GetDigiMatch(uint32_t index)
  {
    fIsLocked              = true;
    const CbmMatch* result = nullptr;
    if (fHasMatches && index < fMatches.size()) result = &(fMatches[index]);
    return result;
  }
 
 
  virtual bool HasMatches() const { return fHasMatches; }
 
 
  virtual uint64_t GetNofDigis() const
  {
    if (fHasMatches) assert(fMatches.size() == fDigis.size());
    return static_cast<uint64_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 fHasMatches = false;
  bool fIsLocked   = false;
 
 
  ClassDef(CbmDigiVector, 1);
};
 
#endif /* CBMDIGIVECTOR_H */