TripletFinder.h

Go to the documentation of this file.

// Christopher Backhouse - bckhouse@fnal.gov

#ifndef RECO3D_TRIPLETFINDER_H
#define RECO3D_TRIPLETFINDER_H

#include "canvas/Persistency/Common/Ptr.h"

#include "larcorealg/Geometry/fwd.h"
#include "larcoreobj/SimpleTypesAndConstants/RawTypes.h" // raw::ChannelID_t
#include "larcoreobj/SimpleTypesAndConstants/geo_types.h"
#include "lardataobj/RecoBase/Hit.h"
namespace detinfo {
  class DetectorPropertiesData;
}

#include <map>
#include <vector>

namespace recob {
  class Hit;
}

namespace reco3d {
  struct HitOrChan {
    HitOrChan(raw::ChannelID_t c) : chan(c), hit(0), xpos(0) {}
    HitOrChan(const recob::Hit* h, double x) : chan(h->Channel()), hit(h), xpos(x) {}

    raw::ChannelID_t chan;
    const recob::Hit* hit; // null for bad channel
    double xpos;           // Only set if hit is set
  };

  struct ChannelDoublet {
    ChannelDoublet(HitOrChan a_, HitOrChan b_, geo::WireIDIntersection p) : a(a_), b(b_), pt(p) {}

    HitOrChan a, b;
    geo::WireIDIntersection pt;
  };

  struct XYZ {
    double x, y, z;
  };

  struct HitTriplet {
    const recob::Hit *x, *u, *v;
    XYZ pt;
  };

  class TripletFinder {
  public:
    TripletFinder(const detinfo::DetectorPropertiesData& detProp,
                  const std::vector<art::Ptr<recob::Hit>>& xhits,
                  const std::vector<art::Ptr<recob::Hit>>& uhits,
                  const std::vector<art::Ptr<recob::Hit>>& vhits,
                  const std::vector<raw::ChannelID_t>& xbad,
                  const std::vector<raw::ChannelID_t>& ubad,
                  const std::vector<raw::ChannelID_t>& vbad,
                  double distThresh,
                  double distThreshDrift,
                  double xhitOffset,
                  int maxTriplets = 0);

    std::vector<HitTriplet> Triplets();
    std::vector<HitTriplet> TripletsTwoView();

  protected:
    const geo::GeometryCore* geom;
    const geo::WireReadoutGeom* wireReadoutGeom;

    void FillHitMap(const detinfo::DetectorPropertiesData& clockData,
                    const std::vector<art::Ptr<recob::Hit>>& hits,
                    std::map<geo::TPCID, std::vector<HitOrChan>>& out);
    void FillBadMap(const std::vector<raw::ChannelID_t>& bads,
                    std::map<geo::TPCID, std::vector<raw::ChannelID_t>>& out);

    bool CloseDrift(double xa, double xb) const;
    bool CloseSpace(geo::WireIDIntersection ra, geo::WireIDIntersection rb) const;

    std::vector<ChannelDoublet> DoubletsXU(geo::TPCID tpc);
    std::vector<ChannelDoublet> DoubletsXV(geo::TPCID tpc);

    std::vector<ChannelDoublet> DoubletHelper(geo::TPCID tpc,
                                              const std::vector<HitOrChan>& ahits,
                                              const std::vector<HitOrChan>& bhits,
                                              const std::vector<raw::ChannelID_t>& bbads) const;

    double fDistThresh;
    double fDistThreshDrift;
    double fXHitOffset;
    int fMaxTriplets;

    std::map<geo::TPCID, std::vector<HitOrChan>> fX_by_tpc;
    std::map<geo::TPCID, std::vector<HitOrChan>> fU_by_tpc;
    std::map<geo::TPCID, std::vector<HitOrChan>> fV_by_tpc;

    // TODO initialize
    std::map<geo::TPCID, std::vector<raw::ChannelID_t>> fXbad_by_tpc;
    std::map<geo::TPCID, std::vector<raw::ChannelID_t>> fUbad_by_tpc;
    std::map<geo::TPCID, std::vector<raw::ChannelID_t>> fVbad_by_tpc;
  };
}

#endif