BlurredClusteringAlg.h

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////
// Implementation of the Blurred Clustering algorithm
//
// Converts a hit map into a 2D image of the hits before convoling
// with a Gaussian function to introduce a weighted blurring.
// Clustering proceeds on this blurred image to create more
// complete clusters.
//
// M Wallbank (m.wallbank@sheffield.ac.uk), May 2015
////////////////////////////////////////////////////////////////////

#ifndef BlurredClustering_h
#define BlurredClustering_h

// Framework includes
#include "art/Framework/Services/Registry/ServiceHandle.h"

// LArSoft includes
#include "larcore/Geometry/Geometry.h"
#include "lardataobj/RecoBase/Hit.h"
#include "larevt/CalibrationDBI/Interface/ChannelStatusService.h"
namespace detinfo {
  class DetectorProperties;
}
namespace fhicl {
  class ParameterSet;
}
namespace lariov {
  class ChannelStatusProvider;
}
namespace geo {
  struct WireID;
}

// ROOT
#include "TString.h"
class TCanvas;
class TH2F;

// c++
#include <array>
#include <string>
#include <vector>

namespace cluster {
  class BlurredClusteringAlg;
}

class cluster::BlurredClusteringAlg {
public:
  BlurredClusteringAlg(fhicl::ParameterSet const& pset);
  ~BlurredClusteringAlg();

  /// Create the PDF to save debug images
  void CreateDebugPDF(int run, int subrun, int event);

  /// Takes a vector of clusters (itself a vector of hits) and turns them into clusters using the initial hit selection
  void ConvertBinsToClusters(std::vector<std::vector<double>> const& image,
                             std::vector<std::vector<int>> const& allClusterBins,
                             std::vector<art::PtrVector<recob::Hit>>& clusters) const;

  /// Takes hit map and returns a 2D vector representing wire and tick, filled with the charge
  std::vector<std::vector<double>> ConvertRecobHitsToVector(
    std::vector<art::Ptr<recob::Hit>> const& hits,
    int readoutWindowSize);

  /// Find clusters in the histogram
  int FindClusters(std::vector<std::vector<double>> const& image,
                   std::vector<std::vector<int>>& allcluster) const;

  /// Find the global wire position
  int GlobalWire(geo::WireID const& wireID) const;

  /// Applies Gaussian blur to image
  std::vector<std::vector<double>> GaussianBlur(
    std::vector<std::vector<double>> const& image) const;

  /// Minimum size of cluster to save
  unsigned int
  GetMinSize() const noexcept
  {
    return fMinSize;
  }

  /// Converts a 2D vector in a histogram for the debug pdf
  TH2F* MakeHistogram(std::vector<std::vector<double>> const& image, TString name) const;

  /// Save the images for debugging
  /// This version takes the final clusters and overlays on the hit map
  void SaveImage(TH2F* image,
                 std::vector<art::PtrVector<recob::Hit>> const& allClusters,
                 int pad,
                 int tpc,
                 int plane);

  /// Save the images for debugging
  void SaveImage(TH2F* image, int pad, int tpc, int plane);

  /// Save the images for debugging
  /// This version takes a vector of bins and overlays the relevant bins on the hit map
  void SaveImage(TH2F* image,
                 std::vector<std::vector<int>> const& allClusterBins,
                 int pad,
                 int tpc,
                 int plane);

private:
  /// Converts a vector of bins into a hit selection - not all the hits in the bins vector are real hits
  art::PtrVector<recob::Hit> ConvertBinsToRecobHits(std::vector<std::vector<double>> const& image,
                                                    std::vector<int> const& bins) const;

  /// Converts a bin into a recob::Hit (not all of these bins correspond to recob::Hits - some are fake hits created by the blurring)
  art::Ptr<recob::Hit> ConvertBinToRecobHit(std::vector<std::vector<double>> const& image,
                                            int bin) const;

  /// Converts an xbin and a ybin to a global bin number
  int ConvertWireTickToBin(std::vector<std::vector<double>> const& image, int xbin, int ybin) const;

  /// Returns the charge stored in the global bin value
  double ConvertBinToCharge(std::vector<std::vector<double>> const& image, int bin) const;

  /// Count how many dead wires there are in the blurring region for a particular hit
  /// Returns a pair of counters representing how many dead wires there are below and above the hit respectively
  std::pair<int, int> DeadWireCount(int wire_bin, int width) const;

  /// Dynamically find the blurring radii and Gaussian sigma in each dimension
  std::array<int, 4> FindBlurringParameters() const;

  /// Returns the hit time of a hit in a particular bin
  double GetTimeOfBin(std::vector<std::vector<double>> const& image, int bin) const;

  /// Makes all the kernels which could be required given the tuned parameters
  std::vector<std::vector<std::vector<double>>> MakeKernels() const;

  /// Determines the number of clustered neighbours of a hit
  unsigned int NumNeighbours(int nx, std::vector<bool> const& used, int bin) const;

  /// Determine if a hit is within a time threshold of any other hits in a cluster
  bool PassesTimeCut(std::vector<double> const& times, double time) const;

  bool fDebug;
  std::string fDetector;

  // Parameters used in the Blurred Clustering algorithm
  int fBlurWire;            // blur radius for Gauss kernel in the wire direction
  int fBlurTick;            // blur radius for Gauss kernel in the tick direction
  double fSigmaWire;        // sigma for Gaussian kernel in the wire direction
  double fSigmaTick;        // sigma for Gaussian kernel in the tick direction
  int fMaxTickWidthBlur;    // maximum distance to blur a hit based on its natural width in time
  int fClusterWireDistance; // how far to cluster from seed in wire direction
  int fClusterTickDistance; // how far to cluster from seed in tick direction
  unsigned int fNeighboursThreshold; // min. number of neighbors to add to cluster
  unsigned int fMinNeighbours;       // minumum number of neighbors to keep in the cluster
  unsigned int fMinSize;             // minimum size for cluster
  double fMinSeed;         // minimum seed after blurring needed before clustering proceeds
  double fTimeThreshold;   // time threshold for clustering
  double fChargeThreshold; // charge threshold for clustering

  // Blurring stuff
  int fKernelWidth, fKernelHeight;
  std::vector<std::vector<std::vector<double>>> fAllKernels;

  // Hit containers
  std::vector<std::vector<art::Ptr<recob::Hit>>> fHitMap;
  std::vector<bool> fDeadWires;

  int fLowerTick, fUpperTick;
  int fLowerWire, fUpperWire;

  // For the debug pdf
  TCanvas* fDebugCanvas{nullptr};
  std::string fDebugPDFName{};

  // art service handles
  art::ServiceHandle<geo::Geometry const> fGeom;
  lariov::ChannelStatusProvider const& fChanStatus{
    art::ServiceHandle<lariov::ChannelStatusService const>()->GetProvider()};
};

#endif