File: GrainCollectionPy.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 (99 lines) | stat: -rw-r--r-- 3,087 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
/////////////////////////////////////////////////////////////
//                                                         //
// 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/GrainCollectionPy.h"
#include "Python/esys/lsm/geometry/SimpleSpherePy.h"

namespace esys
{
  namespace lsm
  {
    GrainCollectionPy::GrainCollectionPy()
      : Inherited()
    {
    }

    GrainCollectionPy::GrainCollectionPy(ParticlePoolPtr particlePoolPtr)
      : Inherited(particlePoolPtr)
    {
    }

    GrainCollectionPy::GrainCollectionPy(
      ParticlePoolPtr particlePoolPtr,
      GrainPoolPtr    grainPoolPtr
    )
      : Inherited(particlePoolPtr, grainPoolPtr)
    {
    }


    GrainPy &GrainCollectionPy::createGrainPy()
    {
      return Inherited::createGrain();
    }

    GrainPy &GrainCollectionPy::createGrainWithIdPy(GrainPy::Id id)
    {
      return Inherited::createGrain(id);
    }

    GrainCollectionPy::GrainIteratorPy GrainCollectionPy::getGrainIteratorPy()
    {
      return GrainIteratorPy(Inherited::getGrainIterator());
    }

    void exportGrainCollection()
    {
      // 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_<GrainCollectionPy>(
        "GrainCollection",
        boost::python::init<>()
      )
      .def("getGrainIterator", &GrainCollectionPy::getGrainIteratorPy)
      .def("getNumParticles", &GrainCollectionPy::getNumParticles)
      .def("getNumGrains", &GrainCollectionPy::getNumGrains)
      .def("__iter__", &GrainCollectionPy::getGrainIteratorPy)
      .def("__len__", &GrainCollectionPy::getNumGrains)
      .def(
        "createGrain",
        &GrainCollectionPy::createGrainPy,
        boost::python::return_value_policy<
          boost::python::reference_existing_object
        >()
      )
      .def(
        "createGrain",
        &GrainCollectionPy::createGrainWithIdPy,
        boost::python::return_value_policy<
          boost::python::reference_existing_object
        >()
      )
      ;

      GrainCollectionPy::GrainIteratorPy::exportIterator(
        "GrainCollectionGrainIterator"
      );
    }
  }
}