trkf::FeatureTracker Class Reference

Inheritance diagram for trkf::FeatureTracker:

Public Member Functions
	FeatureTracker (fhicl::ParameterSet const &pset)

Private Member Functions
void	produce (art::Event &evt) override
void	GetProjectedEnds (detinfo::DetectorPropertiesData const &detProp, TVector3 xyz, std::vector< double > &uvw, std::vector< double > &t, int tpc=0, int cryo=0)
std::map< int, std::vector < double > >	ExtractEndPointTimes (std::vector< recob::EndPoint2D > EndPoints)
std::vector< recob::SpacePoint >	Get3DFeaturePoints (detinfo::DetectorClocksData const &clockData, detinfo::DetectorPropertiesData const &detProp, std::vector< recob::EndPoint2D > const &EndPoints, art::PtrVector< recob::Hit > const &Hits)
std::vector< recob::Seed >	GetValidLines (detinfo::DetectorPropertiesData const &detProp, std::vector< recob::SpacePoint > &sps, bool ApplyFilter=true)
void	RemoveDuplicatePaths (std::vector< recob::Seed > &Seeds, std::vector< int > &ConnectionPoint1, std::vector< int > &ConnectionPoint2)

Private Attributes
trkf::SpacePointAlg	fSP
corner::CornerFinderAlg	fCorner
std::string	fHitModuleLabel
std::string	fCalDataModuleLabel
double	fLineIntFraction
double	fLineIntThreshold
std::map< int, std::vector < double > >	fEndPointTimes
art::ServiceHandle < geo::Geometry const >	fGeometryHandle

Detailed Description

Definition at line 34 of file FeatureTracker_module.cc.

Constructor & Destructor Documentation

trkf::FeatureTracker::FeatureTracker ( fhicl::ParameterSet const &  pset )

explicit

Definition at line 78 of file FeatureTracker_module.cc.

    : EDProducer{pset}
    , fSP(pset.get<fhicl::ParameterSet>("SpacepointPset"))
    , fCorner(pset.get<fhicl::ParameterSet>("CornerPset"))
    , fHitModuleLabel{pset.get<std::string>("HitModuleLabel")}
    , fCalDataModuleLabel{pset.get<std::string>("CornerPset.CalDataModuleLabel")}
    , fLineIntFraction{pset.get<double>("LineIntFraction")}
    , fLineIntThreshold{pset.get<double>("LineIntThreshold")}
  {
    produces<std::vector<recob::Seed>>();
  }

Member Function Documentation

std::map< int, std::vector< double > > trkf::FeatureTracker::ExtractEndPointTimes ( std::vector< recob::EndPoint2D >  EndPoints )

private

Definition at line 134 of file FeatureTracker_module.cc.

  {
    std::map<int, std::vector<double>> EndPointTimesInPlane;
    for (size_t i = 0; i != EndPoints.size(); ++i) {
      EndPointTimesInPlane[EndPoints.at(i).View()].push_back(EndPoints.at(i).DriftTime());
    }

    for (std::map<int, std::vector<double>>::iterator itEpTime = EndPointTimesInPlane.begin();
         itEpTime != EndPointTimesInPlane.end();
         ++itEpTime) {
      std::sort(itEpTime->second.begin(), itEpTime->second.end());
    }
    return EndPointTimesInPlane;
  }

std::vector< recob::SpacePoint > trkf::FeatureTracker::Get3DFeaturePoints ( detinfo::DetectorClocksData const &  clockData, detinfo::DetectorPropertiesData const &  detProp, std::vector< recob::EndPoint2D > const &  EndPoints, art::PtrVector< recob::Hit > const &  Hits  )

private

Definition at line 312 of file FeatureTracker_module.cc.

  {

    art::PtrVector<recob::Hit> HitsForEndPoints;

    // Loop through the hits looking for the ones which match corners
    for (std::vector<recob::EndPoint2D>::const_iterator itEP = EndPoints.begin();
         itEP != EndPoints.end();
         ++itEP) {
      int EPMatchCount = 0;

      for (art::PtrVector<recob::Hit>::const_iterator itHit = Hits.begin(); itHit != Hits.end();
           ++itHit) {

        if ((itEP->View() == (*itHit)->View()) &&
            (itEP->WireID().Wire == (*itHit)->WireID().Wire)) {
          HitsForEndPoints.push_back(*itHit);
          EPMatchCount++;
        }
      }
    }
    std::vector<recob::SpacePoint> spts;
    fSP.makeSpacePoints(clockData, detProp, HitsForEndPoints, spts);

    for (size_t i = 0; i != spts.size(); ++i) {
      for (size_t p = 0; p != 3; ++p) {
        double Closest = 100000;
        double spt_t = detProp.ConvertXToTicks(spts.at(i).XYZ()[0], p, 0, 0);
        for (size_t epTime = 0; epTime != fEndPointTimes[p].size(); ++epTime) {
          if (fabs(fEndPointTimes[p].at(epTime) - spt_t) < Closest) {
            Closest = fabs(fEndPointTimes[p].at(epTime) - spt_t);
          }
        }
      }
    }
    return spts;
  }

void trkf::FeatureTracker::GetProjectedEnds ( detinfo::DetectorPropertiesData const &  detProp, TVector3  xyz, std::vector< double > &  uvw, std::vector< double > &  t, int  tpc = 0, int  cryo = 0  )

private

Definition at line 289 of file FeatureTracker_module.cc.

  {
    art::ServiceHandle<geo::Geometry const> geo;

    int NPlanes = geo->Cryostat(cryo).TPC(tpc).Nplanes();

    uvw.resize(NPlanes);
    t.resize(NPlanes);

    for (int plane = 0; plane != NPlanes; plane++) {
      uvw[plane] = geo->NearestWire(xyz, plane, tpc, cryo);
      t[plane] = detProp.ConvertXToTicks(xyz[0], plane, tpc, cryo);
    }
  }

std::vector< recob::Seed > trkf::FeatureTracker::GetValidLines ( detinfo::DetectorPropertiesData const &  detProp, std::vector< recob::SpacePoint > &  sps, bool  ApplyFilter = true  )

private

Definition at line 152 of file FeatureTracker_module.cc.

  {
    std::vector<recob::Seed> SeedsToReturn;

    std::vector<int> ConnectionPoint1;
    std::vector<int> ConnectionPoint2;
    std::map<int, std::vector<int>> SeedConnections;

    for (size_t i = 0; i != spts.size(); ++i) {
      for (size_t j = 0; j != i; ++j) {

        TVector3 xyz_i;
        TVector3 xyz_j;

        std::vector<double> t_i, t_j;

        std::vector<double> uvw_i;
        std::vector<double> uvw_j;

        for (size_t d = 0; d != 3; ++d) {
          xyz_i[d] = spts.at(i).XYZ()[d];
          xyz_j[d] = spts.at(j).XYZ()[d];
        }

        GetProjectedEnds(detProp, xyz_i, uvw_i, t_i, 0, 0);
        GetProjectedEnds(detProp, xyz_j, uvw_j, t_j, 0, 0);

        bool ThisLineGood = true;

        for (size_t p = 0; p != uvw_i.size(); ++p) {
          TH2F const& RawHist = fCorner.GetWireDataHist(p);

          double lineint = fCorner.line_integral(
            RawHist, uvw_i.at(p), t_i.at(p), uvw_j.at(p), t_j.at(p), fLineIntThreshold);

          if (lineint < fLineIntFraction) { ThisLineGood = false; }
        }
        if (ThisLineGood) {
          double Err[3];
          double Pos[3];
          double Dir[3];

          for (size_t d = 0; d != 3; ++d) {
            Pos[d] = 0.5 * (xyz_i[d] + xyz_j[d]);
            Dir[d] = 0.5 * (xyz_i[d] - xyz_j[d]);
            Err[d] = 0;
          }

          ConnectionPoint1.push_back(i);
          ConnectionPoint2.push_back(j);

          SeedsToReturn.push_back(recob::Seed(Pos, Dir, Err, Err));
        }
      }
    }

    if (ApplyFilter) {
      RemoveDuplicatePaths(SeedsToReturn, ConnectionPoint1, ConnectionPoint2);
      mf::LogInfo("FeatureTracker")
        << "Seeds after filter " << SeedsToReturn.size() << " seeds" << std::endl;
    }

    return SeedsToReturn;
  }

void trkf::FeatureTracker::produce ( art::Event &  evt )

overrideprivate

Definition at line 91 of file FeatureTracker_module.cc.

  {

    // Extract hits PtrVector from event
    art::Handle<std::vector<recob::Hit>> hith;
    evt.getByLabel(fHitModuleLabel, hith);

    art::PtrVector<recob::Hit> hitvec;
    for (unsigned int i = 0; i < hith->size(); ++i) {
      art::Ptr<recob::Hit> prod(hith, i);
      hitvec.push_back(prod);
    }

    //We need to grab out the wires.
    art::Handle<std::vector<recob::Wire>> wireHandle;
    evt.getByLabel(fCalDataModuleLabel, wireHandle);
    std::vector<recob::Wire> const& wireVec(*wireHandle);

    //First, have it process the wires.
    fCorner.GrabWires(wireVec, *fGeometryHandle);

    std::vector<recob::EndPoint2D> EndPoints;
    fCorner.get_feature_points(EndPoints, *fGeometryHandle);

    fEndPointTimes = ExtractEndPointTimes(EndPoints);

    auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
    auto const detProp =
      art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
    std::vector<recob::SpacePoint> sps = Get3DFeaturePoints(clockData, detProp, EndPoints, hitvec);
    std::vector<recob::Seed> SeedsToStore = GetValidLines(detProp, sps, true);

    std::unique_ptr<std::vector<recob::Seed>> seeds(new std::vector<recob::Seed>);

    for (size_t i = 0; i != SeedsToStore.size(); ++i)
      seeds->push_back(SeedsToStore.at(i));

    evt.put(std::move(seeds));
  }

void trkf::FeatureTracker::RemoveDuplicatePaths ( std::vector< recob::Seed > &  Seeds, std::vector< int > &  ConnectionPoint1, std::vector< int > &  ConnectionPoint2  )

private

Definition at line 222 of file FeatureTracker_module.cc.

  {

    std::map<int, bool> MarkedForRemoval;

    std::map<int, std::vector<int>> SeedsSharingPoint;
    for (size_t i = 0; i != Seeds.size(); ++i) {
      SeedsSharingPoint[ConnectionPoint1[i]].push_back(i);
      SeedsSharingPoint[ConnectionPoint2[i]].push_back(i);
    }

    for (size_t s = 0; s != Seeds.size(); ++s) {

      int StartPt = ConnectionPoint1.at(s);
      int EndPt = ConnectionPoint2.at(s);
      int MidPt = -1;

      for (size_t SeedsWithThisStart = 0; SeedsWithThisStart != SeedsSharingPoint[StartPt].size();
           SeedsWithThisStart++) {
        int i = SeedsSharingPoint[StartPt].at(SeedsWithThisStart);
        if (ConnectionPoint1.at(i) == StartPt)
          MidPt = ConnectionPoint2.at(i);
        else if (ConnectionPoint2.at(i) == StartPt)
          MidPt = ConnectionPoint1.at(i);

        for (size_t SeedsWithThisMid = 0; SeedsWithThisMid != SeedsSharingPoint[MidPt].size();
             SeedsWithThisMid++) {
          int j = SeedsSharingPoint[MidPt].at(SeedsWithThisMid);
          if ((ConnectionPoint1.at(j) == EndPt) || (ConnectionPoint2.at(j) == EndPt)) {

            double Lengthi = Seeds.at(i).GetLength();
            double Lengthj = Seeds.at(j).GetLength();
            double Lengths = Seeds.at(s).GetLength();

            if ((Lengths > Lengthi) && (Lengths > Lengthj)) {
              MarkedForRemoval[i] = true;
              MarkedForRemoval[j] = true;
            }

            if ((Lengthi > Lengths) && (Lengthi > Lengthj)) {
              MarkedForRemoval[s] = true;
              MarkedForRemoval[j] = true;
            }
            if ((Lengthj > Lengthi) && (Lengthj > Lengths)) {
              MarkedForRemoval[s] = true;
              MarkedForRemoval[i] = true;
            }
          }
        }
      }
    }
    for (std::map<int, bool>::reverse_iterator itrem = MarkedForRemoval.rbegin();
         itrem != MarkedForRemoval.rend();
         ++itrem) {
      if (itrem->second == true) {
        Seeds.erase(Seeds.begin() + itrem->first);
        ConnectionPoint1.erase(ConnectionPoint1.begin() + itrem->first);
        ConnectionPoint2.erase(ConnectionPoint2.begin() + itrem->first);
      }
    }
  }

Member Data Documentation

std::string trkf::FeatureTracker::fCalDataModuleLabel

private

Definition at line 69 of file FeatureTracker_module.cc.

corner::CornerFinderAlg trkf::FeatureTracker::fCorner

private

Definition at line 66 of file FeatureTracker_module.cc.

std::map<int, std::vector<double> > trkf::FeatureTracker::fEndPointTimes

private

Definition at line 74 of file FeatureTracker_module.cc.

art::ServiceHandle<geo::Geometry const> trkf::FeatureTracker::fGeometryHandle

private

Definition at line 75 of file FeatureTracker_module.cc.

std::string trkf::FeatureTracker::fHitModuleLabel

private

Definition at line 68 of file FeatureTracker_module.cc.

double trkf::FeatureTracker::fLineIntFraction

private

Definition at line 71 of file FeatureTracker_module.cc.

double trkf::FeatureTracker::fLineIntThreshold

private

Definition at line 72 of file FeatureTracker_module.cc.

trkf::SpacePointAlg trkf::FeatureTracker::fSP

private

Definition at line 65 of file FeatureTracker_module.cc.

---

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

• FeatureTracker_module.cc