-
Notifications
You must be signed in to change notification settings - Fork 102
/
yuv2rgb.c
executable file
·87 lines (73 loc) · 2.5 KB
/
yuv2rgb.c
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
/* YUV->RGB conversion for Raspberry Pi camera, because fast video
port doesn't currently support raw RGB capture, only YUV.
python-dev required to build.
Adafruit invests time and resources providing this open source code,
please support Adafruit and open-source development by purchasing
products from Adafruit, thanks!
Written by Phil Burgess / Paint Your Dragon for Adafruit Industries.
Adapted for Python3: Ton van Overbeek, July 2020
BSD license, all text above must be included in any redistribution. */
#include <python3.7/Python.h>
static PyObject *method_convert(PyObject *self, PyObject *args) {
Py_buffer inBuf, outBuf;
short row, col, r, g, b, w, h, rd, gd, bd;
unsigned char *rgbPtr, *yPtr, y;
signed char *uPtr, *vPtr, u, v;
if(!PyArg_ParseTuple(args, "s*s*hh", &inBuf, &outBuf, &w, &h))
return NULL;
if(w & 31) w += 32 - (w & 31); // Round up width to multiple of 32
if(h & 15) h += 16 - (h & 15); // Round up height to multiple of 16
rgbPtr = outBuf.buf;
yPtr = inBuf.buf;
uPtr = (signed char *)&yPtr[w * h];
vPtr = &uPtr[(w * h) >> 2];
w >>= 1; // 2 columns processed per iteration
for(row=0; row<h; row++) {
for(col=0; col<w; col++) {
// U, V (and RGB deltas) updated on even columns
u = uPtr[col] - 128;
v = vPtr[col] - 128;
rd = (359 * v) >> 8;
gd = ((183 * v) + (88 * u)) >> 8;
bd = (454 * u) >> 8;
// Even column
y = *yPtr++;
r = y + rd;
g = y - gd;
b = y + bd;
*rgbPtr++ = (r > 255) ? 255 : (r < 0) ? 0 : r;
*rgbPtr++ = (g > 255) ? 255 : (g < 0) ? 0 : g;
*rgbPtr++ = (b > 255) ? 255 : (b < 0) ? 0 : b;
// Odd column
y = *yPtr++;
r = y + rd;
g = y - gd;
b = y + bd;
*rgbPtr++ = (r > 255) ? 255 : (r < 0) ? 0 : r;
*rgbPtr++ = (g > 255) ? 255 : (g < 0) ? 0 : g;
*rgbPtr++ = (b > 255) ? 255 : (b < 0) ? 0 : b;
}
if(row & 1) {
uPtr += w;
vPtr += w;
}
}
PyBuffer_Release(&inBuf);
PyBuffer_Release(&outBuf);
Py_INCREF(Py_None);
return Py_None;
}
static PyMethodDef yuv2rgb_methods[] = {
{"convert", method_convert, METH_VARARGS, "Convert image data from yuv to rgb"},
{NULL, NULL, 0, NULL}
};
static struct PyModuleDef yuv2rgbmodule = {
PyModuleDef_HEAD_INIT,
"yuv2rgb",
"Python interface for yuv to rgb conversion routine",
-1,
yuv2rgb_methods
};
PyMODINIT_FUNC PyInit_yuv2rgb(void) {
return PyModule_Create(&yuv2rgbmodule);
}