CbmStar2019MonitorPulserTask.cxx

Go to the documentation of this file.

/* Copyright (C) 2019-2020 Facility for Antiproton and Ion Research in Europe, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Pierre-Alain Loizeau [committer] */
 
// -----------------------------------------------------------------------------
// -----                                                                   -----
// -----                   CbmStar2019MonitorPulserTask                    -----
// -----               Created 12.10.2019 by P.-A. Loizeau                 -----
// -----                                                                   -----
// -----------------------------------------------------------------------------
 
#include "CbmStar2019MonitorPulserTask.h"
 
#include "CbmStar2019MonitorPulserAlgo.h"
#include "CbmStar2019TofPar.h"
#include "FairParGenericSet.h"
#include "FairRootManager.h"
#include "FairRun.h"
#include "FairRunOnline.h"
#include "FairRuntimeDb.h"
#include "TCanvas.h"
#include "TClonesArray.h"
#include "THttpServer.h"
#include "TROOT.h"
#include "TString.h"
 
#include <Logger.h>
 
#include <TFile.h>
 
#include <chrono>
#include <cstdint>
#include <fstream>
#include <iomanip>
#include <iostream>
 
Bool_t bStar2019MonitorPulserTaskResetHistos = kFALSE;
Bool_t bStar2019MonitorPulserTaskSaveHistos  = kFALSE;
 
CbmStar2019MonitorPulserTask::CbmStar2019MonitorPulserTask()
  : CbmMcbmUnpack()
  , fbEtofFeeIndexing(kTRUE)
  , fsHistoFileName("data/HistosMonitorPulser.root")
  , fuUpdateFreqTs(100)
  , fuPulserMinTot(90)
  , fuPulserMaxTot(110)
  , fuPulserChannel(3)
  , fiSectorIndex(-1)
  , fuHistoryHistoSize(1800)
  , fParCList(nullptr)
  , fulTsCounter(0)
  , fMonitorPulserAlgo(nullptr)
{
  fMonitorPulserAlgo = new CbmStar2019MonitorPulserAlgo();
}
 
CbmStar2019MonitorPulserTask::~CbmStar2019MonitorPulserTask() { delete fMonitorPulserAlgo; }
 
Bool_t CbmStar2019MonitorPulserTask::Init()
{
  LOG(info) << "CbmStar2019MonitorPulserTask::Init";
  LOG(info) << "Initializing eTOF 2019 Monitor task";
 
  return kTRUE;
}
 
void CbmStar2019MonitorPulserTask::SetParContainers()
{
  LOG(info) << "Setting parameter containers for " << GetName();
 
  fParCList = fMonitorPulserAlgo->GetParList();
 
  for (Int_t iparC = 0; iparC < fParCList->GetEntries(); ++iparC) {
    FairParGenericSet* tempObj = (FairParGenericSet*) (fParCList->At(iparC));
    fParCList->Remove(tempObj);
 
    std::string sParamName {tempObj->GetName()};
    FairParGenericSet* newObj =
      dynamic_cast<FairParGenericSet*>(FairRun::Instance()->GetRuntimeDb()->getContainer(sParamName.data()));
 
    if (nullptr == newObj) {
      LOG(error) << "Failed to obtain parameter container " << sParamName << ", for parameter index " << iparC;
      return;
    }  // if( nullptr == newObj )
 
    fParCList->AddAt(newObj, iparC);
    //      delete tempObj;
  }  // for( Int_t iparC = 0; iparC < fParCList->GetEntries(); ++iparC )
}
 
Bool_t CbmStar2019MonitorPulserTask::InitContainers()
{
  LOG(info) << "Init parameter containers for " << GetName();

  CbmStar2019TofPar* pUnpackPar =
    dynamic_cast<CbmStar2019TofPar*>(FairRun::Instance()->GetRuntimeDb()->getContainer("CbmStar2019TofPar"));
  if (nullptr == pUnpackPar) {
    LOG(error) << "Failed to obtain parameter container CbmStar2019TofPar";
    return kFALSE;
  }  // if( nullptr == pUnpackPar )

  fMonitorPulserAlgo->SetEtofFeeIndexing(fbEtofFeeIndexing);
  fMonitorPulserAlgo->SetSectorIndex(fiSectorIndex);
  fMonitorPulserAlgo->SetUpdateFreqTs(fuUpdateFreqTs);
  fMonitorPulserAlgo->SetPulserTotLimits(fuPulserMinTot, fuPulserMaxTot);
  fMonitorPulserAlgo->SetPulserChannel(fuPulserChannel);
 
  Bool_t initOK = fMonitorPulserAlgo->InitContainers();

  initOK &= fMonitorPulserAlgo->CreateHistograms();

  std::vector<std::pair<TNamed*, std::string>> vHistos = fMonitorPulserAlgo->GetHistoVector();
  std::vector<std::pair<TCanvas*, std::string>> vCanvases = fMonitorPulserAlgo->GetCanvasVector();

  THttpServer* server = FairRunOnline::Instance()->GetHttpServer();
  if (nullptr != server) {
    for (UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto) {
      //         LOG(info) << "Registering  " << vHistos[ uHisto ].first->GetName()
      //                   << " in " << vHistos[ uHisto ].second.data();
      server->Register(Form("/%s", vHistos[uHisto].second.data()), vHistos[uHisto].first);
    }  // for( UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto )
 
    for (UInt_t uCanv = 0; uCanv < vCanvases.size(); ++uCanv) {
      //         LOG(info) << "Registering  " << vCanvases[ uCanv ].first->GetName()
      //                   << " in " << vCanvases[ uCanv ].second.data();
      server->Register(Form("/%s", vCanvases[uCanv].second.data()),
                       gROOT->FindObject((vCanvases[uCanv].first)->GetName()));
    }  //  for( UInt_t uCanv = 0; uCanv < vCanvases.size(); ++uCanv )
 
    server->RegisterCommand("/Reset_Pulser_Hist", "bStar2019MonitorPulserTaskResetHistos=kTRUE");
    server->RegisterCommand("/Save_Pulser_Hist", "bStar2019MonitorPulserTaskSaveHistos=kTRUE");
 
    server->Restrict("/Reset_Moni_Hist", "allow=admin");
    server->Restrict("/Save_Pulser_Hist", "allow=admin");
  }  // if( nullptr != server )
 
  return initOK;
}
 
Bool_t CbmStar2019MonitorPulserTask::ReInitContainers()
{
  LOG(info) << "ReInit parameter containers for " << GetName();
  Bool_t initOK = fMonitorPulserAlgo->ReInitContainers();
 
  return initOK;
}
 
void CbmStar2019MonitorPulserTask::AddMsComponentToList(size_t component, UShort_t usDetectorId)
{
  fMonitorPulserAlgo->AddMsComponentToList(component, usDetectorId);
}
 
Bool_t CbmStar2019MonitorPulserTask::DoUnpack(const fles::Timeslice& ts, size_t /*component*/)
{
  if (bStar2019MonitorPulserTaskResetHistos) {
    LOG(info) << "Reset Monitor histos ";
    fMonitorPulserAlgo->ResetHistograms();
    bStar2019MonitorPulserTaskResetHistos = kFALSE;
  }  // if( bStar2019MonitorPulserTaskResetHistos )
 
  if (bStar2019MonitorPulserTaskSaveHistos) {
    LOG(info) << "Save Monitor histos & canvases";
    SaveHistograms();
    bStar2019MonitorPulserTaskSaveHistos = kFALSE;
  }  // if( bStar2019MonitorPulserTaskSaveHistos )
 
  if (kFALSE == fMonitorPulserAlgo->ProcessTs(ts)) {
    LOG(error) << "Failed processing TS " << ts.index() << " in unpacker algorithm class";
    return kTRUE;
  }  // if( kFALSE == fMonitorPulserAlgo->ProcessTs( ts ) )

  std::vector<CbmTofDigi> vDigi = fMonitorPulserAlgo->GetVector();
  fMonitorPulserAlgo->ClearVector();
 
  if (0 == fulTsCounter % 10000) LOG(info) << "Processed " << fulTsCounter << "TS";
  fulTsCounter++;
 
  return kTRUE;
}
 
void CbmStar2019MonitorPulserTask::Reset() {}
 
void CbmStar2019MonitorPulserTask::Finish()
{
  fMonitorPulserAlgo->UpdateStats();
  SaveHistograms();
}
 
void CbmStar2019MonitorPulserTask::SetIgnoreOverlapMs(Bool_t bFlagIn)
{
  fMonitorPulserAlgo->SetIgnoreOverlapMs(bFlagIn);
}
 
Bool_t CbmStar2019MonitorPulserTask::SaveHistograms()
{
  std::vector<std::pair<TNamed*, std::string>> vHistos  = fMonitorPulserAlgo->GetHistoVector();
  std::vector<std::pair<TCanvas*, std::string>> vCanvas = fMonitorPulserAlgo->GetCanvasVector();

  TDirectory* oldDir = NULL;
  TFile* histoFile   = NULL;
  // Store current directory position to allow restore later
  oldDir = gDirectory;
  // open separate histo file in recreate mode
  histoFile = new TFile(fsHistoFileName, "RECREATE");
 
  if (nullptr == histoFile) return kFALSE;

  for (UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto) {
    TString sFolder = vHistos[uHisto].second.data();
    if (nullptr == gDirectory->Get(sFolder)) gDirectory->mkdir(sFolder);
    gDirectory->cd(sFolder);

    vHistos[uHisto].first->Write();
 
    histoFile->cd();
  }  // for( UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto )
 
  for (UInt_t uCanvas = 0; uCanvas < vCanvas.size(); ++uCanvas) {
    TString sFolder = vCanvas[uCanvas].second.data();
    if (nullptr == gDirectory->Get(sFolder)) gDirectory->mkdir(sFolder);
    gDirectory->cd(sFolder);

    vCanvas[uCanvas].first->Write();
 
    histoFile->cd();
  }  // for( UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto )
 
  // Restore original directory position
  oldDir->cd();
  histoFile->Close();
 
  return kTRUE;
}
ClassImp(CbmStar2019MonitorPulserTask)