--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/plot_mesh.cc	Thu Sep 05 20:49:36 2013 +0200
@@ -0,0 +1,96 @@
+/*
+ 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"
+#include "Plot_2d.h"
+#include "Plot_3d.h"
+
+DEFUN_DLD (plot, args, , "-*- texinfo -*-\n\
+@deftypefn {Function File} \
+plot (@var{Mesh}, @var{Nodal_Values}(OPTIONAL))\n\
+The input parameter is the mesh and optionally also a vector representing\
+the values of a function at each node.\n\
+@seealso{Mesh, save}\n\
+@end deftypefn")
+{
+
+  int nargin = args.length ();
+  octave_value retval;
+  
+  if (nargin < 1 || 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 & msho = static_cast<const mesh&> (args(0).get_rep ());
+
+          if (nargin == 1)
+            {
+              if (!error_state)
+                {
+                  const boost::shared_ptr<const dolfin::Mesh> & mshd = msho.get_pmsh ();
+                  dolfin::plot (*mshd);
+                  dolfin::interactive ();
+                  retval = 0;
+                }
+             }
+
+          else if (nargin == 2)
+            {
+              Array <double> myu = args(1).array_value ();
+
+              if (!error_state)
+                {
+                  const dolfin::Mesh & mshd = msh.get_msh ();
+                  boost::shared_ptr <const dolfin::FunctionSpace> V;
+                  if (mshd.topology ().dim () == 2)
+                    *V = new Plot_2d::FunctionSpace (mshd);
+                  else if (mshd.topology ().dim () () == 3)
+                    *V = new Plot_3d::FunctionSpace (mshd);
+                  else
+                    error ("Plot: The size of the mesh must be 2 or 3");
+
+                  if (V->dim () != myu.length ())
+                    error ("Plot: 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;
+}