CaloDBSCANClustering.h

Go to the documentation of this file.

/*
 * CaloDBSCANClustering.h
 *
 *  Created on: 24 Apr 2019
 *      Author: Valerio Formato
 */
 
#ifndef HERD_CaloDBSCANClustering_H_
#define HERD_CaloDBSCANClustering_H_
 
// HerdSoftware headers
#include "herd/analysis/dataobjects/CaloClusters.h"
#include "herd/analysis/dataobjects/CaloGeoParams.h"
 
// EventAnalysis headers
#include "algorithm/Algorithm.h"
 
namespace Herd {
 
//clang-format off
//clang-format on
 
class CaloDBSCANClustering : public EA::Algorithm {
public:
  CaloDBSCANClustering(const std::string &name);
 
  bool Initialize() override;
 
  bool Process() override;
 
  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, EA::Memory::Status memStatus) override;
 
private:
  // Algorithm parameters
  float m_threshold;
  float m_maxDist;
  unsigned int m_minHits;
  unsigned int m_minShowerHits;
  std::string m_caloHitsName;
 
  // Class members
  CaloClusters m_showers;
  EA::observer_ptr<CaloGeoParams> m_caloGeoParams;
  EA::EventStorePtr m_evStore;
  EA::GlobalStorePtr m_globStore;
 
  int m_nShowers;
  std::map<unsigned int, int> m_labels; // key: volumeID, value: shower number (label)
  void FindShowers(CaloHits &caloHits);
 
  using HitIt = std::vector<Hit>::const_iterator;
  void BuildShowerFromSeed(HitIt seedIt, const CaloHits &hits);
  std::vector<unsigned int> FindNeighborsHitID(HitIt hitIt, const CaloHits &hits);
  unsigned int CountHitNeighbors(HitIt hitIt, const CaloHits &hits);
  std::vector<unsigned int> NeighboringCubesID(unsigned int volID, CaloGeoParams &caloGeoParams);
 
  // value: all neighboring cubes ID to volumeID
  std::vector<std::vector<unsigned int>> m_neighborCubes;
  void BuildNeighborsMap(CaloGeoParams &caloGeoParams);
  const std::vector<unsigned int> &NeighboringCubesID(unsigned int volID) { return m_neighborCubes[volID]; };
};
 
} // namespace Herd
 
#endif /* HERD_CaloDBSCANClustering_H_ */