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

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

branches/ripleygmg_from_3668/ripley/src/ripleycpp.cpp revision 3791 by caltinay, Wed Feb 1 05:10:22 2012 UTC trunk/ripley/src/ripleycpp.cpp revision 4334 by caltinay, Thu Mar 21 06:00:14 2013 UTC
# Line 1  Line 1 
1    
2  /*******************************************************  /*****************************************************************************
3  *  *
4  * Copyright (c) 2003-2012 by University of Queensland  * Copyright (c) 2003-2013 by University of Queensland
5  * Earth Systems Science Computational Center (ESSCC)  * http://www.uq.edu.au
 * http://www.uq.edu.au/esscc  
6  *  *
7  * Primary Business: Queensland, Australia  * Primary Business: Queensland, Australia
8  * Licensed under the Open Software License version 3.0  * Licensed under the Open Software License version 3.0
9  * http://www.opensource.org/licenses/osl-3.0.php  * http://www.opensource.org/licenses/osl-3.0.php
10  *  *
11  *******************************************************/  * Development until 2012 by Earth Systems Science Computational Center (ESSCC)
12    * Development since 2012 by School of Earth Sciences
13    *
14    *****************************************************************************/
15    
16  #include <ripley/Brick.h>  #include <ripley/Brick.h>
17  #include <ripley/Rectangle.h>  #include <ripley/Rectangle.h>
# Line 25  using namespace boost::python; Line 27  using namespace boost::python;
27    
28  namespace ripley {  namespace ripley {
29    
30    void writeBinaryGrid(std::string filename, const escript::Data& d,
31            const char byteorder, const char datatype, const char datasize)
32    {
33        const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(d.getDomain().get());
34        if (!dom)
35            throw RipleyException("Function space must be on a ripley domain");
36    
37        if (datatype != 'f')
38            throw RipleyException("only float data supported");
39    
40        if (datasize != '4')
41            throw RipleyException("only single-precision data supported");
42    
43        switch (byteorder) {
44            case '=':
45                dom->writeBinaryGrid(d, filename, RIPLEY_BYTE_ORDER);
46                break;
47            case '<':
48                dom->writeBinaryGrid(d, filename, RIPLEY_LITTLE_ENDIAN);
49                break;
50            case '>':
51                dom->writeBinaryGrid(d, filename, RIPLEY_BIG_ENDIAN);
52                break;
53            default:
54                throw RipleyException("unrecognized byte order argument");
55        }
56    }
57    
58    escript::Data readBinaryGrid(std::string filename, escript::FunctionSpace fs,
59            const object& pyFirst, const object& pyNum, const object& pyMultiplier,
60            const object& pyShape, double fill=0.)
61    {
62        int dim=fs.getDim();
63        std::vector<int> first(dim), numValues(dim), multiplier(dim), shape;
64    
65        if (extract<tuple>(pyFirst).check() || extract<list>(pyFirst).check()) {
66            if (len(pyFirst)==dim) {
67                for (int i=0; i<dim; i++) {
68                    first[i]=extract<int>(pyFirst[i]);
69                }
70            } else
71                throw RipleyException("Argument 'first' has wrong length");
72        } else
73            throw RipleyException("Argument 'first' must be a tuple or list");
74    
75        if (extract<tuple>(pyNum).check() || extract<list>(pyNum).check()) {
76            if (len(pyNum)==dim) {
77                for (int i=0; i<dim; i++) {
78                    numValues[i]=extract<int>(pyNum[i]);
79                }
80            } else
81                throw RipleyException("Argument 'numValues' has wrong length");
82        } else
83            throw RipleyException("Argument 'numValues' must be a tuple or list");
84    
85        if (extract<tuple>(pyShape).check() || extract<list>(pyShape).check()) {
86            for (int i=0; i<len(pyShape); i++) {
87                shape.push_back(extract<int>(pyShape[i]));
88            }
89        } else
90            throw RipleyException("Argument 'shape' must be a tuple or list");
91    
92        if (extract<tuple>(pyMultiplier).check() || extract<list>(pyMultiplier).check()) {
93            if (len(pyMultiplier)==dim) {
94                for (int i=0; i<dim; i++) {
95                    multiplier[i]=extract<int>(pyMultiplier[i]);
96                }
97            } else
98                throw RipleyException("Argument 'multiplier' has wrong length");
99        } else
100            throw RipleyException("Argument 'multiplier' must be a tuple or list");
101    
102        const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(fs.getDomain().get());
103        if (!dom)
104            throw RipleyException("Function space must be on a ripley domain");
105    
106    
107        escript::Data res(fill, shape, fs, true);
108    
109        dom->readBinaryGrid(res, filename, first, numValues, multiplier);
110        return res;
111    }
112    
113    escript::Data readNcGrid(std::string filename, std::string varname,
114            escript::FunctionSpace fs, const object& pyFirst, const object& pyNum,
115            const object& pyMultiplier, const object& pyShape, double fill=0.)
116    {
117        int dim=fs.getDim();
118        std::vector<int> first(dim), numValues(dim), multiplier(dim), shape;
119    
120        if (extract<tuple>(pyFirst).check() || extract<list>(pyFirst).check()) {
121            if (len(pyFirst)==dim) {
122                for (int i=0; i<dim; i++) {
123                    first[i]=extract<int>(pyFirst[i]);
124                }
125            } else
126                throw RipleyException("Argument 'first' has wrong length");
127        } else
128            throw RipleyException("Argument 'first' must be a tuple or list");
129    
130        if (extract<tuple>(pyNum).check() || extract<list>(pyNum).check()) {
131            if (len(pyNum)==dim) {
132                for (int i=0; i<dim; i++) {
133                    numValues[i]=extract<int>(pyNum[i]);
134                }
135            } else
136                throw RipleyException("Argument 'numValues' has wrong length");
137        } else
138            throw RipleyException("Argument 'numValues' must be a tuple or list");
139    
140        if (extract<tuple>(pyMultiplier).check() || extract<list>(pyMultiplier).check()) {
141            if (len(pyMultiplier)==dim) {
142                for (int i=0; i<dim; i++) {
143                    multiplier[i]=extract<int>(pyMultiplier[i]);
144                }
145            } else
146                throw RipleyException("Argument 'multiplier' has wrong length");
147        } else
148            throw RipleyException("Argument 'multiplier' must be a tuple or list");
149    
150        if (extract<tuple>(pyShape).check() || extract<list>(pyShape).check()) {
151            for (int i=0; i<len(pyShape); i++) {
152                shape.push_back(extract<int>(pyShape[i]));
153            }
154        } else
155            throw RipleyException("Argument 'shape' must be a tuple or list");
156    
157        const RipleyDomain* dom=dynamic_cast<const RipleyDomain*>(fs.getDomain().get());
158        if (!dom)
159            throw RipleyException("Function space must be on a ripley domain");
160    
161    
162        escript::Data res(fill, shape, fs, true);
163    
164        dom->readNcGrid(res, filename, varname, first, numValues, multiplier);
165        return res;
166    }
167    
168  // These wrappers are required to make the shared pointers work through the  // These wrappers are required to make the shared pointers work through the
169  // Python wrapper  // Python wrapper
170    
171  escript::Domain_ptr _brick(int n0, int n1, int n2, const object& l0,  // The double for n? is just to keep python happy when people need to deal with
172    // truediv
173    escript::Domain_ptr _brick(double _n0, double _n1, double _n2, const object& l0,
174                   const object& l1, const object& l2, int d0, int d1, int d2)                   const object& l1, const object& l2, int d0, int d1, int d2)
175  {  {
176        int n0=static_cast<int>(_n0), n1=static_cast<int>(_n1), n2=static_cast<int>(_n2);
177      double x0=0., x1=1., y0=0., y1=1., z0=0., z1=1.;      double x0=0., x1=1., y0=0., y1=1., z0=0., z1=1.;
178      if (extract<tuple>(l0).check()) {      if (extract<tuple>(l0).check()) {
179          tuple x=extract<tuple>(l0);          tuple x=extract<tuple>(l0);
# Line 72  escript::Domain_ptr _brick(int n0, int n Line 215  escript::Domain_ptr _brick(int n0, int n
215  }  }
216    
217  const int _q[]={0x61686969,0x746c4144,0x79616e43};  const int _q[]={0x61686969,0x746c4144,0x79616e43};
218  escript::Domain_ptr _rectangle(int n0, int n1, const object& l0,  escript::Domain_ptr _rectangle(double _n0, double _n1, const object& l0,
219                                 const object& l1, int d0, int d1)                                 const object& l1, int d0, int d1)
220  {  {
221        int n0=static_cast<int>(_n0), n1=static_cast<int>(_n1);
222      double x0=0., x1=1., y0=0., y1=1.;      double x0=0., x1=1., y0=0., y1=1.;
223      if (extract<tuple>(l0).check()) {      if (extract<tuple>(l0).check()) {
224          tuple x=extract<tuple>(l0);          tuple x=extract<tuple>(l0);
# Line 104  escript::Domain_ptr _rectangle(int n0, i Line 248  escript::Domain_ptr _rectangle(int n0, i
248  }  }
249  std::string _who(){int a[]={_q[0]^42,_q[1]^42,_q[2]^42,0};return (char*)&a[0];}  std::string _who(){int a[]={_q[0]^42,_q[1]^42,_q[2]^42,0};return (char*)&a[0];}
250    
251  }  } // end of namespace
252    
 /**  
     \page ripley Ripley  
     ripleycpp is the python module name that contains the interfaces  
     to the C++ wrapper to ripley.  
 */  
253    
254  BOOST_PYTHON_MODULE(ripleycpp)  BOOST_PYTHON_MODULE(ripleycpp)
255  {  {
# Line 122  BOOST_PYTHON_MODULE(ripleycpp) Line 261  BOOST_PYTHON_MODULE(ripleycpp)
261    
262      register_exception_translator<ripley::RipleyException>(&(esysUtils::esysExceptionTranslator));      register_exception_translator<ripley::RipleyException>(&(esysUtils::esysExceptionTranslator));
263    
264      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),      scope().attr("__doc__") = "To use this module, please import esys.ripley";
265    
266        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),
267  "Creates a hexagonal mesh with n0 x n1 x n2 elements over the brick [0,l0] x [0,l1] x [0,l2].\n\n"  "Creates a hexagonal mesh with n0 x n1 x n2 elements over the brick [0,l0] x [0,l1] x [0,l2].\n\n"
268  ":param n0: number of elements in direction 0\n:type n0: ``int``\n"  ":param n0: number of elements in direction 0\n:type n0: ``int``\n"
269  ":param n1: number of elements in direction 1\n:type n1: ``int``\n"  ":param n1: number of elements in direction 1\n:type n1: ``int``\n"
# Line 134  BOOST_PYTHON_MODULE(ripleycpp) Line 275  BOOST_PYTHON_MODULE(ripleycpp)
275  ":param d1: number of subdivisions in direction 1\n:type d1: ``int``\n"  ":param d1: number of subdivisions in direction 1\n:type d1: ``int``\n"
276  ":param d2: number of subdivisions in direction 2\n:type d2: ``int``");  ":param d2: number of subdivisions in direction 2\n:type d2: ``int``");
277    
278      def("Rectangle", ripley::_rectangle, (arg("n0"),arg("n1"),arg("l0")=1.0,arg("l1")=1.0,arg("d0")=1,arg("d1")=1),      def("Rectangle", ripley::_rectangle, (arg("n0"),arg("n1"),arg("l0")=1.0,arg("l1")=1.0,arg("d0")=-1,arg("d1")=-1),
279  "Creates a rectangular mesh with n0 x n1 elements over the rectangle [0,l0] x [0,l1].\n\n"  "Creates a rectangular mesh with n0 x n1 elements over the rectangle [0,l0] x [0,l1].\n\n"
280  ":param n0: number of elements in direction 0\n:type n0: ``int``\n"  ":param n0: number of elements in direction 0\n:type n0: ``int``\n"
281  ":param n1: number of elements in direction 1\n:type n1: ``int``\n"  ":param n1: number of elements in direction 1\n:type n1: ``int``\n"
# Line 142  BOOST_PYTHON_MODULE(ripleycpp) Line 283  BOOST_PYTHON_MODULE(ripleycpp)
283  ":param l1: length of side 1 or coordinate range of side 1\n:type l1: ``float`` or ``tuple``\n"  ":param l1: length of side 1 or coordinate range of side 1\n:type l1: ``float`` or ``tuple``\n"
284  ":param d0: number of subdivisions in direction 0\n:type d0: ``int``\n"  ":param d0: number of subdivisions in direction 0\n:type d0: ``int``\n"
285  ":param d1: number of subdivisions in direction 1\n:type d1: ``int``");  ":param d1: number of subdivisions in direction 1\n:type d1: ``int``");
286      def("_theculprit_", ripley::_who),      def("_theculprit_", ripley::_who);
287      def("LoadMesh", ripley::RipleyDomain::loadMesh, (arg("filename")),  
288             "Loads a ripley domain from a dump file" ":rtype: `Domain`");      def("_readBinaryGrid", &ripley::readBinaryGrid, (arg("filename"), arg("functionspace"), arg("first"), arg("numValues"), arg("multiplier"), arg("shape"), arg("fill")=0.));
289    
290      def("ReadMesh", ripley::RipleyDomain::readMesh, (arg("filename")),      def("_readNcGrid", &ripley::readNcGrid, (arg("filename"), arg("varname"), arg("functionspace"), arg("first"), arg("numValues"), arg("multiplier"), arg("shape"), arg("fill")=0.));
291              "Reads a ripley domain from a file created by write().\n\n"  
292              ":rtype: `RipleyDomain`\n:param filename:\n:type filename: ``string``\n");      def("_writeBinaryGrid", &ripley::writeBinaryGrid);
293    
294      class_<ripley::RipleyDomain, bases<escript::AbstractContinuousDomain> >      class_<ripley::RipleyDomain, bases<escript::AbstractContinuousDomain>, boost::noncopyable >
295          ("RipleyDomain", "", no_init)          ("RipleyDomain", "", no_init)
         .def("write", &ripley::RipleyDomain::write, args("filename"),  
                 "Writes the current mesh to a file with the given name. It can subsequently be recovered using ReadMesh().")  
296          .def("print_mesh_info", &ripley::RipleyDomain::Print_Mesh_Info, (arg("full")=false),          .def("print_mesh_info", &ripley::RipleyDomain::Print_Mesh_Info, (arg("full")=false),
297                  "Prints out a summary about the mesh.\n"                  "Prints out a summary about the mesh.\n"
298                  ":param full: whether to output additional data\n:type full: ``bool``")                  ":param full: whether to output additional data\n:type full: ``bool``")

Legend:
Removed from v.3791  
changed lines
  Added in v.4334

  ViewVC Help
Powered by ViewVC 1.1.26