pma::Element3D Class Reference

#include <PmaElement3D.h>

Inheritance diagram for pma::Element3D:

Public Member Functions
int	TPC (void) const
	TPC index or -1 if out of any TPC. More...
int	Cryo (void) const
	Cryostat index or -1 if out of any cryostat. More...
virtual double	GetDistance2To (const TVector3 &p3d) const =0
	Distance [cm] from the 3D point to the object 3D. More...
virtual double	GetDistance2To (const TVector2 &p2d, unsigned int view) const =0
	Distance [cm] from the 2D point to the object's 2D projection in one of wire views. More...
virtual pma::Vector3D	GetDirection3D (void) const =0
	Get 3D direction cosines corresponding to this element. More...
virtual TVector3	GetUnconstrainedProj3D (const TVector2 &p2d, unsigned int view) const =0
virtual void	SetProjection (pma::Hit3D &h) const =0
virtual double	Length2 (void) const =0
double	Length (void) const
const std::vector< pma::Hit3D * > &	Hits (void) const
bool	HasHit (const pma::Hit3D *h) const
pma::Hit3D &	Hit (size_t index)
void	RemoveHitAt (size_t index)
void	AddHit (pma::Hit3D *h)
size_t	NHits (void) const
size_t	NEnabledHits (unsigned int view=geo::kUnknown) const
size_t	NPrecalcEnabledHits (void) const
TVector3 const &	ReferencePoint (size_t index) const
size_t	NPoints (void) const
void	AddPoint (TVector3 *p)
virtual void	ClearAssigned (pma::Track3D *trk=0)
void	UpdateHitParams (void)
void	UpdateProjection (void)
void	SortHits (void)
double	SumDist2 (void) const
double	SumDist2 (unsigned int view) const
double	SumHitsQ (unsigned int view) const
unsigned int	NHits (unsigned int view) const
unsigned int	NThisHits (unsigned int view) const
double	HitsRadius3D (unsigned int view) const
bool	IsFrozen (void) const
	Check if the vertex 3D position is fixed. More...
void	SetFrozen (bool state)
	Fix / relese vertex 3D position. More...
bool	SelectRndHits (size_t nmax_per_view)
bool	SelectAllHits (void)

Static Public Member Functions
static float	OptFactor (unsigned int view)
static void	SetOptFactor (unsigned int view, float value)

Protected Member Functions
	Element3D (void)
virtual double	SumDist2Hits (void) const =0

Protected Attributes
int	fTPC
int	fCryo
bool	fFrozen
std::vector< pma::Hit3D * >	fAssignedHits
std::vector< TVector3 * >	fAssignedPoints
size_t	fNThisHits [3]
size_t	fNThisHitsEnabledAll
size_t	fNHits [3]
double	fSumHitsQ [3]
double	fHitsRadius

Static Protected Attributes
static float	fOptFactors [3] = { 0.2F, 0.8F, 1.0F }

Detailed Description

Definition at line 30 of file PmaElement3D.h.

Constructor & Destructor Documentation

pma::Element3D::Element3D ( void  )

protected

Definition at line 22 of file PmaElement3D.cxx.

                        :
        fTPC(-1), fCryo(-1),
        fFrozen(false),
        fHitsRadius(0)
{
        fNThisHitsEnabledAll = 0;
        for (unsigned int i = 0; i < 3; i++)
        {
                fNHits[i] = 0;
                fNThisHits[i] = 0;
                fSumHitsQ[i] = 0.0;
        }
}

Member Function Documentation

void pma::Element3D::AddHit ( pma::Hit3D *  h )

inline

Definition at line 68 of file PmaElement3D.h.

        {
                fAssignedHits.push_back(h);
                SetProjection(*h);
        }

void pma::Element3D::AddPoint ( TVector3 *  p )

inline

Definition at line 80 of file PmaElement3D.h.

{ fAssignedPoints.push_back(p); }

void pma::Element3D::ClearAssigned ( pma::Track3D *  trk = 0 )

virtual

Clear hits/points vectors of this element, optionally only those which are owned by given track.

Reimplemented in pma::Node3D.

Definition at line 50 of file PmaElement3D.cxx.

{
        fAssignedPoints.clear();
        fAssignedHits.clear();
        fHitsRadius = 0.0;
}

int pma::Element3D::Cryo ( void  ) const

inline

Cryostat index or -1 if out of any cryostat.

Definition at line 36 of file PmaElement3D.h.

{ return fCryo; }

virtual pma::Vector3D pma::Element3D::GetDirection3D ( void  ) const

pure virtual

Get 3D direction cosines corresponding to this element.

Implemented in pma::Node3D, and pma::Segment3D.

virtual double pma::Element3D::GetDistance2To ( const TVector3 &  p3d ) const

pure virtual

Distance [cm] from the 3D point to the object 3D.

Implemented in pma::Node3D, and pma::Segment3D.

virtual double pma::Element3D::GetDistance2To ( const TVector2 &  p2d, unsigned int  view  ) const

pure virtual

Distance [cm] from the 2D point to the object's 2D projection in one of wire views.

Implemented in pma::Node3D, and pma::Segment3D.

virtual TVector3 pma::Element3D::GetUnconstrainedProj3D ( const TVector2 &  p2d, unsigned int  view  ) const

pure virtual

Implemented in pma::Node3D, and pma::Segment3D.

bool pma::Element3D::HasHit ( const pma::Hit3D *  h ) const

inline

Definition at line 56 of file PmaElement3D.h.

    {
        for (const auto a : fAssignedHits) { if (h == a) return true; }
        return false;
    }

pma::Hit3D& pma::Element3D::Hit ( size_t  index )

inline

Definition at line 62 of file PmaElement3D.h.

{ return *(fAssignedHits[index]); }

const std::vector< pma::Hit3D* >& pma::Element3D::Hits ( void  ) const

inline

Definition at line 54 of file PmaElement3D.h.

{ return fAssignedHits; }

double pma::Element3D::HitsRadius3D

(

unsigned int

view

)

const

Definition at line 225 of file PmaElement3D.cxx.

{
        if (fTPC < 0)
        {
                if (!fAssignedHits.empty()) mf::LogWarning("pma::Element3D") << "Hits assigned to TPC-crossing element.";
                return 0.0F;
        }

        TVector3 mean3D(0, 0, 0);
        size_t nHits = 0;
        for (auto h : fAssignedHits)
                if (h->View2D() == view)
                        { mean3D += h->Point3D(); nHits++; }
        if (!nHits) return 0.0;
        mean3D *= (1.0 / nHits);

        double r2, maxR2 = 0.0;
        for (auto h : fAssignedHits)
                if (h->View2D() == view)
                {
                        r2 = pma::Dist2(h->Point3D(), mean3D);
                        if (r2 > maxR2) maxR2 = r2;
                }
        return sqrt(maxR2);
}

bool pma::Element3D::IsFrozen ( void  ) const

inline

Check if the vertex 3D position is fixed.

Definition at line 99 of file PmaElement3D.h.

{ return fFrozen; }

double pma::Element3D::Length ( void  ) const

inline

Definition at line 52 of file PmaElement3D.h.

{ return sqrt(Length2()); }

virtual double pma::Element3D::Length2 ( void  ) const

pure virtual

Implemented in pma::Node3D, and pma::Segment3D.

size_t pma::Element3D::NEnabledHits

(

unsigned int

view = geo::kUnknown

)

const

Definition at line 36 of file PmaElement3D.cxx.

{
        size_t n = 0;
        for (size_t i = 0; i < fAssignedHits.size(); i++)
                if (fAssignedHits[i]->IsEnabled() &&
                    ((view == geo::kUnknown) || (view == fAssignedHits[i]->View2D()))) n++;
        return n;
}

size_t pma::Element3D::NHits ( void  ) const

inline

Definition at line 74 of file PmaElement3D.h.

{ return fAssignedHits.size(); }

unsigned int pma::Element3D::NHits ( unsigned int  view ) const

inline

Definition at line 93 of file PmaElement3D.h.

{ return fNHits[view]; }

size_t pma::Element3D::NPoints ( void  ) const

inline

Definition at line 79 of file PmaElement3D.h.

{ return fAssignedPoints.size(); }

size_t pma::Element3D::NPrecalcEnabledHits ( void  ) const

inline

Definition at line 76 of file PmaElement3D.h.

{ return fNThisHitsEnabledAll; }

unsigned int pma::Element3D::NThisHits ( unsigned int  view ) const

inline

Definition at line 94 of file PmaElement3D.h.

{ return fNThisHits[view]; }

static float pma::Element3D::OptFactor ( unsigned int  view )

inlinestatic

Definition at line 106 of file PmaElement3D.h.

{ return fOptFactors[view]; }

TVector3 const& pma::Element3D::ReferencePoint ( size_t  index ) const

inline

Definition at line 78 of file PmaElement3D.h.

{ return *(fAssignedPoints[index]); }

void pma::Element3D::RemoveHitAt ( size_t  index )

inline

Definition at line 63 of file PmaElement3D.h.

        {
                if (index < fAssignedHits.size())
                        fAssignedHits.erase(fAssignedHits.begin() + index);
        }

bool pma::Element3D::SelectAllHits

(

void

)

Definition at line 297 of file PmaElement3D.cxx.

{
        bool changed = false;
        for (auto h : fAssignedHits)
        {
                changed |= !(h->IsEnabled());
                h->SetEnabled(true);
        }
        return changed;
}

bool pma::Element3D::SelectRndHits

(

size_t

nmax_per_view

)

Definition at line 251 of file PmaElement3D.cxx.

{
        if (!nmax_per_view) { return SelectAllHits(); }

        size_t nhits[3];
        for (size_t i = 0; i < 3; ++i) nhits[i] = NHits(i);

        int m[3], count[3];
        bool state[3];
        for (size_t i = 0; i < 3; ++i)
        {
                if (nhits[i] >= 2 * nmax_per_view)
                {
                        m[i] = nhits[i] / nmax_per_view;
                        state[i] = true;
                }
                else if (nhits[i] > nmax_per_view)
                {
                        m[i] = nhits[i] / (nhits[i] - nmax_per_view);
                        state[i] = false;
                }
                else { m[i] = 0; state[i] = false; }

                count[i] = 0;
        }

        bool b, changed = false;
        for (auto h : fAssignedHits)
        {
                b = h->IsEnabled();

                size_t view = h->View2D();
                if (m[view])
                {
                        if (count[view] % m[view] == 0) h->SetEnabled(state[view]);
                        else h->SetEnabled(!(state[view]));

                        ++count[view];
                }
                else h->SetEnabled(true);

                changed |= (b != h->IsEnabled());
        }
        return changed;
}

void pma::Element3D::SetFrozen ( bool  state )

inline

Fix / relese vertex 3D position.

Definition at line 101 of file PmaElement3D.h.

{ fFrozen = state; }

static void pma::Element3D::SetOptFactor ( unsigned int  view, float  value  )

inlinestatic

Definition at line 107 of file PmaElement3D.h.

{ fOptFactors[view] = value; }

virtual void pma::Element3D::SetProjection ( pma::Hit3D &  h ) const

pure virtual

Implemented in pma::Node3D, and pma::Segment3D.

void pma::Element3D::SortHits

(

void

)

Definition at line 45 of file PmaElement3D.cxx.

{
        std::sort(fAssignedHits.begin(), fAssignedHits.end(), pma::bTrajectory3DOrderLess());
}

double pma::Element3D::SumDist2

(

void

)

const

Definition at line 172 of file PmaElement3D.cxx.

{
        if (fTPC < 0)
        {
                if (!fAssignedHits.empty()) mf::LogWarning("pma::Element3D") << "Hits assigned to TPC-crossing element.";
                return 0.0F;
        }

    double hit_sum = SumDist2Hits();

        if (fAssignedPoints.size())
        {
                double d, ref_sum = 0.0F;
                for (auto p : fAssignedPoints)
                {
                        d = sqrt( GetDistance2To(*p) ) - 0.5; // guide by ref points up to ~ 3D resolution
                        if (d > 0.0) ref_sum += d * d;
                }
                if (fAssignedHits.size())
                {
                        ref_sum *= 0.2 * fAssignedHits.size() / fAssignedPoints.size();
                }
                hit_sum += ref_sum;
        }

        return hit_sum;
}

double pma::Element3D::SumDist2

(

unsigned int

view

)

const

Definition at line 200 of file PmaElement3D.cxx.

{
        if (fTPC < 0)
        {
                if (!fAssignedHits.empty()) mf::LogWarning("pma::Element3D") << "Hits assigned to TPC-crossing element.";
                return 0.0F;
        }

        double hit_sum = 0.0F;
        for (auto h : fAssignedHits)
        {
                if (h->IsEnabled())
                {
                        unsigned int hitView = h->View2D();
                        if ((view == geo::kUnknown) || (view == hitView))
                        {
                                hit_sum += OptFactor(hitView) *              // alpha_i
                                        h->GetSigmaFactor() *                    // hit_amp / hit_max_amp
                                        GetDistance2To(h->Point2D(), hitView);   // hit_to_fit_dist^2
                        }
                }
        }
        return hit_sum;
}

virtual double pma::Element3D::SumDist2Hits ( void  ) const

protectedpure virtual

Implemented in pma::Node3D, and pma::Segment3D.

double pma::Element3D::SumHitsQ ( unsigned int  view ) const

inline

Definition at line 92 of file PmaElement3D.h.

{ return fSumHitsQ[view]; }

int pma::Element3D::TPC ( void  ) const

inline

TPC index or -1 if out of any TPC.

Definition at line 34 of file PmaElement3D.h.

{ return fTPC; }

void pma::Element3D::UpdateHitParams

(

void

)

Definition at line 57 of file PmaElement3D.cxx.

{
        std::vector< pma::Hit3D* > hitsColl, hitsInd1, hitsInd2;
        for (size_t i = 0; i < 3; ++i) fNThisHitsEnabledAll = 0;
        for (auto h : fAssignedHits)
        {
                if (h->IsEnabled()) fNThisHitsEnabledAll++;
                switch (h->View2D())
                {
                        case geo::kZ: hitsColl.push_back(h); break;
                        case geo::kV: hitsInd2.push_back(h); break;
                        case geo::kU: hitsInd1.push_back(h); break;
                }
        }
        fNThisHits[0] = hitsInd1.size();
        fNThisHits[1] = hitsInd2.size();
        fNThisHits[2] = hitsColl.size();

        pma::SortedObjectBase const * chain = dynamic_cast< pma::SortedObjectBase* >(this);
        pma::Element3D* el = 0;
        for (size_t b = 0; b < chain->NextCount(); b++)
        {
                el = dynamic_cast< pma::Element3D* >(chain->Next(b));
                if (el)
                        for (auto h : el->fAssignedHits)
                        {
                                switch (h->View2D())
                                {
                                        case geo::kZ: hitsColl.push_back(h); break;
                                        case geo::kV: hitsInd2.push_back(h); break;
                                        case geo::kU: hitsInd1.push_back(h); break;
                                }
                        }
        }
        el = dynamic_cast< pma::Element3D* >(chain->Prev());
        if (el)
        {
                for (auto h : el->fAssignedHits)
                {
                        switch (h->View2D())
                        {
                                case geo::kZ: hitsColl.push_back(h); break;
                                case geo::kV: hitsInd2.push_back(h); break;
                                case geo::kU: hitsInd1.push_back(h); break;
                        }
                }
        }

        fHitsRadius = GetHitsRadius2D(hitsColl);
        double r = GetHitsRadius2D(hitsInd2);
        if (r > fHitsRadius) fHitsRadius = r;
        r = GetHitsRadius2D(hitsInd1);
        if (r > fHitsRadius) fHitsRadius = r;

        float amp, sigmaMax = 0.0F;
        fSumHitsQ[0] = 0.0; fNHits[0] = hitsInd1.size();
        for (size_t i = 0; i < hitsInd1.size(); i++)
        {
                amp = hitsInd1[i]->GetAmplitude();
                if (amp > sigmaMax) sigmaMax = amp;
                fSumHitsQ[0] += amp;
        }
        for (size_t i = 0; i < hitsInd1.size(); i++)
        {
                if (sigmaMax > 0.0F)
                {
                        amp = hitsInd1[i]->GetAmplitude();
                        if (amp > 0.0F)
                                hitsInd1[i]->SetSigmaFactor((float)sqrt(amp / sigmaMax));
                        else hitsInd1[i]->SetSigmaFactor(0.01F);
                }
                else hitsInd1[i]->SetSigmaFactor(1.0F);
        }

        sigmaMax = 0.0F;
        fSumHitsQ[1] = 0.0; fNHits[1] = hitsInd2.size();
        for (size_t i = 0; i < hitsInd2.size(); i++)
        {
                amp = hitsInd2[i]->GetAmplitude();
                if (amp > sigmaMax) sigmaMax = amp;
                fSumHitsQ[1] += amp;
        }
        for (size_t i = 0; i < hitsInd2.size(); i++)
        {
                if (sigmaMax > 0.0F)
                {
                        amp = hitsInd2[i]->GetAmplitude();
                        if (amp > 0.0F)
                                hitsInd2[i]->SetSigmaFactor((float)sqrt(amp / sigmaMax));
                        else hitsInd2[i]->SetSigmaFactor(0.01F);
                }
                else hitsInd2[i]->SetSigmaFactor(1.0F);
        }

        sigmaMax = 0.0F;
        fSumHitsQ[2] = 0.0; fNHits[2] = hitsColl.size();
        for (size_t i = 0; i < hitsColl.size(); i++)
        {
                amp = hitsColl[i]->SummedADC();
                if (amp > sigmaMax) sigmaMax = amp;
                fSumHitsQ[2] += amp;
        }
        for (size_t i = 0; i < hitsColl.size(); i++)
        {
                if (sigmaMax > 0.0F)
                {
                        amp = hitsColl[i]->SummedADC();
                        if (amp > 0.0F)
                                hitsColl[i]->SetSigmaFactor((float)sqrt(amp / sigmaMax));
                        else hitsColl[i]->SetSigmaFactor(0.01F);
                }
                else hitsColl[i]->SetSigmaFactor(1.0F);
        }
}

void pma::Element3D::UpdateProjection ( void  )

inline

Definition at line 87 of file PmaElement3D.h.

{ for (auto h : fAssignedHits) SetProjection(*h); }

Member Data Documentation

std::vector< pma::Hit3D* > pma::Element3D::fAssignedHits

protected

Definition at line 116 of file PmaElement3D.h.

std::vector< TVector3* > pma::Element3D::fAssignedPoints

protected

Definition at line 117 of file PmaElement3D.h.

int pma::Element3D::fCryo

protected

Definition at line 111 of file PmaElement3D.h.

bool pma::Element3D::fFrozen

protected

Definition at line 115 of file PmaElement3D.h.

double pma::Element3D::fHitsRadius

protected

Definition at line 122 of file PmaElement3D.h.

size_t pma::Element3D::fNHits[3]

protected

Definition at line 120 of file PmaElement3D.h.

size_t pma::Element3D::fNThisHits[3]

protected

Definition at line 118 of file PmaElement3D.h.

size_t pma::Element3D::fNThisHitsEnabledAll

protected

Definition at line 119 of file PmaElement3D.h.

float pma::Element3D::fOptFactors = { 0.2F, 0.8F, 1.0F }

staticprotected

Definition at line 124 of file PmaElement3D.h.

double pma::Element3D::fSumHitsQ[3]

protected

Definition at line 121 of file PmaElement3D.h.

int pma::Element3D::fTPC

protected

Definition at line 111 of file PmaElement3D.h.

---

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

• larreco/larreco/RecoAlg/PMAlg/PmaElement3D.h
• larreco/larreco/RecoAlg/PMAlg/PmaElement3D.cxx