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

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1034 - (hide annotations)
Wed Mar 14 23:49:20 2007 UTC (12 years, 8 months ago) by gross
File size: 11286 byte(s)
In VC++ boost has problems with numarray arguments from python. This
fixes that problem by taking python::object arguments from the python
level and converting it into python::numeric::array on the C++ level.
This hasn't been tested with VC++ yet.

Moreover the two Data methods dealing with big numarrays as argument and
return value have been removed.


1 jgs 102 //$Id$
2 elspeth 615 /*
3     ************************************************************
4     * Copyright 2006 by ACcESS MNRF *
5     * *
6     * http://www.access.edu.au *
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     ************************************************************
12     */
13 jgs 102
14 jgs 474 #include "Data.h"
15     #include "FunctionSpace.h"
16     #include "FunctionSpaceFactory.h"
17     #include "DataFactory.h"
18     #include "AbstractContinuousDomain.h"
19     #include "AbstractDomain.h"
20     #include "Utils.h"
21 jgs 480 #include "AbstractSystemMatrix.h"
22 gross 797 #include "DataVector.h"
23 jgs 474
24 robwdcock 682 #include "esysUtils/esysExceptionTranslator.h"
25 jgs 102
26     #include <boost/python.hpp>
27     #include <boost/python/module.hpp>
28     #include <boost/python/def.hpp>
29     #include <boost/python/object.hpp>
30     #include <boost/python/tuple.hpp>
31 gross 285 #include <boost/python/numeric.hpp>
32    
33 jgs 102 using namespace boost::python;
34    
35 jgs 121 /*! \mainpage Esys Documentation
36     *
37     * \version 1.0.0
38     *
39     * - \ref escript
40     *
41     * - \ref esys_exception "Esys Exception"
42     *
43     * - \ref finley
44     *
45 cochrane 358 * - <a href=http://iservo.edu.au/esys/epydoc/index.html>Python module documentation (epydoc generated)</a>
46 jgs 122 *
47 jgs 121 */
48 jgs 102
49 jgs 121 /*! \page escript Escript
50     * Escript is the python module that contains the interfaces
51     * to the C++ side of escript.
52     *
53     * \version 1.0.0
54     *
55     * \section class_desc Class Description:
56     * Data
57     *
58     * \section class_limits Class Limitations:
59     * None
60     *
61     * \section class_conds Class Conditions of Use:
62     * None
63     *
64     * \section class_throws Throws:
65     * None
66     *
67     */
68 jgs 102
69     BOOST_PYTHON_MODULE(escriptcpp)
70     {
71 gross 391 def("setNumberOfThreads",escript::setNumberOfThreads);
72     def("getNumberOfThreads",escript::getNumberOfThreads);
73 gross 797 def("releaseUnusedMemory",escript::releaseUnusedMemory);
74 jgs 102
75 gross 797
76 jgs 102 //
77     // Interface for AbstractDomain
78     //
79     class_<escript::AbstractDomain>("Domain",no_init)
80     .def("getX",&escript::AbstractDomain::getX)
81     .def("getNormal",&escript::AbstractDomain::getNormal)
82     .def("getSize",&escript::AbstractDomain::getSize)
83 jgs 153 .def("saveVTK",&escript::AbstractDomain::saveVTK)
84     .def("saveDX",&escript::AbstractDomain::saveDX)
85 jgs 102 .def(self == self)
86     .def(self != self);
87    
88     //
89     // Interface for AbstractContinuousDomain
90     //
91     class_<escript::AbstractContinuousDomain, bases<escript::AbstractDomain> >("ContinuousDomain",no_init)
92     .def("getSystemMatrixTypeId",&escript::AbstractContinuousDomain::getSystemMatrixTypeId);
93    
94     //
95     // Interface for FunctionSpace
96     //
97     class_<escript::FunctionSpace>("FunctionSpace",init<>())
98     .def("getDim",&escript::FunctionSpace::getDim)
99 jgs 153 .def("getDomain",&escript::FunctionSpace::getDomain,return_internal_reference<>())
100 jgs 102 .def("getX",&escript::FunctionSpace::getX)
101     .def("getNormal",&escript::FunctionSpace::getNormal)
102     .def("getSize",&escript::FunctionSpace::getSize)
103 gross 767 .def("setTags",&escript::FunctionSpace::setTags)
104 jgs 149 .def("getTagFromDataPointNo",&escript::FunctionSpace::getTagFromDataPointNo)
105 jgs 147 .def("__str__",&escript::FunctionSpace::toString)
106 jgs 102 .def(self == self)
107     .def(self != self);
108     //
109     // Interface for Data
110     //
111 jgs 149 class_<escript::Data>("Data","TEST DOCUMENTATION",init<>())
112 jgs 102 // various constructors for Data objects
113     .def(init<const numeric::array&, optional<const escript::FunctionSpace&, bool> >(args("value","what","expand")))
114     .def(init<const object&, optional<const escript::FunctionSpace&, bool> >(args("value","what","expand")))
115     .def(init<const double, const tuple&, optional<const escript::FunctionSpace&, bool> >(args("value","shape","what","expand")))
116     .def(init<const escript::Data&, const escript::FunctionSpace&>(args("value","what")))
117     .def(init<const escript::Data&>())
118     // Note for Lutz, Need to specify the call policy in order to return a
119     // reference. In this case return_internal_reference.
120 gross 285 .def("__str__",&escript::Data::toString)
121 jgs 102 .def("getDomain",&escript::Data::getDomain,return_internal_reference<>())
122     .def("getFunctionSpace",&escript::Data::getFunctionSpace,return_internal_reference<>())
123     .def("isEmpty",&escript::Data::isEmpty)
124 gross 783 .def("isProtected",&escript::Data::isProtected)
125     .def("setProtection",&escript::Data::setProtection)
126 jgs 102 .def("getShape",&escript::Data::getShapeTuple)
127     .def("getRank",&escript::Data::getDataPointRank)
128 gross 950 .def("dump",&escript::Data::dump)
129 jgs 102 .def("copyWithMask",&escript::Data::copyWithMask)
130     .def("setTaggedValue",&escript::Data::setTaggedValue)
131 gross 921 .def("getNumberOfDataPoints",&escript::Data::getNumDataPoints)
132 jgs 119 .def("expand",&escript::Data::expand)
133     .def("tag",&escript::Data::tag)
134 jgs 126 .def("copy",&escript::Data::copy)
135 gross 1034 .def("setValueOfDataPoint",&escript::Data::setValueOfDataPointToPyObject)
136     .def("setValueOfDataPoint",&escript::Data::setValueOfDataPointToArray)
137 gross 921 .def("setValueOfDataPoint",&escript::Data::setValueOfDataPoint)
138     .def("getValueOfDataPoint",&escript::Data::getValueOfDataPoint)
139     .def("getValueOfGlobalDataPoint",&escript::Data::getValueOfGlobalDataPoint)
140 gross 285 .def("interpolate",&escript::Data::interpolate)
141 gross 921 .def("minGlobalDataPoint",&escript::Data::minGlobalDataPoint)
142 gross 285 .def("saveDX",&escript::Data::saveDX)
143     .def("saveVTK",&escript::Data::saveVTK)
144 jgs 149 .def("getTagNumber",&escript::Data::getTagNumber)
145 jgs 119 .def("archiveData",&escript::Data::archiveData)
146     .def("extractData",&escript::Data::extractData)
147 gross 285 // Unary functions for Data
148     .def("_interpolate",&escript::Data::interpolate)
149     .def("_grad",&escript::Data::gradOn)
150     .def("_grad",&escript::Data::grad)
151     .def("_transpose",&escript::Data::transpose)
152     .def("_trace",&escript::Data::trace)
153     .def("_maxval",&escript::Data::maxval)
154     .def("_minval",&escript::Data::minval)
155 gross 698 .def("_wherePositive",&escript::Data::wherePositive)
156     .def("_whereNegative",&escript::Data::whereNegative)
157     .def("_whereNonNegative",&escript::Data::whereNonNegative)
158     .def("_whereNonPositive",&escript::Data::whereNonPositive)
159 jgs 571 .def("_whereZero",&escript::Data::whereZero,(arg("tol")=0.0))
160     .def("_whereNonZero",&escript::Data::whereNonZero,(arg("tol")=0.0))
161 ksteube 876 .def("_erf",&escript::Data::erf)
162 gross 285 .def("_sin",&escript::Data::sin)
163     .def("_cos",&escript::Data::cos)
164     .def("_tan",&escript::Data::tan)
165     .def("_asin",&escript::Data::asin)
166     .def("_acos",&escript::Data::acos)
167     .def("_atan",&escript::Data::atan)
168     .def("_sinh",&escript::Data::sinh)
169     .def("_cosh",&escript::Data::cosh)
170     .def("_tanh",&escript::Data::tanh)
171     .def("_asinh",&escript::Data::asinh)
172     .def("_acosh",&escript::Data::acosh)
173     .def("_atanh",&escript::Data::atanh)
174     .def("_exp",&escript::Data::exp)
175     .def("_sqrt",&escript::Data::sqrt)
176     .def("_log10",&escript::Data::log10)
177     .def("_log",&escript::Data::log)
178     .def("_sign",&escript::Data::sign)
179 ksteube 775 .def("_symmetric",&escript::Data::symmetric)
180     .def("_nonsymmetric",&escript::Data::nonsymmetric)
181 gross 800 .def("_trace",&escript::Data::trace)
182 gross 804 .def("_swap_axes",&escript::Data::swapaxes)
183 gross 576 .def("_eigenvalues",&escript::Data::eigenvalues)
184     .def("_eigenvalues_and_eigenvectors",&escript::Data::eigenvalues_and_eigenvectors,(arg("tol")=1.e-13))
185 gross 285 // functions returning a single real number:
186     .def("_Lsup",&escript::Data::Lsup)
187     .def("_sup",&escript::Data::sup)
188     .def("_inf",&escript::Data::inf)
189     .def("_integrate",&escript::Data::integrate)
190    
191     // following implements the python abs operator
192 jgs 108 .def("__abs__",&escript::Data::abs)
193 jgs 102 // following implements the python "-" negation operator
194     .def("__neg__",&escript::Data::neg)
195     // following implements the python "+" identity operator
196     .def("__pos__",&escript::Data::pos)
197     // following two functions implement the python [] operator
198     .def("__getitem__",&escript::Data::getItem)
199     .def("__setitem__",&escript::Data::setItemO)
200     .def("__setitem__",&escript::Data::setItemD)
201     // following two functions implement the python ** operator
202     .def("__pow__",&escript::Data::powO)
203     .def("__pow__",&escript::Data::powD)
204 gross 699 .def("__rpow__",&escript::Data::rpowO)
205 jgs 102 // NOTE:: The order of these declarations is important. Anything
206     // declared before the generic declaration isn't found so the generic
207     // version will be called.
208     .def(self + other<object>())
209     .def(other<object>() + self)
210     .def(self + self)
211 gross 285 .def(self += other<object>())
212     .def(self += self)
213    
214 jgs 102 .def(self - other<object>())
215     .def(other<object>() - self)
216     .def(self - self)
217 gross 285 .def(self -= other<object>())
218     .def(self -= self)
219    
220 jgs 102 .def(self * other<object>())
221     .def(other<object>() * self)
222     .def(self * self)
223 gross 285 .def(self *= other<object>())
224     .def(self *= self)
225    
226 jgs 102 .def(self / other<object>())
227     .def(other<object>() / self)
228     .def(self / self)
229 gross 285 .def(self /= other<object>())
230     .def(self /= self)
231 jgs 102 // Need scope resolution due to a bug either in the compiler or
232     // the boost code. This calls operator << for Data.
233     .def(self_ns::str(self));
234    
235     //
236     // Factory methods for function space
237     //
238     def("ContinuousFunction",escript::continuousFunction);
239     def("Function",escript::function);
240     def("FunctionOnBoundary",escript::functionOnBoundary);
241     def("FunctionOnContactZero",escript::functionOnContactZero);
242     def("FunctionOnContactOne",escript::functionOnContactOne);
243     def("Solution",escript::solution);
244     def("ReducedSolution",escript::reducedSolution);
245     def("DiracDeltaFunction",escript::diracDeltaFunction);
246    
247     //
248     // Factory methods for Data
249     //
250 gross 950 def("load",escript::load);
251 gross 1023 def("loadIsConfigured",escript::loadConfigured);
252 jgs 102 def("Scalar",escript::Scalar,
253     (arg("value")=0.0,
254     arg("what")=escript::FunctionSpace(),
255     arg("expanded")=false));
256     def("Vector",escript::Vector,
257     (arg("value")=0.0,
258     arg("what")=escript::FunctionSpace(),
259     arg("expanded")=false));
260     def("Tensor",escript::Tensor,
261     (arg("value")=0.0,
262     arg("what")=escript::FunctionSpace(),
263     arg("expanded")=false));
264     def("Tensor3",escript::Tensor3,
265     (arg("value")=0.0,
266     arg("what")=escript::FunctionSpace(),
267     arg("expanded")=false));
268     def("Tensor4",escript::Tensor4,
269     (arg("value")=0.0,
270     arg("what")=escript::FunctionSpace(),
271     arg("expanded")=false));
272    
273     //
274 ksteube 813 // Binary operators
275     //
276     def("C_GeneralTensorProduct",escript::C_GeneralTensorProduct,
277     (arg("arg0")=escript::Data(),
278     arg("arg1")=escript::Data(),
279     arg("axis_offset")=0,
280     arg("transpose")=0));
281    
282     //
283 jgs 102 // Interface for AbstractSystemMatrix
284     //
285     class_<escript::AbstractSystemMatrix>("Operator",init<>())
286     .def("isEmpty",&escript::AbstractSystemMatrix::isEmpty)
287     .def("solve",&escript::AbstractSystemMatrix::solve)
288     .def("of",&escript::AbstractSystemMatrix::vectorMultiply)
289     .def("saveMM",&escript::AbstractSystemMatrix::saveMM)
290 jgs 123 .def("saveHB",&escript::AbstractSystemMatrix::saveHB)
291 jgs 149 .def("resetValues",&escript::AbstractSystemMatrix::resetValues)
292 jgs 102 .def(self*other<escript::Data>());
293    
294     //
295     // Register esysExceptionTranslator
296     //
297     register_exception_translator<esysUtils::EsysException>(&esysUtils::esysExceptionTranslator);
298    
299     }

Properties

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

  ViewVC Help
Powered by ViewVC 1.1.26