Mercurial > fem-fenics-eugenio
annotate src/plot_mesh.cc @ 128:d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
* plot_mesh.cc: obtained merging Plot_2d.cc Plot_3d.cc fem_plot.cc
* plot_func.cc: obtined from fem_plot.cc
* fem_save.cc moved to sace.cc
author | gedeone-octave <marcovass89@hotmail.it> |
---|---|
date | Thu, 05 Sep 2013 20:49:36 +0200 |
parents | |
children | d003a8d6ac58 |
rev | line source |
---|---|
128
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
1 /* |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
2 Copyright (C) 2013 Marco Vassallo |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
3 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
4 This program is free software; you can redistribute it and/or modify it under |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
5 the terms of the GNU General Public License as published by the Free Software |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
6 Foundation; either version 2 of the License, or (at your option) any later |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
7 version. |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
8 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
9 This program is distributed in the hope that it will be useful, but WITHOUT |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
10 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
11 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
12 details. |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
13 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
14 You should have received a copy of the GNU General Public License along with |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
15 this program; if not, see <http://www.gnu.org/licenses/>. |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
16 */ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
17 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
18 #include "function.h" |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
19 #include "mesh.h" |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
20 #include "Plot_2d.h" |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
21 #include "Plot_3d.h" |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
22 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
23 DEFUN_DLD (plot, args, , "-*- texinfo -*-\n\ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
24 @deftypefn {Function File} \ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
25 plot (@var{Mesh}, @var{Nodal_Values}(OPTIONAL))\n\ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
26 The input parameter is the mesh and optionally also a vector representing\ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
27 the values of a function at each node.\n\ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
28 @seealso{Mesh, save}\n\ |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
29 @end deftypefn") |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
30 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
31 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
32 int nargin = args.length (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
33 octave_value retval; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
34 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
35 if (nargin < 1 || nargin > 2) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
36 print_usage (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
37 else |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
38 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
39 if (! mesh_type_loaded) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
40 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
41 mesh::register_type (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
42 mesh_type_loaded = true; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
43 mlock (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
44 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
45 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
46 if (args(0).type_id () == mesh::static_type_id ()) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
47 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
48 const mesh & msho = static_cast<const mesh&> (args(0).get_rep ()); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
49 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
50 if (nargin == 1) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
51 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
52 if (!error_state) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
53 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
54 const boost::shared_ptr<const dolfin::Mesh> & mshd = msho.get_pmsh (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
55 dolfin::plot (*mshd); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
56 dolfin::interactive (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
57 retval = 0; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
58 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
59 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
60 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
61 else if (nargin == 2) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
62 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
63 Array <double> myu = args(1).array_value (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
64 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
65 if (!error_state) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
66 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
67 const dolfin::Mesh & mshd = msh.get_msh (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
68 boost::shared_ptr <const dolfin::FunctionSpace> V; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
69 if (mshd.topology ().dim () == 2) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
70 *V = new Plot_2d::FunctionSpace (mshd); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
71 else if (mshd.topology ().dim () () == 3) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
72 *V = new Plot_3d::FunctionSpace (mshd); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
73 else |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
74 error ("Plot: The size of the mesh must be 2 or 3"); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
75 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
76 if (V->dim () != myu.length ()) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
77 error ("Plot: The size of the functional space and of the vector must agree"); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
78 else |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
79 { |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
80 dolfin::Vector du(myu.length ()); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
81 for (std::size_t i = 0; i < myu.length (); ++i) |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
82 du.setitem (i, myu(i)); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
83 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
84 boost::shared_ptr<dolfin::Vector> uu (new dolfin::Vector(du)); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
85 boost::shared_ptr <const dolfin::Function> u (new dolfin::Function(V, uu)); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
86 dolfin::plot (*u); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
87 dolfin::interactive (); |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
88 retval = 0; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
89 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
90 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
91 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
92 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
93 |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
94 } |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
95 return retval; |
d3c29e38e5d2
New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff
changeset
|
96 } |