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)
Public Member Functions inherited from pma::PMAlgTrackingBase
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

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

Additional Inherited Members
Protected Member Functions inherited from pma::PMAlgTrackingBase
	PMAlgTrackingBase (const std::vector< art::Ptr< recob::Hit >> &allhitlist, const pma::ProjectionMatchingAlg::Config &pmalgConfig, const pma::PMAlgVertexing::Config &pmvtxConfig)
	~PMAlgTrackingBase ()
void	guideEndpoints (detinfo::DetectorPropertiesData const &detProp, pma::TrkCandidateColl &tracks)
Protected Attributes inherited from pma::PMAlgTrackingBase
pma::cryo_tpc_view_hitmap	fHitMap
pma::ProjectionMatchingAlg	fProjectionMatchingAlg
pma::PMAlgVertexing	fPMAlgVertexing
pma::TrkCandidateColl	fResult

Detailed Description

Definition at line 101 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 103 of file PMAlgTracking.cxx.

  : 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 150 of file PMAlgTracking.cxx.

{
  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 231 of file PMAlgTracking.cxx.

{
  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 175 of file PMAlgTracking.cxx.

{
  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();
      }
    }
  }
}

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

inlineprivate

Definition at line 140 of file PMAlgTracking.h.

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

Member Data Documentation

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

private

Definition at line 148 of file PMAlgTracking.h.

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

private

Definition at line 149 of file PMAlgTracking.h.

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

private

Definition at line 151 of file PMAlgTracking.h.

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

private

Definition at line 150 of file PMAlgTracking.h.

bool pma::PMAlgFitter::fRunVertexing

private

Definition at line 158 of file PMAlgTracking.h.

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

private

Definition at line 154 of file PMAlgTracking.h.

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

private

Definition at line 156 of file PMAlgTracking.h.

---

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

• PMAlgTracking.h
• PMAlgTracking.cxx