ToFPositionFinder.h

Go to the documentation of this file.

#pragma once

#include <vector>
#include <iostream>
#include <cmath>
#include <map>

#include "TH2D.h"

int HIST_N_BINS = 100;

namespace beamlinereco {
  //useful
  struct posn_t { double x; double y; };
  struct testset_t { double tA; double tB; double tC; double tD; double s; posn_t expected; };
  
  //for floating point comparisons
  const double EPSILON = 1e-6;

  class ToFPositionFinder {
  public:
    ToFPositionFinder();
    virtual ~ToFPositionFinder() = default;
    
    // Find location in x and y (in ns) given 3 hit times
    //  A = (0,s), B = (0,0), C = (s,0)
    posn_t Get3CornerPosition(double tA, double tB, double tC, double s);
    
    // Find location in x and y (in ns) given 4 hit times (by taking average)
    //  A = (0,s), B = (0,0), C = (s,0), D = (s,s)
    posn_t Get4CornerPosition(double tA, double tB, double tC, double tD, double s);
    
    // Find the spread of the four reconstructed positions
    // Calculates the sum of the squares of the distance
    //  from each reconstructed point to the center of mass for the 4 
    //  reconstructed points.
    double Get4CornerSpread(double tA, double tB, double tC, double tD, double s);

    // Produce a 2D histogram with vals of 1 for corners, val of 4 for mean.
    TH2D GetSummaryHist(TString name, TString title, double tA, double tB, double tC, double tD, double s);
    
    // Test whether the given posn is physical (inside the counter) or not.
    bool IsPhysical(posn_t p, double s);
    
    // Run some tests to make sure methods are doing what we think they are
    bool RunTests();

    // Find the distance to the nearest corner
    double GetDistanceToNearestCorner(posn_t p, double s);
    
  private:
  private:
    // Posn utilities
    posn_t PosnAvg(std::vector<posn_t> posns);
    double PosnDist(posn_t a, posn_t b);
    double GetSpread(std::vector<posn_t> posns);
    posn_t Transform90posn(posn_t p, double s);
    posn_t RotateCoords(posn_t p, double s, int n);
    double GetClosestDist(posn_t p, std::vector<posn_t> others);

    // Helpers for 4-corner method
    std::vector<posn_t> Get4Posns(double tA, double tB, double tC, double tD, double s);

    // Helpers for RunTests
    bool Run3CornerTest(testset_t ts);
    bool Run3CornerTests();
    bool Run4CornerTest(testset_t ts);
    bool Run4CornerTests();
  };
}