#include <TripletFinder.h>

Public Member Functions
	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)

std::vector< HitTriplet >	Triplets ()

std::vector< HitTriplet >	TripletsTwoView ()
	Only search for XU intersections. More...

Protected Member Functions
void	FillHitMap (const detinfo::DetectorPropertiesData &clockData, const std::vector< art::Ptr< recob::Hit >> &hits, std::map< geo::TPCID, std::vector< HitOrChan >> &out)
	Helper for constructor. More...

void	FillBadMap (const std::vector< raw::ChannelID_t > &bads, std::map< geo::TPCID, std::vector< raw::ChannelID_t >> &out)
	Helper for constructor. More...

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

Protected Attributes
const geo::GeometryCore *	geom

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

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

Detailed Description

Definition at line 51 of file TripletFinder.h.

Constructor & Destructor Documentation

reco3d::TripletFinder::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
	)

Definition at line 13 of file TripletFinder.cxx.

     : geom(art::ServiceHandle<geo::Geometry const>()->provider())
     , fDistThresh(distThresh)
     , fDistThreshDrift(distThreshDrift)
     , fXHitOffset(xhitOffset)
   {
     FillHitMap(detProp, xhits, fX_by_tpc);
     FillHitMap(detProp, uhits, fU_by_tpc);
     FillHitMap(detProp, vhits, fV_by_tpc);
 
     FillBadMap(xbad, fXbad_by_tpc);
     FillBadMap(ubad, fUbad_by_tpc);
     FillBadMap(vbad, fVbad_by_tpc);
   }

Member Function Documentation

bool reco3d::TripletFinder::CloseDrift	(	double	xa,
		double	xb
	)		const

protected

Definition at line 122 of file TripletFinder.cxx.

   {
     return fabs(xa - xb) < fDistThreshDrift;
   }

bool reco3d::TripletFinder::CloseSpace	(	geo::WireIDIntersection	ra,
		geo::WireIDIntersection	rb
	)		const

protected

Definition at line 129 of file TripletFinder.cxx.

   {
     const TVector3 pa(ra.y, ra.z, 0);
     const TVector3 pb(rb.y, rb.z, 0);
 
     return (pa - pb).Mag() < fDistThresh;
   }

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

protected

Definition at line 302 of file TripletFinder.cxx.

   {
     std::vector<ChannelDoublet> ret;
 
     IntersectionCache isect(tpc);
 
     auto b_begin = bhits.begin();
 
     for (const HitOrChan& a : ahits) {
       // Bad channels are easy because there's no timing constraint
       for (raw::ChannelID_t b : bbads) {
         geo::WireIDIntersection pt;
         if (isect(a.chan, b, pt)) { ret.emplace_back(a, b, pt); }
       }
 
       while (b_begin != bhits.end() && b_begin->xpos < a.xpos && !CloseDrift(b_begin->xpos, a.xpos))
         ++b_begin;
 
       for (auto bit = b_begin; bit != bhits.end(); ++bit) {
         const HitOrChan& b = *bit;
 
         if (b.xpos > a.xpos && !CloseDrift(b.xpos, a.xpos)) break;
 
         geo::WireIDIntersection pt;
         if (!isect(a.chan, b.chan, pt)) continue;
 
         ret.emplace_back(a, b, pt);
       } // end for b
     }   // end for a
 
     return ret;
   }

std::vector< ChannelDoublet > reco3d::TripletFinder::DoubletsXU ( geo::TPCID tpc )

protected

Definition at line 272 of file TripletFinder.cxx.

   {
     std::vector<ChannelDoublet> ret =
       DoubletHelper(tpc, fX_by_tpc[tpc], fU_by_tpc[tpc], fUbad_by_tpc[tpc]);
 
     // Find X(bad)+U(good) doublets, have to flip them for the final result
     for (auto it : DoubletHelper(tpc, fU_by_tpc[tpc], {}, fXbad_by_tpc[tpc])) {
       ret.push_back({it.b, it.a, it.pt});
     }
 
     return ret;
   }

std::vector< ChannelDoublet > reco3d::TripletFinder::DoubletsXV ( geo::TPCID tpc )

protected

Definition at line 287 of file TripletFinder.cxx.

   {
     std::vector<ChannelDoublet> ret =
       DoubletHelper(tpc, fX_by_tpc[tpc], fV_by_tpc[tpc], fVbad_by_tpc[tpc]);
 
     // Find X(bad)+V(good) doublets, have to flip them for the final result
     for (auto it : DoubletHelper(tpc, fV_by_tpc[tpc], {}, fXbad_by_tpc[tpc])) {
       ret.push_back({it.b, it.a, it.pt});
     }
 
     return ret;
   }

void reco3d::TripletFinder::FillBadMap	(	const std::vector< raw::ChannelID_t > &	bads,
		std::map< geo::TPCID, std::vector< raw::ChannelID_t >> &	out
	)

protected

Helper for constructor.

Definition at line 62 of file TripletFinder.cxx.

   {
     for (raw::ChannelID_t chan : bads) {
       for (geo::TPCID tpc : geom->ROPtoTPCs(geom->ChannelToROP(chan))) {
         out[tpc].push_back(chan);
       }
     }
   }

void reco3d::TripletFinder::FillHitMap	(	const detinfo::DetectorPropertiesData &	clockData,
		const std::vector< art::Ptr< recob::Hit >> &	hits,
		std::map< geo::TPCID, std::vector< HitOrChan >> &	out
	)

protected

Helper for constructor.

Definition at line 39 of file TripletFinder.cxx.

   {
     for (const art::Ptr<recob::Hit>& hit : hits) {
       for (geo::TPCID tpc : geom->ROPtoTPCs(geom->ChannelToROP(hit->Channel()))) {
         double xpos = 0;
         for (geo::WireID wire : geom->ChannelToWire(hit->Channel())) {
           if (geo::TPCID(wire) == tpc) {
             xpos = detProp.ConvertTicksToX(hit->PeakTime(), wire);
             if (geom->SignalType(wire) == geo::kCollection) xpos += fXHitOffset;
           }
         }
 
         out[tpc].emplace_back(hit.get(), xpos);
       }
     }
     for (auto& it : out)
       std::sort(it.second.begin(), it.second.end(), [](auto a, auto b) { return a.xpos < b.xpos; });
   }

std::vector< HitTriplet > reco3d::TripletFinder::Triplets ( )

Definition at line 155 of file TripletFinder.cxx.

   {
     std::vector<HitTriplet> ret;
 
     for (const auto& it : fX_by_tpc) {
       const geo::TPCID& tpc = it.first;
 
       std::vector<ChannelDoublet> xus = DoubletsXU(tpc);
       std::vector<ChannelDoublet> xvs = DoubletsXV(tpc);
 
       // Cache to prevent repeating the same questions
       IntersectionCache isectUV(tpc);
 
       // For the efficient looping below to work we need to sort the doublet
       // lists so the X hits occur in the same order.
       std::sort(xus.begin(), xus.end(), LessThanXHit);
       std::sort(xvs.begin(), xvs.end(), LessThanXHit);
 
       auto xvit_begin = xvs.begin();
 
       int nxuv = 0;
       for (const ChannelDoublet& xu : xus) {
         const HitOrChan& x = xu.a;
         const HitOrChan& u = xu.b;
 
         // Catch up until we're looking at the same X hit in XV
         while (xvit_begin != xvs.end() && LessThanXHit(*xvit_begin, xu))
           ++xvit_begin;
 
         // Loop through all those matching hits
         for (auto xvit = xvit_begin; xvit != xvs.end() && SameXHit(*xvit, xu); ++xvit) {
           const HitOrChan& v = xvit->b;
 
           // Only allow one bad channel per triplet
           if (!x.hit && !u.hit) continue;
           if (!x.hit && !v.hit) continue;
           if (!u.hit && !v.hit) continue;
 
           if (u.hit && v.hit && !CloseDrift(u.xpos, v.xpos)) continue;
 
           geo::WireIDIntersection ptUV;
           if (!isectUV(u.chan, v.chan, ptUV)) continue;
 
           if (!CloseSpace(xu.pt, xvit->pt) || !CloseSpace(xu.pt, ptUV) ||
               !CloseSpace(xvit->pt, ptUV))
             continue;
 
           double xavg = 0;
           int nx = 0;
           if (x.hit) {
             xavg += x.xpos;
             ++nx;
           }
           if (u.hit) {
             xavg += u.xpos;
             ++nx;
           }
           if (v.hit) {
             xavg += v.xpos;
             ++nx;
           }
           xavg /= nx;
 
           const XYZ pt{
             xavg, (xu.pt.y + xvit->pt.y + ptUV.y) / 3, (xu.pt.z + xvit->pt.z + ptUV.z) / 3};
 
           ret.emplace_back(HitTriplet{x.hit, u.hit, v.hit, pt});
           ++nxuv;
         } // end for xv
       }   // end for xu
 
       std::cout << tpc << " " << xus.size() << " XUs and " << xvs.size() << " XVs -> " << nxuv
                 << " XUVs" << std::endl;
 
     } // end for tpc
 
     std::cout << ret.size() << " XUVs total" << std::endl;
 
     return ret;
   }

std::vector< HitTriplet > reco3d::TripletFinder::TripletsTwoView ( )

Only search for XU intersections.

Definition at line 238 of file TripletFinder.cxx.

   {
     std::vector<HitTriplet> ret;
 
     for (const auto& it : fX_by_tpc) {
       const geo::TPCID& tpc = it.first;
 
       std::vector<ChannelDoublet> xus = DoubletsXU(tpc);
 
       for (const ChannelDoublet& xu : xus) {
         const HitOrChan& x = xu.a;
         const HitOrChan& u = xu.b;
 
         double xavg = x.xpos;
         int nx = 1;
         if (u.hit) {
           xavg += u.xpos;
           ++nx;
         }
         xavg /= nx;
 
         const XYZ pt{xavg, xu.pt.y, xu.pt.z};
 
         ret.emplace_back(HitTriplet{x.hit, u.hit, 0, pt});
       } // end for xu
     }   // end for tpc
 
     std::cout << ret.size() << " XUs total" << std::endl;
 
     return ret;
   }

Member Data Documentation

double reco3d::TripletFinder::fDistThresh

protected

Definition at line 90 of file TripletFinder.h.

double reco3d::TripletFinder::fDistThreshDrift

protected

Definition at line 91 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<HitOrChan> > reco3d::TripletFinder::fU_by_tpc

protected

Definition at line 95 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<raw::ChannelID_t> > reco3d::TripletFinder::fUbad_by_tpc

protected

Definition at line 100 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<HitOrChan> > reco3d::TripletFinder::fV_by_tpc

protected

Definition at line 96 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<raw::ChannelID_t> > reco3d::TripletFinder::fVbad_by_tpc

protected

Definition at line 101 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<HitOrChan> > reco3d::TripletFinder::fX_by_tpc

protected

Definition at line 94 of file TripletFinder.h.

std::map<geo::TPCID, std::vector<raw::ChannelID_t> > reco3d::TripletFinder::fXbad_by_tpc

protected

Definition at line 99 of file TripletFinder.h.

double reco3d::TripletFinder::fXHitOffset

protected

Definition at line 92 of file TripletFinder.h.

const geo::GeometryCore* reco3d::TripletFinder::geom

protected

Definition at line 69 of file TripletFinder.h.

The documentation for this class was generated from the following files:

Public Member Functions

Protected Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation