PndTrackCombiner.h

Go to the documentation of this file.

/*
 * PndTrackCombiner.h
 *
 *  Created on: Jun 7, 2016
 *      Author: kibellus
 */

#ifndef PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOMBINER_H_
#define PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOMBINER_H_

#include <map>
#include <vector>
#include <iostream>

#include "PndFtsExpandedTrackCand.h"
#include "PndLineApproximation.h"
#include "PndFtsLineComparator.h"
#include "PndPlane.h"
#include "PndTrack.h"
#include "FairTrackParP.h"
#include "PndTrackCand.h"
#include "PndLineCombination.h"

#include "TLine.h"
#include "TMath.h"
#include "TClonesArray.h"

using namespace std;

class PndTrackCombiner {
public:
        static Double_t compDist;
        static Double_t compAngle;
        static Double_t compQuali;

        PndTrackCombiner(TClonesArray* a,TClonesArray *b): fComparator(compDist,compAngle), planes(a), corrHits(b), firstRun(0), angleLimit(20){}
        virtual ~PndTrackCombiner();
        void init(map<Int_t, vector<PndFtsExpandedTrackCand>> c){
                fExpandedTrackCands=c;
                trackNum=0;
                firstRun=0;
                ftsPos[0]=fts1ZPos;
                ftsPos[1]=fts2ZPos;
                ftsPos[2]=fts3ZPos;
                ftsPos[3]=fts4ZPos;
                ftsPos[4]=fts5ZPos;
                ftsPos[5]=fts6ZPos;
        }
        vector<PndLineApproximation> combine(Int_t bigLayer);
        vector<PndLineApproximation> combine(vector<PndFtsExpandedTrackCand> cands1, vector<PndFtsExpandedTrackCand> cands2, Int_t layer1, Int_t layer2);
        vector<PndLineApproximation> combine(PndFtsExpandedTrackCand c1, PndFtsExpandedTrackCand c2, Int_t layer1, Int_t layer2);
private:
        map<Int_t, vector<PndFtsExpandedTrackCand>> fExpandedTrackCands;
        PndFtsLineComparator fComparator;
        void addLine(vector<PndLineCombination> &lines, PndLineApproximation &l, Int_t max);
        void addLineOld(vector<PndLineApproximation> &lines, PndLineApproximation &l,Int_t end);
        void refitLines(vector<PndLineCombination> &lines);
        vector<PndFtsExpandedTrackCand> compareAndRefit(Int_t layer1, Int_t layer2);
        vector<PndLineApproximation> compareAndRefit(vector<PndLineApproximation> l1, vector<PndLineApproximation> l2);
        TClonesArray *planes;
        TClonesArray *corrHits;
        Int_t firstRun = 0;
        Int_t trackNum;
        Double_t angleLimit;

        Double_t fts1ZPos = 295.4;
        Double_t fts2ZPos = 327.4;
        Double_t fts3ZPos = 394.5;
        Double_t fts4ZPos = 438.5;
        Double_t fts5ZPos = 607.5;
        Double_t fts6ZPos = 747.5;
        Double_t ftsPos[6];
};

#endif /* PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOMBINER_H_ */