PndLineApproximation Class Reference

#include <PndLineApproximation.h>

Public Member Functions
	PndLineApproximation (vector< PndFtsHit * > hits)
	PndLineApproximation ()
	PndLineApproximation (PndLine lineApprox, vector< PndFtsHit * > correctedHits)
	PndLineApproximation (PndLine lineApprox, vector< PndFtsHit * > ch1, vector< PndFtsHit * > ch2)
virtual	~PndLineApproximation ()
PndLine &	getLine ()
void	setLine (PndLine l)
vector< PndFtsHit * >	getHits ()
void	setHits (vector< PndFtsHit * > h)
PndLineApproximation	newApproximation (PndLineApproximation &approx2)
PndLine	linearRegression (vector< PndFtsHit * > hits)
PndLine	linearRegressionXZ (vector< PndFtsHit * > hits)
PndLine	linearRegressionZX (vector< PndFtsHit * > hits)
PndLine	linearRegressionYZ (vector< PndFtsHit * > hits)
PndLine	linearRegressionZY (vector< PndFtsHit * > hits)
PndTrack	plot (Double_t zVal1, Double_t zVal2, TClonesArray *hitArr)
void	correctHits3DAndAdd (vector< PndFtsHit * > correctedHits)
void	setYInfo (Bool_t i)
Bool_t	HasYInfo ()

Private Member Functions
TVector2	linearRegression (Double_t matrix[2][3])
PndFtsHit *	copyHitWithNewPosition (PndFtsHit *h, TVector3 pos)

Private Attributes
PndLine	fLineApprox
vector< PndFtsHit * >	fCorrectedHits
Bool_t	hasYInfo

Detailed Description

Definition at line 23 of file PndLineApproximation.h.

Constructor & Destructor Documentation

PndLineApproximation::PndLineApproximation ( vector< PndFtsHit * >  hits )

inline

Definition at line 25 of file PndLineApproximation.h.

: fLineApprox(linearRegression(hits)), fCorrectedHits(hits), hasYInfo(kTRUE){}

PndLineApproximation::PndLineApproximation ( )

inline

Definition at line 26 of file PndLineApproximation.h.

{}

PndLineApproximation::PndLineApproximation ( PndLine  lineApprox, vector< PndFtsHit * >  correctedHits  )

inline

Definition at line 27 of file PndLineApproximation.h.

                :fLineApprox(lineApprox),fCorrectedHits(correctedHits) {}

PndLineApproximation::PndLineApproximation	(	PndLine	lineApprox,
		vector< PndFtsHit * >	ch1,
		vector< PndFtsHit * >	ch2
	)

Definition at line 15 of file PndLineApproximation.cxx.

References correctHits3DAndAdd(), and fLineApprox.

                                                                                                            {
        fLineApprox = lineApprox;
        correctHits3DAndAdd(ch1);
        correctHits3DAndAdd(ch2);
}

PndLineApproximation::~PndLineApproximation ( )

virtual

Definition at line 11 of file PndLineApproximation.cxx.

                                            {
        // TODO Auto-generated destructor stub
}

Member Function Documentation

PndFtsHit * PndLineApproximation::copyHitWithNewPosition ( PndFtsHit *  h, TVector3  pos  )

private

Definition at line 192 of file PndLineApproximation.cxx.

References PndFtsHit::GetChamberID(), PndFtsHit::GetLayerID(), PndFtsHit::GetTubeID(), h2, PndFtsHit::SetChamberID(), PndFtsHit::SetLayerID(), and PndFtsHit::SetTubeID().

Referenced by correctHits3DAndAdd().

                                                                                 {
        PndFtsHit* h2 = new PndFtsHit();
        h2->SetDetectorID(h->GetDetectorID());
        h2->SetTubeID(h->GetTubeID());
        h2->SetChamberID(h->GetChamberID());
        h2->SetLayerID(h->GetLayerID());
        h2->SetXYZ(pos[0],pos[1],pos[2]);
        h2->SetRefIndex(h->GetRefIndex());
        h2->SetEntryNr(h->GetEntryNr());
        return h2;
}

void PndLineApproximation::correctHits3DAndAdd

(

vector< PndFtsHit * >

correctedHits

)

Definition at line 21 of file PndLineApproximation.cxx.

References Bool_t, copyHitWithNewPosition(), CAMath::Cos(), fCorrectedHits, fLineApprox, PndLine::getPerpendicular(), i, and CAMath::Sin().

Referenced by PndWayFollower::getBestSkewedApproxTo(), newApproximation(), and PndLineApproximation().

                                                                   {
        for (size_t i = 0; i < ch.size(); i++) {
                //check to duplicate hits
                Bool_t duplicated = kFALSE;
                for(size_t j=0;j<fCorrectedHits.size();j++){
                        Int_t id1 = fCorrectedHits[j]->GetTubeID();
                        Int_t id2 = ch[i]->GetTubeID();
                        if(id1==id2){
                                duplicated = kTRUE;
                                break;
                        }
                }
                if(duplicated) continue;
                Int_t layer = (ch[i]->GetLayerID() - 1) / 2;
                TVector3 base2;
                ch[i]->Position(base2);
                TVector3 dir2;
                if (layer % 4 == 1) {
                        dir2 = TVector3(TMath::Sin(-5 * TMath::DegToRad()),
                                        TMath::Cos(-5 * TMath::DegToRad()), 0);
                } else if (layer % 4 == 2) {
                        dir2 = TVector3(TMath::Sin(5 * TMath::DegToRad()),
                                        TMath::Cos(5 * TMath::DegToRad()), 0);
                } else {
                        dir2 = TVector3(0, 1, 0);
                }
                PndLine line2(base2, dir2);
                TVector3 newPos = line2.getPerpendicular(fLineApprox);
                PndFtsHit* newHit = copyHitWithNewPosition(ch[i], newPos);
                fCorrectedHits.push_back(newHit);
        }
}

vector<PndFtsHit*> PndLineApproximation::getHits ( )

inline

Definition at line 33 of file PndLineApproximation.h.

Referenced by PndTrackCollection::add(), PndWayFollower::getBestApproxTo(), PndWayFollower::getBestSkewedApproxTo(), PndTrackCollection::getDistTo(), PndTrackCollection::getPndTrack(), PndForwardTrackFinderTask::saveCombined(), PndForwardTrackFinderTask::saveCorrectedTracklets(), and PndForwardTrackFinderTask::saveTrackCollection().

{return fCorrectedHits;}

PndLine& PndLineApproximation::getLine ( )

inline

Definition at line 31 of file PndLineApproximation.h.

Referenced by PndTrackCollection::add(), PndFtsLineApproximator::addLine(), PndTrackCombiner::addLine(), PndTrackCombiner::addLineOld(), PndTrackCombiner::combine(), PndModuleCombiner::combineModules(), PndTrackCombiner::compareAndRefit(), PndFtsLineApproximator::createExpandedTrackCand(), PndForwardTrackFinderTask::createStatictcs(), PndForwardTrackFinderTask::getBest(), PndWayFollower::getBestApproxTo(), PndWayFollower::getBestSkewedApproxTo(), PndTrackCollection::getDistTo(), PndTrackCollection::refitAllTracks(), PndForwardTrackFinderTask::saveCombined(), PndForwardTrackFinderTask::saveCorrectedTracklets(), and PndForwardTrackFinderTask::saveTrackCollection().

{return fLineApprox;}

Bool_t PndLineApproximation::HasYInfo ( )

inline

Definition at line 44 of file PndLineApproximation.h.

Referenced by PndModuleCombiner::combineModules().

{return hasYInfo;}

PndLine PndLineApproximation::linearRegression

(

vector< PndFtsHit * >

hits

)

Definition at line 69 of file PndLineApproximation.cxx.

References PndLine::getDir(), PndLine::getP1(), linearRegressionZX(), linearRegressionZY(), p1, and p2.

Referenced by linearRegressionXZ(), linearRegressionYZ(), linearRegressionZX(), linearRegressionZY(), and newApproximation().

                                                                     {
        PndLine zxLine = linearRegressionZX(hits);
        PndLine zyLine = linearRegressionZY(hits);
        PndPlane p1(zxLine.getP1(),zxLine.getDir(),TVector3(0,1,0));
        PndPlane p2(zyLine.getP1(),zyLine.getDir(),TVector3(1,0,0));
        PndLine newLine = p1.getIntersection(p2);
        return newLine;
}

TVector2 PndLineApproximation::linearRegression ( Double_t  matrix[2][3] )

private

Definition at line 182 of file PndLineApproximation.cxx.

References a, b, Double_t, and i.

                                                                    {
        Double_t fac = -matrix[1][0]/matrix[0][0];
        for(int i=0;i<3;i++){
                matrix[1][i]+= (fac*matrix[0][i]);
        }
        Double_t b = matrix[1][2]/matrix[1][1];
        Double_t a = (matrix[0][2] - b*matrix[0][1])/matrix[0][0];
        return TVector2(a,b);
}

PndLine PndLineApproximation::linearRegressionXZ

(

vector< PndFtsHit * >

hits

)

Definition at line 78 of file PndLineApproximation.cxx.

References Double_t, i, linearRegression(), p1, and p2.

                                                                       {
        Double_t sumX = 0;
        Double_t sumX2 = 0;
        Double_t sumZ = 0;
        Double_t sumXZ = 0;
        for(size_t i=0;i<hits.size();i++){
                cout.precision(10);
                sumX+=hits[i]->GetX();
                sumX2+=hits[i]->GetX()*hits[i]->GetX();
                sumZ+=hits[i]->GetZ();
                sumXZ+=hits[i]->GetX()*hits[i]->GetZ();
        }
        Double_t matrix[2][3];
        matrix[0][0]=hits.size();
        matrix[1][0]=sumX;
        matrix[0][1]=sumX;
        matrix[1][1]=sumX2;
        matrix[0][2]=sumZ;
        matrix[1][2]=sumXZ;
        TVector2 result = linearRegression(matrix);
        TVector3 p1(20,0,result.X()+20*result.Y());
        TVector3 p2(40,0,result.X()+40*result.Y());
        //TVector3 p1(hits[0]->GetX(),0,result.X()+hits[0]->GetX()*result.Y());
        //TVector3 p2(hits[0]->GetX()+20,0,result.X()+(hits[0]->GetX())+20*result.Y());
        TVector3 dir = p1-p2;
        return PndLine(p1,dir);
}

PndLine PndLineApproximation::linearRegressionYZ

(

vector< PndFtsHit * >

hits

)

Definition at line 131 of file PndLineApproximation.cxx.

References Double_t, i, linearRegression(), p1, and p2.

                                                                       {
        Double_t sumY = 0;
        Double_t sumY2 = 0;
        Double_t sumZ = 0;
        Double_t sumYZ = 0;
        for(size_t i=0;i<hits.size();i++){
                sumY+=hits[i]->GetY();
                sumY2+=hits[i]->GetY()*hits[i]->GetY();
                sumZ+=hits[i]->GetZ();
                sumYZ+=hits[i]->GetY()*hits[i]->GetZ();
        }
        Double_t matrix[2][3];
        matrix[0][0]=hits.size();
        matrix[1][0]=sumY;
        matrix[0][1]=sumY;
        matrix[1][1]=sumY2;
        matrix[0][2]=sumZ;
        matrix[1][2]=sumYZ;
        TVector2 result = linearRegression(matrix);
        TVector3 p1(0,-20,result.X()-20*result.Y());
        TVector3 p2(0,20,result.X()+20*result.Y());
        TVector3 dir = p1-p2;
        return PndLine(p1,dir);
}

PndLine PndLineApproximation::linearRegressionZX

(

vector< PndFtsHit * >

hits

)

Definition at line 106 of file PndLineApproximation.cxx.

References Double_t, i, linearRegression(), p1, and p2.

Referenced by linearRegression().

                                                                       {
        Double_t sumZ = 0;
        Double_t sumZ2 = 0;
        Double_t sumX = 0;
        Double_t sumZX = 0;
        for(size_t i=0;i<hits.size();i++){
                sumZ+=hits[i]->GetZ();
                sumZ2+=hits[i]->GetZ()*hits[i]->GetZ();
                sumX+=hits[i]->GetX();
                sumZX+=hits[i]->GetZ()*hits[i]->GetX();
        }
        Double_t matrix[2][3];
        matrix[0][0]=hits.size();
        matrix[1][0]=sumZ;
        matrix[0][1]=sumZ;
        matrix[1][1]=sumZ2;
        matrix[0][2]=sumX;
        matrix[1][2]=sumZX;
        TVector2 result = linearRegression(matrix);
        TVector3 p1(result.X()+hits[0]->GetZ()*result.Y(),0,hits[0]->GetZ());
        TVector3 p2(result.X()+(hits[0]->GetZ()+50)*result.Y(),0,hits[0]->GetZ()+50);
        TVector3 dir = p1-p2;
        return PndLine(p1,dir);
}

PndLine PndLineApproximation::linearRegressionZY

(

vector< PndFtsHit * >

hits

)

Definition at line 156 of file PndLineApproximation.cxx.

References Double_t, i, linearRegression(), p1, and p2.

Referenced by linearRegression().

                                                                       {
        Double_t sumZ = 0;
        Double_t sumZ2 = 0;
        Double_t sumY = 0;
        Double_t sumZY = 0;
        for(size_t i=0;i<hits.size();i++){
                sumZ+=hits[i]->GetZ();
                sumZ2+=hits[i]->GetZ()*hits[i]->GetZ();
                sumY+=hits[i]->GetY();
                sumZY+=hits[i]->GetZ()*hits[i]->GetY();
        }
        Double_t matrix[2][3];
        matrix[0][0]=hits.size();
        matrix[1][0]=sumZ;
        matrix[0][1]=sumZ;
        matrix[1][1]=sumZ2;
        matrix[0][2]=sumY;
        matrix[1][2]=sumZY;
        TVector2 result = linearRegression(matrix);
        TVector3 p1(0,result.X()+hits[0]->GetZ()*result.Y(),hits[0]->GetZ());
        TVector3 p2(0,result.X()+(hits[0]->GetZ()+300)*result.Y(),hits[0]->GetZ()+300);
        TVector3 dir = p1-p2;
        return PndLine(p1,dir);
}

PndLineApproximation PndLineApproximation::newApproximation

(

PndLineApproximation &

approx2

)

Definition at line 54 of file PndLineApproximation.cxx.

References correctHits3DAndAdd(), fCorrectedHits, fLineApprox, PndLine::getRating(), hits, i, linearRegression(), and PndLine::setRating().

Referenced by PndTrackCombiner::combine(), PndModuleCombiner::combineModules(), PndTrackCombiner::compareAndRefit(), PndFtsLineApproximator::createExpandedTrackCand(), and PndTrackCombiner::refitLines().

                                                                                        {
        vector<PndFtsHit*> hits;
        for(size_t i=0;i<fCorrectedHits.size();i++)
                hits.push_back(fCorrectedHits[i]);
        for(size_t i=0;i<approx2.fCorrectedHits.size();i++){
                hits.push_back(approx2.fCorrectedHits[i]);
        }
        PndLine newLine = linearRegression(hits);
        newLine.setRating(fLineApprox.getRating()+approx2.fLineApprox.getRating());
        vector<PndFtsHit*> hits2;
        PndLineApproximation newApprox(newLine,hits2);
        newApprox.correctHits3DAndAdd(hits);
        return newApprox;
}

PndTrack PndLineApproximation::plot	(	Double_t	zVal1,
		Double_t	zVal2,
		TClonesArray *	hitArr
	)

Definition at line 204 of file PndLineApproximation.cxx.

References PndTrackCand::AddHit(), fCorrectedHits, fLineApprox, i, PndLine::plot(), t, and v.

                                                                                       {
        FairTrackParP tp1 = fLineApprox.plot(zVal1,zVal2);
        TVector3 v(1, 1, 1);
        FairTrackParP tp2(v, 0 * v, v, v, 1, v, v, v);
        PndTrackCand cand;
        Int_t j = hitArr->GetEntries();
        for(size_t i=0;i<fCorrectedHits.size();i++){
                fCorrectedHits[i]->SetEntryNr(FairLink(-1, FairRootManager::Instance()->GetEntryNr(),FairRootManager::Instance()->GetBranchId("CorrectedHits"), j));
                new ((*hitArr)[j++]) PndFtsHit(*fCorrectedHits[i]);//PndFtsHit* myHit =  //[R.K.03/2017] unused variable
                cand.AddHit(fCorrectedHits[i]->GetEntryNr(),i);
        }
        PndTrack t(tp1,tp2,cand);
        return t;
}

void PndLineApproximation::setHits ( vector< PndFtsHit * >  h )

inline

Definition at line 34 of file PndLineApproximation.h.

References h.

Referenced by PndTrackCollection::add().

{fCorrectedHits = h;}

void PndLineApproximation::setLine ( PndLine  l )

inline

Definition at line 32 of file PndLineApproximation.h.

{fLineApprox = l;}

void PndLineApproximation::setYInfo ( Bool_t  i )

inline

Definition at line 43 of file PndLineApproximation.h.

References i.

Referenced by PndTrackCombiner::combine().

{hasYInfo=i;}

Member Data Documentation

vector<PndFtsHit*> PndLineApproximation::fCorrectedHits

private

Definition at line 48 of file PndLineApproximation.h.

Referenced by correctHits3DAndAdd(), newApproximation(), and plot().

PndLine PndLineApproximation::fLineApprox

private

Definition at line 47 of file PndLineApproximation.h.

Referenced by correctHits3DAndAdd(), newApproximation(), plot(), and PndLineApproximation().

Bool_t PndLineApproximation::hasYInfo

private

Definition at line 50 of file PndLineApproximation.h.

---

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

• PndLineApproximation.h
• PndLineApproximation.cxx