pma::TrkCandidateColl Class Reference

#include "PmaTrkCandidate.h"

Public Member Functions
size_t	size (void) const
void	resize (size_t n)
bool	empty (void) const
void	push_back (const TrkCandidate &trk)
void	erase_at (size_t pos)
void	clear (void)
TrkCandidate &	operator[] (size_t i)
TrkCandidate const &	operator[] (size_t i) const
TrkCandidate &	front (void)
TrkCandidate const &	front (void) const
TrkCandidate &	back (void)
TrkCandidate const &	back (void) const
std::vector< TrkCandidate > const &	tracks (void) const
std::vector< TrkCandidate > &	tracks (void)
std::vector< TrkCandidate > const &	parents (void) const
int	getCandidateIndex (pma::Track3D const *candidate) const
int	getCandidateTreeId (pma::Track3D const *candidate) const
void	merge (size_t idx1, size_t idx2)
void	setParentDaughterConnections (void)
void	setTreeId (int id, size_t trkIdx, bool isRoot=true)
int	setTreeIds (void)
void	flipTreesToCoordinate (size_t coordinate)
void	flipTreesByDQdx ()
bool	setTreeOriginAtFront (pma::Track3D *trk)
bool	setTreeOriginAtBack (pma::Track3D *trk)
pma::Track3D *	getTreeCopy (pma::TrkCandidateColl &dst, size_t trkIdx, bool isRoot=true)

Private Attributes
std::vector< TrkCandidate >	fCandidates
std::vector< TrkCandidate >	fParents

Detailed Description

Definition at line 80 of file PmaTrkCandidate.h.

Member Function Documentation

TrkCandidate& pma::TrkCandidateColl::back ( void  )

inline

Definition at line 97 of file PmaTrkCandidate.h.

{ return fCandidates.back(); }

TrkCandidate const& pma::TrkCandidateColl::back ( void  ) const

inline

Definition at line 98 of file PmaTrkCandidate.h.

{ return fCandidates.back(); }

void pma::TrkCandidateColl::clear ( void  )

inline

Definition at line 89 of file PmaTrkCandidate.h.

Referenced by pma::PMAlgVertexing::cleanTracks(), and pma::PMAlgVertexing::collectTracks().

{ fCandidates.clear(); }

bool pma::TrkCandidateColl::empty ( void  ) const

inline

Definition at line 85 of file PmaTrkCandidate.h.

Referenced by pma::PMAlgTracker::matchCluster().

{ return fCandidates.empty(); }

void pma::TrkCandidateColl::erase_at ( size_t  pos )

inline

Definition at line 88 of file PmaTrkCandidate.h.

Referenced by pma::VtxCandidate::JoinTracks(), and pma::PMAlgTracker::mergeCoLinear().

{ fCandidates.erase(fCandidates.begin() + pos); }

void pma::TrkCandidateColl::flipTreesByDQdx

(

void

)

Definition at line 286 of file PmaTrkCandidate.cxx.

References pma::Track3D::AutoFlip(), pma::Track3D::CanFlip(), and pma::Track3D::kForward.

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

{
        std::map< int, std::vector< pma::Track3D* > > trkMap;

        setTreeIds();
        for (auto const & t : fCandidates)
        {
            if (t.IsValid()) trkMap[t.TreeId()].push_back(t.Track());
        }

        for (auto & tEntry : trkMap)
        {
                std::sort(tEntry.second.begin(), tEntry.second.end(), pma::bTrack3DLonger());
                for (size_t i = tEntry.second.size(); i > 0; --i)
                {
                        pma::Track3D* trk = tEntry.second[i-1];
                        if (trk->CanFlip())
                        {
                        trk->AutoFlip(pma::Track3D::kForward, 0.15);
            }

            // This could e.g. break muon by delta ray, so for
            // the moment use conservative way and think of
            // a better strategy...
                        //std::vector< pma::Track3D* > newTracks;
                        //trk->AutoFlip(newTracks, pma::Track3D::kForward, 0.15);
                        //for (const auto ts : newTracks)
            //{
            //    fCandidates.emplace_back(ts, -1, tEntry.first);
            //}
                }
        }
}

void pma::TrkCandidateColl::flipTreesToCoordinate

(

size_t

coordinate

)

Definition at line 183 of file PmaTrkCandidate.cxx.

References pma::Track3D::back(), pma::Track3D::front(), and pma::Hit3D::Point3D().

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

{
        std::map< int, std::vector< pma::Track3D* > > toFlip;
        std::map< int, double > minVal;

        setTreeIds();
        for (auto & t : fCandidates)
        {
            if (!t.IsValid()) continue;

                int tid = t.TreeId();
                if (minVal.find(tid) == minVal.end()) minVal[tid] = 1.0e12;

                TVector3 pFront(t.Track()->front()->Point3D()); pFront.SetX(-pFront.X()); pFront.SetY(-pFront.Y());
                TVector3 pBack(t.Track()->back()->Point3D()); pBack.SetX(-pBack.X()); pBack.SetY(-pBack.Y());

                bool pushed = false;
                if (pFront[coordinate] < minVal[tid]) { minVal[tid] = pFront[coordinate]; toFlip[tid].push_back(t.Track()); pushed = true; }
                if (pBack[coordinate] < minVal[tid]) { minVal[tid] = pBack[coordinate]; if (!pushed) toFlip[tid].push_back(t.Track()); }
        }

        for (auto & tEntry : toFlip)
                if (tEntry.first >= 0)
        {
                size_t attempts = 0;
                while (!tEntry.second.empty())
                {
                        pma::Track3D* trk = tEntry.second.back();
                        tEntry.second.pop_back();

                        TVector3 pFront(trk->front()->Point3D()); pFront.SetX(-pFront.X()); pFront.SetY(-pFront.Y());
                        TVector3 pBack(trk->back()->Point3D()); pBack.SetX(-pBack.X()); pBack.SetY(-pBack.Y());

                        if (pFront[coordinate] > pBack[coordinate])
                        {
                                if (setTreeOriginAtBack(trk)) { break; }
                                else { mf::LogWarning("pma::TrkCandidateColl") << "Flip to coordinate failed."; }
                        }
                        else
                        {
                            setTreeOriginAtFront(trk);
                            break; // good orientation, go to the next tree
                        }

                        if (attempts++ > 2) break; // do not try all the tracks in the queue...
                }
        }
}

TrkCandidate& pma::TrkCandidateColl::front ( void  )

inline

Definition at line 94 of file PmaTrkCandidate.h.

{ return fCandidates.front(); }

TrkCandidate const& pma::TrkCandidateColl::front ( void  ) const

inline

Definition at line 95 of file PmaTrkCandidate.h.

{ return fCandidates.front(); }

int pma::TrkCandidateColl::getCandidateIndex

(

pma::Track3D const *

candidate

)

const

Definition at line 49 of file PmaTrkCandidate.cxx.

References pma::TrkCandidate::Track().

Referenced by pma::VtxCandidate::JoinTracks(), and pma::PMAlgStitching::StitchTracks().

{
        for (size_t t = 0; t < fCandidates.size(); ++t)
                if (fCandidates[t].Track() == candidate) return t;
        return -1;
}

int pma::TrkCandidateColl::getCandidateTreeId

(

pma::Track3D const *

candidate

)

const

Definition at line 56 of file PmaTrkCandidate.cxx.

Referenced by pma::PMAlgStitching::StitchTracks().

{
    int id = getCandidateIndex(candidate);
        if (id >= 0) return fCandidates[id].TreeId();
        else return -1;
}

pma::Track3D * pma::TrkCandidateColl::getTreeCopy	(	pma::TrkCandidateColl &	dst,
		size_t	trkIdx,
		bool	isRoot = true
	)

Definition at line 332 of file PmaTrkCandidate.cxx.

References pma::Track3D::AttachTo(), pma::SortedObjectBase::Next(), pma::SortedBranchBase::Next(), pma::SortedBranchBase::NextCount(), pma::Track3D::NextSegment(), pma::Track3D::Nodes(), pma::Segment3D::Parent(), and tracks().

Referenced by pma::VtxCandidate::JoinTracks().

{
        pma::Track3D* trk = fCandidates[trkIdx].Track();
        pma::Node3D* vtx = trk->Nodes().front();
        pma::Segment3D* segThis = 0;
        pma::Segment3D* seg = 0;

        int key = fCandidates[trkIdx].Key();
        int tid = fCandidates[trkIdx].TreeId();

        pma::Track3D* trkCopy = new pma::Track3D(*trk);
        pma::Node3D* vtxCopy = trkCopy->Nodes().front();
        pma::Segment3D* segThisCopy = 0;

        dst.tracks().emplace_back(trkCopy, key, tid);

        if (!isRoot)
        {
                segThis = trk->NextSegment(vtx);
                if (segThis) vtx = static_cast< pma::Node3D* >(segThis->Next());

                segThisCopy = trkCopy->NextSegment(vtxCopy);
                if (segThisCopy) vtxCopy = static_cast< pma::Node3D* >(segThisCopy->Next());
        }

        while (vtx)
        {
                segThis = trk->NextSegment(vtx);
                segThisCopy = trkCopy->NextSegment(vtxCopy);

                for (size_t i = 0; i < vtx->NextCount(); i++)
                {
                        seg = static_cast< pma::Segment3D* >(vtx->Next(i));
                        if (seg != segThis)
                        {
                                int idx = getCandidateIndex(seg->Parent());

                                if (idx >= 0)
                                {
                                        pma::Track3D* branchCopy = getTreeCopy(dst, idx, false);
                                        if (!branchCopy->AttachTo(vtxCopy, true)) // no flip
                                        mf::LogError("pma::getTreeCopy") << "Branch copy cannot be attached to the tree.";
                                }
                                else mf::LogError("pma::getTreeCopy") << "Branch of the tree not found in source collection.";
                        }
                }

                if (segThis) vtx = static_cast< pma::Node3D* >(segThis->Next());
                else break;

                if (segThisCopy) vtxCopy = static_cast< pma::Node3D* >(segThisCopy->Next());
        }

        return trkCopy;
}

void pma::TrkCandidateColl::merge	(	size_t	idx1,
		size_t	idx2
	)

Definition at line 321 of file PmaTrkCandidate.cxx.

References pma::TrkCandidate::Clusters(), pma::TrkCandidate::Track(), and pma::TrkCandidate::TreeId().

Referenced by pma::PMAlgStitching::StitchTracks().

{
    fCandidates[idx1].Track()->ExtendWith(fCandidates[idx2].Track()); // deletes track at idx2

    for (auto c : fCandidates[idx2].Clusters()) { fCandidates[idx1].Clusters().push_back(c); }

    fCandidates.erase(fCandidates.begin() + idx2);

    setTreeId(fCandidates[idx1].TreeId(), idx1);
}

TrkCandidate& pma::TrkCandidateColl::operator[] ( size_t  i )

inline

Definition at line 91 of file PmaTrkCandidate.h.

{ return fCandidates[i]; }

TrkCandidate const& pma::TrkCandidateColl::operator[] ( size_t  i ) const

inline

Definition at line 92 of file PmaTrkCandidate.h.

{ return fCandidates[i]; }

std::vector< TrkCandidate > const& pma::TrkCandidateColl::parents ( void  ) const

inline

Definition at line 103 of file PmaTrkCandidate.h.

{ return fParents; }

void pma::TrkCandidateColl::push_back ( const TrkCandidate &  trk )

inline

Definition at line 87 of file PmaTrkCandidate.h.

Referenced by pma::PMAlgTracker::build(), pma::PMAlgFitter::buildShowers(), pma::PMAlgFitter::buildTracks(), pma::PMAlgVertexing::collectTracks(), pma::PMAlgTracker::fromMaxCluster_tpc(), pma::VtxCandidate::JoinTracks(), pma::PMAlgTracker::matchCluster(), and pma::PMAlgTracker::reassignHits_1().

{ fCandidates.push_back(trk); }

void pma::TrkCandidateColl::resize ( size_t  n )

inline

Definition at line 84 of file PmaTrkCandidate.h.

{ return fCandidates.resize(n); }

void pma::TrkCandidateColl::setParentDaughterConnections

(

void

)

Definition at line 63 of file PmaTrkCandidate.cxx.

References pma::TrkCandidate::IsValid(), n, pma::SortedBranchBase::Next(), pma::SortedBranchBase::NextCount(), pma::Track3D::Nodes(), pma::TrkCandidate::Parent(), pma::SortedObjectBase::Prev(), pma::TrkCandidate::SetParent(), and pma::TrkCandidate::TreeId().

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

{
    fParents.clear();

    size_t t = 0;
    while (t < fCandidates.size())
    {
        if (fCandidates[t].IsValid())
        {
            fCandidates[t].SetParent(-1);
            fCandidates[t].Daughters().clear();
            t++;
        }
        else fCandidates.erase(fCandidates.begin() + t);
    }

        for (t = 0; t < fCandidates.size(); ++t)
        {
            if (!fCandidates[t].IsValid()) continue;

                pma::Track3D const * trk = fCandidates[t].Track();
                pma::Node3D const * firstNode = trk->Nodes().front();
                if (firstNode->Prev()) // parent is a reconstructed track
                {
                        pma::Track3D const * parentTrk = static_cast< pma::Segment3D* >(firstNode->Prev())->Parent();
                        fCandidates[t].SetParent(getCandidateIndex(parentTrk));
                }
                else if (fCandidates[t].Parent() < 0) // parent not reconstructed and not yet set, add empty candidate as a "primary" particle
                {
                    fParents.push_back(pma::TrkCandidate());
                    fParents.back().SetTreeId(fCandidates[t].TreeId());
                    size_t pri_idx = fCandidates.size() + fParents.size() - 1;

                    for (size_t i = 0; i < firstNode->NextCount(); ++i)
                    {
                        pma::Track3D const * daughterTrk = static_cast< pma::Segment3D* >(firstNode->Next(i))->Parent();
                        int idx = getCandidateIndex(daughterTrk);
                                if (idx >= 0)
                                {
                                    fCandidates[(size_t)idx].SetParent(pri_idx);
                                    fParents.back().Daughters().push_back((size_t)idx);                             
                                }
                    }
                }

                for (size_t n = 1; n < trk->Nodes().size(); ++n)
                {
                        auto node = trk->Nodes()[n]; 
                        for (size_t i = 0; i < node->NextCount(); ++i)
                        {
                                pma::Track3D const * daughterTrk = static_cast< pma::Segment3D* >(node->Next(i))->Parent();
                                if (daughterTrk != trk)
                                {
                                        int idx = getCandidateIndex(daughterTrk);
                                        if (idx >= 0) fCandidates[t].Daughters().push_back((size_t)idx);
                                }
                        }
                }
        }
}

void pma::TrkCandidateColl::setTreeId	(	int	id,
		size_t	trkIdx,
		bool	isRoot = true
	)

Definition at line 125 of file PmaTrkCandidate.cxx.

References pma::SortedObjectBase::Next(), pma::SortedBranchBase::Next(), pma::SortedBranchBase::NextCount(), pma::Track3D::NextSegment(), pma::Track3D::Nodes(), and pma::Segment3D::Parent().

{
        pma::Track3D* trk = fCandidates[trkIdx].Track();
        pma::Node3D* vtx = trk->Nodes().front();
        pma::Segment3D* segThis = 0;
        pma::Segment3D* seg = 0;

        if (!isRoot)
        {
                segThis = trk->NextSegment(vtx);
                if (segThis) vtx = static_cast< pma::Node3D* >(segThis->Next());
        }

        while (vtx)
        {
                segThis = trk->NextSegment(vtx);

                for (size_t i = 0; i < vtx->NextCount(); i++)
                {
                        seg = static_cast< pma::Segment3D* >(vtx->Next(i));
                        if (seg != segThis)
                        {
                                int idx = getCandidateIndex(seg->Parent());

                                if (idx >= 0) setTreeId(id, idx, false);
                                else mf::LogError("pma::setTreeId") << "Branch of the tree not found in tracks collection.";
                        }
                }

                if (segThis) vtx = static_cast< pma::Node3D* >(segThis->Next());
                else break;
        }

        fCandidates[trkIdx].SetTreeId(id);
}

int pma::TrkCandidateColl::setTreeIds

(

void

)

Definition at line 161 of file PmaTrkCandidate.cxx.

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

{
        for (auto & t : fCandidates) t.SetTreeId(-1);

        int id = 0;
        for (auto & t : fCandidates)
        {
                if (!t.IsValid() || (t.TreeId() >= 0)) continue;

        // index of a valid (reconstructed) track without reconstructed parent track
                int rootTrkIdx = getCandidateIndex(t.Track()->GetRoot());

                if (rootTrkIdx >= 0) setTreeId(id, rootTrkIdx);
                else mf::LogError("pma::setTreeIds") << "Root of the tree not found in tracks collection.";

                id++;
        }

        return id;
}

bool pma::TrkCandidateColl::setTreeOriginAtBack

(

pma::Track3D *

trk

)

Definition at line 269 of file PmaTrkCandidate.cxx.

References pma::Track3D::Flip().

{
    int trkIdx = getCandidateIndex(trk);
    int treeId = getCandidateTreeId(trk);
    if (trkIdx < 0) { throw cet::exception("pma::TrkCandidateColl") << "Track not found in the collection." << std::endl; }

    pma::Track3D* incoming = fCandidates[trkIdx].Track();
    std::vector< pma::Track3D* > newTracks;
    bool done = incoming->Flip(newTracks);
    for (const auto ts : newTracks)
    {
        fCandidates.emplace_back(ts, -1, treeId);
    }
    return done;
}

bool pma::TrkCandidateColl::setTreeOriginAtFront

(

pma::Track3D *

trk

)

Definition at line 233 of file PmaTrkCandidate.cxx.

References pma::Track3D::Flip(), pma::Track3D::index_of(), n, pma::Track3D::NextSegment(), pma::Track3D::Nodes(), pma::Segment3D::Parent(), pma::SortedObjectBase::Prev(), and pma::Track3D::Split().

Referenced by pma::PMAlgStitching::StitchTracks().

{
    int trkIdx = getCandidateIndex(trk);
    int treeId = getCandidateTreeId(trk);
    if (trkIdx < 0) { throw cet::exception("pma::TrkCandidateColl") << "Track not found in the collection." << std::endl; }

    bool done = true;
    pma::Node3D* n = trk->Nodes().front();
    if (n->Prev())
    {
        pma::Segment3D* seg = static_cast< pma::Segment3D* >(n->Prev());
        pma::Track3D* incoming = seg->Parent();
        std::vector< pma::Track3D* > newTracks;
        if (incoming->NextSegment(n)) // upfff, need to break the parent track
        {
                    int idx = incoming->index_of(n);
                    if (idx >= 0)
                    {
                    pma::Track3D* u = incoming->Split(idx, false); // u is in front of incoming
                    if (u)
                    {
                        newTracks.push_back(u);
                        done = u->Flip(newTracks);
                    }
                    else { done = false; }
                }
                else { throw cet::exception("pma::Track3D") << "Node not found." << std::endl; }
        }
        else { done = incoming->Flip(newTracks); } // just flip incoming

        for (const auto ts : newTracks) { fCandidates.emplace_back(ts, -1, treeId); }
    }
    return done;
}

size_t pma::TrkCandidateColl::size ( void  ) const

inline

Definition at line 83 of file PmaTrkCandidate.h.

Referenced by pma::PMAlgFitter::build(), pma::PMAlgTracker::build(), pma::PMAlgVertexing::collectTracks(), pma::PMAlgVertexing::findKinksOnTracks(), pma::PMAlgVertexing::firstPassCandidates(), pma::PMAlgVertexing::getKinks(), pma::PMAlgVertexing::getVertices(), pma::VtxCandidate::JoinTracks(), pma::PMAlgTracker::matchCluster(), pma::PMAlgVertexing::mergeBrokenTracks(), pma::PMAlgTracker::mergeCoLinear(), pma::PMAlgTracker::reassignHits_1(), pma::PMAlgTracker::reassignHits_2(), pma::PMAlgTracker::reassignSingleViewEnds_1(), pma::PMAlgTracker::reassignSingleViewEnds_2(), pma::PMAlgVertexing::run(), pma::PMAlgVertexing::secondPassCandidates(), pma::PMAlgVertexing::sortTracks(), pma::PMAlgVertexing::splitMergedTracks(), pma::PMAlgStitching::StitchTracks(), pma::PMAlgStitching::StitchTracksAPA(), pma::PMAlgStitching::StitchTracksCPA(), and pma::PMAlgCosmicTagger::tag().

{ return fCandidates.size(); }

std::vector< TrkCandidate > const& pma::TrkCandidateColl::tracks ( void  ) const

inline

Definition at line 100 of file PmaTrkCandidate.h.

Referenced by pma::PMAlgVertexing::cleanTracks(), pma::PMAlgVertexing::collectTracks(), pma::PMAlgTracker::freezeBranchingNodes(), pma::PMAlgCosmicTagger::fullCrossingTagger(), getTreeCopy(), pma::PMAlgTrackingBase::guideEndpoints(), pma::VtxCandidate::JoinTracks(), pma::PMAlgTracker::matchTrack(), pma::PMAlgTracker::maxCluster(), pma::PMAlgTracker::mergeCoLinear(), pma::PMAlgCosmicTagger::nonBeamT0Tag(), pma::PMAlgCosmicTagger::outOfDriftWindow(), pma::PMAlgTracker::releaseAllNodes(), pma::PMAlgVertexing::sortTracks(), pma::PMAlgStitching::StitchTracks(), pma::PMAlgCosmicTagger::tagApparentStopper(), pma::PMAlgCosmicTagger::tagTopFrontBack(), and pma::PMAlgTrackingBase::~PMAlgTrackingBase().

{ return fCandidates; }

std::vector< TrkCandidate >& pma::TrkCandidateColl::tracks ( void  )

inline

Definition at line 101 of file PmaTrkCandidate.h.

{ return fCandidates; }

Member Data Documentation

std::vector< TrkCandidate > pma::TrkCandidateColl::fCandidates

private

Definition at line 124 of file PmaTrkCandidate.h.

std::vector< TrkCandidate > pma::TrkCandidateColl::fParents

private

Definition at line 125 of file PmaTrkCandidate.h.

---

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

• larreco/v06_64_02/source/larreco/RecoAlg/PMAlg/PmaTrkCandidate.h
• larreco/v06_64_02/source/larreco/RecoAlg/PMAlg/PmaTrkCandidate.cxx