-
Notifications
You must be signed in to change notification settings - Fork 25
Expand file tree
/
Copy pathwrap_rgba.cpp
More file actions
154 lines (136 loc) · 3.17 KB
/
wrap_rgba.cpp
File metadata and controls
154 lines (136 loc) · 3.17 KB
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
// Copyright (c) 2003, 2004 by Jonathan Brandmeyer and others.
// See the file license.txt for complete license terms.
// See the file authors.txt for a complete list of contributors.
#include "util/rgba.hpp"
#include <boost/python/to_python_converter.hpp>
#include <boost/python/extract.hpp>
#include <boost/python/tuple.hpp>
#include <boost/python/proxy.hpp>
#include <boost/python/class.hpp>
// Python 2/3 compatibility
#ifndef PyInt_Check
#define PyInt_Check PyLong_Check
#endif
namespace cvisual {
namespace py = boost::python;
struct rgb_from_seq
{
rgb_from_seq()
{
py::converter::registry::push_back(
&convertible,
&construct,
py::type_id<rgb>());
}
static void* convertible( PyObject* obj)
{
using py::handle;
using py::allow_null;
handle<> obj_iter( allow_null( PyObject_GetIter(obj)));
if (!obj_iter.get()) {
PyErr_Clear();
return 0;
}
int obj_size = PyObject_Length(obj);
if (obj_size < 0) {
PyErr_Clear();
return 0;
}
if (obj_size != 3)
return 0;
return obj;
}
static void construct(
PyObject* _obj,
py::converter::rvalue_from_python_stage1_data* data)
{
py::object obj = py::object(py::handle<>(py::borrowed(_obj)));
void* storage = (
(py::converter::rvalue_from_python_storage<rgb>*)
data)->storage.bytes;
new (storage) rgb(
py::extract<float>(obj[0]),
py::extract<float>(obj[1]),
py::extract<float>(obj[2]));
data->convertible = storage;
}
};
struct rgba_from_seq
{
rgba_from_seq()
{
py::converter::registry::push_back(
&convertible,
&construct,
py::type_id<rgba>());
}
static void* convertible( PyObject* obj)
{
using py::handle;
using py::allow_null;
if (PyInt_Check(obj) || PyFloat_Check(obj))
return obj;
handle<> obj_iter( allow_null( PyObject_GetIter(obj)));
if (!obj_iter.get()) {
PyErr_Clear();
return 0;
}
int obj_size = PyObject_Length(obj);
if (obj_size < 0) {
PyErr_Clear();
return 0;
}
if (obj_size != 3 && obj_size != 4)
return 0;
return obj;
}
static void construct(
PyObject* _obj,
py::converter::rvalue_from_python_stage1_data* data)
{
py::object obj = py::object(py::handle<>(py::borrowed(_obj)));
void* storage = (
(py::converter::rvalue_from_python_storage<rgba>*)
data)->storage.bytes;
int obj_size = PyObject_Length(_obj);
if (obj_size == 3)
new (storage) rgba(
py::extract<float>(obj[0]),
py::extract<float>(obj[1]),
py::extract<float>(obj[2]));
else
new (storage) rgba(
py::extract<float>(obj[0]),
py::extract<float>(obj[1]),
py::extract<float>(obj[2]),
py::extract<float>(obj[3]));
data->convertible = storage;
}
};
struct rgb_to_tuple
{
static PyObject* convert( const rgb& color)
{
py::tuple ret = py::make_tuple( color.red, color.green, color.blue);
Py_INCREF(ret.ptr());
return ret.ptr();
}
};
struct rgba_to_tuple
{
static PyObject* convert( const rgba& color)
{
py::tuple ret = py::make_tuple( color.red, color.green, color.blue, color.opacity);
Py_INCREF(ret.ptr());
return ret.ptr();
}
};
void
wrap_rgba()
{
rgb_from_seq();
rgba_from_seq();
py::to_python_converter< rgb, rgb_to_tuple>();
py::to_python_converter< rgba, rgba_to_tuple>();
}
} // !namespace cvisual