pma::PMAlgFitter Class Reference

#include "PMAlgTracking.h"

Inheritance diagram for pma::PMAlgFitter:

Classes
struct	Config

Public Member Functions
	PMAlgFitter (const std::vector< art::Ptr< recob::Hit >> &allhitlist, const std::vector< recob::Cluster > &clusters, const std::vector< recob::PFParticle > &pfparticles, const art::FindManyP< recob::Hit > &hitsFromClusters, const art::FindManyP< recob::Cluster > &clusFromPfps, const art::FindManyP< recob::Vertex > &vtxFromPfps, const pma::ProjectionMatchingAlg::Config &pmalgConfig, const pma::PMAlgFitter::Config &pmalgFitterConfig, const pma::PMAlgVertexing::Config &pmvtxConfig)
int	build (detinfo::DetectorPropertiesData const &detProp)
const pma::TrkCandidateColl &	result ()
std::vector< std::pair< TVector3, std::vector< std::pair< size_t, bool > > > >	getVertices (bool onlyBranching=false) const
std::vector< std::pair< TVector3, size_t > >	getKinks () const

Protected Member Functions
void	guideEndpoints (detinfo::DetectorPropertiesData const &detProp, pma::TrkCandidateColl &tracks)

Protected Attributes
pma::cryo_tpc_view_hitmap	fHitMap
pma::ProjectionMatchingAlg	fProjectionMatchingAlg
pma::PMAlgVertexing	fPMAlgVertexing
pma::TrkCandidateColl	fResult

Private Member Functions
void	buildTracks (detinfo::DetectorPropertiesData const &detProp)
void	buildShowers (detinfo::DetectorPropertiesData const &detProp)
bool	has (const std::vector< int > &v, int i) const

Private Attributes
std::vector< std::vector< art::Ptr< recob::Hit > > >	fCluHits
std::map< int, std::vector< art::Ptr< recob::Cluster > > >	fPfpClusters
std::map< int, pma::Vector3D >	fPfpVtx
std::map< int, int >	fPfpPdgCodes
std::vector< int >	fTrackingOnlyPdg
std::vector< int >	fTrackingSkipPdg
bool	fRunVertexing

Detailed Description

Definition at line 96 of file PMAlgTracking.h.

Constructor & Destructor Documentation

pma::PMAlgFitter::PMAlgFitter	(	const std::vector< art::Ptr< recob::Hit >> &	allhitlist,
		const std::vector< recob::Cluster > &	clusters,
		const std::vector< recob::PFParticle > &	pfparticles,
		const art::FindManyP< recob::Hit > &	hitsFromClusters,
		const art::FindManyP< recob::Cluster > &	clusFromPfps,
		const art::FindManyP< recob::Vertex > &	vtxFromPfps,
		const pma::ProjectionMatchingAlg::Config &	pmalgConfig,
		const pma::PMAlgFitter::Config &	pmalgFitterConfig,
		const pma::PMAlgVertexing::Config &	pmvtxConfig
	)

Definition at line 101 of file PMAlgTracking.cxx.

References fCluHits, fPfpClusters, fPfpPdgCodes, and fPfpVtx.

  : PMAlgTrackingBase(allhitlist, pmalgConfig, pmvtxConfig)
  , fTrackingOnlyPdg(pmalgFitterConfig.TrackingOnlyPdg())
  , fTrackingSkipPdg(pmalgFitterConfig.TrackingSkipPdg())
  , fRunVertexing(pmalgFitterConfig.RunVertexing())
{
  mf::LogVerbatim("PMAlgFitter") << "Found " << allhitlist.size() << "hits in the event.";
  mf::LogVerbatim("PMAlgFitter") << "Sort hits by clusters assigned to PFParticles...";

  fCluHits.resize(clusters.size());
  for (size_t i = 0; i < pfparticles.size(); ++i) {
    fPfpPdgCodes[i] = pfparticles[i].PdgCode();

    auto cv = clusFromPfps.at(i);
    for (const auto& c : cv) {
      fPfpClusters[i].push_back(c);
      if (fCluHits[c.key()].empty()) {
        auto hv = hitsFromClusters.at(c.key());
        fCluHits[c.key()].reserve(hv.size());
        for (auto const& h : hv)
          fCluHits[c.key()].push_back(h);
      }
    }

    if (vtxFromPfps.isValid() && vtxFromPfps.at(i).size()) {
      double xyz[3];
      vtxFromPfps.at(i).front()->XYZ(xyz);
      fPfpVtx[i] = pma::Vector3D(xyz[0], xyz[1], xyz[2]);
    }
  }

  mf::LogVerbatim("PMAlgFitter") << "...done, " << fCluHits.size() << " clusters from "
                                 << fPfpClusters.size() << " pfparticles for 3D tracking.";
}

Member Function Documentation

int pma::PMAlgFitter::build

(

detinfo::DetectorPropertiesData const &

detProp

)

Definition at line 147 of file PMAlgTracking.cxx.

References buildShowers(), buildTracks(), fCluHits, fPfpClusters, pma::PMAlgTrackingBase::fPMAlgVertexing, pma::PMAlgTrackingBase::fResult, fRunVertexing, pma::PMAlgTrackingBase::guideEndpoints(), pma::PMAlgVertexing::run(), and pma::TrkCandidateColl::size().

{
  if (!fPfpClusters.empty() && !fCluHits.empty()) {
    // build pm tracks
    buildTracks(detProp);

    // add 3D ref.points for clean endpoints of wire-plae parallel tracks
    guideEndpoints(detProp, fResult);

    if (fRunVertexing) fPMAlgVertexing.run(detProp, fResult);

    // build segment of shower
    buildShowers(detProp);
  }
  else {
    mf::LogWarning("PMAlgFitter") << "no clusters, no pfparticles";
    return -1;
  }

  return fResult.size();
}

void pma::PMAlgFitter::buildShowers ( detinfo::DetectorPropertiesData const &  detProp )

private

Definition at line 226 of file PMAlgTracking.cxx.

References pma::ProjectionMatchingAlg::buildShowerSeg(), pma::TrkCandidate::Clusters(), pma::TrkCandidate::DeleteTrack(), fCluHits, fPfpClusters, fPfpPdgCodes, fPfpVtx, pma::PMAlgTrackingBase::fProjectionMatchingAlg, pma::PMAlgTrackingBase::fResult, fTrackingOnlyPdg, fTrackingSkipPdg, has(), pma::Track3D::HasTwoViews(), pma::TrkCandidate::IsValid(), pma::Track3D::Length(), pma::Track3D::Nodes(), pma::TrkCandidateColl::push_back(), pma::TrkCandidate::SetKey(), pma::TrkCandidate::SetTrack(), util::size(), and pma::TrkCandidate::Track().

Referenced by build().

{
  bool skipPdg = true;
  if (!fTrackingSkipPdg.empty() && (fTrackingSkipPdg.front() == 0)) skipPdg = false;

  bool selectPdg = true;
  if (!fTrackingOnlyPdg.empty() && (fTrackingOnlyPdg.front() == 0)) selectPdg = false;

  for (const auto& pfpCluEntry : fPfpClusters) {
    int pfPartIdx = pfpCluEntry.first;
    int pdg = fPfpPdgCodes[pfPartIdx];

    if (pdg != 11) continue;
    if (skipPdg && has(fTrackingSkipPdg, pdg)) continue;
    if (selectPdg && !has(fTrackingOnlyPdg, pdg)) continue;

    mf::LogVerbatim("PMAlgFitter") << "Process clusters from PFP:" << pfPartIdx << ", pdg:" << pdg;

    std::vector<art::Ptr<recob::Hit>> allHits;

    pma::TrkCandidate candidate;
    std::unordered_map<geo::View_t, size_t> clu_count;
    for (const auto& c : pfpCluEntry.second) {
      if (c->NHits() == 0) { continue; }

      candidate.Clusters().push_back(c.key());
      clu_count[c->View()]++;

      allHits.reserve(allHits.size() + fCluHits.at(c.key()).size());
      for (const auto& h : fCluHits.at(c.key()))
        allHits.push_back(h);
    }
    if (clu_count.size() > 1) // try building only if there are clusters from multiple views
    {
      candidate.SetKey(pfpCluEntry.first);

      mf::LogVerbatim("PMAlgFitter") << "building..."
                                     << ", pdg:" << pdg;

      auto search = fPfpVtx.find(pfPartIdx);
      if (search != fPfpVtx.end()) {
        candidate.SetTrack(fProjectionMatchingAlg.buildShowerSeg(detProp, allHits, search->second));

        if (candidate.IsValid() && candidate.Track()->HasTwoViews() &&
            (candidate.Track()->Nodes().size() > 1) && !std::isnan(candidate.Track()->Length())) {
          fResult.push_back(candidate);
        }
        else {
          candidate.DeleteTrack();
        }
      }
    }
  }
}

void pma::PMAlgFitter::buildTracks ( detinfo::DetectorPropertiesData const &  detProp )

private

Definition at line 171 of file PMAlgTracking.cxx.

References pma::ProjectionMatchingAlg::buildMultiTPCTrack(), pma::TrkCandidate::Clusters(), pma::TrkCandidate::DeleteTrack(), fCluHits, fPfpClusters, fPfpPdgCodes, pma::PMAlgTrackingBase::fProjectionMatchingAlg, pma::PMAlgTrackingBase::fResult, fTrackingOnlyPdg, fTrackingSkipPdg, has(), pma::Track3D::HasTwoViews(), pma::TrkCandidate::IsValid(), pma::Track3D::Length(), pma::Track3D::Nodes(), pma::TrkCandidateColl::push_back(), pma::TrkCandidate::SetKey(), pma::TrkCandidate::SetTrack(), util::size(), and pma::TrkCandidate::Track().

Referenced by build().

{
  bool skipPdg = true;
  if (!fTrackingSkipPdg.empty() && (fTrackingSkipPdg.front() == 0)) skipPdg = false;

  bool selectPdg = true;
  if (!fTrackingOnlyPdg.empty() && (fTrackingOnlyPdg.front() == 0)) selectPdg = false;

  for (const auto& pfpCluEntry : fPfpClusters) {
    int pfPartIdx = pfpCluEntry.first;
    int pdg = fPfpPdgCodes[pfPartIdx];

    //if (pdg == 11) continue;
    if (skipPdg && has(fTrackingSkipPdg, pdg)) continue;
    if (selectPdg && !has(fTrackingOnlyPdg, pdg)) continue;

    mf::LogVerbatim("PMAlgFitter") << "Process clusters from PFP:" << pfPartIdx << ", pdg:" << pdg;

    std::vector<art::Ptr<recob::Hit>> allHits;

    pma::TrkCandidate candidate;
    std::unordered_map<geo::View_t, size_t> clu_count;
    for (const auto& c : pfpCluEntry.second) {
      if (c->NHits() == 0) { continue; }

      candidate.Clusters().push_back(c.key());
      clu_count[c->View()]++;

      allHits.reserve(allHits.size() + fCluHits.at(c.key()).size());
      for (const auto& h : fCluHits.at(c.key())) {
        allHits.push_back(h);
      }
    }
    if (clu_count.size() > 1) // try building only if there are clusters from multiple views
    {
      candidate.SetKey(pfpCluEntry.first);

      candidate.SetTrack(fProjectionMatchingAlg.buildMultiTPCTrack(detProp, allHits));

      if (candidate.IsValid() && candidate.Track()->HasTwoViews() &&
          (candidate.Track()->Nodes().size() > 1)) {
        if (!std::isnan(candidate.Track()->Length())) { fResult.push_back(candidate); }
        else {
          mf::LogError("PMAlgFitter") << "Trajectory fit lenght is nan.";
          candidate.DeleteTrack();
        }
      }
      else {
        candidate.DeleteTrack();
      }
    }
  }
}

std::vector<std::pair<TVector3, size_t> > pma::PMAlgTrackingBase::getKinks ( ) const

inlineinherited

Definition at line 74 of file PMAlgTracking.h.

References lar::dump::vector().

  {
    return fPMAlgVertexing.getKinks(fResult);
  }

std::vector<std::pair<TVector3, std::vector<std::pair<size_t, bool> > > > pma::PMAlgTrackingBase::getVertices ( bool  onlyBranching = false ) const

inlineinherited

Definition at line 68 of file PMAlgTracking.h.

  {
    return fPMAlgVertexing.getVertices(fResult, onlyBranching);
  }

void pma::PMAlgTrackingBase::guideEndpoints ( detinfo::DetectorPropertiesData const &  detProp, pma::TrkCandidateColl &  tracks  )

protectedinherited

Definition at line 79 of file PMAlgTracking.cxx.

References pma::PMAlgTrackingBase::fHitMap, pma::PMAlgTrackingBase::fProjectionMatchingAlg, pma::ProjectionMatchingAlg::guideEndpoints(), pma::Track3D::kBegin, pma::Track3D::kEnd, and pma::TrkCandidateColl::tracks().

Referenced by build(), and pma::PMAlgTracker::build().

{
  for (auto const& t : tracks.tracks()) {
    auto& trk = *(t.Track());

    unsigned int tpc = trk.FrontTPC(), cryo = trk.FrontCryo();
    if ((tpc == trk.BackTPC()) && (cryo == trk.BackCryo())) {
      fProjectionMatchingAlg.guideEndpoints(detProp, trk, fHitMap[cryo][tpc]);
    }
    else {
      fProjectionMatchingAlg.guideEndpoints(
        detProp, trk, pma::Track3D::kBegin, fHitMap[trk.FrontCryo()][trk.FrontTPC()]);
      fProjectionMatchingAlg.guideEndpoints(
        detProp, trk, pma::Track3D::kEnd, fHitMap[trk.BackCryo()][trk.BackTPC()]);
    }
  }
}

bool pma::PMAlgFitter::has ( const std::vector< int > &  v, int  i  ) const

inlineprivate

Definition at line 134 of file PMAlgTracking.h.

Referenced by buildShowers(), and buildTracks().

  {
    for (auto c : v) {
      if (c == i) return true;
    }
    return false;
  }

const pma::TrkCandidateColl& pma::PMAlgTrackingBase::result ( )

inlineinherited

Definition at line 66 of file PMAlgTracking.h.

Referenced by pma::PMAlgTracker::matchCluster(), pma::PMAlgTracker::reassignHits_1(), and pma::PMAlgTracker::reassignSingleViewEnds_1().

{ return fResult; }

Member Data Documentation

std::vector<std::vector<art::Ptr<recob::Hit> > > pma::PMAlgFitter::fCluHits

private

Definition at line 142 of file PMAlgTracking.h.

Referenced by build(), buildShowers(), buildTracks(), and PMAlgFitter().

pma::cryo_tpc_view_hitmap pma::PMAlgTrackingBase::fHitMap

protectedinherited

Definition at line 88 of file PMAlgTracking.h.

Referenced by pma::PMAlgTrackingBase::guideEndpoints(), pma::PMAlgTrackingBase::PMAlgTrackingBase(), and pma::PMAlgTracker::validate().

std::map<int, std::vector<art::Ptr<recob::Cluster> > > pma::PMAlgFitter::fPfpClusters

private

Definition at line 143 of file PMAlgTracking.h.

Referenced by build(), buildShowers(), buildTracks(), and PMAlgFitter().

std::map<int, int> pma::PMAlgFitter::fPfpPdgCodes

private

Definition at line 145 of file PMAlgTracking.h.

Referenced by buildShowers(), buildTracks(), and PMAlgFitter().

std::map<int, pma::Vector3D> pma::PMAlgFitter::fPfpVtx

private

Definition at line 144 of file PMAlgTracking.h.

Referenced by buildShowers(), and PMAlgFitter().

pma::PMAlgVertexing pma::PMAlgTrackingBase::fPMAlgVertexing

protectedinherited

Definition at line 91 of file PMAlgTracking.h.

Referenced by build(), and pma::PMAlgTracker::build().

pma::ProjectionMatchingAlg pma::PMAlgTrackingBase::fProjectionMatchingAlg

protectedinherited

Definition at line 90 of file PMAlgTracking.h.

Referenced by buildShowers(), buildTracks(), pma::PMAlgTracker::extendTrack(), pma::PMAlgTrackingBase::guideEndpoints(), pma::PMAlgTracker::matchCluster(), pma::PMAlgTracker::matchTrack(), pma::PMAlgTracker::mergeCoLinear(), pma::PMAlgTracker::reassignHits_1(), and pma::PMAlgTracker::validate().

pma::TrkCandidateColl pma::PMAlgTrackingBase::fResult

protectedinherited

Definition at line 93 of file PMAlgTracking.h.

Referenced by build(), pma::PMAlgTracker::build(), buildShowers(), buildTracks(), pma::PMAlgTracker::listUsedClusters(), and pma::PMAlgTrackingBase::~PMAlgTrackingBase().

bool pma::PMAlgFitter::fRunVertexing

private

Definition at line 152 of file PMAlgTracking.h.

Referenced by build().

std::vector<int> pma::PMAlgFitter::fTrackingOnlyPdg

private

Definition at line 148 of file PMAlgTracking.h.

Referenced by buildShowers(), and buildTracks().

std::vector<int> pma::PMAlgFitter::fTrackingSkipPdg

private

Definition at line 150 of file PMAlgTracking.h.

Referenced by buildShowers(), and buildTracks().

---

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

• larreco/v09_25_00/source/larreco/RecoAlg/PMAlgTracking.h
• larreco/v09_25_00/source/larreco/RecoAlg/PMAlgTracking.cxx