PndTrackCollection.h

Go to the documentation of this file.

/*
 * PndTrackCollection.h
 *
 *  Created on: Jul 28, 2016
 *      Author: kibellus
 */

#ifndef PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOLLECTION_H_
#define PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOLLECTION_H_

#include <vector>

#include "PndLineApproximation.h"

using namespace std;

class PndTrackCollection {
public:
        PndTrackCollection();
        virtual ~PndTrackCollection();
        void add(PndLineApproximation l,Bool_t skewed);
        PndLine getCurrLine(){return fCurrLine;}
        PndLine getLastLine(){return (*fLines)[fLines->size()-1].getLine();}
        vector<PndLineApproximation> *getLines(){return fLines;}
        PndTrack getPndTrack(map<Int_t,PndFtsHit*> orgHits);
        vector<PndFtsHit*> getHits(){
                vector<PndFtsHit*> result;
                for(size_t i=0;i<fLines->size();i++){
                        for(size_t j=0;j<(*fLines)[i].getHits().size();j++){
                                result.push_back((*fLines)[i].getHits()[j]);
                        }
                }
                return result;
        }
        vector<PndFtsHit*> fHits;
        void addHit(PndFtsHit* h){fHits.push_back(h);}
        Double_t getDistTo(PndLine l,Int_t layer);
private:
        std::vector<PndLineApproximation> *fLines;
        PndLine fCurrLine;
        PndFtsHit* copyHit(PndFtsHit *h);
        void refitAllHits();
        void refitAllTracks();
        void refitHit(PndLine &l,PndFtsHit* hit);
};

#endif /* PNDTOOLS_PNDFORWARDTRACKFINDER_PNDTRACKCOLLECTION_H_ */