/* -*- mia-c++  -*-
 *
 * This file is part of MIA - a toolbox for medical image analysis
 * Copyright (c) Leipzig, Madrid 1999-2017 Gert Wollny
 *
 * MIA 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.
 *
 * This program 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 MIA; if not, see <http://www.gnu.org/licenses/>.
 *
 */

/*
  This plug-in implements the navier-stokes operator like registration model
  that accounts for linear elastic and fluid dynamic registration.
  Which model is used depends on the selected time step.

  To solve the PDE a full SOR is implemented.
*/
#ifndef mia_2d_model_navier_hh
#define mia_2d_model_navier_hh

#include <limits>

#include <mia/2d/model.hh>

NS_BEGIN(naviera_regmodel);

class C2DNavierRegModel: public mia::C2DRegModel
{
public:
       C2DNavierRegModel(float mu, float lambda, size_t maxiter, float omega, float epsilon);

       float  solve_at(unsigned int x, unsigned int y,
                       const mia::C2DFVector& b, mia::C2DFVectorfield::iterator v, int dx)const;

private:
       virtual void do_solve(const mia::C2DFVectorfield& b, mia::C2DFVectorfield& x) const;

       float m_mu;
       float m_lambda;
       float m_a, m_a_b, m_b_4;
       float m_omega;
       float m_epsilon;
       size_t m_max_iter;
};

class C2DNavierRegModelPlugin: public mia::C2DRegModelPlugin
{
public:
       C2DNavierRegModelPlugin();
       mia::C2DRegModel *do_create()const;

private:
       const std::string do_get_descr()const;

       float m_mu;
       float m_lambda;
       float m_omega;
       float m_epsilon;
       int m_maxiter;
};

NS_END

#endif

