CbmTaskTriggerDigi.h

Go to the documentation of this file.

/* Copyright (C) 2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Volker Friese [committer] */
 
#ifndef CBMTASKTRIGGERDIGI_H
#define CBMTASKTRIGGERDIGI_H 1
 
#include "CbmDefs.h"
#include "CbmDigiBranchBase.h"
#include "TimeClusterTrigger.h"
 
#include <FairTask.h>
 
#include <TStopwatch.h>
 
#include <boost/any.hpp>
 
#include <vector>
 
class CbmDigiBranchBase;
class CbmDigiManager;
class CbmDigiTimeslice;
 
using namespace std;
 
class CbmTaskTriggerDigi : public FairTask {
 
 
 public:
  CbmTaskTriggerDigi();
 
 
  CbmTaskTriggerDigi(const CbmTaskTriggerDigi&) = delete;
 
 
  virtual ~CbmTaskTriggerDigi();
 
 
  virtual void Exec(Option_t* opt);
 
 
  virtual void Finish();
 
 
  CbmTaskTriggerDigi& operator=(const CbmTaskTriggerDigi&) = delete;
 
 
  void SetConfig(const cbm::algo::evbuild::DigiTriggerConfig& config)
  {
    fConfig.reset(new cbm::algo::evbuild::DigiTriggerConfig(config));
  }
 
  void AddSystem(ECbmModuleId system)
  {
    if (std::find(fSystems.begin(), fSystems.end(), system) != fSystems.end()) return;
    fSystems.push_back(system);
  }
 
 private:  // methods
  virtual InitStatus Init();
 
  template<class TDigi>
  std::vector<double> GetDigiTimes(const CbmDigiBranchBase* digiBranch)
  {
    TStopwatch timerStep;
    // --- Get input digi vector
    const vector<TDigi>* digiVec = boost::any_cast<const vector<TDigi>*>(digiBranch->GetBranchContainer());
    assert(digiVec);
 
    // --- Extract digi times into to a vector
    timerStep.Start();
    std::vector<double> digiTimes(digiVec->size());
    std::transform(digiVec->begin(), digiVec->end(), digiTimes.begin(),
                   [](const TDigi& digi) { return digi.GetTime(); });
    timerStep.Stop();
    fTimeExtract += timerStep.RealTime();
    return digiTimes;
  }
 
 
  std::vector<double> GetDigiTimes(ECbmModuleId system);
 
 
 private:                                                                    // members
  const CbmDigiTimeslice* fTimeslice = nullptr;                              
  CbmDigiManager* fDigiMan           = nullptr;                              
  std::vector<ECbmModuleId> fSystems{};                                      //  List of detector systems
  std::vector<double>* fTriggers                                 = nullptr;  
  std::unique_ptr<cbm::algo::evbuild::TimeClusterTrigger> fAlgo  = nullptr;  
  std::unique_ptr<cbm::algo::evbuild::DigiTriggerConfig> fConfig = nullptr;  
  double fTriggerWindow                                          = 0.;
  int32_t fMinNumDigis                                           = 0;
  double fDeadTime                                               = 0.;
  size_t fNumTs                                                  = 0;  //  Number of processed time slices
  size_t fNumDigis                                               = 0;  //  Number of digis from trigger detector
  size_t fNumTriggers                                            = 0;  //  Number of found triggers
  double fTimeExtract                                            = 0.;
  double fTimeFind                                               = 0.;
  double fTimeTot                                                = 0.;
 
 
  ClassDef(CbmTaskTriggerDigi, 1);
};
 
#endif /* CBMTASKTRIGGERDIGI_H */