TripletFinder.h

Go to the documentation of this file.

// Christopher Backhouse - bckhouse@fnal.gov

#ifndef RECO3D_TRIPLETFINDER_H
#define RECO3D_TRIPLETFINDER_H

#include "larcore/Geometry/Geometry.h"

#include "larcoreobj/SimpleTypesAndConstants/RawTypes.h" // raw::ChannelID_t

#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"

#include "lardataobj/RecoBase/Hit.h"

#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 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);

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

  protected:
    const geo::GeometryCore* geom;
    const detinfo::DetectorProperties* detprop;

    void FillHitMap(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;

    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