CollectCalibInfoTOF.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 <TTree.h>
#include <cassert>
 
#include <fairlogger/Logger.h>
#include "TOFBase/Geo.h"
 
#include <TFile.h>
#include "TOFCalibration/CollectCalibInfoTOF.h"
 
using namespace o2::globaltracking;
 
ClassImp(CollectCalibInfoTOF);
 
//______________________________________________
void CollectCalibInfoTOF::run()
{
 
  if (!mInitDone) {
    LOG(fatal) << "init() was not done yet";
  }
 
  mTimerTot.Start();
 
  while (loadTOFCalibInfo()) { // fill here all histos you need
    for (int ihit = 0; ihit < mTOFCalibInfo->size(); ihit++) {
      addHit((*mTOFCalibInfo)[ihit]);
    }
  }
  fillTree(); // filling with whatever we have in memory
 
  // fit and extract calibration parameters once histos are filled
  // ...
 
  mTimerTot.Stop();
  printf("Timing:\n");
  printf("Total:        ");
  mTimerTot.Print();
}
 
//______________________________________________
void CollectCalibInfoTOF::init()
{
 
  if (mInitDone) {
    LOG(error) << "Initialization was already done";
    return;
  }
 
  attachInputTrees();
 
  // create output branch with track-tof matching
  if (mOutputTree) {
    mOutputTree->Branch(mOutputBranchName.data(), &mTOFCalibInfoOut);
    LOG(info) << "Accumulated calib info TOF will be stored in " << mOutputBranchName << " branch of tree "
              << mOutputTree->GetName();
  } else {
    LOG(error) << "Output tree is not attached, accumulated CalibInfoTOFshort will not be stored";
  }
  mInitDone = true;
 
  {
    mTimerTot.Stop();
    mTimerTot.Reset();
  }
 
  print();
}
 
//______________________________________________
void CollectCalibInfoTOF::print() const
{
 
  LOG(info) << "****** component for calibration of TOF channels ******";
  if (!mInitDone) {
    LOG(info) << "init is not done yet - nothing to print";
    return;
  }
 
  LOG(info) << "**********************************************************************";
}
 
//______________________________________________
void CollectCalibInfoTOF::attachInputTrees()
{
 
  if (!mTreeTOFCalibInfo) {
    LOG(fatal) << "Input tree with TOF calib infos is not set";
  }
 
  if (!mTreeTOFCalibInfo->GetBranch(mTOFCalibInfoBranchName.data())) {
    LOG(fatal) << "Did not find TOF calib info branch " << mTOFCalibInfoBranchName << " in the input tree";
  }
  mTreeTOFCalibInfo->SetBranchAddress(mTOFCalibInfoBranchName.data(), &mTOFCalibInfo);
  LOG(info) << "Attached tracksTOF calib info " << mTOFCalibInfoBranchName << " branch with " << mTreeTOFCalibInfo->GetEntries()
            << " entries";
 
  mCurrTOFInfoTreeEntry = -1;
}
 
//______________________________________________
bool CollectCalibInfoTOF::loadTOFCalibInfo()
{
  printf("Loading TOF calib infos: number of entries in tree = %lld\n", mTreeTOFCalibInfo->GetEntries());
 
  while (++mCurrTOFInfoTreeEntry < mTreeTOFCalibInfo->GetEntries()) {
    mTreeTOFCalibInfo->GetEntry(mCurrTOFInfoTreeEntry);
    LOG(info) << "Loading TOF calib info entry " << mCurrTOFInfoTreeEntry << " -> " << mTOFCalibInfo->size() << " infos";
 
    if (!mTOFCalibInfo->size()) {
      continue;
    }
    return true;
  }
  --mCurrTOFInfoTreeEntry;
 
  return false;
}
//______________________________________________
void CollectCalibInfoTOF::addHit(o2::dataformats::CalibInfoTOF& calibInfo)
{
 
 
  mTOFCollectedCalibInfo[calibInfo.getTOFChIndex()].emplace_back(calibInfo.getTimestamp(), calibInfo.getDeltaTimePi(), calibInfo.getTot(), calibInfo.getFlags());
  if (mTOFCollectedCalibInfo[calibInfo.getTOFChIndex()].size() == MAXNUMBEROFHITS) { // the current channel has arrived to the limit of hits that we can store between two fills --> filling the tree
    fillTree();
  }
  if (calibInfo.getTimestamp() < mMinTimestamp.GetVal() || mMinTimestamp.GetVal() == -1) {
    mMinTimestamp.SetVal(calibInfo.getTimestamp());
  } else if (calibInfo.getTimestamp() > mMaxTimestamp.GetVal()) {
    mMaxTimestamp.SetVal(calibInfo.getTimestamp());
  }
}
//______________________________________________
void CollectCalibInfoTOF::fillTree()
{
 
 
  for (int ich = 0; ich < Geo::NCHANNELS; ich++) {
    mTOFCalibInfoOut = &mTOFCollectedCalibInfo[ich];
    mOutputTree->Fill();
    mTOFCollectedCalibInfo[ich].clear();
  }
}