/*
 Copyright (C) 2013 Marco Vassallo

 This program 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 2 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
 this program; if not, see <http://www.gnu.org/licenses/>.
*/
#include "mesh.h"
#include "Plot_2d.h"

DEFUN_DLD (Plot_2d, args, , "-*- texinfo -*-\n\
@deftypefn {Function File} \
Plot_2d (@var{msh}, @var{nodal_value})\n\
The input parameters are a Mesh and a vector which specify the values of the \
function that you want to plot at each node\n\
@seealso{fem_func, fem_save}\n\
@end deftypefn")
{

  int nargin = args.length ();
  octave_value retval;
  
  if (nargin < 2 || nargin > 2)
    print_usage ();
  else
    {
      if (! mesh_type_loaded)
        {
          mesh::register_type ();
          mesh_type_loaded = true;
          mlock ();
        }
      if (args(0).type_id () == mesh::static_type_id ())
        {
          const mesh & msh =
            static_cast<const mesh&> (args(0).get_rep ());
          Array <double> myu = args(1).array_value ();

          if (!error_state)
            {
              const dolfin::Mesh & mshd = msh.get_msh ();
              boost::shared_ptr <const dolfin::FunctionSpace> V (new Plot_2d::FunctionSpace (mshd));

              if (V->dim () != myu.length ())
                error ("Plot_2D: The size of the functional space and of the vector must agree");
              else
                {
                  dolfin::Vector du(myu.length ());
                  for (std::size_t i = 0; i < myu.length (); ++i)
                    du.setitem (i, myu(i));

                  boost::shared_ptr<dolfin::Vector> uu (new dolfin::Vector(du));
                  boost::shared_ptr <const dolfin::Function> u (new dolfin::Function(V, uu));
                  dolfin::plot (*u);
                  dolfin::interactive ();
                  retval = 0;
                }
            }
        }
    }
  return retval;
}