FitContext.h

Go to the documentation of this file.

#ifndef FITCONTEXT_H_
#define FITCONTEXT_H_
//_____________________________________________________________________________

#include "TVectorD.h"

#include "CandFitTrackSA/ConvergenceMaster.h"
#include "CandFitTrackSA/DataFT.h"
#include "CandFitTrackSA/FitResult.h"
#include "CandFitTrackSA/MatrixCalculator.h"

class AlgConfig;
class FitState;
class NtpFitSA;
class TrackEstimator;

class FitContext {

public:
    FitContext( const AlgConfig& algcfg,    const TrackContext& context, 
                TrackEstimator* estimator,  SwimSwimmer* swimmer         );
    ~FitContext();
    
    void        Iterate();
    
    void        Print();
    void        PrintCurrentFit();
    
    void        Config(const AlgConfig& ac);
    
    FitState*   GetState() const;
    void        SetState(const FitState* state);

    void        SetFitParams(const TVectorD& fit);
    void        SetLastGoodFitParams(const TVectorD& fit); 
    
    void        SetFromLastGoodFit(); 
    
    Float_t     GetPrange() { return fTrackContext.GetPrange(); };
    
    NtpFitSA    MakeNtpFitSA() const;        
    
    void        FillNtpFitSA(NtpFitSA& ntp) const;   
    void        FillNtpPlaneInfo(NtpFitSA& ntp) const;
    void        FillNtpBFieldCalib(NtpFitSA& ntp) const;
    void        FillNtpFitSR(NtpFitSA& ntp) const;

    const DataFT*   GetData() const { return &fData; };
    
    Double_t    GetCpu() const { return fCpu; };
    void        SetCpu(Double_t cpu) { fCpu = cpu; };
    
//private:

    const TrackContext&   fTrackContext;
    
    const FitState*       fState;
    
    SwimSwimmer*          fSwimmer;
    
    TrackEstimator*       fEstimator;
    
    DataFT          fData;
    
    ConvergenceMaster fConvergenceMaster;
    
    MatrixCalculator fMatCalc;
    
    Int_t           fNIterationsStep;
    Int_t           fNIterationsTotal;
    
    //Int_t          fNPlanesToFit;
    Int_t           fNPlanesFit;
         
    TVectorD        fCurrentFit;

    Int_t           fTimesConverged;
        
    FitResult       fLastGoodFit;

    Double_t        fDChi2;
    
    Int_t           fNTriesDiverges;
    
    Int_t           fNHitsInViewMin;
    
    Int_t           fNMaxIterations;
    
    Int_t           fNMaxDiverging;
    
    Double_t        fConvergenceCond;
    
    Double_t        fCpu;
};

#endif // FITCONTEXT_H_

---