Inheritance diagram for trkf::FeatureTracker:

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

Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)

template<typename Config >
	EDProducer (Table< Config > const &config)

std::string	workerType () const

Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept

	Producer (fhicl::ParameterSet const &)

	Producer (Producer const &)=delete

	Producer (Producer &&)=delete

Producer &	operator= (Producer const &)=delete

Producer &	operator= (Producer &&)=delete

void	doBeginJob (SharedResources const &resources)

void	doEndJob ()

void	doRespondToOpenInputFile (FileBlock const &fb)

void	doRespondToCloseInputFile (FileBlock const &fb)

void	doRespondToOpenOutputFiles (FileBlock const &fb)

void	doRespondToCloseOutputFiles (FileBlock const &fb)

bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)

bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)

bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)

Public Member Functions inherited from art::Modifier
	~Modifier () noexcept

	Modifier ()

	Modifier (Modifier const &)=delete

	Modifier (Modifier &&)=delete

Modifier &	operator= (Modifier const &)=delete

Modifier &	operator= (Modifier &&)=delete

Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept

	ModuleBase ()

ModuleDescription const &	moduleDescription () const

void	setModuleDescription (ModuleDescription const &)

std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const

void	sortConsumables (std::string const &current_process_name)

template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)

template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

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

Additional Inherited Members
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer

using	WorkerType = WorkerT< EDProducer >

Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >

Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >

Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)

Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()

template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)

template<typename T , BranchType = InEvent>
void	consumesMany ()

template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)

template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)

template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

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 )

overrideprivatevirtual

Implements art::EDProducer.

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:

larreco/larreco/TrackFinder/FeatureTracker_module.cc

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation