FitClusteringAlgo.h

Go to the documentation of this file.

/*
 * FitClusteringAlgo.h
 *
 *  Created on: 7 Apr 2020
 *      Author: Valerio Formato
 */
 
#ifndef HERD_FitClusteringAlgo_H_
#define HERD_FitClusteringAlgo_H_
 
// HerdSoftware headers
#include "herd/analysis/dataobjects/FitChannelInfoColl.h"
#include "herd/analysis/dataobjects/FitClustersColl.h"
#include "herd/analysis/dataobjects/FitGeoParams.h"
#include "herd/analysis/dataobjects/FitGeoParamsColl.h"
#include "herd/analysis/dataobjects/FitHits.h"
 
// EventAnalysis headers
#include "algorithm/Algorithm.h"
 
namespace Herd {
 
// clang-format off
// clang-format on
class FitClusteringAlgo : public EA::Algorithm {
public:
  FitClusteringAlgo(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_snThreshold;
  float m_snSeed;
 
  // Class members
  FitClustersColl m_fitClustersColl;
  EA::observer_ptr<FitGeoParamsColl> m_fitGeoParamsColl;
  EA::observer_ptr<FitChannelInfoColl> m_fitChannelInfoColl;
  EA::StorePtr m_evStore, m_globStore;
 
  // other stuff
  bool m_hasChannelInfo = false;
 
  std::vector<Cluster> _Clusterize(std::vector<Hit> &hits, const FitGeoParams &geoParams);
  std::vector<Cluster> _Clusterize(std::vector<Hit> &hits, const FitGeoParams &geoParams,
                                   const FitChannelInfo &channelInfo);
};
 
} // namespace Herd
 
#endif /* HERD_FitClusteringAlgo_H_ */