/[escript]/trunk/escript/src/escriptcpp.cpp
ViewVC logotype

Contents of /trunk/escript/src/escriptcpp.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2841 - (show annotations)
Thu Jan 14 01:02:56 2010 UTC (9 years, 7 months ago) by jfenwick
File size: 33351 byte(s)
Unit tests for 2838 and adding setTags variant which takes a string instead of a tag num.


1
2 /*******************************************************
3 *
4 * Copyright (c) 2003-2009 by University of Queensland
5 * Earth Systems Science Computational Center (ESSCC)
6 * http://www.uq.edu.au/esscc
7 *
8 * Primary Business: Queensland, Australia
9 * Licensed under the Open Software License version 3.0
10 * http://www.opensource.org/licenses/osl-3.0.php
11 *
12 *******************************************************/
13
14
15 #include "Data.h"
16 #include "FunctionSpace.h"
17 #include "FunctionSpaceFactory.h"
18 #include "DataFactory.h"
19 #include "AbstractContinuousDomain.h"
20 #include "AbstractDomain.h"
21 #include "Utils.h"
22 #include "AbstractSystemMatrix.h"
23 #include "AbstractTransportProblem.h"
24 #include "DataVector.h"
25 #include "paso/Paso_MPI.h"
26 #include "EscriptParams.h"
27 #include "TestDomain.h"
28
29
30 extern "C" {
31 #include "esysUtils/blocktimer.h"
32 }
33
34 #include "esysUtils/esysExceptionTranslator.h"
35
36 #include <boost/version.hpp>
37 #include <boost/python.hpp>
38 #include <boost/python/module.hpp>
39 #include <boost/python/def.hpp>
40 #include <boost/python/object.hpp>
41 #include <boost/python/tuple.hpp>
42 #include <boost/smart_ptr.hpp>
43 #include <boost/version.hpp>
44
45 using namespace boost::python;
46
47 /*! \mainpage Esys Documentation
48 *
49 * \version 3.0.0
50 *
51 * - \ref escript
52 *
53 * - \ref esys_exception "Esys Exception"
54 *
55 * - \ref finley
56 *
57 * - <a href="../../epydoc/index.html">Python module documentation (epydoc generated)</a>
58 *
59 */
60
61 /*! \page escript Escript
62 * Escript is the python module that contains the interfaces
63 * to the C++ side of escript.
64 *
65 *
66 *
67 * \section class_desc Class Description:
68 * Data
69 *
70 * \section class_limits Class Limitations:
71 * None
72 *
73 * \section class_conds Class Conditions of Use:
74 * None
75 *
76 * \section class_throws Throws:
77 * None
78 *
79 */
80
81 BOOST_PYTHON_MODULE(escriptcpp)
82 {
83
84 #if BOOST_VERSION >= 103500
85 // params are: bool show_user_defined, bool show_py_signatures, bool show_cpp_signatures
86 docstring_options docopt(true,true,false);
87 #endif
88
89 def("setNumberOfThreads",escript::setNumberOfThreads,"Use of this method is strongly discouraged.");
90 def("getNumberOfThreads",escript::getNumberOfThreads,"Return the maximum number of threads"
91 " available to OpenMP.");
92 def("releaseUnusedMemory",escript::releaseUnusedMemory);
93 def("blocktimer_initialize",blocktimer_initialize);
94 def("blocktimer_reportSortByName",blocktimer_reportSortByName);
95 def("blocktimer_reportSortByTime",blocktimer_reportSortByTime);
96 def("blocktimer_increment",blocktimer_increment);
97 def("blocktimer_time",blocktimer_time);
98 def("getVersion",escript::getSvnVersion,"This method will only report accurate version numbers for clean checkouts.");
99 def("printParallelThreadCounts",escript::printParallelThreadCnt);
100 def("getMPISizeWorld",escript::getMPISizeWorld,"Return number of MPI processes in the job.");
101 def("getMPIRankWorld",escript::getMPIRankWorld,"Return the rank of this process in the MPI World.");
102 def("MPIBarrierWorld",escript::MPIBarrierWorld,"Wait until all MPI processes have reached this point.");
103 def("getMPIWorldMax",escript::getMPIWorldMax,"\nEach MPI process calls this function with a"
104 " value for arg1. The maximum value is computed and returned.\n\n:rtype: int");
105 def("getMPIWorldSum",escript::getMPIWorldSum,"\nEach MPI process calls this function with a"
106 " value for arg1. The values are added up and the total value is returned.\n\n:rtype: int");
107 def("getMachinePrecision",escript::getMachinePrecision);
108 def("getMaxFloat",escript::getMaxFloat);
109 def("_saveDataCSV",escript::saveDataCSV, (args("filename","arg","sep","csep"), arg("append")=false),
110 "Saves data objects passed in a python dictionary to a file.\n"
111 "The data objects must be over the same domain and be able to be interpolated to the same FunctionSpace.\n"
112 "If one of the dictionary keys is named ``mask``, then only samples where ``mask`` has a positive\n"
113 "value will be written to the file.\n\n"
114 "A header line giving the names of each column will be output first.\n"
115 "The keys given in the dictionary will be used to name columns.\n"
116 "Then the data will be output, one line per sample (for all data).\n"
117 "That is, items in each column will be printed in the same order.\n"
118 "So you can be sure that values in the same row correspond to the same input value.\n"
119 "\n:param filename:\n:type filename: ``string``\n"
120 ":param arg: dictionary of named `Data` objects. If one is called ``mask`` it must be scalar data."
121 "\n:type arg: ``dict``\n"
122 ":param sep:separator for columns (defaults to \", \"\n:type sep:``string``\n"
123 ":param csep:separator for fields within data object (defaults to \"_\"\n:type csep:``string``\n"
124 ":param append: If True, write to the end of ``filename``\n:type append: ``string``\n");
125
126
127 //
128 // Interface for AbstractDomain
129 //
130 class_<escript::AbstractDomain, escript::Domain_ptr>("Domain","Base class for all domains.",no_init)
131 .def("getStatus",&escript::AbstractDomain::getStatus,"The status of a domain changes whenever the domain is modified\n\n:rtype: int")
132 .def("setTagMap",&escript::AbstractDomain::setTagMap,args("name","tag"),
133 "Give a tag number a name.\n\n:param name: Name for the tag\n:type name: ``string``\n"
134 ":param tag: numeric id\n:type tag: ``int``\n:note: Tag names must be unique within a domain")
135 .def("getTag",&escript::AbstractDomain::getTag,args("name"),":return: tag id for "
136 "``name``\n:rtype: ``string``")
137 .def("isValidTagName",&escript::AbstractDomain::isValidTagName,args("name"),
138 ":return: True is ``name`` corresponds to a tag\n:rtype: ``bool``")
139 .def("showTagNames",&escript::AbstractDomain::showTagNames,":return: A space separated list of tag names\n:rtype: ``string``")
140 .def("getX",&escript::AbstractDomain::getX,":rtype: `Data`\n:return: Locations in the"
141 "`Domain`. FunctionSpace is chosen appropriately")
142 .def("getDim",&escript::AbstractDomain::getDim,":rtype: `int`\n:return: Spatial dimension of the `Domain`")
143 .def("getNormal",&escript::AbstractDomain::getNormal,":rtype: `escript`\n:return: Boundary normals")
144 .def("getSize",&escript::AbstractDomain::getSize,":return: the local size of samples. The function space is chosen appropriately\n:rtype: `Data`")
145 .def("saveVTK",&escript::AbstractDomain::saveVTK,args("filename","arg", "metadata"
146 , "metadata_schema"),
147 ":param filename: \n:type filename: ``string``\n:param arg: items to be added"
148 "\n:type arg: ``dict``\n:param metadata: string representing some meta data to be added\n:type metadata: ``string``"
149 "\n:param metadata_schema: schema type for metadata\n:type metadata_schema: ``string``")
150 .def("dump",&escript::AbstractDomain::dump,args("filename"),"Dumps the domain to a file"
151 ":param filename:\n:type filename: string")
152 .def("saveDX",&escript::AbstractDomain::saveDX,args("filename","arg"),"Saves a dictonary of Data objects to an OpenDX input file.\n\n:param filename:\n:type filename: ``string``"
153 "\n:param arg:\n:type arg: `dict`")
154 .def("getMPISize",&escript::AbstractDomain::getMPISize,":return: the number of processes used for this `Domain`\n:rtype: ``int``")
155 .def("getMPIRank",&escript::AbstractDomain::getMPIRank,":return: the rank of this process\n:rtype: ``int``")
156 .def("MPIBarrier",&escript::AbstractDomain::MPIBarrier,"Wait until all processes have reached this point")
157 .def("onMasterProcessor",&escript::AbstractDomain::onMasterProcessor,":return: True if this code is executing on the master process\n:rtype: `bool`")
158
159 .def(self == self)
160 .def(self != self);
161
162 //
163 // Interface for AbstractContinuousDomain
164 //
165 class_<escript::AbstractContinuousDomain, bases<escript::AbstractDomain> >("ContinuousDomain","Class representing continuous domains",no_init)
166 .def("getSystemMatrixTypeId",&escript::AbstractContinuousDomain::getSystemMatrixTypeId,
167 args("solver", "preconditioner", "package", "symmetry"),
168 ":return: the identifier of the matrix type to be used for the global stiffness matrix "
169 "when a particular solver package and symmetric matrix is used.\n"
170 ":rtype: int")
171 .def("getTransportTypeId",&escript::AbstractContinuousDomain::getTransportTypeId,
172 args("solver", "preconditioner", "package", "symmetry"));
173
174
175 //
176 // Interface for TestDomain
177 //
178 class_ <escript::TestDomain, bases<escript::AbstractDomain> >("TestDomain", "Test Class for domains with no structure. May be removed from future releases without notice.", init<int,int>());
179
180 // This is the only python visible way to get a TestDomain
181 def("getTestDomainFunctionSpace",&escript::getTestDomainFunctionSpace, args("dpps",
182 "samples"),
183 "For testing only. May be removed without notice.");
184
185 //
186 // Interface for FunctionSpace
187 //
188 class_<escript::FunctionSpace> fs_definer("FunctionSpace","A FunctionSpace describes which points from the `Domain` to use to represent functions.",init<>()); // Doco goes in the empty string param
189 fs_definer.def("getDim",&escript::FunctionSpace::getDim,":return: the spatial dimension of the underlying domain.\n:rtype: int");
190 // fs_definer.def("getDomain",&escript::FunctionSpace::getDomain,
191 // return_internal_reference<>());
192 fs_definer.def("getDomain",&escript::FunctionSpace::getDomainPython,":return: the underlying `Domain` for this FunctionSpace.\n:rtype: `Domain`");
193 fs_definer.def("getX",&escript::FunctionSpace::getX,"\n:return: a function whose values are its input coordinates. ie an identity function.\n:rtype: `Data`");
194 fs_definer.def("getNormal",&escript::FunctionSpace::getNormal,":return: the surface normal field.\n\n:rtype: `Data`");
195 fs_definer.def("getSize",&escript::FunctionSpace::getSize,":return: sample size\n:rtype: `Data`");
196 fs_definer.def("setTags",&escript::FunctionSpace::setTags,args("newtag","mask"),
197 "Set tags according to a mask\n\n:param newtag: tag number to set\n:type newtag: non-zero ``int``\n:param mask: Samples which correspond to positive values in the mask will be set to ``newtag``.\n:type mask: scalar `Data`");
198 fs_definer.def("setTags",&escript::FunctionSpace::setTagsByString,args("newtag","mask"),
199 "Set tags according to a mask\n\n:param newtag: tag name to set\n:type newtag: string\n:param mask: Samples which correspond to positive values in the mask will be set to ``newtag``.\n:type mask: scalar `Data`");
200 fs_definer.def("getTagFromDataPointNo",
201 &escript::FunctionSpace::getTagFromDataPointNo,":return: the tag associated with the given sample number.\n:rtype: int");
202 fs_definer.def("getReferenceIDFromDataPointNo", &escript::FunctionSpace::getReferenceIDFromDataPointNo,args("dataPointNo"),":return: the reference number associated with ``dataPointNo``\n:rtype: int ");
203 fs_definer.def("getListOfTags",&escript::FunctionSpace::getListOfTags,":return: a list of the tags used in this function space\n:rtype: ``list``");
204 fs_definer.def("__str__", &escript::FunctionSpace::toString);
205 fs_definer.def(self == self);
206 fs_definer.def(self != self);
207 //
208 // Interface for Data
209 //
210 class_<escript::Data>("Data","Represents a collection of datapoints. It is used to store the values of a function. For more details please consult the c++ class documentation.",init<>() )
211 // various constructors for Data objects
212 .def(init<const object&, optional<const escript::FunctionSpace&, bool> >(args("value","what","expand")))
213 .def(init<const double, const tuple&, optional<const escript::FunctionSpace&, bool> >(args("value","shape","what","expand")))
214 .def(init<const escript::Data&, const escript::FunctionSpace&>(args("value","what")))
215 .def(init<const escript::Data&>())
216 // Note for Lutz, Need to specify the call policy in order to return a
217 // reference. In this case return_internal_reference.
218 .def("__str__",&escript::Data::toString)
219 .def("getDomain",&escript::Data::getDomainPython,":rtype: `Domain`")
220 .def("getFunctionSpace",&escript::Data::getFunctionSpace,return_value_policy<copy_const_reference>(),":rtype: `FunctionSpace`")
221 .def("isEmpty",&escript::Data::isEmpty,"Is this object an instance of ``DataEmpty``\n\n:rtype: ``bool``\n:note: This is not the same thing as asking if the object contains datapoints.")
222 .def("isProtected",&escript::Data::isProtected,"Can this instance be modified.\n:rtype: ``bool``")
223 .def("setProtection",&escript::Data::setProtection,"Disallow modifications to this data object\n\n:note: This method does not allow you to undo protection.")
224 .def("getShape",&escript::Data::getShapeTuple,"\nReturns the shape of the datapoints in this object as a python tuple. Scalar data has the shape ``()``\n\n:rtype: ``tuple``")
225 .def("getRank",&escript::Data::getDataPointRank,":return: the number of indicies required to address a component of a datapoints\n:rtype: positive ``int``")
226 .def("dump",&escript::Data::dump,args("fileName"),"Save the data as a netCDF file\n\n:param fileName: \n:type fileName: ``string``")
227 .def("toListOfTuples",&escript::Data::toListOfTuples, (arg("scalarastuple")=false),
228 "Return the datapoints of this object in a list. Each datapoint is stored as a tuple.\n\n"
229 ":param scalarastuple: if True, scalar data will be wrapped as a tuple."
230 " True => [(0), (1), (2)]; False => [0, 1, 2]")
231 .def("copyWithMask",&escript::Data::copyWithMask,args("other","mask"),
232 "Selectively copy values from ``other`` `Data`."
233 "Datapoints which correspond to postive values in ``mask`` will be copied from ``other``\n"
234 "\n:param other: source of values\n"
235 ":type other: `Data`\n:param mask:\n:type mask: Scalar `Data`")
236 .def("setTaggedValue",&escript::Data::setTaggedValue,args("tagKey","value"),
237 "Set the value of tagged Data.\n\n:param tagKey: tag to update\n:type tagKey: ``int``\n")
238 .def("setTaggedValue",&escript::Data::setTaggedValueByName,args("name","value"),":param name: tag to update\n:type name: ``string``\n"
239 ":param value: value to set tagged data to\n:type value: ``object`` which acts like an array, ``tuple`` or ``list``\n")
240 .def("getNumberOfDataPoints",&escript::Data::getNumDataPoints,":rtype: ``int``\n:return: Number of datapoints in the object")
241 .def("isExpanded",&escript::Data::isExpanded,":rtype: ``bool``\n:return: True if this ``Data`` is expanded.")
242 .def("isTagged",&escript::Data::isTagged,":rtype: ``bool``\n:return: True if this ``Data`` is expanded.")
243 .def("isConstant",&escript::Data::isConstant,":rtype: ``bool``\n:return: True if this ``Data`` is an instance of ``DataConstant``\n:note: This does not mean the data is immutable.")
244 .def("isLazy",&escript::Data::isLazy,":rtype: ``bool``\n:return: True if this ``Data`` is lazy.")
245 .def("isReady",&escript::Data::isReady,":rtype: ``bool``\n:return: True if this ``Data`` is not lazy.")
246 .def("expand",&escript::Data::expand,"Convert the data to expanded representation if it is not expanded already.")
247 .def("tag",&escript::Data::tag,"Convert data to tagged representation if it is not already tagged or expanded")
248 .def("resolve",&escript::Data::resolve,"Convert the data to non-lazy representation.")
249 .def("copy",&escript::Data::copy,args("other"),"Make this object a copy of ``other``\n"
250 "\n:note: The two objects will act independently from now on. That is, changing ``other`` "
251 "after this call will not change this object and vice versa.")
252 .def("copy",&escript::Data::copySelf,":note: In the no argument form, a new object will be returned which is an independent copy of this object.")
253 .def("delay",&escript::Data::delay,"Convert this object into lazy representation")
254 .def("setValueOfDataPoint",&escript::Data::setValueOfDataPointToPyObject,args("dataPointNo","value"))
255 .def("setValueOfDataPoint",&escript::Data::setValueOfDataPointToArray)
256 .def("setValueOfDataPoint",&escript::Data::setValueOfDataPoint,"\nModify the value of a single datapoint.\n\n:param dataPointNo:\n"
257 ":type dataPointNo: int\n:param value: \n:type value: ``float`` or an object which acts like an array, ``tuple`` or ``list``\n:warning: Use of this operation is discouraged. It prevents some optimisations from operating.")
258 .def("getTupleForDataPoint",&escript::Data::getValueOfDataPointAsTuple,args("dataPointNo"),
259 ":return: Value of the specified datapoint\n:rtype: ``tuple``\n:param dataPointNo: datapoint to access\n:type dataPointNo: ``int``")
260 .def("getTupleForGlobalDataPoint",&escript::Data::getValueOfGlobalDataPointAsTuple,args("procNo","dataPointNo"),"Get a specific datapoint from a specific process\n\n"
261 ":rtype: ``tuple``\n:param procNo: MPI rank of the process\n:type procNo: positive ``int``"
262 "\n:param dataPointNo: datapoint to access\n:type dataPointNo: int")
263 .def("setToZero",&escript::Data::setToZero,"After this call the object will store values of the same shape as before but all components will be zero.")
264 .def("interpolate",&escript::Data::interpolate,args("functionspace"),"Interpolate this object's values into a new functionspace.")
265 .def("interpolateTable", &escript::Data::interpolateFromTable2DP,
266 (arg("table"),arg("Amin"),arg("Astep"), arg("B"), arg("Bmin"), arg("Bstep"), arg("undef")=1.e50, arg("check_boundaries")=false),
267 "Creates a new Data object by interpolating using the source data (which are\n"
268 "looked up in ``table``)\n``A`` must be the outer dimension on the table\n\n"
269 ":param table: two dimensional collection of values\n"
270 ":param Amin: The base of locations in table\n:type Amin: float\n"
271 ":param Astep: size of gap between each item in the table\n:type Astep: float\n"
272 ":param undef: upper bound on interpolated values\n:type undef: float\n"
273 ":param B: Scalar representing the second coordinate to be mapped into the table\n"
274 ":type B: `Data`\n"
275 ":param Bmin: The base of locations in table for 2nd dimension\n:type Bmin: float\n"
276 ":param Bstep: size of gap between each item in the table for 2nd dimension\n:type Bstep: float\n"
277 ":param check_boundaries: if true, then values outside the boundaries will be rejected. If false, then boundary values will be used.\n"
278 ":raise RuntimeError(DataException): if the cordinates do not map into the table or if the interpolated value is above ``undef``"
279 "\n:rtype: `Data`"
280 )
281 .def("interpolateTable", &escript::Data::interpolateFromTable1DP,
282 (arg("table"),arg("Amin"),arg("Astep"), arg("undef")=1.e50, arg("check_boundaries")=false),
283 "Creates a new Data object by interpolating using the source data (which are\n"
284 "looked up in ``table``)\n\n"
285 ":param table: one dimensional collection of values\n"
286 ":param Amin: The base of locations in table\n:type Amin: float\n"
287 ":param Astep: size of gap between each item in the table\n:type Astep: float\n"
288 ":param undef: upper bound on interpolated values\n:type undef: float\n"
289 ":param check_boundaries: if true, then values outside the boundaries will be rejected. If false, then boundary values will be used.\n"
290 ":raise RuntimeError(DataException): if the cordinates do not map into the table or if the interpolated value is above ``undef``"
291 "\n:rtype: `Data`"
292 )
293
294
295 .def("minGlobalDataPoint",&escript::Data::minGlobalDataPoint,"Please consider using getInfLocator() from pdetools instead.")
296 .def("maxGlobalDataPoint",&escript::Data::maxGlobalDataPoint, "Please consider using getSupLocator() from pdetools instead.")
297 .def("saveDX",&escript::Data::saveDX,args("fileName"),"Save the object in DX format.\n\n"
298 ":param fileName: filename\n:type fileName: string")
299 .def("saveVTK",&escript::Data::saveVTK, args("fileName"),"Save the object in VTK format.\n\n"
300 ":param fileName: \n:type fileName: string")
301 .def("getTagNumber",&escript::Data::getTagNumber,args("dpno"),"Return tag number for the specified datapoint\n\n:rtype: int\n:param dpno: datapoint number\n:type dpno: int")
302 // Unary functions for Data
303 .def("_interpolate",&escript::Data::interpolate)
304 .def("_grad",&escript::Data::gradOn)
305 .def("_grad",&escript::Data::grad)
306 .def("_transpose",&escript::Data::transpose)
307 .def("_trace",&escript::Data::trace)
308 .def("_maxval",&escript::Data::maxval)
309 .def("_minval",&escript::Data::minval)
310 .def("_wherePositive",&escript::Data::wherePositive)
311 .def("_whereNegative",&escript::Data::whereNegative)
312 .def("_whereNonNegative",&escript::Data::whereNonNegative)
313 .def("_whereNonPositive",&escript::Data::whereNonPositive)
314 .def("_whereZero",&escript::Data::whereZero,(arg("tol")=0.0))
315 .def("_whereNonZero",&escript::Data::whereNonZero,(arg("tol")=0.0))
316 .def("_erf",&escript::Data::erf)
317 .def("_sin",&escript::Data::sin)
318 .def("_cos",&escript::Data::cos)
319 .def("_tan",&escript::Data::tan)
320 .def("_asin",&escript::Data::asin)
321 .def("_acos",&escript::Data::acos)
322 .def("_atan",&escript::Data::atan)
323 .def("_sinh",&escript::Data::sinh)
324 .def("_cosh",&escript::Data::cosh)
325 .def("_tanh",&escript::Data::tanh)
326 .def("_asinh",&escript::Data::asinh)
327 .def("_acosh",&escript::Data::acosh)
328 .def("_atanh",&escript::Data::atanh)
329 .def("_exp",&escript::Data::exp)
330 .def("_sqrt",&escript::Data::sqrt)
331 .def("_log10",&escript::Data::log10)
332 .def("_log",&escript::Data::log)
333 .def("_sign",&escript::Data::sign)
334 .def("_symmetric",&escript::Data::symmetric)
335 .def("_nonsymmetric",&escript::Data::nonsymmetric)
336 .def("_trace",&escript::Data::trace)
337 .def("_swap_axes",&escript::Data::swapaxes)
338 .def("_eigenvalues",&escript::Data::eigenvalues)
339 .def("_eigenvalues_and_eigenvectors",&escript::Data::eigenvalues_and_eigenvectors,(arg("tol")=1.e-13))
340 // functions returning a single real number:
341 .def("_Lsup",&escript::Data::Lsup,":return: the Lsup-norm of the object\n:rtype: float\n:note: If the ``Data`` contains no values, zero will be returned instead.")
342 .def("_sup",&escript::Data::sup,":return: the maximum value over all data points.\n:rtype: float\n:note: If the ``Data`` contains no values a large negative value will be returned instead.")
343 .def("_inf",&escript::Data::inf,":return: minimum value over all components and all data points\n:rtype: float\n:note: If the ``Data`` contains no values a large positive value will be returned instead.")
344 .def("_integrateToTuple",&escript::Data::integrateToTuple,":return: Calculate the integral over the function space domain as a python tuple\n:rtype: tuple")
345 // following implements the python abs operator
346 .def("__abs__",&escript::Data::abs,":return: absolute value\n\n:rtype: `Data`")
347 // following implements the python "-" negation operator
348 .def("__neg__",&escript::Data::neg, ":return: negation of the values in this object\n:rtype: `Data`")
349 // following implements the python "+" identity operator
350 .def("__pos__",&escript::Data::pos, "\nThe unary + operator\n\n:rtype: `Data`")
351 // following two functions implement the python [] operator
352 .def("__getitem__",&escript::Data::getItem,"Used by the python [] operator\n\n:rtype: `Data`")
353 .def("__setitem__",&escript::Data::setItemO,"Used by the python [] operator")
354 .def("__setitem__",&escript::Data::setItemD,"Used by the python [] operator")
355 // following two functions implement the python ** operator
356 .def("__pow__",&escript::Data::powO,"Used by the python ** operator\n\n:rtype: `Data`")
357 .def("__pow__",&escript::Data::powD)
358 .def("__rpow__",&escript::Data::rpowO,"\nUsed by the python ** operator\n\n:rtype: `Data`")
359 // NOTE:: The order of these declarations is important. Anything
360 // declared before the generic declaration isn't found so the generic
361 // version will be called.
362 .def(self + other<object>())
363 .def(other<object>() + self)
364 .def(self + self)
365 .def(self += other<object>())
366 .def(self += self)
367
368 .def(self - other<object>())
369 .def(other<object>() - self)
370 .def(self - self)
371 .def(self -= other<object>())
372 .def(self -= self)
373
374 .def(self * other<object>())
375 .def(other<object>() * self)
376 .def(self * self)
377 .def(self *= other<object>())
378 .def(self *= self)
379
380 .def(self / other<object>())
381 .def(other<object>() / self)
382 .def(self / self)
383 .def(self /= other<object>())
384 .def(self /= self)
385 // Need scope resolution due to a bug either in the compiler or
386 // the boost code. This calls operator << for Data.
387 .def(self_ns::str(self))
388 .def("_inverse", &escript::Data::matrixInverse, ":return: inverse of square matricies\n");
389
390
391 //
392 // Factory methods for function space
393 //
394 def("ContinuousFunction",escript::continuousFunction,args("domain"),
395 ":return: a continuous FunctionSpace (overlapped node values)\n"
396 ":rtype: `FunctionSpace`");
397 def("ReducedContinuousFunction",escript::reducedContinuousFunction,args("domain"),
398 ":return: a continuous with reduced order FunctionSpace (overlapped node values on reduced element order)\n"
399 ":rtype: `FunctionSpace`");
400 def("Function",escript::function,args("domain"),":return: a function `FunctionSpace`\n"
401 ":rtype: `FunctionSpace`");
402 def("ReducedFunction",escript::reducedFunction, args("domain"),":return: a function FunctionSpace with reduced integration order\n:rtype: `FunctionSpace`");
403 def("FunctionOnBoundary",escript::functionOnBoundary, args("domain"), ":return: a function on boundary FunctionSpace\n:rtype: `FunctionSpace`");
404 def("ReducedFunctionOnBoundary",escript::reducedFunctionOnBoundary, args("domain"),
405 ":return: a function on boundary FunctionSpace with reduced integration order\n"
406 ":rtype: `FunctionSpace`");
407 def("FunctionOnContactZero",escript::functionOnContactZero, args("domain"), ":return: Return a FunctionSpace on left side of contact\n:rtype: `FunctionSpace`");
408 def("ReducedFunctionOnContactZero",escript::reducedFunctionOnContactZero, args("domain"),
409 ":return: a FunctionSpace on left side of contact with reduced integration order\n:rtype: `FunctionSpace`");
410 def("FunctionOnContactOne",escript::functionOnContactOne, args("domain"), ":return: Return a FunctionSpace on right side of contact\n:rtype: `FunctionSpace`");
411 def("ReducedFunctionOnContactOne",escript::reducedFunctionOnContactOne, args("domain"),
412 ":return: Return a FunctionSpace on right side of contact with reduced integration order\n"
413 ":rtype: `FunctionSpace`");
414 def("Solution",escript::solution, args("domain"), ":rtype: `FunctionSpace`");
415 def("ReducedSolution",escript::reducedSolution, args("domain"), ":rtype: `FunctionSpace`");
416 def("DiracDeltaFunction",escript::diracDeltaFunction, args("domain"), ":rtype: `FunctionSpace`");
417
418
419
420
421
422 //
423 // Factory methods for Data
424 //
425 def("load",escript::load, args("fileName","domain"), "reads Data on domain from file in netCDF format\n\n:param fileName:\n:type fileName: ``string``\n:param domain:\n:type domain: `Domain`");
426 def("loadIsConfigured",escript::loadConfigured,":return: True if the load function is configured.");
427 def("Scalar",escript::Scalar,
428 (arg("value")=0.0,
429 arg("what")=escript::FunctionSpace(),
430 arg("expanded")=false),
431 "Construct a Data object containing scalar data-points.\n\n:param value: scalar value for all points\n"
432 "\n:rtype: `Data`\n:type value: float\n:param what: FunctionSpace for Data\n:type what: `FunctionSpace`\n"
433 ":param expanded: If True, a value is stored for each point. If False, more efficient representations may be used\n"
434 ":type expanded: ``bool``");
435 def("Vector",escript::Vector,
436 (arg("value")=0.0,
437 arg("what")=escript::FunctionSpace(),
438 arg("expanded")=false),
439 "Construct a Data object containing rank1 data-points.\n\n:param value: scalar value for all points\n"
440 "\n:rtype: `Data`\n:type value: float\n:param what: FunctionSpace for Data\n:type what: `FunctionSpace`\n"
441 ":param expanded: If True, a value is stored for each point. If False, more efficient representations may be used\n"
442 ":type expanded: ``bool``");
443 def("Vector", escript::VectorFromObj,
444 (arg("value"),
445 arg("what")=escript::FunctionSpace(),
446 arg("expanded")=false));
447 def("Tensor",escript::Tensor,
448 (arg("value")=0.0,
449 arg("what")=escript::FunctionSpace(),
450 arg("expanded")=false),
451 "Construct a Data object containing rank2 data-points.\n\n:param value: scalar value for all points\n"
452 "\n:rtype: `Data`\n:type value: float\n:param what: FunctionSpace for Data\n:type what: `FunctionSpace`\n"
453 ":param expanded: If True, a value is stored for each point. If False, more efficient representations may be used\n"
454 ":type expanded: ``bool``");
455 def("Tensor", escript::TensorFromObj,
456 (arg("value"),
457 arg("what")=escript::FunctionSpace(),
458 arg("expanded")=false));
459 def("Tensor3",escript::Tensor3,
460 (arg("value")=0.0,
461 arg("what")=escript::FunctionSpace(),
462 arg("expanded")=false),
463 "Construct a Data object containing rank3 data-points.\n\n:param value: scalar value for all points\n"
464 "\n:rtype: `Data`\n:type value: float\n:param what: FunctionSpace for Data\n:type what: `FunctionSpace`\n"
465 ":param expanded: If True, a value is stored for each point. If False, more efficient representations may be used\n"
466 ":type expanded: ``bool``"
467 );
468 def("Tensor3", escript::Tensor3FromObj,
469 (arg("value"),
470 arg("what")=escript::FunctionSpace(),
471 arg("expanded")=false));
472 def("Tensor4",escript::Tensor4,
473 (arg("value")=0.0,
474 arg("what")=escript::FunctionSpace(),
475 arg("expanded")=false),
476 "Construct a Data object containing rank4 data-points.\n\n:param value: scalar value for all points\n"
477 "\n:rtype: `Data`\n:type value: float\n:param what: FunctionSpace for Data\n:type what: `FunctionSpace`\n"
478 ":param expanded: If True, a value is stored for each point. If False, more efficient representations may be used\n"
479 ":type expanded: ``bool``"
480 );
481 def("Tensor4", escript::Tensor4FromObj,
482 (arg("value"),
483 arg("what")=escript::FunctionSpace(),
484 arg("expanded")=false));
485
486 //
487 // Binary operators
488 //
489 def("C_GeneralTensorProduct",escript::C_GeneralTensorProduct,
490 (arg("arg0"),
491 arg("arg1"),
492 arg("axis_offset")=0,
493 arg("transpose")=0),
494 "Compute a tensor product of two Data objects.\n\n:rtype: `Data`\n:param arg0:\n"
495 ":param arg1:\n:param axis_offset:\n:type axis_offset: ``int``\n"
496 ":param transpose: 0: transpose neither, 1: transpose arg0, 2: transpose arg1\n"
497 ":type transpose: int");
498
499 //
500 // Interface for AbstractSystemMatrix
501 //
502 class_<escript::AbstractSystemMatrix>("Operator","",init<>()) // Doco goes in the empty string param
503 .def("isEmpty",&escript::AbstractSystemMatrix::isEmpty,":rtype: ``bool``\n"
504 ":return: True if matrix is empty")
505 .def("solve",&escript::AbstractSystemMatrix::solve, args("in","options"),
506 ":return: the solution *u* of the linear system *this*u=in*\n\n:param in:\n:type in: `Data`")
507 .def("of",&escript::AbstractSystemMatrix::vectorMultiply,args("right"),
508 "matrix*vector multiplication")
509 .def("saveMM",&escript::AbstractSystemMatrix::saveMM, args("fileName"),
510 "writes the matrix to a file using the Matrix Market file format")
511 .def("saveHB",&escript::AbstractSystemMatrix::saveHB, args("filename"),
512 "writes the matrix to a file using the Harwell-Boeing file format")
513 .def("resetValues",&escript::AbstractSystemMatrix::resetValues, "resets the matrix entries")
514 .def(self*other<escript::Data>());
515 //
516 // Interface for AbstractTransportProblem
517 //
518 class_<escript::AbstractTransportProblem>("TransportProblem","",init<>()) // Doco goes in the empty string param
519 .def("isEmpty",&escript::AbstractTransportProblem::isEmpty,":rtype: ``int``")
520 .def("solve",&escript::AbstractTransportProblem::solve, args("source","dt", "options"),
521 "returns the solution *u* for a time step *dt>0*\n\n:rtype: `Data`\n"
522 ":param source:\n:type source: `Data`")
523 .def("setInitialValue",&escript::AbstractTransportProblem::setInitialValue, args("u"),
524 "sets the value for *u* at time *t=0*.")
525 .def("insertConstraint",&escript::AbstractTransportProblem::insertConstraint,
526 args("source", "q", "r","factor"),
527 "inserts constraint *u_{,t}=r* where *q>0* into the problem using a weighting factor")
528 .def("reset",&escript::AbstractTransportProblem::resetTransport,
529 "resets the transport operator typically as they have been updated.")
530 .def("resetValues",&escript::AbstractTransportProblem::resetTransport)
531 .def("getSafeTimeStepSize",&escript::AbstractTransportProblem::getSafeTimeStepSize)
532 .def("getUnlimitedTimeStepSize",&escript::AbstractTransportProblem::getUnlimitedTimeStepSize);
533
534 // Functions to modify global parameters
535 def("setEscriptParamInt",escript::setEscriptParamInt,
536 (arg("name"), arg("value")=0), "Modify the value of an escript tuning parameter\n\n"
537 ":param name:\n:type name: ``string``\n:param value:\n:type value: ``int``");
538 def("getEscriptParamInt",escript::getEscriptParamInt,
539 (arg("name"),arg("sentinel")=0), "Read the value of an escript tuning paramter\n\n"
540 ":param name: parameter to lookup\n:type name: ``string``\n:param sentinel: Value to be returned if ``name`` is not a known parameter\n"
541 ":type sentinel: ``int``");
542 def("listEscriptParams",escript::listEscriptParams,":return: A list of pairs (p,d) where p is the name of a parameter for escript and d is a description.");
543
544
545 def("resolveGroup", escript::resolveGroup);
546
547 #ifdef IKNOWWHATIMDOING
548
549 def("applyBinaryCFunction", escript::applyBinaryCFunction, (arg("function"), arg("outshape"),
550 arg("in1"),
551 arg("in2"))
552 );
553 #endif
554
555 //
556 // Register esysExceptionTranslator
557 //
558 register_exception_translator<esysUtils::EsysException>(&esysUtils::esysExceptionTranslator);
559 }

Properties

Name Value
svn:eol-style native
svn:keywords Author Date Id Revision

  ViewVC Help
Powered by ViewVC 1.1.26