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 "dataobjects/FitChannelInfoColl.h"
#include "dataobjects/FitClustersColl.h"
#include "dataobjects/FitGeoParams.h"
#include "dataobjects/FitGeoParamsColl.h"
#include "dataobjects/FitHits.h"

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

using namespace EA;

namespace Herd {

// clang-format off
// clang-format on
class FitClusteringAlgo : public Algorithm {
public:
  FitClusteringAlgo(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 _snThreshold;
  float _snSeed;

  // Class members
  FitClustersColl _fitClustersColl;
  observer_ptr<FitGeoParamsColl> _fitGeoParamsColl;
  observer_ptr<FitChannelInfoColl> _fitChannelInfoColl;
  StorePtr _evStore, _globStore;

  // other stuff
  bool _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_ */