#include <octave/oct.h>
#include <octave/xdiv.h>
#ifdef OCTAVE_DEV
#define VECTOR_VALUE column_vector_value
#else // !OCTAVE_DEV
#define VECTOR_VALUE vector_value
#endif // OCTAVE_DEV
// Code generation directives
#define STANDALONE 0
#define OCTAVEDLD 1
#if (! defined (CODEGENTARGET))
#define CODEGENTARGET STANDALONE
#endif // (! defined (CODEGENTARGET))
#if (CODEGENTARGET == STANDALONE)
ColumnVector Fmtt_implicit ( ColumnVector &x,
ColumnVector &dx,
Matrix &AA,
ColumnVector &AAx,
const double &t,
const int &Nx,
const ColumnVector &openx)
{
#elif (CODEGENTARGET == OCTAVEDLD)
DEFUN_DLD (mtt_implicit, args, ,
"implicit integration method")
{
ColumnVector x = args(0).VECTOR_VALUE ();
ColumnVector dx = args(1).VECTOR_VALUE ();
Matrix AA = args(2).matrix_value ();
ColumnVector AAx = args(3).VECTOR_VALUE ();
const double t = args(4).double_value ();
const int Nx = (int) (args(5).double_value ());
const ColumnVector openx = args(6).VECTOR_VALUE ();
#endif // (CODEGENTARGET == STANDALONE)
register int row, col;
for (row = 0; row < Nx; row++)
{
if (openx (row) > 0.5)
{
AAx (row) = 0.0;
dx (row) = 0.0;
for (col = 0; col < Nx; col++)
{
AA (row,col) = 0.0;
AA (col,row) = 0.0;
}
}
}
#ifdef OCTAVE_DEV
x = xleftdiv (AA, Matrix(AAx + dx * t)).column (0);
#else // !OCTAVE_DEV
Matrix tmp = xleftdiv (AA, static_cast<Matrix>(static_cast<ColumnVector>(AAx + dx * t)));
for (row = 0; row < Nx; row++)
{
x (row) = tmp (row,0);
}
#endif // OCTAVE_DEV
for (row = 0; row < Nx; row++)
{
if (openx (row) > 0.5)
{
x (row) = 0.0;
}
}
#if (CODEGENTARGET == STANDALONE)
return x;
#elif (CODEGENTARGET == OCTAVEDLD)
return octave_value (x);
#endif // (CODEGENTARGET == STANDALONE)
}