sbn::TwoPlaneStubMerge Class Reference

Art tool for merging stubs across planes. More...

Inheritance diagram for sbn::TwoPlaneStubMerge:

Public Member Functions
	TwoPlaneStubMerge (fhicl::ParameterSet const &pset)
	Constructor. More...
std::vector< sbn::StubInfo >	Merge (const std::vector< sbn::StubInfo > &stubs, const geo::GeometryCore *geo, const spacecharge::SpaceCharge *sce, const detinfo::DetectorClocksData &dclock, const detinfo::DetectorPropertiesData &dprop) override
sbn::StubInfo	MergeStubs (const sbn::StubInfo &A, const sbn::StubInfo &B, const geo::GeometryCore *geo, const spacecharge::SpaceCharge *sce, const detinfo::DetectorPropertiesData &dprop)
bool	SortStubs (const sbn::StubInfo &A, const sbn::StubInfo &B)
Public Member Functions inherited from sbn::IStubMerge
virtual	~IStubMerge () noexcept=default
	Virtual Destructor. More...

Private Attributes
double	fMaxMergeTOff
double	fMaxMergeQOff
bool	fRemoveDuplicateMerges
bool	fSaveOldStubs

Detailed Description

Art tool for merging stubs across planes.

Uses requirements on total charge and hit timing. Re-computes location information of successful merges.

Definition at line 38 of file TwoPlaneStubMerge_tool.cc.

Constructor & Destructor Documentation

sbn::TwoPlaneStubMerge::TwoPlaneStubMerge

(

fhicl::ParameterSet const &

pset

)

Constructor.

Definition at line 65 of file TwoPlaneStubMerge_tool.cc.

                                                                 :
  fMaxMergeTOff(pset.get<double>("MaxMergeTOff")),
  fMaxMergeQOff(pset.get<double>("MaxMergeQOff")),
  fRemoveDuplicateMerges(pset.get<bool>("RemoveDuplicateMerges")),
  fSaveOldStubs(pset.get<bool>("SaveOldStubs"))
{
}

Member Function Documentation

std::vector< sbn::StubInfo > sbn::TwoPlaneStubMerge::Merge ( const std::vector< sbn::StubInfo > &  stubs, const geo::GeometryCore *  geo, const spacecharge::SpaceCharge *  sce, const detinfo::DetectorClocksData &  dclock, const detinfo::DetectorPropertiesData &  dprop  )

overridevirtual

Implements sbn::IStubMerge.

Definition at line 138 of file TwoPlaneStubMerge_tool.cc.

                                                {

  // Internal struct
  struct MergeInfo {
    float toff;
    float qoff;
    unsigned i;
    unsigned j;
  };

  std::vector<MergeInfo> merges;

  // Compute all of the possible cross-plane matches
  for (unsigned i_stub = 0; i_stub < stubs.size(); i_stub++) {
    for (unsigned j_stub = 0; j_stub < stubs.size(); j_stub++) {
      const sbn::StubInfo &A = stubs[i_stub];     
      const sbn::StubInfo &B = stubs[j_stub];     
      if (A.stub.plane.size() != 1 || B.stub.plane.size() != 1) continue; // Only match 1-plane-stubs

      if (A.stub.plane.front().Plane == B.stub.plane.front().Plane) continue; // Match across planes
      if (A.stub.plane.front().asTPCID() != B.stub.plane.front().asTPCID()) continue; // But should be in the same TPC

      float toff = sbn::StubTimeOffset(A, B, dclock, dprop);
      float qoff = sbn::StubChargeOffset(A, B);

      merges.push_back({toff, qoff, i_stub, j_stub});
    }
  }

  // Greedily do the best matches
  //
  // Sort by the time-offset -- this one is more precise
  std::sort(merges.begin(), merges.end(), [](auto const &lhs, auto const &rhs) {return lhs.toff < rhs.toff;});

  // keep track of which index stubs have been merged
  std::set<unsigned> hasmerged;

  // keep track of which index stubs should not be saved
  std::set<unsigned> dontsave;

  // Output list
  std::vector<sbn::StubInfo> ret;

  for (unsigned i_mrg = 0; i_mrg < merges.size(); i_mrg++) {
    const MergeInfo &thismrg = merges[i_mrg];
    if (thismrg.toff < fMaxMergeTOff && thismrg.qoff < fMaxMergeQOff) {
      if (!hasmerged.count(thismrg.i) && !hasmerged.count(thismrg.j)) { 
        ret.push_back(MergeStubs(stubs[thismrg.i], stubs[thismrg.j], geo, sce, dprop));
        hasmerged.insert(thismrg.i);
        hasmerged.insert(thismrg.j);
      }
      else if (!hasmerged.count(thismrg.i) && fRemoveDuplicateMerges) {
        dontsave.insert(thismrg.i);
      }
      else if (!hasmerged.count(thismrg.j) && fRemoveDuplicateMerges) {
        dontsave.insert(thismrg.j);
      }
    }
  }

  // Save the stubs which were not merged across planes and not marked to be removed
  for (unsigned i_stub = 0; i_stub < stubs.size(); i_stub++) {
    if (fSaveOldStubs || (!hasmerged.count(i_stub) && !dontsave.count(i_stub))) {
      ret.push_back(stubs[i_stub]);
    }
  }

  return ret;
}

sbn::StubInfo sbn::TwoPlaneStubMerge::MergeStubs	(	const sbn::StubInfo &	A,
		const sbn::StubInfo &	B,
		const geo::GeometryCore *	geo,
		const spacecharge::SpaceCharge *	sce,
		const detinfo::DetectorPropertiesData &	dprop
	)

Definition at line 86 of file TwoPlaneStubMerge_tool.cc.

                                                {
  sbn::StubInfo ret; 

  // Metadata

  // Combine the hits
  ret.hits.insert(ret.hits.end(), A.hits.begin(), A.hits.end());
  ret.hits.insert(ret.hits.end(), B.hits.begin(), B.hits.end());

  // Figure out which is the "better" stub-plane -- probably the one with the lower pitch
  const sbn::StubInfo &best = SortStubs(A, B) ? A : B;
  const sbn::StubInfo &othr = SortStubs(A, B) ? B : A;

  ret.pfp = best.pfp;
  ret.vhit = best.vhit;
  ret.vhit_hit = best.vhit_hit;

  // The stub info
  // Vertex should be the same between the two
  ret.stub.vtx = A.stub.vtx;
  // The real thing -- combine the two positions to get a new endpoint
  ret.stub.end = sbn::TwoStubEndPosition(best, othr, geo, sce, dprop);

  // Save the EField at the start and end point
  ret.stub.efield_vtx = sbn::GetEfield(dprop, sce, ret.stub.vtx, ret.vhit_hit->WireID(), false);
  ret.stub.efield_end = sbn::GetEfield(dprop, sce, ret.stub.end, ret.vhit_hit->WireID(), false);

  // In all the plane info stuff put the best stub first
  ret.stub.plane.push_back(best.stub.plane.front());
  ret.stub.plane.push_back(othr.stub.plane.front());

  ret.stub.hits.push_back(best.stub.hits.front());
  ret.stub.hits.push_back(othr.stub.hits.front());

  ret.stub.pitch.push_back(best.stub.pitch.front());
  ret.stub.pitch.push_back(othr.stub.pitch.front());

  ret.stub.trkpitch.push_back(best.stub.trkpitch.front());
  ret.stub.trkpitch.push_back(othr.stub.trkpitch.front());

  ret.stub.vtx_w.push_back(best.stub.vtx_w.front());
  ret.stub.vtx_w.push_back(othr.stub.vtx_w.front());

  ret.stub.hit_w.push_back(best.stub.hit_w.front());
  ret.stub.hit_w.push_back(othr.stub.hit_w.front());

  return ret;
}

bool sbn::TwoPlaneStubMerge::SortStubs	(	const sbn::StubInfo &	A,
		const sbn::StubInfo &	B
	)

Definition at line 74 of file TwoPlaneStubMerge_tool.cc.

                                                                            {
  // Count the hits on the main stub
  int A_nhit = A.stub.CoreNHit();
  int B_nhit = B.stub.CoreNHit();

  // First -- try more hits
  if (A_nhit != B_nhit) return A_nhit > B_nhit;

  // Sort by desirability of plane: 2 better than 0 better than 1
  return ((A.vhit_hit->WireID().Plane + 1) % 3) < ((B.vhit_hit->WireID().Plane + 1) % 3);
}

Member Data Documentation

double sbn::TwoPlaneStubMerge::fMaxMergeQOff

private

Definition at line 60 of file TwoPlaneStubMerge_tool.cc.

double sbn::TwoPlaneStubMerge::fMaxMergeTOff

private

Definition at line 59 of file TwoPlaneStubMerge_tool.cc.

bool sbn::TwoPlaneStubMerge::fRemoveDuplicateMerges

private

Definition at line 61 of file TwoPlaneStubMerge_tool.cc.

bool sbn::TwoPlaneStubMerge::fSaveOldStubs

private

Definition at line 62 of file TwoPlaneStubMerge_tool.cc.

---

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

• TwoPlaneStubMerge_tool.cc