HoughFinder2DAlgo.h

Go to the documentation of this file.

/*
 * HoughFinder2DAlgo.h
 *
 *  Created on: 20 Nov 2018
 *      Author: Valerio Formato
 */
 
#ifndef HoughFinder2DAlgo_H_
#define HoughFinder2DAlgo_H_
 
// HERD headers
#include "herd/analysis/algorithms/tracking/HoughFinderTools/HFCluster.h"
#include "herd/analysis/algorithms/tracking/HoughFinderTools/HoughTrackStub.h"
#include "herd/analysis/common/ViewArray.h"
#include "herd/analysis/dataobjects/ScdClusters.h"
#include "herd/analysis/dataobjects/ScdGeoParamsColl.h"
#include "herd/analysis/dataobjects/Track2DColl.h"
 
// Event Analysis headers
#include "algorithm/Algorithm.h"
#include "herd/analysis/dataobjects/Track.h"
 
// headers for debug only
// #include "TCanvas.h"
// #include "TH2D.h"
 
// c++ headers
#include <tuple>
#include <variant>
 
namespace Herd {
// clang-format off
// clang-format on
 
using HFClusters = std::vector<std::vector<HFCluster>>;
using ClusterColl = std::variant<FitClustersColl, ScdClustersColl>;
using GeoParamsColl = std::variant<FitGeoParamsColl, ScdGeoParamsColl>;
 
class HoughFinder2DAlgo : public EA::Algorithm {
public:
  HoughFinder2DAlgo(const std::string &name);
 
  bool Initialize() override;
 
  bool Process() override;
 
  bool Finalize() override;
 
  int VotingIndex(float R, float Theta);
 
  std::pair<float, float> RTheta(int votingIndex);
 
  void FindTracks(RefFrame::View view, const HFClusters &clusters);
 
private:
  std::string m_collName = "tracks2DColl_hough";
  bool m_publishAlias = true;
 
  bool m_useSCD = true;
  bool m_useFIT = true;
 
  unsigned int m_nThetaDiv;
  unsigned int m_nRDiv;
  float m_RMax;
  unsigned int m_nVotes;
 
  HFClusters m_clXZ;
  HFClusters m_clYZ;
  HFClusters m_clXY;
 
  // TH2D *_debugHist;
  // TCanvas *_debugCanv;
  void ImportHits(const ClusterColl &clColl, const GeoParamsColl &geoParamsColl, HFCluster::DetectorType detType);
 
  // Pretty self explanatory...
  inline void ClearVotingSpace() { m_votingSpace.clear(); };
 
  // Fill voting space with clusters. Pass the normal axis to the view to
  // determine which coordinates will be used.
  void FillVotingSpace(RefFrame::View view, const HFClusters &clusters);
 
  // Remove a cluster votes from the voting space. Avoids a cluster partecipating
  // in more than one track.
  // void RemoveFromVotingSpace(HoughTrackStub trStub);
  void RemoveFromVotingSpace(const Cluster &cluster);
 
  // Find out the voting space cell with highest number of votes
  int FindCandidateTrack();
 
  // Aggregate all compatible votes around maximum
  std::vector<HoughTrackStub> AccumulateVotes(int votingIndex);
 
  // Search around vote for compatible stubs
  void SearchVotes(int votingIndex, std::vector<int> &vIndexes);
 
  // Find back a cluster given the seed ID
  std::pair<bool, HFCluster> FindClusterFromSeedIndex(const HFClusters &clColl, unsigned int clID);
 
  // Get average direction from list of votes
  std::pair<float, float> RThetaAverage(const std::vector<HoughTrackStub> &votes);
 
  // Build a track from a set of votes and eventually add it to the collection
  Track2D BuildTrackFromVotes(std::vector<HoughTrackStub> &votes, const HFClusters &clColl);
 
  // Separate internal representation of clusters into separate detector collections, depending on a list
  // of votes
  std::tuple<ScdClustersColl, FitClustersColl> FillClusterCollections(const std::vector<HoughTrackStub> &votes,
                                                                      const HFClusters &clColl);
 
  // int index -> list of track stubs that voted for that index
  std::unordered_map<int, std::vector<HoughTrackStub>> m_votingSpace;
 
  EA::observer_ptr<ScdGeoParamsColl> m_scdGeoParamsColl;
  EA::observer_ptr<FitGeoParamsColl> m_fitGeoParamsColl;
  Track2DColl m_trackColl;
  EA::StorePtr m_globStore; // Pointer to the global data store
  EA::StorePtr m_evStore;   // Pointer to the event data store
};
} // namespace Herd
#endif