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
|
/*************************************************************************
* Copyright (C) 2004 by Olivier Galizzi *
* olivier.galizzi@imag.fr *
* Copyright (C) 2004 by Janek Kozicki *
* cosurgi@berlios.de *
* *
* This program is free software; it is licensed under the terms of the *
* GNU General Public License v2 or later. See file LICENSE for details. *
*************************************************************************/
#include "Serializable.hpp"
namespace yade { // Cannot have #include directive inside.
void Serializable::pyRegisterClass(boost::python::object _scope)
{
checkPyClassRegistersItself("Serializable");
boost::python::scope thisScope(_scope);
boost::python::class_<Serializable, shared_ptr<Serializable>, boost::noncopyable>("Serializable")
.def("__str__", &Serializable::pyStr)
.def("__repr__", &Serializable::pyStr)
.def("dict", &Serializable::pyDict, "Return dictionary of attributes.")
.def("updateAttrs", &Serializable::pyUpdateAttrs, "Update object attributes from given dictionary")
/* boost::python pickling support, as per http://www.boost.org/doc/libs/1_42_0/libs/python/doc/v2/pickle.html */
.def("__getstate__", &Serializable::pyDict)
.def("__setstate__", &Serializable::pyUpdateAttrs)
.add_property("__safe_for_unpickling__", &Serializable::getClassName, "just define the attr, return some bogus data")
.add_property("__getstate_manages_dict__", &Serializable::getClassName, "just define the attr, return some bogus data")
// constructor with dictionary of attributes
.def("__init__", boost::python::raw_constructor(Serializable_ctor_kwAttrs<Serializable>))
// comparison operators
.def(boost::python::self == boost::python::self)
.def(boost::python::self != boost::python::self);
}
void Serializable::checkPyClassRegistersItself(const std::string& thisClassName) const
{
if (getClassName() != thisClassName)
throw std::logic_error(
("Class " + getClassName() + " does not register with YADE_CLASS_BASE_DOC_ATTR*, would not be accessible from python.").c_str());
}
void Serializable::pyUpdateAttrs(const boost::python::dict& d)
{
boost::python::list l = d.items();
size_t ll = boost::python::len(l);
if (ll == 0) return;
for (size_t i = 0; i < ll; i++) {
boost::python::tuple t = boost::python::extract<boost::python::tuple>(l[i]);
string key = boost::python::extract<string>(t[0]);
pySetAttr(key, t[1]);
}
callPostLoad();
}
} // namespace yade
|
