CbmCosy2019UnpackerTaskHodo.cxx

Go to the documentation of this file.

/* Copyright (C) 2019-2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Pierre-Alain Loizeau, Florian Uhlig [committer] */
 
// -----------------------------------------------------------------------------
// -----                                                                   -----
// -----                   CbmCosy2019UnpackerTaskHodo                     -----
// -----              Created 31/07/19  by P.-A. Loizeau                   -----
// -----                                                                   -----
// -----------------------------------------------------------------------------
 
#include "CbmCosy2019UnpackerTaskHodo.h"
 
#include "CbmCosy2019HodoPar.h"
#include "CbmCosy2019UnpackerAlgoHodo.h"
 
//#include "CbmMcbm2018StsPar.h"
#include "CbmMcbm2018UnpackerAlgoSts.h"
#include "FairParGenericSet.h"
#include "FairRootManager.h"
#include "FairRun.h"
#include "FairRunOnline.h"
#include "FairRuntimeDb.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 bCosy2019UnpackerTaskHodoResetHistos = kFALSE;
 
CbmCosy2019UnpackerTaskHodo::CbmCosy2019UnpackerTaskHodo(UInt_t /*uNbGdpb*/)
  : CbmMcbmUnpack()
  , fbMonitorMode(kFALSE)
  , fbWriteOutput(kTRUE)
  , fvChanMasks()
  , fulTsCounter(0)
  , fUnpackerAlgo(nullptr)
{
  fUnpackerAlgo    = new CbmCosy2019UnpackerAlgoHodo();
  fUnpackerAlgoSts = new CbmMcbm2018UnpackerAlgoSts();
}
 
CbmCosy2019UnpackerTaskHodo::~CbmCosy2019UnpackerTaskHodo()
{
  delete fUnpackerAlgo;
  delete fUnpackerAlgoSts;
}
 
Bool_t CbmCosy2019UnpackerTaskHodo::Init()
{
  LOG(info) << "CbmCosy2019UnpackerTaskHodo::Init";
  LOG(info) << "Initializing mCBM STS 2018 Unpacker";
 
  FairRootManager* ioman = FairRootManager::Instance();
  if (NULL == ioman) { LOG(fatal) << "No FairRootManager instance"; }
  if (nullptr != ioman->InitObjectAs<std::vector<CbmStsDigi> const*>("StsDigi"))
    LOG(fatal) << "CbmCosy2019UnpackerTaskHodo::Init => output vector already "
                  "registered,"
               << " probably by CbmMcbm2018UnpackerTaskSts" << std::endl
               << " THESE TWO CLASSES ARE INCOMPATIBLE!";

  fpvDigiSts = new std::vector<CbmStsDigi>();
  ioman->RegisterAny("StsDigi", fpvDigiSts, fbWriteOutput);

  fpvErrorSts = new std::vector<CbmErrorMessage>();
  ioman->RegisterAny("CbmStsError", fpvErrorSts, fbWriteOutput);
 
  return kTRUE;
}
 
void CbmCosy2019UnpackerTaskHodo::SetParContainers()
{
  LOG(info) << "Setting parameter containers for " << GetName();
 
  TList* fParCList = fUnpackerAlgo->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 )
 
  fParCList = fUnpackerAlgoSts->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 CbmCosy2019UnpackerTaskHodo::InitContainers()
{
  LOG(info) << "Init parameter containers for " << GetName();

  CbmCosy2019HodoPar* pUnpackPar =
    dynamic_cast<CbmCosy2019HodoPar*>(FairRun::Instance()->GetRuntimeDb()->getContainer("CbmCosy2019HodoPar"));
  if (nullptr == pUnpackPar) {
    LOG(error) << "Failed to obtain parameter container CbmCosy2019HodoPar";
    return kFALSE;
  }  // if( nullptr == pUnpackPar )
     /*
   fbMonitorMode = pUnpackPar->GetMonitorMode();
   LOG(info) << "Monitor mode:       "
             << ( fbMonitorMode ? "ON" : "OFF" );
 
   fbDebugMonitorMode = pUnpackPar->GetDebugMonitorMode();
   LOG(info) << "Debug Monitor mode: "
             << ( fbDebugMonitorMode ? "ON" : "OFF" );
*/
  Bool_t initOK = fUnpackerAlgo->InitContainers();
  initOK &= fUnpackerAlgoSts->InitContainers();

  if (kTRUE == fbMonitorMode) {
    initOK &= fUnpackerAlgo->CreateHistograms();
    initOK &= fUnpackerAlgoSts->CreateHistograms();

    std::vector<std::pair<TNamed*, std::string>> vHistos    = fUnpackerAlgo->GetHistoVector();
    std::vector<std::pair<TNamed*, std::string>> vHistosSts = fUnpackerAlgoSts->GetHistoVector();
    vHistos.insert(vHistos.end(), vHistosSts.begin(), vHistosSts.end());

    THttpServer* server = FairRunOnline::Instance()->GetHttpServer();
    if (nullptr != server) {
      for (UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto) {
        server->Register(Form("/%s", vHistos[uHisto].second.data()), vHistos[uHisto].first);
      }  // for( UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto )
 
      server->RegisterCommand("/Reset_UnpHodo_Hist", "bCosy2019UnpackerTaskHodoResetHistos=kTRUE");
      server->Restrict("/Reset_UnpHodo_Hist", "allow=admin");
    }  // if( nullptr != server )
 
  }  // if( kTRUE == fbMonitorMode )
 
  fUnpackerAlgo->SetMonitorMode(fbMonitorMode);
 
  fUnpackerAlgoSts->SetMonitorMode(fbMonitorMode);
  for (std::vector<FebChanMaskSts>::iterator it = fvChanMasks.begin(); it < fvChanMasks.end(); ++it)
    fUnpackerAlgoSts->MaskNoisyChannel((*it).uFeb, (*it).uChan, (*it).bMasked);
 
  return initOK;
}
 
Bool_t CbmCosy2019UnpackerTaskHodo::ReInitContainers()
{
  LOG(info) << "ReInit parameter containers for " << GetName();
  Bool_t initOK = fUnpackerAlgo->ReInitContainers();
  initOK &= fUnpackerAlgoSts->ReInitContainers();
 
  return initOK;
}
 
void CbmCosy2019UnpackerTaskHodo::AddMsComponentToList(size_t component, UShort_t usDetectorId)
{
  fUnpackerAlgo->AddMsComponentToList(component, usDetectorId);
  fUnpackerAlgoSts->AddMsComponentToList(component, usDetectorId);
}
 
Bool_t CbmCosy2019UnpackerTaskHodo::DoUnpack(const fles::Timeslice& ts, size_t /*component*/)
{
  if (fbMonitorMode && bCosy2019UnpackerTaskHodoResetHistos) {
    LOG(info) << "Reset Hodo + STS unpacker histos ";
    fUnpackerAlgo->ResetHistograms();
    fUnpackerAlgoSts->ResetHistograms();
    bCosy2019UnpackerTaskHodoResetHistos = kFALSE;
  }  // if( fbMonitorMode && bCosy2019UnpackerTaskHodoResetHistos )
 
  if (kFALSE == fUnpackerAlgo->ProcessTs(ts)) {
    LOG(error) << "Failed processing TS " << ts.index() << " in HODO unpacker algorithm class";
    return kTRUE;
  }  // if( kFALSE == fUnpackerAlgo->ProcessTs( ts ) )
 
  if (kFALSE == fUnpackerAlgoSts->ProcessTs(ts)) {
    LOG(error) << "Failed processing TS " << ts.index() << " in STS unpacker algorithm class";
    return kTRUE;
  }  // if( kFALSE == fUnpackerAlgoSts->ProcessTs( ts ) )

  fpvDigiSts->insert(fpvDigiSts->end(), fUnpackerAlgo->GetVector().begin(), fUnpackerAlgo->GetVector().end());
  fpvDigiSts->insert(fpvDigiSts->end(), fUnpackerAlgoSts->GetVector().begin(), fUnpackerAlgoSts->GetVector().end());

  sort(fpvDigiSts->begin(), fpvDigiSts->end(),
       [](const CbmStsDigi& a, const CbmStsDigi& b) -> bool { return a.GetTime() < b.GetTime(); });
 
  fUnpackerAlgo->ClearVector();
  fUnpackerAlgoSts->ClearVector();

  fpvErrorSts->insert(fpvErrorSts->end(), fUnpackerAlgo->GetErrorVector().begin(),
                      fUnpackerAlgo->GetErrorVector().end());
  fpvErrorSts->insert(fpvErrorSts->end(), fUnpackerAlgoSts->GetErrorVector().begin(),
                      fUnpackerAlgoSts->GetErrorVector().end());
 
  fUnpackerAlgo->ClearErrorVector();
  fUnpackerAlgoSts->ClearErrorVector();
 
  if (0 == fulTsCounter % 10000) LOG(info) << "Processed " << fulTsCounter << "TS";
  fulTsCounter++;
 
  return kTRUE;
}
 
void CbmCosy2019UnpackerTaskHodo::Reset()
{
  fpvDigiSts->clear();
  fpvErrorSts->clear();
}
 
void CbmCosy2019UnpackerTaskHodo::Finish()
{
  if (kTRUE == fbMonitorMode) {
    std::vector<std::pair<TNamed*, std::string>> vHistos    = fUnpackerAlgo->GetHistoVector();
    std::vector<std::pair<TNamed*, std::string>> vHistosSts = fUnpackerAlgoSts->GetHistoVector();
    vHistos.insert(vHistos.end(), vHistosSts.begin(), vHistosSts.end());

    TFile* oldFile     = gFile;
    TDirectory* oldDir = gDirectory;
 
    TFile* histoFile = nullptr;
 
    // open separate histo file in recreate mode
    histoFile = new TFile("data/HistosUnpackerSts.root", "RECREATE");
    histoFile->cd();

    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 )

    gFile      = oldFile;
    gDirectory = oldDir;
 
    histoFile->Close();
  }  // if( kTRUE == fbMonitorMode )
}
 
void CbmCosy2019UnpackerTaskHodo::SetIgnoreOverlapMs(Bool_t bFlagIn)
{
  fUnpackerAlgo->SetIgnoreOverlapMs(bFlagIn);
  fUnpackerAlgoSts->SetIgnoreOverlapMs(bFlagIn);
}
 
void CbmCosy2019UnpackerTaskHodo::SetTimeOffsetNs(Double_t dOffsetIn) { fUnpackerAlgo->SetTimeOffsetNs(dOffsetIn); }
 
void CbmCosy2019UnpackerTaskHodo::SetTimeOffsetNsSts(Double_t dOffsetIn)
{
  fUnpackerAlgoSts->SetTimeOffsetNs(dOffsetIn);
}
 
void CbmCosy2019UnpackerTaskHodo::SetTimeOffsetNsAsic(UInt_t uAsicIdx, Double_t dOffsetIn)
{
  fUnpackerAlgo->SetTimeOffsetNsAsic(uAsicIdx, dOffsetIn);
}
 
void CbmCosy2019UnpackerTaskHodo::SetTimeOffsetNsAsicSts(UInt_t uAsicIdx, Double_t dOffsetIn)
{
  fUnpackerAlgoSts->SetTimeOffsetNsAsic(uAsicIdx, dOffsetIn);
}
 
void CbmCosy2019UnpackerTaskHodo::MaskNoisyChannelSts(UInt_t uFeb, UInt_t uChan, Bool_t bMasked)
{
  fvChanMasks.push_back(FebChanMaskSts {uFeb, uChan, bMasked});
}
 
void CbmCosy2019UnpackerTaskHodo::SetAdcCutSts(UInt_t uAdc) { fUnpackerAlgoSts->SetAdcCut(uAdc); }
 
ClassImp(CbmCosy2019UnpackerTaskHodo)