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 "algorithms/tracking/HoughFinderTools/HFCluster.h"
#include "algorithms/tracking/HoughFinderTools/HoughTrackStub.h"
#include "common/ViewArray.h"
#include "dataobjects/ScdClusters.h"
#include "dataobjects/ScdGeoParamsColl.h"
#include "dataobjects/StkClusters.h"
#include "dataobjects/StkGeoParamsColl.h"
#include "dataobjects/Track2DColl.h"

// Event Analysis headers
#include "algorithm/Algorithm.h"
#include "dataobjects/Track.h"

// headers for debug only
// #include "TCanvas.h"
// #include "TH2D.h"

// c++ headers
#include <tuple>

using namespace EA;

namespace Herd {
// clang-format off
// clang-format on

using HFClusters = std::vector<std::vector<HFCluster>>;

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

  bool Initialize() override;

  bool Process() override;

  bool Finalize() override;

  int GetVotingIndex(float R, float Theta);

  std::pair<float, float> GetRTheta(int votingIndex);

  void FindTracks(RefFrame::View view, const HFClusters &clusters);

private:
  std::string _collName = "tracks2DColl_hough";
  bool _publishAlias = true;

  bool _useSCD = true;
  bool _useFIT = true;

  unsigned int _nThetaDiv;
  unsigned int _nRDiv;
  float _RMax;
  unsigned int _nVotes;

  HFClusters _clXZ;
  HFClusters _clYZ;
  HFClusters _clXY;

  // TH2D *_debugHist;
  // TCanvas *_debugCanv;

  // handle both stk and scd hits and import them
  void ImportSiliconDetectorHits(const SiliconDetectorClustersColl &clColl,
                                 const SiliconDetectorGeoParamsColl &geoParamsColl, HFCluster::DetectorType detType);

  // handle fit hits and import them
  void ImportFitHits(const FitClustersColl &clColl, const FitGeoParamsColl &geoParamsColl);

  // Pretty self explanatory...
  inline void ClearVotingSpace() { _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> GetAvRTheta(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<StkClustersColl, 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>> _votingSpace;

  EA::observer_ptr<StkGeoParamsColl> _stkGeoParamsColl;
  EA::observer_ptr<ScdGeoParamsColl> _scdGeoParamsColl;
  EA::observer_ptr<FitGeoParamsColl> _fitGeoParamsColl;
  Track2DColl _trackColl;
  StorePtr _globStore; // Pointer to the global data store
  StorePtr _evStore;   // Pointer to the event data store
};
} // namespace Herd
#endif