MCLabelAccumulator.h

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.
 
 
#ifndef O2_GPU_MC_LABEL_ACCUMULATOR_H
#define O2_GPU_MC_LABEL_ACCUMULATOR_H
 
#include "clusterFinderDefs.h"
#include "Array2D.h"
#include <bitset>
#include <vector>
 
namespace o2
{
class MCCompLabel;
namespace dataformats
{
class MCCompLabel;
template <typename T>
class ConstMCTruthContainerView;
using ConstMCLabelContainerView = o2::dataformats::ConstMCTruthContainerView<o2::MCCompLabel>;
} // namespace dataformats
} // namespace o2
 
namespace o2::gpu
{
 
class GPUTPCClusterFinder;
struct GPUTPCClusterMCInterimArray;
 
class MCLabelAccumulator
{
 
 public:
  MCLabelAccumulator(GPUTPCClusterFinder&);
 
  void collect(const ChargePos&, tpccf::Charge);
 
  bool engaged() const { return mLabels != nullptr && mOutput != nullptr; }
 
  void commit(tpccf::Row, uint32_t, uint32_t);
 
 private:
  Array2D<const uint32_t> mIndexMap;
  const o2::dataformats::ConstMCLabelContainerView* mLabels = nullptr;
  GPUTPCClusterMCInterimArray* mOutput = nullptr;
 
  std::bitset<64> mMaybeHasLabel;
  std::vector<o2::MCCompLabel> mClusterLabels;
};
 
} // namespace o2::gpu
 
#endif