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 "dataobjects/CaloClusters.h"
#include "dataobjects/CaloGeoParams.h"

// EventAnalysis headers
#include "algorithm/Algorithm.h"

using namespace EA;

namespace Herd {

//clang-format off
//clang-format on

class CaloDBSCANClustering : public Algorithm {
public:
  CaloDBSCANClustering(const std::string &name);

  bool Initialize() override;

  bool Process() override;

  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, Memory::Status memStatus) override;

private:
  // Algorithm parameters
  float _threshold;
  float _maxDist;
  unsigned int _minHits;
  unsigned int _minShowerHits;
  std::string _caloHitsName;

  // Class members
  CaloClusters _showers;
  observer_ptr<CaloGeoParams> _caloGeoParams;
  EventStorePtr _evStore;
  GlobalStorePtr _globStore;

  int 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> GetNeighboringCubesID(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> &GetNeighboringCubesID(unsigned int volID) { return m_neighborCubes[volID]; };
};

} // namespace Herd

#endif /* HERD_CaloDBSCANClustering_H_ */