CbmSourceDigiTimeslice.h

Go to the documentation of this file.

/* Copyright (C) 2024 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Pierre-Alain Loizeau [committer] */
 
#pragma once  // include this header only once per compilation unit
 
#include "DigiData.h"
#include "RecoResultsInputArchive.h"
 
#include <FairSource.h>
 
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
class CbmTimeSlice;
class CbmTsEventHeader;
class FairRootManager;
 
class CbmSourceDigiTimeslice : public FairSource {
 
 public:
  CbmSourceDigiTimeslice(const char* fileName = "");
 
  CbmSourceDigiTimeslice(const CbmSourceDigiTimeslice&) = delete;
 
  CbmSourceDigiTimeslice& operator=(const CbmSourceDigiTimeslice&) = delete;
 
  virtual ~CbmSourceDigiTimeslice() {}
 
  virtual void Close();
 
  virtual Source_Type GetSourceType() { return fSourceType; }
 
  virtual Bool_t Init();
 
  virtual Bool_t InitUnpackers() { return kTRUE; }
 
  Int_t ReadEvent(UInt_t = 0);
 
  virtual Bool_t ReInitUnpackers() { return kTRUE; }
 
  virtual void Reset() {}
 
  virtual void SetParUnpackers() {}
 
  void SetSourceType(Source_Type type) { fSourceType = type; }
 
  Bool_t SpecifyRunId() { return kTRUE; }
 
 
 private:
  std::string fInputFileName = {};
 
  std::unique_ptr<cbm::algo::RecoResultsInputArchive> fArchive = nullptr;
 
  std::vector<CbmBmonDigi>* fBmonDigis = nullptr;
  std::vector<CbmStsDigi>* fStsDigis   = nullptr;
  std::vector<CbmMuchDigi>* fMuchDigis = nullptr;
  std::vector<CbmTrdDigi>* fTrdDigis   = nullptr;
  std::vector<CbmTofDigi>* fTofDigis   = nullptr;
  std::vector<CbmRichDigi>* fRichDigis = nullptr;
 
  Source_Type fSourceType = Source_Type::kONLINE;
 
  CbmTimeSlice* fTimeslice = nullptr;
 
  CbmTsEventHeader* fTsEventHeader = nullptr;
 
 
  size_t fNumTs = 0;
 
  template<typename TVecobj>
  Bool_t RegisterVector(FairRootManager* ioman, std::vector<TVecobj>*& vec);
 
  template<typename TVecobj>
  typename std::enable_if<std::is_member_function_pointer<decltype(&TVecobj::GetTime)>::value, void>::type
  Timesort(std::vector<TVecobj>* vec = nullptr)
  {
    if (vec == nullptr) return;
    std::sort(vec->begin(), vec->end(),
              [](const TVecobj& a, const TVecobj& b) -> bool { return a.GetTime() < b.GetTime(); });
  }
 
 
  ClassDef(CbmSourceDigiTimeslice, 1)
};