ClustersPerLayer.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 ALICEO2_ITSMFT_CLUSTERSPERLAYER_H
#define ALICEO2_ITSMFT_CLUSTERSPERLAYER_H
 
#include "DataFormatsITSMFT/ClusterID.h"
#include "Framework/Logger.h"
#include <array>
#include <vector>
 
namespace o2::itsmft
{
 
template <typename T>
class ClustersPerLayer
{
 public:
  void init(int nLr)
  {
    if (nLr < 1 || nLr > MaxClusLayers) {
      LOGP(fatal, "Clusters container cannot be initialized for {} layers, must be within 1:{}", nLr, MaxClusLayers);
    }
    mClusters.clear();
    mLrFirst.fill(0);
    mNLayers = nLr;
  }
 
  void beginLayer(int lr)
  {
    if (lr < 0 || lr >= mNLayers) {
      LOGP(fatal, "Clusters container was initialized for {} layers, cannot fill the layer {}", mNLayers, lr);
    }
    mLrFirst[lr] = int(mClusters.size());
  }
 
  void finalize()
  {
    for (int lr = mNLayers; lr <= MaxClusLayers; lr++) { // the slots above the filled ones are empty
      mLrFirst[lr] = int(mClusters.size());
    }
  }
 
  void clear()
  {
    mClusters.clear();
    mLrFirst.fill(0);
    mNLayers = 1;
  }
 
  auto& getClusters() { return mClusters; }
  const auto& getClusters() const { return mClusters; }
 
  auto getNLayers() const { return mNLayers; }
  auto getFirstIndex(int lr) const { return mLrFirst[lr]; }
  auto getNClusters(int lr) const { return mLrFirst[lr + 1] - mLrFirst[lr]; }
  auto size() const { return mClusters.size(); }
  bool empty() const { return mClusters.empty(); }
 
  int flatIndex(int composedID) const
  {
    return mLrFirst[clusID2Layer(composedID)] + clusID2Index(composedID);
  }
  const T& operator[](int composedID) const { return mClusters[flatIndex(composedID)]; }
 
 private:
  std::vector<T> mClusters{};                    
  std::array<int, MaxClusLayers + 1> mLrFirst{}; 
  int mNLayers = 1;                              
};
 
} // namespace o2::itsmft
 
#endif