Skip to content
Snippets Groups Projects
minimize.h 2.29 KiB
Newer Older
Mark Woolrich's avatar
Mark Woolrich committed
/*  minimize
 
    Tim Behrens, FMRIB Image Analysis Group

    Copyright (C) 1999-2000 University of Oxford  */

/*  CCOPYRIGHT  */

#if !defined(minimize_h)
#define minimize_h

#include <string>
#include <iostream>
#include <fstream>
#include <unistd.h>
#include <vector>
#include <algorithm>
#include "newmatap.h"
#include "newmatio.h"
Mark Woolrich's avatar
Mark Woolrich committed


#define WANT_STREAM
#define WANT_MATH

using namespace MISCMATHS;
using namespace NEWMAT;
using namespace std;
///////////////////////////////////////////////////////

//fminsearch.m

namespace MISCMATHS {

class pair_comparer
{
public:
  bool operator()(const pair<float,ColumnVector>& p1,const pair<float,ColumnVector>& p2) const
  {
    return p1.first < p2.first;
  }    
};

class EvalFunction
Mark Woolrich's avatar
Mark Woolrich committed
{//Function where gradient is not analytic (or you are too lazy to work it out) (required for fminsearch)
Mark Woolrich's avatar
Mark Woolrich committed
public:
  EvalFunction(){}
Tim Behrens's avatar
Tim Behrens committed
  virtual float evaluate(const ColumnVector& x) const = 0; //evaluate the function
Mark Woolrich's avatar
Mark Woolrich committed
  virtual ~EvalFunction(){};

Mark Woolrich's avatar
Mark Woolrich committed
private:    
Mark Woolrich's avatar
Mark Woolrich committed
  const EvalFunction& operator=(EvalFunction& par);
  EvalFunction(const EvalFunction&);
};

Tim Behrens's avatar
Tim Behrens committed
class gEvalFunction : public EvalFunction
{//Function where gradient is analytic (required for scg)
public:
  gEvalFunction() : EvalFunction(){}
  // evaluate is inherited from EvalFunction
Mark Woolrich's avatar
Mark Woolrich committed
  
  virtual ReturnMatrix g_evaluate(const ColumnVector& x) const = 0; //evaluate the gradient
Tim Behrens's avatar
Tim Behrens committed
  virtual ~gEvalFunction(){};

private:
    
  const gEvalFunction& operator=(gEvalFunction& par);
  gEvalFunction(const gEvalFunction&);
};


float diff1(const ColumnVector& x, const EvalFunction& func, int i,float h,int errorord=4);// finite diff derivative
Tim Behrens's avatar
Tim Behrens committed
float diff2(const ColumnVector& x, const EvalFunction& func, int i,float h,int errorord=4);// finite diff 2nd derivative
Tim Behrens's avatar
Tim Behrens committed
float diff2(const ColumnVector& x, const EvalFunction& func, int i,int j,float h,int errorord=4);// finite diff cross derivative

Mark Woolrich's avatar
Mark Woolrich committed
ReturnMatrix gradient(const ColumnVector& x, const EvalFunction& func,float h,int errorord=4);// finite diff derivative vector 

Tim Behrens's avatar
Tim Behrens committed
ReturnMatrix hessian(const ColumnVector& x, const EvalFunction& func,float h,int errorord=4);// finite diff hessian
Mark Woolrich's avatar
Mark Woolrich committed
void fminsearch(ColumnVector& x, const EvalFunction& func);
Mark Woolrich's avatar
Mark Woolrich committed
void scg(ColumnVector& x, const gEvalFunction& func, float tol = 0.0001, float eps=1e-16, int niters=100);
Tim Behrens's avatar
Tim Behrens committed
}