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

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 775 - (show annotations)
Mon Jul 10 04:00:08 2006 UTC (13 years, 2 months ago) by ksteube
File size: 10612 byte(s)
Modified the following python methods in escript/py_src/util.py to
call faster C++ methods:
	escript_trace
	escript_transpose
	escript_symmetric
	escript_nonsymmetric

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

Properties

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

  ViewVC Help
Powered by ViewVC 1.1.26