GFGbl.h

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

/* Copyright 2013
 *   Authors: Sergey Yashchenko and Tadeas Bilka
 *
 *   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 GFGBL_H
#define GFGBL_H

#include "GblTrajectory.h"
#include "AbsFitter.h"

#include <map>
#include <iostream>

#include <TMatrixD.h>
#include <assert.h>
#include <sstream>

#include <TMath.h>
#include <TVector3.h>


namespace genfit {
  
  
  class GFGbl : public AbsFitter {
    
  private:
    GFGbl(const GFGbl&);
    GFGbl& operator=(GFGbl const&);
    
    std::string m_milleFileName;
    std::string m_gblInternalIterations;
    double m_pValueCut;
    int m_minNdf;
    double m_chi2Cut;
    bool m_enableScatterers;
    bool m_enableIntermediateScatterer;
    
    
  public:
    
    GFGbl();
    
    virtual ~GFGbl() {;}
    
    void beginRun();
    
    void endRun();
    
    
    void setGBLOptions(std::string internalIterations = "THC", bool enableScatterers = true, bool enableIntermediateScatterer = true) {
      m_gblInternalIterations = internalIterations;
      if (!enableScatterers)
        enableIntermediateScatterer = false;
      m_enableScatterers = enableScatterers;
      m_enableIntermediateScatterer = enableIntermediateScatterer;
    }
    
    void setMP2Options(double pValueCut = 0., unsigned int minNdf = 1, std::string mille_file_name = "millefile.dat", double chi2Cut = 0.) {
      m_pValueCut = pValueCut;
      m_minNdf = minNdf;
      m_milleFileName = mille_file_name;
      m_chi2Cut = chi2Cut;
    }
    
    void processTrackWithRep(Track* trk, const AbsTrackRep* rep, bool resortHits = false) override;
    
    
  public:
    
    ClassDef(GFGbl, 1)
    
  };
  
}  /* End of namespace genfit */
#endif // GFGBL_H