PmaVtxCandidate.h

Go to the documentation of this file.

#ifndef VtxCandidate_h
#define VtxCandidate_h

#include "larreco/RecoAlg/PMAlg/PmaTrkCandidate.h"

namespace pma
{
        class VtxCandidate;
}

class pma::VtxCandidate
{
public:
        static const double kMaxDistToTrack;
        static const double kMinDistToNode;

        VtxCandidate(double segMinLength = 0.5) :
                tracksJoined(false),
                fSegMinLength(segMinLength),
                fMse(0.0), fMse2D(0.0),
                fCenter(0., 0., 0.),
                fErr(0., 0., 0.)
        {}

        bool Has(pma::Track3D* trk) const;

        bool Has(const VtxCandidate& other) const;

        bool IsAttached(pma::Track3D* trk) const;

        bool IsAttached(const VtxCandidate& other) const;

        bool HasLoops(void) const;

        bool Add(const pma::TrkCandidate & trk);

        double ComputeMse2D(void);

        double Test(const VtxCandidate& other) const;

        double MaxAngle(double minLength = 0.0) const;

        size_t Size(void) const { return fAssigned.size(); }
        size_t Size(double minLength) const;

        bool MergeWith(const VtxCandidate& other);

        double Compute(void);

        bool JoinTracks(pma::TrkCandidateColl & tracks, pma::TrkCandidateColl & src);

        const TVector3& Center(void) const { return fCenter; }
        double Mse(void) const { return fMse; }
        double Mse2D(void) const { return fMse2D; }

        std::pair< pma::Track3D*, size_t > Track(size_t i) const
        {
                return std::pair< pma::Track3D*, size_t >(fAssigned[i].first.Track(), fAssigned[i].second);
        }

private:
        bool has(const std::vector<int>& v, int id) const
        {
                for (auto c : v) if (c == id) return true;
                return false;
        }

        bool tracksJoined;
        double fSegMinLength, fMse, fMse2D;
        std::vector< std::pair< pma::TrkCandidate, size_t > > fAssigned;
        TVector3 fCenter, fErr;
};

#endif