File: CubicBlockPy.cpp

package info (click to toggle)
esys-particle 2.1-4
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 7,284 kB
  • sloc: cpp: 77,304; python: 5,647; makefile: 1,176; sh: 10
file content (121 lines) | stat: -rw-r--r-- 4,038 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/////////////////////////////////////////////////////////////
//                                                         //
// Copyright (c) 2003-2011 by The University of Queensland //
// Earth Systems Science Computational Centre (ESSCC)      //
// http://www.uq.edu.au/esscc                              //
//                                                         //
// Primary Business: Brisbane, Queensland, Australia       //
// Licensed under the Open Software License version 3.0    //
// http://www.opensource.org/licenses/osl-3.0.php          //
//                                                         //
/////////////////////////////////////////////////////////////


#include <boost/version.hpp>
#include <boost/python.hpp>
#include <boost/noncopyable.hpp>
#include "Python/esys/lsm/geometry/CubicBlockPy.h"
#include "Python/esys/lsm/geometry/SimpleSphereCollectionPy.h"
#include "Python/esys/lsm/geometry/SimpleSpherePy.h"
#include "Geometry/CubicBlock.h"

namespace esys
{
  namespace lsm
  {
    CubicBlockPy::CubicBlockPy() : SimpleSphereCollectionPy()
    {
    }
    
    CubicBlockPy::CubicBlockPy(
      const boost::python::list &dimCount,
      double radius,
      const ClosePackOrientation &orientation
    )
      : SimpleSphereCollectionPy()
    {
      CubicBlock<SimpleSpherePy>::BlockGenerator
        generator = CubicBlock<SimpleSpherePy>::BlockGenerator(
            boost::python::extract<int>(dimCount[0]),
            boost::python::extract<int>(dimCount[1]),
            boost::python::extract<int>(dimCount[2]),
            radius,
            orientation
        );
      generator.createParticles(*this);
    }

    using boost::python::arg;
    void exportOrientation()
    {
      // Check that Boost 1.34.0 or higher is being used.
      // If so, disable auto-generation of C++ signatures for Epydoc
      // (which stumbles over indentation in the auto-generated strings).
      #if ((BOOST_VERSION / 100000 >= 1) \
          && (BOOST_VERSION / 100 % 1000 >= 34)) \
          || (BOOST_VERSION / 100000 >= 2)
        boost::python::docstring_options no_autogen(true,false);
      #endif

      boost::python::enum_<ClosePackOrientation>(
        "Orientation"
      )
        .value("DEFAULT", DEFAULT_ORIENT)
        .value("XYZ", XYZ)
        .value("XZY", XZY)
        .value("YXZ", YXZ)
        .value("YZX", YZX)
        .value("ZXY", ZXY)
        .value("ZYX", ZYX)
      ;
    }
    
    void exportCubicBlock()
    {
      // Check that Boost 1.34.0 or higher is being used.
      // If so, disable auto-generation of C++ signatures for Epydoc
      // (which stumbles over indentation in the auto-generated strings).
      #if ((BOOST_VERSION / 100000 >= 1) \
          && (BOOST_VERSION / 100 % 1000 >= 34)) \
          || (BOOST_VERSION / 100000 >= 2)
        boost::python::docstring_options no_autogen(true,false);
      #endif

      boost::python::class_<
        CubicBlockPy,
        boost::python::bases<SimpleSphereCollectionPy>,
        boost::noncopyable
      >(
        "CubicBlock",
        "Collection of particles arranged in a cubic close "
        " (face centred cubic) packing.",
        boost::python::init<
          const boost::python::list &,
          double,
          const ClosePackOrientation &
        >(
          (
            arg("dimCount"),
            arg("radius")=0.5,
            arg("orientation")=DEFAULT_ORIENT
          )
        )
      )
      .def(
        boost::python::init<
        >(
          "Creates a collection of C{dimCount[0]*dimCount[2]*dimCount[2]}"
          " spheres arranged in a cubic close packing.\n"
          "@type dimCount: list of 3 elements\n"
          "@kwarg dimCount: number of particles in each packing dimension\n"
          "@type radius: float\n"
          "@kwarg radius: Radius of spheres in the packing\n"
          "@type orientation: L{Orientation}\n"
          "@kwarg orientation: orientation of the axis-aligned layers"
          " in the packing."
        )
      )
      ;
    }
  }
}