oOptimizerManager.hh

Go to the documentation of this file.

/*
This file is part of CASToR.

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

    CASToR 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 General Public License for more
    details.

    You should have received a copy of the GNU General Public License along with
    CASToR (in file GNU_GPL.TXT). If not, see <http://www.gnu.org/licenses/>.

Copyright 2017-2018 all CASToR contributors listed below:

    --> current contributors: Thibaut MERLIN, Simon STUTE, Didier BENOIT, Claude COMTAT, Marina FILIPOVIC, Mael MILLARDET
    --> past contributors: Valentin VIELZEUF

This is CASToR version 2.0.
*/

#ifndef OOPTIMIZERMANAGER_HH
#define OOPTIMIZERMANAGER_HH 1

#include "gVariables.hh"
#include "vOptimizer.hh"
#include "vPenalty.hh"
#include "oImageSpace.hh"
#include "vDataFile.hh"

class oOptimizerManager
{
  // -------------------------------------------------------------------
  // Constructor & Destructor
  public:
    oOptimizerManager();
    ~oOptimizerManager();


  // -------------------------------------------------------------------
  // Public member functions
  public:
    int CheckParameters();
    int Initialize();
    int PreDataUpdateStep(int a_iteration, int a_nbIterations, int a_subset, int* ap_nbSubsets);
    int PostDataUpdateStep(int a_iteration, int a_nbIterations, int a_subset, int* ap_nbSubsets);
    int DataUpdateStep( oProjectionLine* ap_Line, vEvent* ap_Event,
                        int a_bed, int a_timeFrame, int a_respGate, int a_cardGate,
                        int a_iteration, int a_thread );
    int ImageUpdateStep( int a_iteration, int a_nbSubsets );


  // -------------------------------------------------------------------
  // Private member functions
  private:
    int ParseOptionsAndInitializeOptimizerAndPenalty();


  // -------------------------------------------------------------------
  // Public Get & Set functions
  public:
    inline void SetVerbose(int a_verboseLevel)
           {m_verbose = a_verboseLevel;}
    inline void SetOptionsOptimizer(const string& a_optionsOptimizer)
           {m_optionsOptimizer = a_optionsOptimizer;}
    inline void SetOptionsPenalty(const string& a_optionsPenalty)
           {m_optionsPenalty = a_optionsPenalty;}
    inline void SetImageDimensionsAndQuantification(oImageDimensionsAndQuantification* ap_ImageDimensionsAndQuantification)
           {mp_ImageDimensionsAndQuantification = ap_ImageDimensionsAndQuantification;}
    inline void SetImageSpace(oImageSpace* ap_ImageSpace)
           {mp_ImageSpace = ap_ImageSpace;}
    inline void SetNbTOFBins(int a_nbTOFBins)
           {m_nbTOFBins = a_nbTOFBins;}
    inline void SetDataMode(int a_dataMode)
           {m_dataMode = a_dataMode;}
    inline void SetDataType(int a_dataType)
           {m_dataType = a_dataType;}
    inline void SetDataSpec(int a_dataSpec)
           {m_dataSpec = a_dataSpec;}
    inline void SetOptimizerFOMFlag(bool a_optimizerFOMFlag)
           {m_optimizerFOMFlag = a_optimizerFOMFlag;}
    inline void SetOptimizerImageStatFlag(bool a_optimizerImageStatFlag)
           {m_optimizerImageStatFlag = a_optimizerImageStatFlag;}
    inline int GetNbBackwardImages()
           {return mp_Optimizer->GetNbBackwardImages();}
    inline FLTNB GetInitialValue()
           {return mp_Optimizer->GetInitialValue();}


  // -------------------------------------------------------------------
  // Data members
  private:
    oImageDimensionsAndQuantification* 
      mp_ImageDimensionsAndQuantification; 
    oImageSpace* mp_ImageSpace;                 
    int m_dataMode;                        
    int m_dataType;                        
    int m_dataSpec;                        
    int m_nbTOFBins;                       
    string m_optionsOptimizer;             
    string m_optionsPenalty;               
    bool m_optimizerFOMFlag;               
    bool m_optimizerImageStatFlag;         
    vOptimizer* mp_Optimizer;              
    vPenalty* mp_Penalty;                  
    int m_verbose;                         
};

#endif