CbmMcbm2018UnpackerTaskRich.cxx

Go to the documentation of this file.

 
//TODO: check that all data members are initialized in the constructor
 
#include "CbmMcbm2018UnpackerTaskRich.h"
 
// ROOT
#include <TCanvas.h>
#include <TClonesArray.h>
#include <TFile.h>
#include <THttpServer.h>
#include <TROOT.h>
 
// FairRoot
#include <FairLogger.h>
#include <FairParGenericSet.h>
#include <FairRootManager.h>
#include <FairRun.h>
#include <FairRunOnline.h>
#include <FairRuntimeDb.h>
 
// CbmRoot
#include "CbmMcbm2018RichPar.h"
#include "CbmMcbm2018UnpackerAlgoRich.h"
 
#include <chrono>
#include <thread>
 
//TODO global variable, really?
Bool_t bMcbm2018UnpackerTaskRichResetHistos = kFALSE;
 
CbmMcbm2018UnpackerTaskRich::CbmMcbm2018UnpackerTaskRich()
  : CbmMcbmUnpack()
  , fbMonitorMode(kFALSE)
  , fbDebugMonitorMode(kFALSE)
  , fbWriteOutput(kTRUE)
  , fUnpackerAlgo(new CbmMcbm2018UnpackerAlgoRich()) {}
 
CbmMcbm2018UnpackerTaskRich::~CbmMcbm2018UnpackerTaskRich() {
  if (fUnpackerAlgo) {
    delete fUnpackerAlgo;
    fUnpackerAlgo = nullptr;
  }
}
 
Bool_t CbmMcbm2018UnpackerTaskRich::Init() {
  LOG(info) << "CbmMcbm2018UnpackerTaskRich::Init";
  LOG(info) << "Initializing mCBM RICH 2018 Unpacker";
 
  FairRootManager* ioman = FairRootManager::Instance();
  if (ioman == NULL) { LOG(fatal) << "No FairRootManager instance"; }
 
  fpvDigiRich = &(fUnpackerAlgo->GetVector());
  ioman->RegisterAny("RichDigi", fpvDigiRich, fbWriteOutput);
 
  return kTRUE;
}
 
Bool_t CbmMcbm2018UnpackerTaskRich::DoUnpack(const fles::Timeslice& ts,
                                             size_t component) {
  //std::this_thread::sleep_for(std::chrono::milliseconds(10));
  if (fbMonitorMode && bMcbm2018UnpackerTaskRichResetHistos) {
    LOG(info) << "Reset RICH unpacker histos ";
    fUnpackerAlgo->ResetHistograms();
    bMcbm2018UnpackerTaskRichResetHistos = kFALSE;
  }  // if( fbMonitorMode && bMcbm2018UnpackerTaskRichResetHistos )
 
  if (kFALSE == fUnpackerAlgo->ProcessTs(ts, component)) {
    LOG(error) << "Failed processing TS " << ts.index()
               << " in mRICH unpacker algorithm class";
    return kTRUE;
  }  // if( kFALSE == fUnpackerAlgo->ProcessTs( ts, component ) )
 
  /*
   //=> Done in the algo!!!
   sort(fpvDigiRich->begin(), fpvDigiRich->end(),
        [](const CbmRichDigi & a, const CbmRichDigi & b) -> bool
        {
          return a.GetTime() < b.GetTime();
        });
*/
 
  if (0 == fulTsCounter % 10000)
    LOG(info) << "Processed " << fulTsCounter << "TS";
  fulTsCounter++;
 
  return kTRUE;
}
 
void CbmMcbm2018UnpackerTaskRich::Reset() { fUnpackerAlgo->ClearVector(); }
 
void CbmMcbm2018UnpackerTaskRich::Finish() {
  if (kTRUE == fbMonitorMode) {
    std::vector<std::pair<TNamed*, std::string>> vHistos =
      fUnpackerAlgo->GetHistoVector();
 
    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("data/HistosUnpackerRich.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 )
 
    // Restore original directory position
    oldDir->cd();
    histoFile->Close();
  }  // if( kTRUE == fbMonitorMode )
}
 
void CbmMcbm2018UnpackerTaskRich::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 )
}
 
Bool_t CbmMcbm2018UnpackerTaskRich::InitContainers() {
  LOG(info) << "Init parameter containers for " << GetName();
 
  CbmMcbm2018RichPar* pUnpackPar = dynamic_cast<CbmMcbm2018RichPar*>(
    FairRun::Instance()->GetRuntimeDb()->getContainer("CbmMcbm2018RichPar"));
  if (nullptr == pUnpackPar) {
    LOG(error) << "Failed to obtain parameter container CbmMcbm2018RichPar";
    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();
 
  if (kTRUE == fbMonitorMode) {
    initOK &= fUnpackerAlgo->CreateHistograms();
 
    std::vector<std::pair<TNamed*, std::string>> vHistos =
      fUnpackerAlgo->GetHistoVector();
    std::vector<std::pair<TCanvas*, std::string>> vCanvases =
      fUnpackerAlgo->GetCanvasVector();
 
    THttpServer* server = FairRunOnline::Instance()->GetHttpServer();
    if (nullptr != server) {
      for (UInt_t uHisto = 0; uHisto < vHistos.size(); ++uHisto) {
        server->Register(Form("/rich/%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) {
        server->Register(
          Form("/rich/%s", vCanvases[uCanv].second.data()),
          gROOT->FindObject((vCanvases[uCanv].first)->GetName()));
      }  //  for( UInt_t uCanv = 0; uCanv < vCanvases.size(); ++uCanv )
 
 
      server->RegisterCommand("/Reset_UnpRich_Hist",
                              "bMcbm2018UnpackerTaskRichResetHistos=kTRUE");
      server->Restrict("/Reset_UnpRich_Hist", "allow=admin");
    }  // if( nullptr != server )
 
  }  // if( kTRUE == fbMonitorMode )
 
  fUnpackerAlgo->SetMonitorMode(fbMonitorMode);
 
  return initOK;
}
 
Bool_t CbmMcbm2018UnpackerTaskRich::ReInitContainers() {
  LOG(info) << "ReInit parameter containers for " << GetName();
  Bool_t initOK = fUnpackerAlgo->ReInitContainers();
 
  return initOK;
}
 
void CbmMcbm2018UnpackerTaskRich::AddMsComponentToList(size_t component,
                                                       UShort_t usDetectorId) {
  fUnpackerAlgo->AddMsComponentToList(component, usDetectorId);
}
 
void CbmMcbm2018UnpackerTaskRich::SetNbMsInTs(size_t /*uCoreMsNb*/,
                                              size_t /*uOverlapMsNb*/) {}
 
void CbmMcbm2018UnpackerTaskRich::SetIgnoreOverlapMs(Bool_t bFlagIn) {
  fUnpackerAlgo->SetIgnoreOverlapMs(bFlagIn);
}
 
void CbmMcbm2018UnpackerTaskRich::SetTimeOffsetNs(Double_t dOffsetIn) {
  fUnpackerAlgo->SetTimeOffsetNs(dOffsetIn);
}
 
void CbmMcbm2018UnpackerTaskRich::DoTotCorr(Bool_t bDoToTCorr) {
  fUnpackerAlgo->DoTotCorr(bDoToTCorr);
}
 
ClassImp(CbmMcbm2018UnpackerTaskRich)