source: NonGTP/Boost/boost/python/enum.hpp @ 857

Revision 857, 2.9 KB checked in by igarcia, 19 years ago (diff)
Line 
1// Copyright David Abrahams 2002.
2// Distributed under the Boost Software License, Version 1.0. (See
3// accompanying file LICENSE_1_0.txt or copy at
4// http://www.boost.org/LICENSE_1_0.txt)
5#ifndef ENUM_DWA200298_HPP
6# define ENUM_DWA200298_HPP
7
8# include <boost/python/detail/prefix.hpp>
9
10# include <boost/python/object/enum_base.hpp>
11# include <boost/python/converter/rvalue_from_python_data.hpp>
12# include <boost/python/converter/registered.hpp>
13
14namespace boost { namespace python {
15
16template <class T>
17struct enum_ : public objects::enum_base
18{
19    typedef objects::enum_base base;
20
21    // Declare a new enumeration type in the current scope()
22    enum_(char const* name);
23
24    // Add a new enumeration value with the given name and value.
25    inline enum_<T>& value(char const* name, T);
26
27    // Add all of the defined enumeration values to the current scope with the
28    // same names used here.
29    inline enum_<T>& export_values();
30 private:
31    static PyObject* to_python(void const* x);
32    static void* convertible_from_python(PyObject* obj);
33    static void construct(PyObject* obj, converter::rvalue_from_python_stage1_data* data);
34};
35
36template <class T>
37inline enum_<T>::enum_(char const* name)
38    : base(
39        name
40        , &enum_<T>::to_python
41        , &enum_<T>::convertible_from_python
42        , &enum_<T>::construct
43        , type_id<T>())
44{
45}
46
47// This is the conversion function that gets registered for converting
48// these enums to Python.
49template <class T>
50PyObject* enum_<T>::to_python(void const* x)
51{
52    return base::to_python(
53        converter::registered<T>::converters.m_class_object
54        , static_cast<long>(*(T const*)x));
55}
56
57//
58// The following two static functions serve as the elements of an
59// rvalue from_python converter for the enumeration type.
60//
61
62// This checks that a given Python object can be converted to the
63// enumeration type.
64template <class T>
65void* enum_<T>::convertible_from_python(PyObject* obj)
66{
67    return PyObject_IsInstance(
68        obj
69        , upcast<PyObject>(
70            converter::registered<T>::converters.m_class_object))
71       
72        ? obj : 0;
73}
74
75// Constructs an instance of the enumeration type in the from_python
76// data.
77template <class T>
78void enum_<T>::construct(PyObject* obj, converter::rvalue_from_python_stage1_data* data)
79{
80    T x = static_cast<T>(PyInt_AS_LONG(obj));
81    void* const storage = ((converter::rvalue_from_python_storage<T>*)data)->storage.bytes;
82    new (storage) T(x);
83    data->convertible = storage;
84}
85
86template <class T>
87inline enum_<T>& enum_<T>::value(char const* name, T x)
88{
89    this->add_value(name, static_cast<long>(x));
90    return *this;
91}
92
93template <class T>
94inline enum_<T>& enum_<T>::export_values()
95{
96    this->base::export_values();
97    return *this;
98}
99
100}} // namespace boost::python
101
102#endif // ENUM_DWA200298_HPP
Note: See TracBrowser for help on using the repository browser.