nmrLSNonLinSolver.h

Go to the documentation of this file.

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-    */
/* ex: set filetype=cpp softtabstop=4 shiftwidth=4 tabstop=4 cindent expandtab: */

/*
  
  Author(s):    Ankur Kapoor
  Created on: 2004-10-26

  (C) Copyright 2004-2007 Johns Hopkins University (JHU), All Rights
  Reserved.

--- begin cisst license - do not edit ---

This software is provided "as is" under an open source license, with
no warranty.  The complete license can be found in license.txt and
http://www.cisst.org/cisst/license.txt.

--- end cisst license ---
*/


#ifndef _nmrLSNonLinSolver_h
#define _nmrLSNonLinSolver_h


#include <cisstVector/vctDynamicVector.h>
#include <cisstVector/vctDynamicMatrix.h>

#include <cisstNumerical/nmrCallBack.h>

//include <cnetlib.h>

class nmrLSNonLinSolver {
    // we have this class so that we reserve memory only one
    // would help if svd of a same size matrix (or a matrix)
    // that doesnt change much is desired.

protected:
    CISSTNETLIB_INTEGER M;
    CISSTNETLIB_INTEGER N;
    CISSTNETLIB_DOUBLE Tolerance;
    CISSTNETLIB_INTEGER Info;
    CISSTNETLIB_INTEGER Lwork;
    vctDynamicVector<CISSTNETLIB_INTEGER> IWork;
    vctDynamicVector<CISSTNETLIB_DOUBLE> Work;

public:
    nmrLSNonLinSolver(void)
    {
        Allocate(0, 0);
    }


    nmrLSNonLinSolver(CISSTNETLIB_INTEGER m, CISSTNETLIB_INTEGER n)
    {
        Allocate(m, n);
    }

    
    nmrLSNonLinSolver(vctDynamicVector<CISSTNETLIB_DOUBLE> &X, vctDynamicVector<CISSTNETLIB_DOUBLE> &F) {
        Allocate(X, F);
    }


    inline void Allocate(CISSTNETLIB_INTEGER m, CISSTNETLIB_INTEGER n) {
        M = m;
        N = n;
        Lwork = M * N + 5 * N + M;
        Tolerance = 1e-6;
        Work.SetSize(Lwork);
        IWork.SetSize(N);
    }


    inline void Allocate(vctDynamicVector<CISSTNETLIB_DOUBLE> &X, vctDynamicVector<CISSTNETLIB_DOUBLE> &F) {
        Allocate(X.size(), F.size());
    }


    template <int __instanceLine, class __elementType>
    inline void Solve(nmrCallBackFunctionF<__instanceLine, __elementType> &callBack, vctDynamicVector<CISSTNETLIB_DOUBLE> &X,
                      vctDynamicVector<CISSTNETLIB_DOUBLE> &F, CISSTNETLIB_DOUBLE tolerance) throw (std::runtime_error) {
        if ((N != static_cast<CISSTNETLIB_INTEGER>(X.size())) || (M != static_cast<CISSTNETLIB_INTEGER>(F.size()))) {
            cmnThrow(std::runtime_error("nmrLSNonLinSolver Solve: Size used for Allocate was different"));
        }
        Tolerance = tolerance;
        /* call the MINPACK C function */
        lmdif1_((U_fp)callBack.FunctionFlmdif, &M, &N,
                X.Pointer(), F.Pointer(), 
                &Tolerance, &Info,
                IWork.Pointer(), Work.Pointer(), &Lwork);
    }
    
};


#endif // _nmrLSNonLinSolver_h