Mercurial > fem-fenics-eugenio

annotate src/plot_mesh.cc @ 129:d003a8d6ac58

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix error made in the previous commit.
author gedeone-octave <marcovass89@hotmail.it>
date Thu, 05 Sep 2013 21:18:58 +0200
parents d3c29e38e5d2
children 5fe2a157f4eb
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
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 {
129
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
67 const dolfin::Mesh & mshd = msho.get_msh ();
128
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;
129
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
69 uint D = mshd.topology ().dim ();
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
70 if (D == 2)
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
71 {
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
72 boost::shared_ptr <const dolfin::FunctionSpace> Vt (new Plot_2d::FunctionSpace (mshd));
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
73 V = Vt;
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
74 }
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
75
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
76 else if(D == 3)
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
77 {
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
78 boost::shared_ptr <const dolfin::FunctionSpace> Vt (new Plot_3d::FunctionSpace (mshd));
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
79 V = Vt;
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
80 }
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
81
128
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
82 else
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
83 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
84
129
d003a8d6ac58 Fix error made in the previous commit.
gedeone-octave <marcovass89@hotmail.it>
parents: 128
diff changeset
85
128
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
86 if (V->dim () != myu.length ())
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
87 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
88 else
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 dolfin::Vector du(myu.length ());
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
91 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
92 du.setitem (i, myu(i));
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 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
95 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
96 dolfin::plot (*u);
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
97 dolfin::interactive ();
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
98 retval = 0;
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
99 }
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
100 }
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
101 }
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
102 }
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
103
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
104 }
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
105 return retval;
d3c29e38e5d2 New naming convention adopted and Polymorphism for plot and save.
gedeone-octave <marcovass89@hotmail.it>
parents:
diff changeset
106 }