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 (void)
const pma::TrkCandidateColl &	result (void)
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 (void) const

Protected Member Functions
void	guideEndpoints (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 (void)
void	buildShowers (void)
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 86 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 94 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

(

void

)

Definition at line 144 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();

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

                        if (fRunVertexing) fPMAlgVertexing.run(fResult);

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

void pma::PMAlgFitter::buildShowers ( void  )

private

Definition at line 229 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(), 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(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 ( void  )

private

Definition at line 170 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(), 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(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 ( void  ) const

inlineinherited

Definition at line 65 of file PMAlgTracking.h.

References pma::PMAlgTrackingBase::fPMAlgVertexing, pma::PMAlgTrackingBase::fResult, pma::PMAlgVertexing::getKinks(), pma::PMAlgTrackingBase::guideEndpoints(), pma::PMAlgTrackingBase::PMAlgTrackingBase(), lar::dump::vector(), and pma::PMAlgTrackingBase::~PMAlgTrackingBase().

        { 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 62 of file PMAlgTracking.h.

References pma::PMAlgTrackingBase::fPMAlgVertexing, pma::PMAlgTrackingBase::fResult, and pma::PMAlgVertexing::getVertices().

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

void pma::PMAlgTrackingBase::guideEndpoints ( pma::TrkCandidateColl &  tracks )

protectedinherited

Definition at line 70 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(), pma::PMAlgTracker::build(), and pma::PMAlgTrackingBase::getKinks().

{
        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(trk, fHitMap[cryo][tpc]);
                }
                else
                {
                        fProjectionMatchingAlg.guideEndpoints(trk, pma::Track3D::kBegin,
                                fHitMap[trk.FrontCryo()][trk.FrontTPC()]);
                        fProjectionMatchingAlg.guideEndpoints(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 127 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 ( void  )

inlineinherited

Definition at line 59 of file PMAlgTracking.h.

References pma::PMAlgTrackingBase::fResult.

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

{ return fResult; }

Member Data Documentation

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

private

Definition at line 133 of file PMAlgTracking.h.

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

pma::cryo_tpc_view_hitmap pma::PMAlgTrackingBase::fHitMap

protectedinherited

Definition at line 77 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 134 of file PMAlgTracking.h.

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

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

private

Definition at line 136 of file PMAlgTracking.h.

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

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

private

Definition at line 135 of file PMAlgTracking.h.

Referenced by buildShowers(), and PMAlgFitter().

pma::PMAlgVertexing pma::PMAlgTrackingBase::fPMAlgVertexing

protectedinherited

Definition at line 80 of file PMAlgTracking.h.

Referenced by build(), pma::PMAlgTracker::build(), pma::PMAlgTrackingBase::getKinks(), and pma::PMAlgTrackingBase::getVertices().

pma::ProjectionMatchingAlg pma::PMAlgTrackingBase::fProjectionMatchingAlg

protectedinherited

Definition at line 79 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(), pma::PMAlgTracker::reassignHits_2(), and pma::PMAlgTracker::validate().

pma::TrkCandidateColl pma::PMAlgTrackingBase::fResult

protectedinherited

Definition at line 82 of file PMAlgTracking.h.

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

bool pma::PMAlgFitter::fRunVertexing

private

Definition at line 141 of file PMAlgTracking.h.

Referenced by build().

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

private

Definition at line 139 of file PMAlgTracking.h.

Referenced by buildShowers(), and buildTracks().

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

private

Definition at line 140 of file PMAlgTracking.h.

Referenced by buildShowers(), and buildTracks().

---

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

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