testmisfit.cpp

Go to the documentation of this file.

#include "CTestMisfit.h"
#include "CVariableMetric.h"
#include "CSimpleLineSearch.h"
#include <iostream>
#include <vector>
#include <fstream>

using namespace std;
namespace ublas = boost::numeric::ublas;
int main()
{
        const int ndata = 1;
        const int nmodel = 2;
        const int maxit = 100;
        
        ofstream ModelFile("models.out");
        CTestMisfit MyMisfit;
        CSimpleLineSearch LineSearch;
        CVariableMetric Minimizer(ndata,nmodel);
        ublas::vector<double> StartData(ndata);
        ublas::vector<double> StartModel(nmodel);
        cout << "Start X: ";
        cin >> StartModel(0);
        cout << "Start Y: ";
        cin >> StartModel(1);
        ublas::matrix<double> DataCov(ndata,ndata);
        ublas::matrix<double> ModelCov(nmodel,nmodel);
        DataCov(0,0) = 1;
        ModelCov(1,1) = 1;
        ModelCov(0,0) = 1;
        Minimizer.InputData = StartData;
        Minimizer.StartModel = StartModel;
        Minimizer.MisfitCalculator = &(MyMisfit);
        LineSearch.MisfitCalculator = &(MyMisfit);
        Minimizer.LineSearch = &LineSearch;
        
        Minimizer.DataCovar.assign(DataCov);
        Minimizer.ModelCovar.assign(ModelCov);
        Minimizer.Prepare();
        for (int j = 0; j < Minimizer.CurrentModel.size(); ++j)
                        ModelFile << Minimizer.CurrentModel(j) << " ";
        ModelFile << endl;
        try
        {
                for (int i = 0; i < maxit; ++i)
                {
                        Minimizer.DoIteration();
                        for (int j = 0; j < Minimizer.CurrentModel.size(); ++j)
                                ModelFile << Minimizer.CurrentModel(j) << " ";
                        ModelFile << endl;
                }
        }
        catch (const exception& e)
        {
                cerr << "Code has converged !" << endl;
                cerr << e.what() << endl;
                for (int j = 0; j < Minimizer.CurrentModel.size(); ++j)
                                ModelFile << Minimizer.CurrentModel(j) << " ";
                ModelFile << endl;
        }
        
}