ClusterLabeler.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 "MIDSimulation/ClusterLabeler.h"
 
#include <map>
 
namespace o2
{
namespace mid
{
 
void ClusterLabeler::process(gsl::span<const PreCluster> preClusters, const o2::dataformats::MCTruthContainer<MCCompLabel>& inMCContainer, gsl::span<const Cluster> clusters, gsl::span<const std::array<size_t, 2>> correlations)
{
  mMCContainer.clear();
 
  std::map<size_t, std::vector<size_t>> reordered;
  for (auto& corr : correlations) {
    reordered[corr[0]].emplace_back(corr[1]);
  }
 
  for (auto& corr : reordered) {
    for (auto& pcIdx : corr.second) {
      int cathode = static_cast<int>(preClusters[pcIdx].cathode);
      auto labels = inMCContainer.getLabels(pcIdx);
      addLabels(corr.first, cathode, labels);
    }
  }
}
 
MCClusterLabel* ClusterLabeler::findLabel(size_t idx, const MCCompLabel& pcLabel)
{
  if (idx >= mMCContainer.getIndexedSize()) {
    return nullptr;
  }
 
  for (auto& cLabel : mMCContainer.getLabels(idx)) {
    if (pcLabel.compare(cLabel) == 1) {
      return &cLabel;
    }
  }
 
  return nullptr;
}
 
void ClusterLabeler::addLabels(size_t idx, int cathode, gsl::span<const MCCompLabel>& labels)
{
  for (auto& pcLabel : labels) {
    MCClusterLabel* foundLabel = findLabel(idx, pcLabel);
    if (foundLabel) {
      if (cathode == 0) {
        foundLabel->setIsFiredBP(true);
      } else {
        foundLabel->setIsFiredNBP(true);
      }
    } else {
      bool isFiredBP = (cathode == 0);
      bool isFiredNBP = (cathode == 1);
      MCClusterLabel cLabel(pcLabel.getTrackID(), pcLabel.getEventID(), pcLabel.getSourceID(), isFiredBP, isFiredNBP);
      mMCContainer.addElement(idx, cLabel);
    }
  }
}
 
} // namespace mid
} // namespace o2