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 "CbmMvdDigi.h"
#include "algo/base/DigiData.h"
#include "algo/global/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<CbmMvdDigi>* fMvdDigis   = 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;
  std::vector<CbmFsdDigi>* fFsdDigis   = 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)
};