GFDaf.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GFDaf.h

/* 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() { 
      std::cout<<"Warning: Using deprecated GFDaf::getWeights()! The weights of the hits are accessible in the bookkeeping of the track which was fitted, the key is \"dafWeight\""<<std::endl;
      return fWeights; 
    };

    void setProbCut(double prob_cut);

    void setBetas(double b1,double b2=-1,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);

    void saveWeights(GFTrack* trk, const std::vector<std::vector<std::vector<double> > >& weights) const;

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

};

#endif