ClustererTask.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 "TPCReconstruction/ClustererTask.h"
 
#include <fairlogger/Logger.h> // for LOG
#include "FairRootManager.h" // for FairRootManager
 
ClassImp(o2::tpc::ClustererTask);
 
using namespace o2::tpc;
using namespace o2::dataformats;
 
//_____________________________________________________________________
ClustererTask::ClustererTask(int sectorid)
  : FairTask("TPCClustererTask"),
    mClusterSector(sectorid),
    mHwClusterer(),
    mDigitsArray(),
    mDigitMCTruthArray(),
    mHwClustersArray(),
    mHwClustersMCTruthArray()
{
}
 
//_____________________________________________________________________
InitStatus ClustererTask::Init()
{
  LOG(debug) << "Enter Initializer of ClustererTask";
 
  FairRootManager* mgr = FairRootManager::Instance();
  if (!mgr) {
    LOG(error) << "Could not instantiate FairRootManager. Exiting ...";
    return kERROR;
  }
 
  if (mClusterSector < 0 || mClusterSector >= Sector::MAXSECTOR) {
    LOG(error) << "Sector ID " << mClusterSector << " is not supported. Exiting ...";
    return kERROR;
  }
 
  // Register input container
  std::stringstream sectornamestr;
  sectornamestr << "TPCDigit" << mClusterSector;
  LOG(info) << "FETCHING DIGITS FOR SECTOR " << mClusterSector << "\n";
  mDigitsArray = std::unique_ptr<const std::vector<Digit>>(
    mgr->InitObjectAs<const std::vector<Digit>*>(sectornamestr.str().c_str()));
  if (!mDigitsArray) {
    LOG(error) << "TPC points not registered in the FairRootManager. Exiting ...";
    return kERROR;
  }
  std::stringstream mcsectornamestr;
  mcsectornamestr << "TPCDigitMCTruth" << mClusterSector;
  mDigitMCTruthArray = std::unique_ptr<const ConstMCLabelContainerView>(
    mgr->InitObjectAs<const ConstMCLabelContainerView*>(mcsectornamestr.str().c_str()));
  if (!mDigitMCTruthArray) {
    LOG(error) << "TPC MC Truth not registered in the FairRootManager. Exiting ...";
    return kERROR;
  }
 
  // Register output container
  mHwClustersArray = std::make_unique<std::vector<OutputType>>();
  // a trick to register the unique pointer with FairRootManager
  static auto clusterArrayTmpPtr = mHwClustersArray.get();
  mgr->RegisterAny(Form("TPCClusterHW%i", mClusterSector), clusterArrayTmpPtr, kTRUE);
 
  // Register MC Truth output container
  mHwClustersMCTruthArray = std::make_unique<MCLabelContainer>();
  // a trick to register the unique pointer with FairRootManager
  static auto clusterMcTruthTmpPtr = mHwClustersMCTruthArray.get();
  mgr->RegisterAny(Form("TPCClusterHWMCTruth%i", mClusterSector), clusterMcTruthTmpPtr, kTRUE);
 
  // create clusterer and pass output pointer
  mHwClusterer = std::make_unique<HwClusterer>(mHwClustersArray.get(), mClusterSector, mHwClustersMCTruthArray.get());
  mHwClusterer->setContinuousReadout(mIsContinuousReadout);
 
  // TODO: implement noise/pedestal objects
  //    mHwClusterer->setNoiseObject(...);
  //    mHwClusterer->setPedestalObject(...);
 
  return kSUCCESS;
}
 
//_____________________________________________________________________
void ClustererTask::Exec(Option_t* option)
{
  LOG(debug) << "Running clusterization on event " << mEventCount << " with " << mDigitsArray->size() << " digits.";
 
  if (mHwClustersArray) {
    mHwClustersArray->clear();
  }
  if (mHwClustersMCTruthArray) {
    mHwClustersMCTruthArray->clear();
  }
 
  mHwClusterer->process(gsl::span<o2::tpc::Digit const>(mDigitsArray->data(), mDigitsArray->size()), *mDigitMCTruthArray.get());
  LOG(debug) << "Hw clusterer delivered " << mHwClustersArray->size() << " cluster container";
 
  ++mEventCount;
}
 
//_____________________________________________________________________
void ClustererTask::FinishTask()
{
  LOG(debug) << "Finish clusterization";
 
  if (mHwClustersArray) {
    mHwClustersArray->clear();
  }
  if (mHwClustersMCTruthArray) {
    mHwClustersMCTruthArray->clear();
  }
 
  mHwClusterer->finishProcess(*mDigitsArray.get(), *mDigitMCTruthArray.get());
  LOG(debug) << "Hw clusterer delivered " << mHwClustersArray->size() << " cluster container";
 
  ++mEventCount;
}