DigitReaderSpec.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include <vector>
 
#include "TTree.h"
 
#include "Framework/ControlService.h"
#include "Framework/ConfigParamRegistry.h"
#include "Framework/Logger.h"
#include "TOFWorkflowIO/DigitReaderSpec.h"
#include "DataFormatsParameters/GRPObject.h"
#include "CommonUtils/NameConf.h"
#include "DetectorsBase/TFIDInfoHelper.h"
 
using namespace o2::framework;
using namespace o2::tof;
 
namespace o2
{
namespace tof
{
 
void DigitReader::init(InitContext& ic)
{
  LOG(debug) << "Init Digit reader!";
  auto filename = o2::utils::Str::concat_string(o2::utils::Str::rectifyDirectory(ic.options().get<std::string>("input-dir")),
                                                ic.options().get<std::string>("tof-digit-infile"));
  mDelayInMuSec1TF = atof(ic.options().get<std::string>("delay-1st-tf").c_str()) * 1E6;
 
  mFile.reset(TFile::Open(filename.c_str()));
  if (!mFile->IsOpen()) {
    LOG(error) << "Cannot open the " << filename.c_str() << " file !";
    mState = 0;
    return;
  }
  mState = 1;
}
 
void DigitReader::run(ProcessingContext& pc)
{
  static bool firstCall = true;
  if (firstCall) {
    usleep(mDelayInMuSec1TF);
  }
  firstCall = false;
 
  if (mState != 1) {
    return;
  }
 
  std::unique_ptr<TTree> treeDig((TTree*)mFile->Get("o2sim"));
 
  if (treeDig) {
    treeDig->SetBranchAddress("TOFDigit", &mPdigits);
    treeDig->SetBranchAddress("TOFReadoutWindow", &mProw);
    treeDig->SetBranchAddress("TOFPatterns", &mPpatterns);
 
    if (mUseMC) {
      treeDig->SetBranchAddress("TOFDigitMCTruth", &mPlabels);
    }
 
    treeDig->GetEntry(mCurrentEntry);
 
    // fill diagnostic frequencies
    mFiller.clearCounts();
    for (auto digit : mDigits) {
      mFiller.addCount(digit.getChannel());
    }
    mFiller.setReadoutWindowData(mRow, mPatterns);
    mFiller.fillDiagnosticFrequency();
    mDiagnostic = mFiller.getDiagnosticFrequency();
    auto creationTime = pc.services().get<o2::framework::TimingInfo>().creation;
    mDiagnostic.setTimeStamp(creationTime / 1000);
    // add TFIDInfo
    o2::dataformats::TFIDInfo tfinfo;
    o2::base::TFIDInfoHelper::fillTFIDInfo(pc, tfinfo);
    mDiagnostic.setTFIDInfo(tfinfo);
 
    // add digits loaded in the output snapshot
    pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "DIGITS", 0}, mDigits);
    pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "READOUTWINDOW", 0}, mRow);
    pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "PATTERNS", 0}, mPatterns);
    pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "DIAFREQ", 0}, mDiagnostic);
    if (mUseMC) {
      pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "DIGITSMCTR", 0}, mLabels);
    }
 
    static o2::parameters::GRPObject::ROMode roMode = o2::parameters::GRPObject::CONTINUOUS;
 
    LOG(debug) << "TOF: Sending ROMode= " << roMode << " to GRPUpdater";
    pc.outputs().snapshot(Output{o2::header::gDataOriginTOF, "ROMode", 0}, roMode);
  } else {
    LOG(error) << "Cannot read the TOF digits !";
    return;
  }
 
  mCurrentEntry++;
 
  if (mCurrentEntry >= treeDig->GetEntries()) {
    mState = 2;
    //pc.services().get<ControlService>().readyToQuit(QuitRequest::Me);
    pc.services().get<ControlService>().endOfStream();
  }
}
 
DataProcessorSpec getDigitReaderSpec(bool useMC)
{
  std::vector<OutputSpec> outputs;
  outputs.emplace_back(o2::header::gDataOriginTOF, "DIGITS", 0, Lifetime::Timeframe);
  outputs.emplace_back(o2::header::gDataOriginTOF, "READOUTWINDOW", 0, Lifetime::Timeframe);
  outputs.emplace_back(o2::header::gDataOriginTOF, "DIAFREQ", 0, Lifetime::Timeframe);
  if (useMC) {
    outputs.emplace_back(o2::header::gDataOriginTOF, "DIGITSMCTR", 0, Lifetime::Timeframe);
  }
  outputs.emplace_back(o2::header::gDataOriginTOF, "PATTERNS", 0, Lifetime::Timeframe);
  outputs.emplace_back(o2::header::gDataOriginTOF, "ROMode", 0, Lifetime::Timeframe);
 
  return DataProcessorSpec{
    "tof-digit-reader",
    Inputs{},
    outputs,
    AlgorithmSpec{adaptFromTask<DigitReader>(useMC)},
    Options{
      {"tof-digit-infile", VariantType::String, "tofdigits.root", {"Name of the input file"}},
      {"delay-1st-tf", VariantType::String, "none", {"delay in seconds before 1st TF"}},
      {"input-dir", VariantType::String, "none", {"Input directory"}}}};
}
 
} // namespace tof
} // namespace o2