animaNNLSOptimizer.h

Go to the documentation of this file.

#pragma once

#include <vnl/vnl_matrix.h>
#include <vector>

#include <itkOptimizer.h>

#include <animaCholeskyDecomposition.h>
#include "AnimaOptimizersExport.h"

namespace anima
{
class ANIMAOPTIMIZERS_EXPORT NNLSOptimizer : public itk::Optimizer
{
public:
    typedef NNLSOptimizer Self;
    typedef itk::Optimizer Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    typedef Superclass::ParametersType ParametersType;

    itkNewMacro(Self)

    
    itkTypeMacro(NNLSOptimizer, Optimizer)

    
    typedef vnl_matrix <double> MatrixType;
    typedef vnl_vector <double> VectorType;

    void StartOptimization() ITK_OVERRIDE;

    void SetDataMatrix(const MatrixType &data) {m_DataMatrix = data;}
    void SetPoints(const ParametersType &points) {m_Points = points;}

    double GetCurrentResidual();

    itkSetMacro(SquaredProblem, bool)

protected:
    NNLSOptimizer()
    {
        m_SquaredProblem = false;
    }

    virtual ~NNLSOptimizer() ITK_OVERRIDE {}

private:
    ITK_DISALLOW_COPY_AND_ASSIGN(NNLSOptimizer);

    unsigned int UpdateProcessedIndexes();
    void ComputeSPVector();
    void ComputeWVector();

    MatrixType m_DataMatrix;
    ParametersType m_Points;

    static const double m_EpsilonValue;

    bool m_SquaredProblem;

    // Working values
    std::vector <unsigned short> m_TreatedIndexes;
    std::vector <unsigned int> m_ProcessedIndexes;
    std::vector <double> m_WVector;
    VectorType m_SPVector;
    MatrixType m_DataMatrixP;

    anima::CholeskyDecomposition m_CholeskySolver;
};

} // end of namespace anima