buffers.c 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400
  1. /*
  2. * Source updates from:
  3. * - https://wayland-book.com/xdg-shell-basics/example-code.html
  4. * - libdrm's "modetest"
  5. */
  6. /*
  7. * DRM based mode setting test program
  8. * Copyright 2008 Tungsten Graphics
  9. * Jakob Bornecrantz <jakob@tungstengraphics.com>
  10. * Copyright 2008 Intel Corporation
  11. * Jesse Barnes <jesse.barnes@intel.com>
  12. *
  13. * Permission is hereby granted, free of charge, to any person obtaining a
  14. * copy of this software and associated documentation files (the "Software"),
  15. * to deal in the Software without restriction, including without limitation
  16. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  17. * and/or sell copies of the Software, and to permit persons to whom the
  18. * Software is furnished to do so, subject to the following conditions:
  19. *
  20. * The above copyright notice and this permission notice shall be included in
  21. * all copies or substantial portions of the Software.
  22. *
  23. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  24. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  25. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  26. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  27. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  28. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  29. * IN THE SOFTWARE.
  30. */
  31. #include <errno.h>
  32. #include <fcntl.h>
  33. #include <getopt.h>
  34. #include <limits.h>
  35. #include <stdbool.h>
  36. #include <stdint.h>
  37. #include <stdio.h>
  38. #include <stdlib.h>
  39. #include <string.h>
  40. #include <sys/mman.h>
  41. #include <time.h>
  42. #include <unistd.h>
  43. #include "drm/drm.h"
  44. #include "drm/drm_fourcc.h"
  45. #include "xf86drm.h"
  46. #include "buffers.h"
  47. long int raw_offset = 0;
  48. static void randname(char *buf) {
  49. struct timespec ts;
  50. clock_gettime(CLOCK_REALTIME, &ts);
  51. long r = ts.tv_nsec;
  52. for (int i = 0; i < 6; ++i) {
  53. buf[i] = 'A' + (r & 15) + (r & 16) * 2;
  54. r >>= 5;
  55. }
  56. }
  57. static int create_shm_file(void) {
  58. int retries = 100;
  59. do {
  60. char name[] = "/wl_shm-XXXXXX";
  61. randname(name + sizeof(name) - 7);
  62. --retries;
  63. int fd = shm_open(name, O_RDWR | O_CREAT | O_EXCL, 0600);
  64. if (fd >= 0) {
  65. shm_unlink(name);
  66. return fd;
  67. }
  68. } while (retries > 0 && errno == EEXIST);
  69. return -1;
  70. }
  71. int allocate_shm_file(size_t size) {
  72. int fd = create_shm_file();
  73. if (fd < 0)
  74. return -1;
  75. int ret;
  76. do {
  77. ret = ftruncate(fd, size);
  78. } while (ret < 0 && errno == EINTR);
  79. if (ret < 0) {
  80. close(fd);
  81. return -1;
  82. }
  83. return fd;
  84. }
  85. int fill_with_raw(FILE *fp, uint32_t width, uint32_t height,
  86. uint8_t *data) {
  87. fseek(fp, 0, SEEK_END);
  88. uint32_t file_size = ftell(fp);
  89. uint32_t frame_size = (width * height + 2 * ((width + 1) / 2) * ((height + 1) / 2));
  90. fseek(fp, raw_offset, SEEK_SET);
  91. size_t result = fread(data, 1, frame_size, fp);
  92. if (result != frame_size) {
  93. fprintf(stderr, "Error reading yuv image\n");
  94. return -1;
  95. }
  96. raw_offset = raw_offset + frame_size;
  97. if (raw_offset >= file_size)
  98. raw_offset = 0;
  99. return 0;
  100. }
  101. struct bo *bo_create_dumb(int fd, unsigned int width,
  102. unsigned int height, unsigned int bpp) {
  103. struct drm_mode_create_dumb arg;
  104. struct bo *bo;
  105. int ret;
  106. bo = calloc(1, sizeof(*bo));
  107. if (bo == NULL) {
  108. fprintf(stderr, "failed to allocate buffer object\n");
  109. return NULL;
  110. }
  111. memset(&arg, 0, sizeof(arg));
  112. arg.bpp = bpp;
  113. arg.width = width;
  114. arg.height = height;
  115. ret = drmIoctl(fd, DRM_IOCTL_MODE_CREATE_DUMB, &arg);
  116. if (ret) {
  117. fprintf(stderr, "failed to create dumb buffer: %s\n", strerror(errno));
  118. free(bo);
  119. return NULL;
  120. }
  121. bo->fd = fd;
  122. bo->handle = arg.handle;
  123. bo->size = arg.size;
  124. bo->pitch = arg.pitch;
  125. return bo;
  126. }
  127. int bo_map(struct bo *bo, void **out) {
  128. struct drm_mode_map_dumb arg;
  129. void *map;
  130. int ret;
  131. memset(&arg, 0, sizeof(arg));
  132. arg.handle = bo->handle;
  133. ret = drmIoctl(bo->fd, DRM_IOCTL_MODE_MAP_DUMB, &arg);
  134. if (ret)
  135. return ret;
  136. map =
  137. mmap(0, bo->size, PROT_READ | PROT_WRITE, MAP_SHARED, bo->fd, arg.offset);
  138. bo->ptr = map;
  139. *out = map;
  140. return 0;
  141. }
  142. void bo_unmap(struct bo *bo) {
  143. if (!bo->ptr)
  144. return;
  145. munmap(bo->ptr, bo->size);
  146. bo->ptr = NULL;
  147. }
  148. struct bo *bo_create(int fd, unsigned int format, unsigned int width,
  149. unsigned int height, unsigned int handles[4],
  150. unsigned int pitches[4], unsigned int offsets[4]) {
  151. unsigned int virtual_height;
  152. struct bo *bo;
  153. unsigned int bpp;
  154. void *planes[3] = {
  155. 0,
  156. };
  157. void *virtual;
  158. int ret;
  159. switch (format) {
  160. case DRM_FORMAT_C8:
  161. case DRM_FORMAT_NV12:
  162. case DRM_FORMAT_NV21:
  163. case DRM_FORMAT_NV16:
  164. case DRM_FORMAT_NV61:
  165. case DRM_FORMAT_YUV420:
  166. case DRM_FORMAT_YVU420:
  167. bpp = 8;
  168. break;
  169. case DRM_FORMAT_ARGB4444:
  170. case DRM_FORMAT_XRGB4444:
  171. case DRM_FORMAT_ABGR4444:
  172. case DRM_FORMAT_XBGR4444:
  173. case DRM_FORMAT_RGBA4444:
  174. case DRM_FORMAT_RGBX4444:
  175. case DRM_FORMAT_BGRA4444:
  176. case DRM_FORMAT_BGRX4444:
  177. case DRM_FORMAT_ARGB1555:
  178. case DRM_FORMAT_XRGB1555:
  179. case DRM_FORMAT_ABGR1555:
  180. case DRM_FORMAT_XBGR1555:
  181. case DRM_FORMAT_RGBA5551:
  182. case DRM_FORMAT_RGBX5551:
  183. case DRM_FORMAT_BGRA5551:
  184. case DRM_FORMAT_BGRX5551:
  185. case DRM_FORMAT_RGB565:
  186. case DRM_FORMAT_BGR565:
  187. case DRM_FORMAT_UYVY:
  188. case DRM_FORMAT_VYUY:
  189. case DRM_FORMAT_YUYV:
  190. case DRM_FORMAT_YVYU:
  191. bpp = 16;
  192. break;
  193. case DRM_FORMAT_BGR888:
  194. case DRM_FORMAT_RGB888:
  195. bpp = 24;
  196. break;
  197. case DRM_FORMAT_ARGB8888:
  198. case DRM_FORMAT_XRGB8888:
  199. case DRM_FORMAT_ABGR8888:
  200. case DRM_FORMAT_XBGR8888:
  201. case DRM_FORMAT_RGBA8888:
  202. case DRM_FORMAT_RGBX8888:
  203. case DRM_FORMAT_BGRA8888:
  204. case DRM_FORMAT_BGRX8888:
  205. case DRM_FORMAT_ARGB2101010:
  206. case DRM_FORMAT_XRGB2101010:
  207. case DRM_FORMAT_ABGR2101010:
  208. case DRM_FORMAT_XBGR2101010:
  209. case DRM_FORMAT_RGBA1010102:
  210. case DRM_FORMAT_RGBX1010102:
  211. case DRM_FORMAT_BGRA1010102:
  212. case DRM_FORMAT_BGRX1010102:
  213. bpp = 32;
  214. break;
  215. case DRM_FORMAT_XRGB16161616F:
  216. case DRM_FORMAT_XBGR16161616F:
  217. case DRM_FORMAT_ARGB16161616F:
  218. case DRM_FORMAT_ABGR16161616F:
  219. bpp = 64;
  220. break;
  221. default:
  222. fprintf(stderr, "unsupported format 0x%08x\n", format);
  223. return NULL;
  224. }
  225. switch (format) {
  226. case DRM_FORMAT_NV12:
  227. case DRM_FORMAT_NV21:
  228. case DRM_FORMAT_YUV420:
  229. case DRM_FORMAT_YVU420:
  230. virtual_height = height * 3 / 2;
  231. break;
  232. case DRM_FORMAT_NV16:
  233. case DRM_FORMAT_NV61:
  234. virtual_height = height * 2;
  235. break;
  236. default:
  237. virtual_height = height;
  238. break;
  239. }
  240. bo = bo_create_dumb(fd, width, virtual_height, bpp);
  241. if (!bo)
  242. return NULL;
  243. ret = bo_map(bo, &virtual);
  244. if (ret) {
  245. fprintf(stderr, "failed to map buffer: %s\n", strerror(-errno));
  246. bo_destroy(bo);
  247. return NULL;
  248. }
  249. /* just testing a limited # of formats to test single
  250. * and multi-planar path.. would be nice to add more..
  251. */
  252. switch (format) {
  253. case DRM_FORMAT_UYVY:
  254. case DRM_FORMAT_VYUY:
  255. case DRM_FORMAT_YUYV:
  256. case DRM_FORMAT_YVYU:
  257. offsets[0] = 0;
  258. handles[0] = bo->handle;
  259. pitches[0] = bo->pitch;
  260. planes[0] = virtual;
  261. break;
  262. case DRM_FORMAT_NV12:
  263. case DRM_FORMAT_NV21:
  264. case DRM_FORMAT_NV16:
  265. case DRM_FORMAT_NV61:
  266. offsets[0] = 0;
  267. handles[0] = bo->handle;
  268. pitches[0] = bo->pitch;
  269. pitches[1] = pitches[0];
  270. offsets[1] = pitches[0] * height;
  271. handles[1] = bo->handle;
  272. planes[0] = virtual;
  273. planes[1] = virtual + offsets[1];
  274. break;
  275. case DRM_FORMAT_YUV420:
  276. case DRM_FORMAT_YVU420:
  277. offsets[0] = 0;
  278. handles[0] = bo->handle;
  279. pitches[0] = bo->pitch;
  280. pitches[1] = pitches[0] / 2;
  281. offsets[1] = pitches[0] * height;
  282. handles[1] = bo->handle;
  283. pitches[2] = pitches[1];
  284. offsets[2] = offsets[1] + pitches[1] * height / 2;
  285. handles[2] = bo->handle;
  286. planes[0] = virtual;
  287. planes[1] = virtual + offsets[1];
  288. planes[2] = virtual + offsets[2];
  289. break;
  290. case DRM_FORMAT_C8:
  291. case DRM_FORMAT_ARGB4444:
  292. case DRM_FORMAT_XRGB4444:
  293. case DRM_FORMAT_ABGR4444:
  294. case DRM_FORMAT_XBGR4444:
  295. case DRM_FORMAT_RGBA4444:
  296. case DRM_FORMAT_RGBX4444:
  297. case DRM_FORMAT_BGRA4444:
  298. case DRM_FORMAT_BGRX4444:
  299. case DRM_FORMAT_ARGB1555:
  300. case DRM_FORMAT_XRGB1555:
  301. case DRM_FORMAT_ABGR1555:
  302. case DRM_FORMAT_XBGR1555:
  303. case DRM_FORMAT_RGBA5551:
  304. case DRM_FORMAT_RGBX5551:
  305. case DRM_FORMAT_BGRA5551:
  306. case DRM_FORMAT_BGRX5551:
  307. case DRM_FORMAT_RGB565:
  308. case DRM_FORMAT_BGR565:
  309. case DRM_FORMAT_BGR888:
  310. case DRM_FORMAT_RGB888:
  311. case DRM_FORMAT_ARGB8888:
  312. case DRM_FORMAT_XRGB8888:
  313. case DRM_FORMAT_ABGR8888:
  314. case DRM_FORMAT_XBGR8888:
  315. case DRM_FORMAT_RGBA8888:
  316. case DRM_FORMAT_RGBX8888:
  317. case DRM_FORMAT_BGRA8888:
  318. case DRM_FORMAT_BGRX8888:
  319. case DRM_FORMAT_ARGB2101010:
  320. case DRM_FORMAT_XRGB2101010:
  321. case DRM_FORMAT_ABGR2101010:
  322. case DRM_FORMAT_XBGR2101010:
  323. case DRM_FORMAT_RGBA1010102:
  324. case DRM_FORMAT_RGBX1010102:
  325. case DRM_FORMAT_BGRA1010102:
  326. case DRM_FORMAT_BGRX1010102:
  327. case DRM_FORMAT_XRGB16161616F:
  328. case DRM_FORMAT_XBGR16161616F:
  329. case DRM_FORMAT_ARGB16161616F:
  330. case DRM_FORMAT_ABGR16161616F:
  331. offsets[0] = 0;
  332. handles[0] = bo->handle;
  333. pitches[0] = bo->pitch;
  334. planes[0] = virtual;
  335. break;
  336. }
  337. bo_unmap(bo);
  338. return bo;
  339. }
  340. void bo_destroy(struct bo *bo) {
  341. struct drm_mode_destroy_dumb arg;
  342. int ret;
  343. memset(&arg, 0, sizeof(arg));
  344. arg.handle = bo->handle;
  345. ret = drmIoctl(bo->fd, DRM_IOCTL_MODE_DESTROY_DUMB, &arg);
  346. if (ret)
  347. fprintf(stderr, "failed to destroy dumb buffer: %s\n", strerror(errno));
  348. free(bo);
  349. }