GFDaf.h

Go to the documentation of this file.

/* Copyright 2011, Technische Universitaet Muenchen,
Authors: Karl Bicker, Christian Hoeppner

This file is part of GENFIT.

GENFIT is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

GENFIT is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef GFDAF_H
#define GFDAF_H

#include<assert.h>
#include<cmath>
#include<GFAbsRecoHit.h>
#include<GFDafHit.h>
#include<GFKalman.h>
#include<GFTrack.h>
#include<stdlib.h>
#include<vector>

class GFDaf: GFKalman {
        public:

                GFDaf();
                ~GFDaf() { };

                void processTrack(GFTrack* trk);

                const std::vector<std::vector<std::vector<double> > > getWeights() { return fWeights; };

                void setProbCut(double prob_cut);

                void setBetas(double b1,double b2,double b3=-1.,double b4=-1.,double b5=-1.,double b6=-1.,double b7=-1.,double b8=-1.,double b9=-1.,double b10=-1.);

        private:

                std::vector<GFDafHit*> initHitsWeights(GFTrack* trk);

                std::vector<std::vector<double> > calcWeights(GFTrack* trk, double beta);

                void copySmoothing(GFTrack* source, GFTrack* target, int target_ire);

                std::vector<std::vector<std::vector<double> > > fWeights;
                std::vector<double> fBeta;
                std::map<int,double>  fchi2Cuts;

};

#endif