PreClustersDE.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_MID_PRECLUSTERSDE_H
#define O2_MID_PRECLUSTERSDE_H
 
#include <cstdint>
#include <cstddef>
#include <array>
#include <vector>
#include "MIDBase/MpArea.h"
 
namespace o2
{
namespace mid
{
 
// Internal pre-cluster structure for MID clustering
class PreClustersDE
{
 public:
  struct NBP {
    size_t index;               
    int paired;                 
    std::array<MpArea, 7> area; 
  };
 
  struct BP {
    size_t index; 
    int paired;   
    MpArea area;  
  };
 
  NBP& getPreClusterNBP(int idx) { return mPreClustersNBP[idx]; }
 
  const NBP& getPreClusterNBP(int idx) const { return mPreClustersNBP[idx]; }
 
  BP& getPreClusterBP(int icolumn, int idx) { return mPreClustersBP[icolumn][idx]; }
 
  const BP& getPreClusterBP(int icolumn, int idx) const { return mPreClustersBP[icolumn][idx]; }
 
  size_t getNPreClustersNBP() const { return mPreClustersNBP.size(); }
 
  size_t getNPreClustersBP(int icolumn) const { return mPreClustersBP[icolumn].size(); }
 
  void setDEId(uint8_t deId) { mDEId = deId; }
 
  uint8_t getDEId() const { return mDEId; }
 
  std::vector<int> getNeighbours(int icolumn, int idx) const;
 
  std::vector<NBP>& getPreClustersNBP() { return mPreClustersNBP; }
 
  std::vector<BP>& getPreClustersBP(int icolumn) { return mPreClustersBP[icolumn]; }
 
  bool init();
  void reset();
 
 private:
  uint8_t mDEId;                                 
  std::vector<NBP> mPreClustersNBP;              
  std::array<std::vector<BP>, 7> mPreClustersBP; 
};
} // namespace mid
} // namespace o2
 
#endif /* O2_MID_PRECLUSTERSDE_H */