Clusterizer.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_CLUSTERIZER_H
#define O2_MID_CLUSTERIZER_H
 
#include <functional>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#include <gsl/gsl>
#include "DataFormatsMID/Cluster.h"
#include "DataFormatsMID/ROFRecord.h"
#include "MIDBase/MpArea.h"
#include "MIDClustering/PreCluster.h"
#include "MIDClustering/PreClusterHelper.h"
#include "MIDClustering/PreClustersDE.h"
 
namespace o2
{
namespace mid
{
class Clusterizer
{
 public:
  bool init(std::function<void(size_t, size_t)> func = [](size_t, size_t) {});
 
  void process(gsl::span<const PreCluster> preClusters, bool accumulate = false);
 
  void process(gsl::span<const PreCluster> preClusters, gsl::span<const ROFRecord> rofRecords);
 
  const std::vector<Cluster>& getClusters() { return mClusters; }
 
  const std::vector<ROFRecord>& getROFRecords() { return mROFRecords; }
 
 private:
  void reset();
 
  bool loadPreClusters(gsl::span<const PreCluster>& preClusters);
 
  bool makeClusters(PreClustersDE& pcs);
 
  void makeCluster(const MpArea& areaBP, const MpArea& areaNBP, uint8_t deId);
 
  void makeCluster(const MpArea& area, const uint8_t deId, int cathode);
 
  void makeCluster(const PreClustersDE::BP& pcBP, const PreClustersDE::BP& pcBPNeigh, const PreClustersDE::NBP& pcNBP, uint8_t deId);
 
  gsl::span<const PreCluster> mPreClusters;                    
  std::unordered_map<uint8_t, PreClustersDE> mPreClustersDE{}; 
  std::unordered_set<uint8_t> mActiveDEs{};                    
  PreClusterHelper mPreClusterHelper{};                        
  std::vector<Cluster> mClusters{};                            
  std::vector<ROFRecord> mROFRecords{};                        
  size_t mPreClusterOffset{0};                                 
 
  std::function<void(size_t, size_t)> mFunction{[](size_t, size_t) {}}; 
};
} // namespace mid
} // namespace o2
 
#endif /* O2_MID_CLUSTERIZER_H */