/*
 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 "function.h"
#include "mesh.h"

DEFUN_DLD (fem_plot, args, , "-*- texinfo -*-\n\
@deftypefn {Function File} \
fem_func (@var{Function})\n\
The input parameter is the object that you want to plot\n\
It can be either a Function or a Mesh \n\
@seealso{fem_func, fem_save}\n\
@end deftypefn")
{

  int nargin = args.length ();
  octave_value retval;
  
  if (nargin < 1 || nargin > 1)
    print_usage ();
  else
    {
      if (! function_type_loaded)
        {
          function::register_type ();
          function_type_loaded = true;
          mlock ();
        }

      if (! mesh_type_loaded)
        {
          mesh::register_type ();
          mesh_type_loaded = true;
          mlock ();
        }

      if (args(0).type_id () == function::static_type_id ())
        {
          const function & uo =
            static_cast<const function&> (args(0).get_rep ());

          if (!error_state)
            {
              const boost::shared_ptr<const dolfin::Function> & u = uo.get_pfun ();
              dolfin::plot (*u);
              dolfin::interactive ();
              retval = 0;
            }
        }

      else if (args(0).type_id () == mesh::static_type_id ())
        {
          const mesh & msho = static_cast<const mesh&> (args(0).get_rep ());

          if (!error_state)
            {
              const boost::shared_ptr<const dolfin::Mesh> & mshd = msho.get_pmsh ();
              dolfin::plot (*mshd);
              dolfin::interactive ();
              retval = 0;
            }
        }

    }
  return retval;
}