/[escript]/trunk/ripley/src/ripleycpp.cpp
ViewVC logotype

Annotation of /trunk/ripley/src/ripleycpp.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 4906 - (hide annotations)
Tue Apr 29 05:27:34 2014 UTC (4 years, 11 months ago) by sshaw
File size: 23443 byte(s)
restoring lost changes
1 caltinay 3670
2 jfenwick 3981 /*****************************************************************************
3 caltinay 3670 *
4 jfenwick 4657 * Copyright (c) 2003-2014 by University of Queensland
5 jfenwick 3981 * http://www.uq.edu.au
6 caltinay 3670 *
7     * Primary Business: Queensland, Australia
8     * Licensed under the Open Software License version 3.0
9     * http://www.opensource.org/licenses/osl-3.0.php
10     *
11 jfenwick 3981 * Development until 2012 by Earth Systems Science Computational Center (ESSCC)
12 jfenwick 4657 * Development 2012-2013 by School of Earth Sciences
13     * Development from 2014 by Centre for Geoscience Computing (GeoComp)
14 jfenwick 3981 *
15     *****************************************************************************/
16 caltinay 3670
17 caltinay 3691 #include <ripley/Brick.h>
18     #include <ripley/Rectangle.h>
19 caltinay 3670 #include <esysUtils/esysExceptionTranslator.h>
20    
21 sshaw 4906 #include <boost/python.hpp>
22 caltinay 3670 #include <boost/python/module.hpp>
23     #include <boost/python/def.hpp>
24     #include <boost/python/detail/defaults_gen.hpp>
25     #include <boost/version.hpp>
26    
27     using namespace boost::python;
28    
29 caltinay 3691 namespace ripley {
30 caltinay 3670
31 caltinay 4616 template<typename T>
32     std::vector<T> extractPyArray(const object& obj, const std::string& name,
33     int expectedLength=0)
34     {
35     std::vector<T> result;
36     if (extract<tuple>(obj).check() || extract<list>(obj).check()) {
37     if (expectedLength==0 || len(obj)==expectedLength) {
38     for (int i=0; i<len(obj); i++) {
39     result.push_back(extract<T>(obj[i]));
40     }
41     } else {
42     std::stringstream ssError;
43     ssError << "argument '" << name << "' has wrong length";
44     std::string error(ssError.str());
45     throw RipleyException(error.c_str());
46     }
47     } else {
48     std::stringstream ssError;
49     ssError << "argument '" << name << "' must be a tuple or list";
50     std::string error(ssError.str());
51     throw RipleyException(error.c_str());
52     }
53    
54     return result;
55     }
56    
57 caltinay 3971 escript::Data readBinaryGrid(std::string filename, escript::FunctionSpace fs,
58 caltinay 4616 const object& pyShape, double fill, int byteOrder, int dataType,
59 caltinay 4277 const object& pyFirst, const object& pyNum, const object& pyMultiplier,
60 caltinay 4616 const object& pyReverse)
61 caltinay 3971 {
62     int dim=fs.getDim();
63 caltinay 4618 ReaderParameters params;
64 caltinay 3971
65 caltinay 4616 params.first = extractPyArray<int>(pyFirst, "first", dim);
66     params.numValues = extractPyArray<int>(pyNum, "numValues", dim);
67     params.multiplier = extractPyArray<int>(pyMultiplier, "multiplier", dim);
68     params.reverse = extractPyArray<int>(pyReverse, "reverse", dim);
69     params.byteOrder = byteOrder;
70     params.dataType = dataType;
71     std::vector<int> shape(extractPyArray<int>(pyShape, "shape"));
72 caltinay 3971
73     const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(fs.getDomain().get());
74     if (!dom)
75     throw RipleyException("Function space must be on a ripley domain");
76    
77     escript::Data res(fill, shape, fs, true);
78 caltinay 4615 dom->readBinaryGrid(res, filename, params);
79 caltinay 3971 return res;
80     }
81    
82 sshaw 4738 #ifdef USE_BOOSTIO
83     escript::Data readBinaryGridFromZipped(std::string filename, escript::FunctionSpace fs,
84     const object& pyShape, double fill, int byteOrder, int dataType,
85     const object& pyFirst, const object& pyNum, const object& pyMultiplier,
86     const object& pyReverse)
87     {
88     int dim=fs.getDim();
89     ReaderParameters params;
90    
91     params.first = extractPyArray<int>(pyFirst, "first", dim);
92     params.numValues = extractPyArray<int>(pyNum, "numValues", dim);
93     params.multiplier = extractPyArray<int>(pyMultiplier, "multiplier", dim);
94     params.reverse = extractPyArray<int>(pyReverse, "reverse", dim);
95     params.byteOrder = byteOrder;
96     params.dataType = dataType;
97     std::vector<int> shape(extractPyArray<int>(pyShape, "shape"));
98    
99     const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(fs.getDomain().get());
100     if (!dom)
101     throw RipleyException("Function space must be on a ripley domain");
102    
103     escript::Data res(fill, shape, fs, true);
104     dom->readBinaryGridFromZipped(res, filename, params);
105     return res;
106     }
107     #endif
108    
109 caltinay 4013 escript::Data readNcGrid(std::string filename, std::string varname,
110 caltinay 4616 escript::FunctionSpace fs, const object& pyShape, double fill,
111     const object& pyFirst, const object& pyNum, const object& pyMultiplier,
112     const object& pyReverse)
113 caltinay 4013 {
114     int dim=fs.getDim();
115 caltinay 4618 ReaderParameters params;
116 caltinay 4013
117 caltinay 4616 params.first = extractPyArray<int>(pyFirst, "first", dim);
118     params.numValues = extractPyArray<int>(pyNum, "numValues", dim);
119     params.multiplier = extractPyArray<int>(pyMultiplier, "multiplier", dim);
120     params.reverse = extractPyArray<int>(pyReverse, "reverse", dim);
121     std::vector<int> shape(extractPyArray<int>(pyShape, "shape"));
122 caltinay 4013
123     const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(fs.getDomain().get());
124     if (!dom)
125     throw RipleyException("Function space must be on a ripley domain");
126    
127     escript::Data res(fill, shape, fs, true);
128 caltinay 4615 dom->readNcGrid(res, filename, varname, params);
129 caltinay 4013 return res;
130     }
131    
132 caltinay 3691 // These wrappers are required to make the shared pointers work through the
133     // Python wrapper
134 caltinay 3670
135 jfenwick 3892 // The double for n? is just to keep python happy when people need to deal with
136 caltinay 3971 // truediv
137 jfenwick 3892 escript::Domain_ptr _brick(double _n0, double _n1, double _n2, const object& l0,
138 sshaw 4622 const object& l1, const object& l2, int d0, int d1, int d2,
139     const object& objpoints, const object& objtags)
140 caltinay 3691 {
141 jfenwick 3892 int n0=static_cast<int>(_n0), n1=static_cast<int>(_n1), n2=static_cast<int>(_n2);
142 caltinay 3781 double x0=0., x1=1., y0=0., y1=1., z0=0., z1=1.;
143     if (extract<tuple>(l0).check()) {
144     tuple x=extract<tuple>(l0);
145     if (len(x)==2) {
146     x0=extract<double>(x[0]);
147     x1=extract<double>(x[1]);
148     } else
149     throw RipleyException("Argument l0 must be a float or 2-tuple");
150     } else if (extract<double>(l0).check()) {
151     x1=extract<double>(l0);
152     } else
153     throw RipleyException("Argument l0 must be a float or 2-tuple");
154    
155     if (extract<tuple>(l1).check()) {
156     tuple y=extract<tuple>(l1);
157     if (len(y)==2) {
158     y0=extract<double>(y[0]);
159     y1=extract<double>(y[1]);
160     } else
161     throw RipleyException("Argument l1 must be a float or 2-tuple");
162     } else if (extract<double>(l1).check()) {
163     y1=extract<double>(l1);
164     } else
165     throw RipleyException("Argument l1 must be a float or 2-tuple");
166    
167     if (extract<tuple>(l2).check()) {
168     tuple z=extract<tuple>(l2);
169     if (len(z)==2) {
170     z0=extract<double>(z[0]);
171     z1=extract<double>(z[1]);
172     } else
173     throw RipleyException("Argument l2 must be a float or 2-tuple");
174     } else if (extract<double>(l2).check()) {
175     z1=extract<double>(l2);
176     } else
177     throw RipleyException("Argument l2 must be a float or 2-tuple");
178 sshaw 4622 boost::python::list pypoints=extract<boost::python::list>(objpoints);
179     boost::python::list pytags=extract<boost::python::list>(objtags);
180     int numpts=extract<int>(pypoints.attr("__len__")());
181     int numtags=extract<int>(pytags.attr("__len__")());
182     std::vector<double> points;
183     std::vector<int> tags;
184     tags.resize(numtags, -1);
185     for (int i=0;i<numpts;++i) {
186 sshaw 4828 tuple temp = extract<tuple>(pypoints[i]);
187 sshaw 4622 int l=extract<int>(temp.attr("__len__")());
188 sshaw 4828 if (l != 3)
189     throw RipleyException("Number of coordinates for each dirac point must match dimensions.");
190 sshaw 4622 for (int k=0;k<l;++k) {
191     points.push_back(extract<double>(temp[k]));
192     }
193     }
194     std::map<std::string, int> tagstonames;
195     int curmax=40;
196     // but which order to assign tags to names?????
197     for (int i=0;i<numtags;++i) {
198     extract<int> ex_int(pytags[i]);
199     extract<std::string> ex_str(pytags[i]);
200     if (ex_int.check()) {
201     tags[i]=ex_int();
202     if (tags[i]>= curmax) {
203     curmax=tags[i]+1;
204     }
205     } else if (ex_str.check()) {
206     std::string s=ex_str();
207     std::map<std::string, int>::iterator it=tagstonames.find(s);
208     if (it!=tagstonames.end()) {
209     // we have the tag already so look it up
210     tags[i]=it->second;
211     } else {
212     tagstonames[s]=curmax;
213     tags[i]=curmax;
214     curmax++;
215     }
216     } else {
217     throw RipleyException("Error - Unable to extract tag value.");
218     }
219     }
220 sshaw 4872 if (numtags != numpts)
221     throw RipleyException("Number of tags does not match number of points.");
222 sshaw 4622 return escript::Domain_ptr(new Brick(n0,n1,n2, x0,y0,z0, x1,y1,z1, d0,d1,d2,
223     points, tags, tagstonames));
224 caltinay 3691 }
225 caltinay 3670
226 jfenwick 4779 //const int _q[]={0x61686969,0x746c4144,0x79616e43};
227     const int _q[]={0x62207363, 0x6574735F, 0x2020214e};
228 jfenwick 3892 escript::Domain_ptr _rectangle(double _n0, double _n1, const object& l0,
229 sshaw 4622 const object& l1, int d0, int d1,
230     const object& objpoints, const object& objtags)
231 caltinay 3691 {
232 jfenwick 3892 int n0=static_cast<int>(_n0), n1=static_cast<int>(_n1);
233 caltinay 3781 double x0=0., x1=1., y0=0., y1=1.;
234     if (extract<tuple>(l0).check()) {
235     tuple x=extract<tuple>(l0);
236     if (len(x)==2) {
237     x0=extract<double>(x[0]);
238     x1=extract<double>(x[1]);
239     } else
240     throw RipleyException("Argument l0 must be a float or 2-tuple");
241     } else if (extract<double>(l0).check()) {
242     x1=extract<double>(l0);
243     } else
244     throw RipleyException("Argument l0 must be a float or 2-tuple");
245    
246     if (extract<tuple>(l1).check()) {
247     tuple y=extract<tuple>(l1);
248     if (len(y)==2) {
249     y0=extract<double>(y[0]);
250     y1=extract<double>(y[1]);
251     } else
252     throw RipleyException("Argument l1 must be a float or 2-tuple");
253     } else if (extract<double>(l1).check()) {
254     y1=extract<double>(l1);
255     } else
256     throw RipleyException("Argument l1 must be a float or 2-tuple");
257 sshaw 4622 boost::python::list pypoints=extract<boost::python::list>(objpoints);
258     boost::python::list pytags=extract<boost::python::list>(objtags);
259     int numpts=extract<int>(pypoints.attr("__len__")());
260     int numtags=extract<int>(pytags.attr("__len__")());
261     std::vector<double> points;
262     std::vector<int> tags;
263     tags.resize(numtags, -1);
264     for (int i=0;i<numpts;++i) {
265 sshaw 4828 tuple temp = extract<tuple>(pypoints[i]);
266 sshaw 4622 int l=extract<int>(temp.attr("__len__")());
267 sshaw 4828 if (l != 2)
268     throw RipleyException("Number of coordinates for each dirac point must match dimensions.");
269 sshaw 4622 for (int k=0;k<l;++k) {
270     points.push_back(extract<double>(temp[k]));
271     }
272     }
273     std::map<std::string, int> tagstonames;
274     int curmax=40;
275     // but which order to assign tags to names?????
276     for (int i=0;i<numtags;++i) {
277     extract<int> ex_int(pytags[i]);
278     extract<std::string> ex_str(pytags[i]);
279     if (ex_int.check()) {
280     tags[i]=ex_int();
281     if (tags[i]>= curmax) {
282     curmax=tags[i]+1;
283     }
284     } else if (ex_str.check()) {
285     std::string s=ex_str();
286     std::map<std::string, int>::iterator it=tagstonames.find(s);
287     if (it!=tagstonames.end()) {
288     // we have the tag already so look it up
289     tags[i]=it->second;
290     } else {
291     tagstonames[s]=curmax;
292     tags[i]=curmax;
293     curmax++;
294     }
295     } else {
296     throw RipleyException("Error - Unable to extract tag value.");
297     }
298     }
299 sshaw 4872 if (numtags != numpts)
300     throw RipleyException("Number of tags does not match number of points.");
301 sshaw 4622 return escript::Domain_ptr(new Rectangle(n0,n1, x0,y0, x1,y1, d0,d1,
302     points, tags, tagstonames));
303 caltinay 3691 }
304 caltinay 3791 std::string _who(){int a[]={_q[0]^42,_q[1]^42,_q[2]^42,0};return (char*)&a[0];}
305 caltinay 3670
306 caltinay 4357 } // end of namespace ripley
307 caltinay 3670
308    
309     BOOST_PYTHON_MODULE(ripleycpp)
310     {
311     // This feature was added in boost v1.34
312     #if ((BOOST_VERSION/100)%1000 > 34) || (BOOST_VERSION/100000 >1)
313 caltinay 3691 // params are: bool show_user_defined, bool show_py_signatures, bool show_cpp_signatures
314     docstring_options docopt(true, true, false);
315 caltinay 3670 #endif
316    
317 sshaw 4821 register_exception_translator<ripley::RipleyException>(&(esysUtils::RuntimeErrorTranslator));
318 caltinay 3670
319 caltinay 4174 scope().attr("__doc__") = "To use this module, please import esys.ripley";
320 caltinay 4357 scope().attr("BYTEORDER_NATIVE") = (int)ripley::BYTEORDER_NATIVE;
321     scope().attr("BYTEORDER_LITTLE_ENDIAN") = (int)ripley::BYTEORDER_LITTLE_ENDIAN;
322     scope().attr("BYTEORDER_BIG_ENDIAN") = (int)ripley::BYTEORDER_BIG_ENDIAN;
323     scope().attr("DATATYPE_INT32") = (int)ripley::DATATYPE_INT32;
324     scope().attr("DATATYPE_FLOAT32") = (int)ripley::DATATYPE_FLOAT32;
325     scope().attr("DATATYPE_FLOAT64") = (int)ripley::DATATYPE_FLOAT64;
326 caltinay 4174
327 sshaw 4622 def("Brick", ripley::_brick, (arg("n0"),arg("n1"),arg("n2"),arg("l0")=1.0,arg("l1")=1.0,arg("l2")=1.0,arg("d0")=-1,arg("d1")=-1,arg("d2")=-1,arg("diracPoints")=list(),arg("diracTags")=list()),
328 caltinay 3691 "Creates a hexagonal mesh with n0 x n1 x n2 elements over the brick [0,l0] x [0,l1] x [0,l2].\n\n"
329     ":param n0: number of elements in direction 0\n:type n0: ``int``\n"
330     ":param n1: number of elements in direction 1\n:type n1: ``int``\n"
331     ":param n2: number of elements in direction 2\n:type n2: ``int``\n"
332 caltinay 3781 ":param l0: length of side 0 or coordinate range of side 0\n:type l0: ``float`` or ``tuple``\n"
333     ":param l1: length of side 1 or coordinate range of side 1\n:type l1: ``float`` or ``tuple``\n"
334     ":param l2: length of side 2 or coordinate range of side 2\n:type l2: ``float`` or ``tuple``\n"
335 caltinay 3691 ":param d0: number of subdivisions in direction 0\n:type d0: ``int``\n"
336     ":param d1: number of subdivisions in direction 1\n:type d1: ``int``\n"
337     ":param d2: number of subdivisions in direction 2\n:type d2: ``int``");
338 caltinay 3670
339 sshaw 4622 def("Rectangle", ripley::_rectangle, (arg("n0"),arg("n1"),arg("l0")=1.0,arg("l1")=1.0,arg("d0")=-1,arg("d1")=-1,arg("diracPoints")=list(),arg("diracTags")=list()),
340 caltinay 3691 "Creates a rectangular mesh with n0 x n1 elements over the rectangle [0,l0] x [0,l1].\n\n"
341     ":param n0: number of elements in direction 0\n:type n0: ``int``\n"
342     ":param n1: number of elements in direction 1\n:type n1: ``int``\n"
343 caltinay 3781 ":param l0: length of side 0 or coordinate range of side 0\n:type l0: ``float`` or ``tuple``\n"
344     ":param l1: length of side 1 or coordinate range of side 1\n:type l1: ``float`` or ``tuple``\n"
345 caltinay 3691 ":param d0: number of subdivisions in direction 0\n:type d0: ``int``\n"
346     ":param d1: number of subdivisions in direction 1\n:type d1: ``int``");
347 caltinay 3971 def("_theculprit_", ripley::_who);
348 caltinay 3670
349 caltinay 4616 def("_readBinaryGrid", &ripley::readBinaryGrid, (arg("filename"),
350     arg("functionspace"), arg("shape"), arg("fill")=0.,
351     arg("byteOrder"), arg("dataType"), arg("first"),
352     arg("numValues"), arg("multiplier"), arg("reverse")),
353     "Reads a binary Grid");
354 sshaw 4738 #ifdef USE_BOOSTIO
355     def("_readBinaryGridFromZipped", &ripley::readBinaryGridFromZipped, (arg("filename"),
356     arg("functionspace"), arg("shape"), arg("fill")=0.,
357     arg("byteOrder"), arg("dataType"), arg("first"),
358     arg("numValues"), arg("multiplier"), arg("reverse")),
359     "Reads a binary Grid");
360     #endif
361 caltinay 4616 def("_readNcGrid", &ripley::readNcGrid, (arg("filename"), arg("varname"),
362     arg("functionspace"), arg("shape"), arg("fill"), arg("first"),
363     arg("numValues"), arg("multiplier"), arg("reverse")),
364     "Reads a grid from a netCDF file");
365 caltinay 4013
366 jfenwick 4242 class_<ripley::RipleyDomain, bases<escript::AbstractContinuousDomain>, boost::noncopyable >
367 caltinay 3691 ("RipleyDomain", "", no_init)
368     .def("print_mesh_info", &ripley::RipleyDomain::Print_Mesh_Info, (arg("full")=false),
369     "Prints out a summary about the mesh.\n"
370     ":param full: whether to output additional data\n:type full: ``bool``")
371 caltinay 4357 .def("writeBinaryGrid", &ripley::RipleyDomain::writeBinaryGrid)
372    
373 caltinay 3691 .def("dump", &ripley::RipleyDomain::dump, args("filename"),
374     "Dumps the mesh to a file with the given name.")
375 caltinay 4340 .def("getGridParameters", &ripley::RipleyDomain::getGridParameters,
376     "Returns the tuple (origin, spacing, elements) where the entries are tuples:\n"
377     "``origin``=the coordinates of the domain's global origin,\n"
378     "``spacing``=the element size (=node spacing) of the domain,\n"
379     "``elements``=the global number of elements in all dimensions\n\n"
380     ":rtype: ``tuple``")
381 caltinay 3691 .def("getDescription", &ripley::RipleyDomain::getDescription,
382 caltinay 3670 ":return: a description for this domain\n:rtype: ``string``")
383 caltinay 3691 .def("getDim", &ripley::RipleyDomain::getDim, ":rtype: ``int``")
384     .def("getDataShape", &ripley::RipleyDomain::getDataShape, args("functionSpaceCode"),
385 caltinay 3670 ":return: a pair (dps, ns) where dps=the number of data points per sample, and ns=the number of samples\n:rtype: ``tuple``")
386 caltinay 3691 .def("getNumDataPointsGlobal", &ripley::RipleyDomain::getNumDataPointsGlobal,
387 caltinay 3670 ":return: the number of data points summed across all MPI processes\n"
388     ":rtype: ``int``")
389 sshaw 4622 .def("addToSystem",&ripley::RipleyDomain::addToSystemFromPython,
390     args("mat", "rhs", "data"),
391     "adds a PDE to the system, results depend on domain\n\n"
392     ":param mat:\n:type mat: `OperatorAdapter`\n"
393     ":param rhs:\n:type rhs: `Data`\n"
394     ":param data:\ntype data: `list`")
395 caltinay 3691 .def("addPDEToSystem",&ripley::RipleyDomain::addPDEToSystem,
396     args("mat", "rhs", "A", "B", "C", "D", "X", "Y", "d", "y", "d_contact", "y_contact"),
397 caltinay 3670 "adds a PDE onto the stiffness matrix mat and a rhs\n\n"
398     ":param mat:\n:type mat: `OperatorAdapter`\n:param rhs:\n:type rhs: `Data`\n"
399     ":param A:\n:type A: `Data`\n"
400     ":param B:\n:type B: `Data`\n"
401     ":param C:\n:type C: `Data`\n"
402     ":param D:\n:type D: `Data`\n"
403     ":param X:\n:type X: `Data`\n"
404     ":param Y:\n:type Y: `Data`\n"
405     ":param d:\n:type d: `Data`\n"
406     ":param d_contact:\n:type d_contact: `Data`\n"
407 caltinay 3791 ":param y_contact:\n:type y_contact: `Data`"
408 caltinay 3670 )
409 sshaw 4622
410 caltinay 3691 .def("addPDEToRHS",&ripley::RipleyDomain::addPDEToRHS,
411 caltinay 3670 args("rhs", "X", "Y", "y", "y_contact"),
412     "adds a PDE onto the stiffness matrix mat and a rhs\n\n"
413     ":param rhs:\n:type rhs: `Data`\n"
414     ":param X:\n:type X: `Data`\n"
415     ":param Y:\n:type Y: `Data`\n"
416     ":param y:\n:type y: `Data`\n"
417     ":param y_contact:\n:type y_contact: `Data`"
418     )
419 sshaw 4622 .def("addToRHS",&ripley::RipleyDomain::addToRHSFromPython,
420     args("rhs", "data"),
421     "adds a PDE onto the stiffness matrix mat and a rhs, "
422     "results depends on domain\n\n"
423     ":param rhs:\n:type rhs: `Data`\n"
424     ":param data:\ntype data: `list`")
425     .def("setAssembler", &ripley::RipleyDomain::setAssemblerFromPython,
426     args("typename", "options"),
427     "sets the domain to use the named assembler, if supported, using"
428     "the options if provided"
429     ":param typename:\n:type typename: `string`\n"
430     ":param options:\n:type options: `list`\n")
431 caltinay 3691 .def("addPDEToTransportProblem",&ripley::RipleyDomain::addPDEToTransportProblem,
432 caltinay 3670 args( "tp", "source", "M", "A", "B", "C", "D", "X", "Y", "d", "y", "d_contact", "y_contact"),
433     ":param tp:\n:type tp: `TransportProblemAdapter`\n"
434     ":param source:\n:type source: `Data`\n"
435     ":param M:\n:type M: `Data`\n"
436     ":param A:\n:type A: `Data`\n"
437     ":param B:\n:type B: `Data`\n"
438     ":param C:\n:type C: `Data`\n"
439     ":param D:\n:type D: `Data`\n"
440     ":param X:\n:type X: `Data`\n"
441     ":param Y:\n:type Y: `Data`\n"
442     ":param d:\n:type d: `Data`\n"
443     ":param y:\n:type y: `Data`\n"
444     ":param d_contact:\n:type d_contact: `Data`\n"
445 caltinay 3791 ":param y_contact:\n:type y_contact: `Data`"
446 caltinay 3670 )
447 caltinay 3691 .def("newOperator",&ripley::RipleyDomain::newSystemMatrix,
448 caltinay 3670 args("row_blocksize", "row_functionspace", "column_blocksize", "column_functionspace", "type"),
449     "creates a SystemMatrixAdapter stiffness matrix and initializes it with zeros\n\n"
450     ":param row_blocksize:\n:type row_blocksize: ``int``\n"
451     ":param row_functionspace:\n:type row_functionspace: `FunctionSpace`\n"
452     ":param column_blocksize:\n:type column_blocksize: ``int``\n"
453     ":param column_functionspace:\n:type column_functionspace: `FunctionSpace`\n"
454 caltinay 3791 ":param type:\n:type type: ``int``"
455 caltinay 3670 )
456 caltinay 3691 .def("newTransportProblem",&ripley::RipleyDomain::newTransportProblem,
457 caltinay 3670 args("theta", "blocksize", "functionspace", "type"),
458     "creates a TransportProblemAdapter\n\n"
459     ":param theta:\n:type theta: ``float``\n"
460     ":param blocksize:\n:type blocksize: ``int``\n"
461     ":param functionspace:\n:type functionspace: `FunctionSpace`\n"
462 caltinay 3791 ":param type:\n:type type: ``int``"
463 caltinay 3670 )
464 caltinay 3691 .def("getSystemMatrixTypeId",&ripley::RipleyDomain::getSystemMatrixTypeId,
465 caltinay 3670 args("solver", "preconditioner", "package", "symmetry"),
466 caltinay 3697 ":return: the identifier of the matrix type to be used for the global stiffness matrix when a particular solver, package, preconditioner, and symmetric matrix is used.\n"
467 caltinay 3670 ":rtype: ``int``\n"
468     ":param solver:\n:type solver: ``int``\n"
469     ":param preconditioner:\n:type preconditioner: ``int``\n"
470     ":param package:\n:type package: ``int``\n"
471 caltinay 3791 ":param symmetry:\n:type symmetry: ``int``"
472 caltinay 3670 )
473 caltinay 3691 .def("getTransportTypeId",&ripley::RipleyDomain::getTransportTypeId,
474 caltinay 3670 args("solver", "preconditioner", "package", "symmetry"),
475 caltinay 3697 ":return: the identifier of the transport problem type to be used when a particular solver, preconditioner, package and symmetric matrix is used.\n"
476 caltinay 3670 ":rtype: ``int``\n"
477     ":param solver:\n:type solver: ``int``\n"
478     ":param preconditioner:\n:type preconditioner: ``int``\n"
479     ":param package:\n:type package: ``int``\n"
480 caltinay 3791 ":param symmetry:\n:type symmetry: ``int``"
481 caltinay 3670 )
482 caltinay 3691 .def("getX",&ripley::RipleyDomain::getX, ":return: locations in the FEM nodes\n\n"
483 caltinay 3670 ":rtype: `Data`")
484 caltinay 3691 .def("getNormal",&ripley::RipleyDomain::getNormal,
485 caltinay 3670 ":return: boundary normals at the quadrature point on the face elements\n"
486     ":rtype: `Data`")
487 caltinay 3691 .def("getSize",&ripley::RipleyDomain::getSize,":return: the element size\n"
488 caltinay 3670 ":rtype: `Data`")
489 caltinay 3691 .def("setTagMap",&ripley::RipleyDomain::setTagMap,args("name","tag"),
490 caltinay 3670 "Give a tag number a name.\n\n:param name: Name for the tag\n:type name: ``string``\n"
491     ":param tag: numeric id\n:type tag: ``int``\n:note: Tag names must be unique within a domain")
492 caltinay 3691 .def("getTag",&ripley::RipleyDomain::getTag,args("name"),":return: tag id for "
493 caltinay 3670 "``name``\n:rtype: ``string``")
494 caltinay 3691 .def("isValidTagName",&ripley::RipleyDomain::isValidTagName,args("name"),
495     ":return: True if ``name`` corresponds to a tag, otherwise False\n:rtype: ``bool``")
496     .def("showTagNames",&ripley::RipleyDomain::showTagNames,":return: A space separated list of tag names\n:rtype: ``string``")
497     .def("getMPISize",&ripley::RipleyDomain::getMPISize,":return: the number of processes used for this `Domain`\n:rtype: ``int``")
498     .def("getMPIRank",&ripley::RipleyDomain::getMPIRank,":return: the rank of this process\n:rtype: ``int``")
499     .def("MPIBarrier",&ripley::RipleyDomain::MPIBarrier,"Wait until all processes have reached this point")
500     .def("onMasterProcessor",&ripley::RipleyDomain::onMasterProcessor,":return: True if this code is executing on the master process\n:rtype: `bool`");
501 jfenwick 4825 /* These two class exports are necessary to ensure that the extra methods added by ripley make it to python.
502     * This change became necessary when the Brick and Rectangle constructors turned into factories instead of classes */
503 jfenwick 4687 class_<ripley::Brick, bases<ripley::RipleyDomain> >("RipleyBrick", "", no_init);
504     class_<ripley::Rectangle, bases<ripley::RipleyDomain> >("RipleyRectangle", "", no_init);
505 caltinay 3670 }
506    

  ViewVC Help
Powered by ViewVC 1.1.26