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