92f9a0c302285b7e8abfe9f9a3cccccaaec5622e
[glsdk/omapdrmtest.git] / util / util.c
1 /*
2  * Copyright (C) 2011 Texas Instruments
3  * Author: Rob Clark <rob.clark@linaro.org>
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 as published by
7  * the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
12  * more details.
13  *
14  * You should have received a copy of the GNU General Public License along with
15  * this program.  If not, see <http://www.gnu.org/licenses/>.
16  */
18 #include "util.h"
20 #include <drm.h>
22 void disp_kms_usage(void);
23 struct display * disp_kms_open(int argc, char **argv);
25 #ifdef HAVE_X11
26 void disp_x11_usage(void);
27 struct display * disp_x11_open(int argc, char **argv);
28 #endif
30 void
31 disp_usage(void)
32 {
33 #ifdef HAVE_X11
34         disp_x11_usage();
35 #endif
36         disp_kms_usage();
37 }
39 struct display *
40 disp_open(int argc, char **argv)
41 {
42         struct display *disp;
44 #ifdef HAVE_X11
45         disp = disp_x11_open(argc, argv);
46         if (disp)
47                 return disp;
48 #endif
50         disp = disp_kms_open(argc, argv);
52         if (!disp) {
53                 ERROR("unable to create display");
54         }
56         return disp;
57 }
59 struct buffer *
60 disp_get_fb(struct display *disp)
61 {
62         struct buffer **bufs = disp_get_buffers(disp, 1);
63         if (!bufs)
64                 return NULL;
65         fill(bufs[0], 42);
66         disp_post_buffer(disp, bufs[0]);
67         return bufs[0];
68 }
71 int
72 check_args(int argc, char **argv)
73 {
74         int i;
75         for (i = 1; i < argc; i++) {
76                 if (argv[i]) {
77                         ERROR("invalid arg: %s", argv[i]);
78                         return -1;
79                 }
80         }
81         return 0;
82 }
84 /* stolen from modetest.c */
85 static void
86 fillRGB4(char *virtual, int n, int width, int height, int stride)
87 {
88         int i, j;
89         /* paint the buffer with colored tiles */
90         for (j = 0; j < height; j++) {
91                 uint32_t *fb_ptr = (uint32_t*)((char*)virtual + j * stride);
92                 for (i = 0; i < width; i++) {
93                         div_t d = div(n+i+j, width);
94                         fb_ptr[i] =
95                                         0x00130502 * (d.quot >> 6) +
96                                         0x000a1120 * (d.rem >> 6);
97                 }
98         }
99 }
102 /* swap these for big endian.. */
103 #define RED   2
104 #define GREEN 1
105 #define BLUE  0
107 static void
108 fill420(unsigned char *y, unsigned char *u, unsigned char *v,
109                 int cs /*chroma pixel stride */,
110                 int n, int width, int height, int stride)
112         int i, j;
114         /* paint the buffer with colored tiles, in blocks of 2x2 */
115         for (j = 0; j < height; j+=2) {
116                 unsigned char *y1p = y + j * stride;
117                 unsigned char *y2p = y1p + stride;
118                 unsigned char *up = u + (j/2) * stride * cs / 2;
119                 unsigned char *vp = v + (j/2) * stride * cs / 2;
121                 for (i = 0; i < width; i+=2) {
122                         div_t d = div(n+i+j, width);
123                         uint32_t rgb = 0x00130502 * (d.quot >> 6) + 0x000a1120 * (d.rem >> 6);
124                         unsigned char *rgbp = (unsigned char *)&rgb;
125                         unsigned char y = (0.299 * rgbp[RED]) + (0.587 * rgbp[GREEN]) + (0.114 * rgbp[BLUE]);
127                         *(y2p++) = *(y1p++) = y;
128                         *(y2p++) = *(y1p++) = y;
130                         *up = (rgbp[BLUE] - y) * 0.565 + 128;
131                         *vp = (rgbp[RED] - y) * 0.713 + 128;
132                         up += cs;
133                         vp += cs;
134                 }
135         }
138 static void
139 fill422(unsigned char *virtual, int n, int width, int height, int stride)
141         int i, j;
142         /* paint the buffer with colored tiles */
143         for (j = 0; j < height; j++) {
144                 uint8_t *ptr = (uint8_t*)((char*)virtual + j * stride);
145                 for (i = 0; i < width; i++) {
146                         div_t d = div(n+i+j, width);
147                         uint32_t rgb = 0x00130502 * (d.quot >> 6) + 0x000a1120 * (d.rem >> 6);
148                         unsigned char *rgbp = (unsigned char *)&rgb;
149                         unsigned char y = (0.299 * rgbp[RED]) + (0.587 * rgbp[GREEN]) + (0.114 * rgbp[BLUE]);
151                         *(ptr++) = y;
152                         *(ptr++) = (rgbp[BLUE] - y) * 0.565 + 128;
153                         *(ptr++) = y;
154                         *(ptr++) = (rgbp[RED] - y) * 0.713 + 128;
155                 }
156         }
160 void
161 fill(struct buffer *buf, int n)
163         int i;
165         for (i = 0; i < buf->nbo; i++)
166                 omap_bo_cpu_prep(buf->bo[i], OMAP_GEM_WRITE);
168         switch(buf->fourcc) {
169         case 0: {
170                 assert(buf->nbo == 1);
171                 fillRGB4(omap_bo_map(buf->bo[0]), n,
172                                 buf->width, buf->height, buf->pitches[0]);
173                 break;
174         }
175         case FOURCC('Y','U','Y','V'): {
176                 assert(buf->nbo == 1);
177                 fill422(omap_bo_map(buf->bo[0]), n,
178                                 buf->width, buf->height, buf->pitches[0]);
179                 break;
180         }
181         case FOURCC('N','V','1','2'): {
182                 unsigned char *y, *u, *v;
183                 assert(buf->nbo == 2);
184                 y = omap_bo_map(buf->bo[0]);
185                 u = omap_bo_map(buf->bo[1]);
186                 v = u + 1;
187                 fill420(y, u, v, 2, n, buf->width, buf->height, buf->pitches[0]);
188                 break;
189         }
190         case FOURCC('I','4','2','0'): {
191                 unsigned char *y, *u, *v;
192                 assert(buf->nbo == 3);
193                 y = omap_bo_map(buf->bo[0]);
194                 u = omap_bo_map(buf->bo[1]);
195                 v = omap_bo_map(buf->bo[2]);
196                 fill420(y, u, v, 1, n, buf->width, buf->height, buf->pitches[0]);
197                 break;
198         }
199         default:
200                 ERROR("invalid format: 0x%08x", buf->fourcc);
201                 break;
202         }
204         for (i = 0; i < buf->nbo; i++)
205                 omap_bo_cpu_fini(buf->bo[i], OMAP_GEM_WRITE);