CbmSpillCheck.cxx

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#include "CbmSpillCheck.h"
 
#include <TFitResult.h>
 
#include <ctime>
#include <iostream>
#include <typeinfo>
 
CbmSpillCheck::CbmSpillCheck() : fRefSystem{ECbmModuleId::kBmon}, fRateMinPercent{0.2}, fRateMaxPercent{0.5}
{
  LOG(debug) << "Creating an instance of CbmSpillCheck ...";
}
 
CbmSpillCheck::CbmSpillCheck(ECbmModuleId ref, double lvl_min, double lvl_max)
  : fRefSystem{ref}
  , fRateMinPercent{lvl_min}
  , fRateMaxPercent{lvl_max}
{
  LOG(debug) << "Creating an instance of CbmSpillCheck ...";
}
 
void CbmSpillCheck::BookHistograms(ECbmModuleId system)
{
  std::string h_name = Form("%s Digi Rate", std::string(cbm::util::ToString(system)).c_str());
  LOG(debug) << "Booking rate for : " << h_name;
  fG1D[h_name] = std::make_unique<TGraphErrors>();
}
 
InitStatus CbmSpillCheck::Init()
{
  FairRootManager* ioman = FairRootManager::Instance();
  if (ioman != nullptr) {
    fDigiManager = CbmDigiManager::Instance();
    fDigiManager->Init();
 
    if (!fDigiManager->IsPresent(fRefSystem)) {
      LOG(fatal) << GetName() << ": No " << fRefSystem << " branch in input!";
    }
 
    for (ECbmModuleId system = ECbmModuleId::kRef; system != ECbmModuleId::kLastModule; ++system) {
      if (fDigiManager->IsPresent(system)) BookHistograms(system);
    }
 
    return kSUCCESS;
  }
  return kERROR;
}
 
void CbmSpillCheck::Exec(Option_t*)
{
  LOG(info) << "Running CbmSpillCheck ...";
 
  size_t nb_ref_digis = fDigiManager->GetNofDigis(fRefSystem);
  fRateMin            = fRateMin == -1 ? nb_ref_digis : std::min(nb_ref_digis, size_t(fRateMin));
  fRateMax            = fRateMax == -1 ? nb_ref_digis : std::max(nb_ref_digis, size_t(fRateMax));
  fRefRate.push_back(nb_ref_digis);
 
  for (ECbmModuleId system = ECbmModuleId::kRef; system != ECbmModuleId::kLastModule; ++system) {
    if (fDigiManager->IsPresent(system)) {
      size_t n_of_digis  = fDigiManager->GetNofDigis(system);
      std::string h_name = Form("%s Digi Rate", std::string(cbm::util::ToString(system)).c_str());
      fG1D[h_name]->SetPoint(fNbPoints, fNbPoints, n_of_digis);
    }
  }
  fNbPoints++;
}
 
void CbmSpillCheck::Finish()
{
  double spill_lvl_off = fRateMin + fRateMinPercent * (fRateMax - fRateMin);
  double spill_lvl_on  = fRateMin + fRateMaxPercent * (fRateMax - fRateMin);
  fSpillStatus         = std::vector<int>(fRefRate.size(), 0);
  for (size_t ts_idx = 0; ts_idx < fRefRate.size(); ts_idx++) {
    fSpillStatus[ts_idx] = fRefRate[ts_idx] <= spill_lvl_off ? -1 : (fRefRate[ts_idx] <= spill_lvl_on ? 0 : 1);
    std::cout << fSpillStatus[ts_idx] << std::endl;
  }
 
  std::cout << spill_lvl_off << std::endl;
  std::cout << spill_lvl_on << std::endl;
 
  SaveToFile();
  fG1D.clear();
  fH1D.clear();
  fH2D.clear();
}