StubBuilder.cxx File Reference

#include "StubBuilder.h"
#include "sbncode/TPCReco/VertexStub/StubMergeAlgorithms.h"

Go to the source code of this file.

Functions
bool	HitOnTrack (const art::Ptr< recob::Hit > &hit, const art::Ptr< recob::Track > &trk, const std::vector< art::Ptr< recob::Hit >> &trk_hits, const std::vector< const recob::TrackHitMeta * > &trk_thms)
int	FindPFPInd (const std::vector< art::Ptr< recob::Hit >> &hits, const std::vector< art::Ptr< recob::PFParticle >> &pfps, const std::vector< std::vector< art::Ptr< recob::Hit >>> &pfp_hits)
std::vector< art::Ptr < recob::Hit > >	CollectHits (const std::vector< art::Ptr< recob::Hit >> &hits, const sbn::VertexHit &vhit, const recob::Hit &vhit_hit, const geo::GeometryCore *geo, const detinfo::DetectorPropertiesData &dprop)

Function Documentation

std::vector<art::Ptr<recob::Hit> > CollectHits	(	const std::vector< art::Ptr< recob::Hit >> &	hits,
		const sbn::VertexHit &	vhit,
		const recob::Hit &	vhit_hit,
		const geo::GeometryCore *	geo,
		const detinfo::DetectorPropertiesData &	dprop
	)

Definition at line 56 of file StubBuilder.cxx.

                                                {

  // project the vertex onto the wireplane
  float vert_wf = vhit.vtxw;
  float vert_x = vhit.vtxx;

  // get the vertex hit-coordinates
  int hit_w = vhit_hit.WireID().Wire;
  float hit_x = dprop.ConvertTicksToX(vhit_hit.PeakTime(), vhit_hit.WireID());
  geo::PlaneID vhit_plane = vhit_hit.WireID();

  // get the line-segment slope / intercept
  float slope = (hit_x - vert_x) / (hit_w - vert_wf);
  float intercept = hit_x - slope * hit_w;

  // check for "bad" (infinite) slope
  //
  // In this case, only hits overlapping with the vertex hit should be
  // included in the returned list
  bool badslope = abs(hit_w - vert_wf) < 1e-5;
  if (badslope) {
    std::vector<art::Ptr<recob::Hit>> ret;
    for (art::Ptr<recob::Hit> h: hits) {
      int this_hit_w = h->WireID().Wire;
      geo::PlaneID this_hit_plane = h->WireID();
      if (this_hit_plane == vhit_plane &&  // check plane
          this_hit_w == hit_w /* check wire */) {
        float h_time_lo = h->PeakTime() - h->RMS();
        float h_time_hi = h->PeakTime() + h->RMS();

        if (h_time_lo < vhit_hit.PeakTime() && h_time_hi > vhit_hit.PeakTime()) {
          ret.push_back(h);
        }
      }
    }
    return ret;
  }

  // Include wires one after the hit and one before the vtx
  int vert_w = (int)((hit_w > vert_wf) ? std::floor(vert_wf) : std::ceil(vert_wf));
  hit_w = hit_w + ((hit_w > vert_wf) ? 1 : -1);

  // get all the hits that overlap between these two points
  std::vector<art::Ptr<recob::Hit>> ret;
  for (art::Ptr<recob::Hit> h: hits) {
    int this_hit_w = h->WireID().Wire;
    geo::PlaneID this_hit_plane = h->WireID();
    if (this_hit_plane == vhit_plane &&  // check plane
       ((this_hit_w <= hit_w && this_hit_w >= vert_w) || // check overlap
        (this_hit_w >= hit_w && this_hit_w <= vert_w))) {
      float this_proj_x = slope * this_hit_w + intercept;

      float h_time_lo = h->PeakTime() - h->RMS();
      float h_x_A = dprop.ConvertTicksToX(h_time_lo, h->WireID());
      float h_time_hi = h->PeakTime() + h->RMS();
      float h_x_B = dprop.ConvertTicksToX(h_time_hi, h->WireID());

      if ((h_x_A >= this_proj_x && h_x_B <= this_proj_x) ||
          (h_x_A <= this_proj_x && h_x_B >= this_proj_x)) {
        ret.push_back(h);
      }
    }
  }
  
  return ret;
}

int FindPFPInd	(	const std::vector< art::Ptr< recob::Hit >> &	hits,
		const std::vector< art::Ptr< recob::PFParticle >> &	pfps,
		const std::vector< std::vector< art::Ptr< recob::Hit >>> &	pfp_hits
	)

Definition at line 25 of file StubBuilder.cxx.

                                                          {

  std::vector<unsigned> pfp_nhit(pfps.size(), 0);

  for (const art::Ptr<recob::Hit> h: hits) {
    for (unsigned i_pfp = 0; i_pfp < pfps.size(); i_pfp++) {
      const std::vector<art::Ptr<recob::Hit>> &thispfpHits = pfp_hits[i_pfp];
      bool found = false;
      for (const art::Ptr<recob::Hit> &pfp_h: thispfpHits) {
        if (h == pfp_h) {
          found = true;
          pfp_nhit[i_pfp] ++;
          break;
        }
      }
      if (found) break;
    }
  }

  // Look to see if more than half of the hits on the stub match to a single
  // PFParticle
  for (unsigned i_pfp = 0; i_pfp < pfps.size(); i_pfp++) {
    if (pfp_nhit[i_pfp] >= hits.size() / 2) return i_pfp;
  }

  return -1;
}

bool HitOnTrack	(	const art::Ptr< recob::Hit > &	hit,
		const art::Ptr< recob::Track > &	trk,
		const std::vector< art::Ptr< recob::Hit >> &	trk_hits,
		const std::vector< const recob::TrackHitMeta * > &	trk_thms
	)

Definition at line 5 of file StubBuilder.cxx.

                                                          {

  if (!trk) return false; // no track: hit can't be on a track
    
  for (unsigned i = 0; i < trk_hits.size(); i++) {
    // Found the hit on the track!
    if (hit == trk_hits[i]) { 
      // Is the hit part of the track Calo?
      if (trk->HasValidPoint(trk_thms[i]->Index())) { 
        return true;
      }
      break;
    }
  }
  return false;
}