summaryrefslogtreecommitdiffstats
path: root/spa/plugins/videotestsrc/draw.c
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 18:28:17 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 18:28:17 +0000
commit7a46c07230b8d8108c0e8e80df4522d0ac116538 (patch)
treed483300dab478b994fe199a5d19d18d74153718a /spa/plugins/videotestsrc/draw.c
parentInitial commit. (diff)
downloadpipewire-upstream/0.3.65.tar.xz
pipewire-upstream/0.3.65.zip
Adding upstream version 0.3.65.upstream/0.3.65upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'spa/plugins/videotestsrc/draw.c')
-rw-r--r--spa/plugins/videotestsrc/draw.c288
1 files changed, 288 insertions, 0 deletions
diff --git a/spa/plugins/videotestsrc/draw.c b/spa/plugins/videotestsrc/draw.c
new file mode 100644
index 0000000..20fed49
--- /dev/null
+++ b/spa/plugins/videotestsrc/draw.c
@@ -0,0 +1,288 @@
+/* Spa Video Test Source
+ *
+ * Copyright © 2018 Wim Taymans
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+
+#include <errno.h>
+
+typedef enum {
+ GRAY = 0,
+ YELLOW,
+ CYAN,
+ GREEN,
+ MAGENTA,
+ RED,
+ BLUE,
+ BLACK,
+ NEG_I,
+ WHITE,
+ POS_Q,
+ DARK_BLACK,
+ LIGHT_BLACK,
+ N_COLORS
+} Color;
+
+typedef struct _Pixel Pixel;
+
+struct _Pixel {
+ unsigned char R;
+ unsigned char G;
+ unsigned char B;
+ unsigned char Y;
+ unsigned char U;
+ unsigned char V;
+};
+
+static Pixel colors[N_COLORS] = {
+ {191, 191, 191, 0, 0, 0}, /* GRAY */
+ {191, 191, 0, 0, 0, 0}, /* YELLOW */
+ {0, 191, 191, 0, 0, 0}, /* CYAN */
+ {0, 191, 0, 0, 0, 0}, /* GREEN */
+ {191, 0, 191, 0, 0, 0}, /* MAGENTA */
+ {191, 0, 0, 0, 0, 0}, /* RED */
+ {0, 0, 191, 0, 0, 0}, /* BLUE */
+ {19, 19, 19, 0, 0, 0}, /* BLACK */
+ {0, 33, 76, 0, 0, 0}, /* NEGATIVE I */
+ {255, 255, 255, 0, 0, 0}, /* WHITE */
+ {49, 0, 107, 0, 0, 0}, /* POSITIVE Q */
+ {9, 9, 9, 0, 0, 0}, /* DARK BLACK */
+ {29, 29, 29, 0, 0, 0}, /* LIGHT BLACK */
+};
+
+/* YUV values are computed in init_colors() */
+
+typedef struct _DrawingData DrawingData;
+
+typedef void (*DrawPixelFunc) (DrawingData * dd, int x, Pixel * pixel);
+
+struct _DrawingData {
+ char *line;
+ int width;
+ int height;
+ int stride;
+ DrawPixelFunc draw_pixel;
+};
+
+static inline void update_yuv(Pixel * pixel)
+{
+ uint16_t y, u, v;
+
+ /* see https://en.wikipedia.org/wiki/YUV#Studio_swing_for_BT.601 */
+
+ y = 76 * pixel->R + 150 * pixel->G + 29 * pixel->B;
+ u = -43 * pixel->R - 84 * pixel->G + 127 * pixel->B;
+ v = 127 * pixel->R - 106 * pixel->G - 21 * pixel->B;
+
+ y = (y + 128) >> 8;
+ u = (u + 128) >> 8;
+ v = (v + 128) >> 8;
+
+ pixel->Y = y;
+ pixel->U = u + 128;
+ pixel->V = v + 128;
+}
+
+static void init_colors(void)
+{
+ int i;
+
+ if (colors[WHITE].Y != 0) {
+ /* already computed */
+ return;
+ }
+
+ for (i = 0; i < N_COLORS; i++) {
+ update_yuv(&colors[i]);
+ }
+}
+
+static void draw_pixel_rgb(DrawingData * dd, int x, Pixel * color)
+{
+ dd->line[3 * x + 0] = color->R;
+ dd->line[3 * x + 1] = color->G;
+ dd->line[3 * x + 2] = color->B;
+}
+
+static void draw_pixel_uyvy(DrawingData * dd, int x, Pixel * color)
+{
+ if (x & 1) {
+ /* odd pixel */
+ dd->line[2 * (x - 1) + 3] = color->Y;
+ } else {
+ /* even pixel */
+ dd->line[2 * x + 0] = color->U;
+ dd->line[2 * x + 1] = color->Y;
+ dd->line[2 * x + 2] = color->V;
+ }
+}
+
+static int drawing_data_init(DrawingData * dd, struct impl *this, char *data)
+{
+ struct port *port = &this->port;
+ struct spa_video_info *format = &port->current_format;
+ struct spa_rectangle *size = &format->info.raw.size;
+
+ if ((format->media_type != SPA_MEDIA_TYPE_video) ||
+ (format->media_subtype != SPA_MEDIA_SUBTYPE_raw))
+ return -ENOTSUP;
+
+ if (format->info.raw.format == SPA_VIDEO_FORMAT_RGB) {
+ dd->draw_pixel = draw_pixel_rgb;
+ } else if (format->info.raw.format == SPA_VIDEO_FORMAT_UYVY) {
+ dd->draw_pixel = draw_pixel_uyvy;
+ } else
+ return -ENOTSUP;
+
+ dd->line = data;
+ dd->width = size->width;
+ dd->height = size->height;
+ dd->stride = port->stride;
+
+ return 0;
+}
+
+static inline void draw_pixels(DrawingData * dd, int offset, Color color, int length)
+{
+ int x;
+
+ for (x = offset; x < offset + length; x++) {
+ dd->draw_pixel(dd, x, &colors[color]);
+ }
+}
+
+static inline void next_line(DrawingData * dd)
+{
+ dd->line += dd->stride;
+}
+
+static void draw_smpte_snow(DrawingData * dd)
+{
+ int h, w;
+ int y1, y2;
+ int i, j;
+
+ w = dd->width;
+ h = dd->height;
+ y1 = 2 * h / 3;
+ y2 = 3 * h / 4;
+
+ for (i = 0; i < y1; i++) {
+ for (j = 0; j < 7; j++) {
+ int x1 = j * w / 7;
+ int x2 = (j + 1) * w / 7;
+ draw_pixels(dd, x1, j, x2 - x1);
+ }
+ next_line(dd);
+ }
+
+ for (i = y1; i < y2; i++) {
+ for (j = 0; j < 7; j++) {
+ int x1 = j * w / 7;
+ int x2 = (j + 1) * w / 7;
+ Color c = (j & 1) ? BLACK : BLUE - j;
+
+ draw_pixels(dd, x1, c, x2 - x1);
+ }
+ next_line(dd);
+ }
+
+ for (i = y2; i < h; i++) {
+ int x = 0;
+
+ /* negative I */
+ draw_pixels(dd, x, NEG_I, w / 6);
+ x += w / 6;
+
+ /* white */
+ draw_pixels(dd, x, WHITE, w / 6);
+ x += w / 6;
+
+ /* positive Q */
+ draw_pixels(dd, x, POS_Q, w / 6);
+ x += w / 6;
+
+ /* pluge */
+ draw_pixels(dd, x, DARK_BLACK, w / 12);
+ x += w / 12;
+ draw_pixels(dd, x, BLACK, w / 12);
+ x += w / 12;
+ draw_pixels(dd, x, LIGHT_BLACK, w / 12);
+ x += w / 12;
+
+ /* war of the ants (a.k.a. snow) */
+ for (j = x; j < w; j++) {
+ Pixel p;
+ unsigned char r = rand();
+
+ p.R = r;
+ p.G = r;
+ p.B = r;
+ update_yuv(&p);
+ dd->draw_pixel(dd, j, &p);
+ }
+
+ next_line(dd);
+ }
+}
+
+static void draw_snow(DrawingData * dd)
+{
+ int x, y;
+
+ for (y = 0; y < dd->height; y++) {
+ for (x = 0; x < dd->width; x++) {
+ Pixel p;
+ unsigned char r = rand();
+
+ p.R = r;
+ p.G = r;
+ p.B = r;
+ update_yuv(&p);
+ dd->draw_pixel(dd, x, &p);
+ }
+
+ next_line(dd);
+ }
+}
+
+static int draw(struct impl *this, char *data)
+{
+ DrawingData dd;
+ int res;
+
+ init_colors();
+
+ if ((res = drawing_data_init(&dd, this, data)) < 0)
+ return res;
+
+ switch (this->props.pattern) {
+ case PATTERN_SMPTE_SNOW:
+ draw_smpte_snow(&dd);
+ break;
+ case PATTERN_SNOW:
+ draw_snow(&dd);
+ break;
+ default:
+ return -ENOTSUP;
+ }
+ return 0;
+}