opengl_enc.c 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243
  1. /*
  2. * Copyright (c) 2014 Lukasz Marek
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. //TODO: support for more formats
  21. //TODO: support for more systems.
  22. //TODO: implement X11, Windows, Mac OS native default window. SDL 1.2 doesn't allow to render to custom thread.
  23. #include <stdio.h>
  24. #include <stdlib.h>
  25. #include <string.h>
  26. #include <unistd.h>
  27. #include <stddef.h>
  28. #include "config.h"
  29. #if HAVE_OPENGL_GL3_H
  30. #include <OpenGL/gl3.h>
  31. #elif HAVE_ES2_GL_H
  32. #include <ES2/gl.h>
  33. #else
  34. #include <GL/gl.h>
  35. #include <GL/glext.h>
  36. #endif
  37. #if HAVE_GLXGETPROCADDRESS
  38. #include <GL/glx.h>
  39. #endif
  40. #if HAVE_WINDOWS_H
  41. #include <windows.h>
  42. #endif
  43. #if HAVE_SDL
  44. #include <SDL.h>
  45. #endif
  46. #include "libavutil/common.h"
  47. #include "libavutil/pixdesc.h"
  48. #include "libavutil/log.h"
  49. #include "libavutil/opt.h"
  50. #include "libavutil/avassert.h"
  51. #include "libavutil/avstring.h"
  52. #include "libavformat/avformat.h"
  53. #include "libavdevice/avdevice.h"
  54. #include "opengl_enc_shaders.h"
  55. #ifndef APIENTRY
  56. #define APIENTRY
  57. #endif
  58. /* FF_GL_RED_COMPONENT is used for plannar pixel types.
  59. * Only red component is sampled in shaders.
  60. * On some platforms GL_RED is not availabe and GL_LUMINANCE have to be used,
  61. * but since OpenGL 3.0 GL_LUMINANCE is deprecated.
  62. * GL_RED produces RGBA = value, 0, 0, 1.
  63. * GL_LUMINANCE produces RGBA = value, value, value, 1.
  64. * Note: GL_INTENSITY may also be used which produce RGBA = value, value, value, value. */
  65. #if defined(GL_RED)
  66. #define FF_GL_RED_COMPONENT GL_RED
  67. #elif defined(GL_LUMINANCE)
  68. #define FF_GL_RED_COMPONENT GL_LUMINANCE
  69. #else
  70. #define FF_GL_RED_COMPONENT 0x1903; //GL_RED
  71. #endif
  72. /* Constants not defined for iOS */
  73. #define FF_GL_UNSIGNED_BYTE_3_3_2 0x8032
  74. #define FF_GL_UNSIGNED_BYTE_2_3_3_REV 0x8362
  75. #define FF_GL_UNSIGNED_SHORT_1_5_5_5_REV 0x8366
  76. /* MinGW exposes only OpenGL 1.1 API */
  77. #define FF_GL_ARRAY_BUFFER 0x8892
  78. #define FF_GL_ELEMENT_ARRAY_BUFFER 0x8893
  79. #define FF_GL_STATIC_DRAW 0x88E4
  80. #define FF_GL_FRAGMENT_SHADER 0x8B30
  81. #define FF_GL_VERTEX_SHADER 0x8B31
  82. #define FF_GL_COMPILE_STATUS 0x8B81
  83. #define FF_GL_LINK_STATUS 0x8B82
  84. #define FF_GL_INFO_LOG_LENGTH 0x8B84
  85. typedef void (APIENTRY *FF_PFNGLACTIVETEXTUREPROC) (GLenum texture);
  86. typedef void (APIENTRY *FF_PFNGLGENBUFFERSPROC) (GLsizei n, GLuint *buffers);
  87. typedef void (APIENTRY *FF_PFNGLDELETEBUFFERSPROC) (GLsizei n, const GLuint *buffers);
  88. typedef void (APIENTRY *FF_PFNGLBUFFERDATAPROC) (GLenum target, ptrdiff_t size, const GLvoid *data, GLenum usage);
  89. typedef void (APIENTRY *FF_PFNGLBINDBUFFERPROC) (GLenum target, GLuint buffer);
  90. typedef GLint (APIENTRY *FF_PFNGLGETATTRIBLOCATIONPROC) (GLuint program, const char *name);
  91. typedef void (APIENTRY *FF_PFNGLENABLEVERTEXATTRIBARRAYPROC) (GLuint index);
  92. typedef void (APIENTRY *FF_PFNGLVERTEXATTRIBPOINTERPROC) (GLuint index, GLint size, GLenum type, GLboolean normalized, GLsizei stride, uintptr_t pointer);
  93. typedef GLint (APIENTRY *FF_PFNGLGETUNIFORMLOCATIONPROC) (GLuint program, const char *name);
  94. typedef void (APIENTRY *FF_PFNGLUNIFORM1FPROC) (GLint location, GLfloat v0);
  95. typedef void (APIENTRY *FF_PFNGLUNIFORM1IPROC) (GLint location, GLint v0);
  96. typedef void (APIENTRY *FF_PFNGLUNIFORMMATRIX4FVPROC) (GLint location, GLsizei count, GLboolean transpose, const GLfloat *value);
  97. typedef GLuint (APIENTRY *FF_PFNGLCREATEPROGRAMPROC) (void);
  98. typedef void (APIENTRY *FF_PFNGLDELETEPROGRAMPROC) (GLuint program);
  99. typedef void (APIENTRY *FF_PFNGLUSEPROGRAMPROC) (GLuint program);
  100. typedef void (APIENTRY *FF_PFNGLLINKPROGRAMPROC) (GLuint program);
  101. typedef void (APIENTRY *FF_PFNGLGETPROGRAMIVPROC) (GLuint program, GLenum pname, GLint *params);
  102. typedef void (APIENTRY *FF_PFNGLGETPROGRAMINFOLOGPROC) (GLuint program, GLsizei bufSize, GLsizei *length, char *infoLog);
  103. typedef void (APIENTRY *FF_PFNGLATTACHSHADERPROC) (GLuint program, GLuint shader);
  104. typedef GLuint (APIENTRY *FF_PFNGLCREATESHADERPROC) (GLenum type);
  105. typedef void (APIENTRY *FF_PFNGLDELETESHADERPROC) (GLuint shader);
  106. typedef void (APIENTRY *FF_PFNGLCOMPILESHADERPROC) (GLuint shader);
  107. typedef void (APIENTRY *FF_PFNGLSHADERSOURCEPROC) (GLuint shader, GLsizei count, const char* *string, const GLint *length);
  108. typedef void (APIENTRY *FF_PFNGLGETSHADERIVPROC) (GLuint shader, GLenum pname, GLint *params);
  109. typedef void (APIENTRY *FF_PFNGLGETSHADERINFOLOGPROC) (GLuint shader, GLsizei bufSize, GLsizei *length, char *infoLog);
  110. typedef struct FFOpenGLFunctions {
  111. FF_PFNGLACTIVETEXTUREPROC glActiveTexture; //Require GL ARB multitexture
  112. FF_PFNGLGENBUFFERSPROC glGenBuffers; //Require GL_ARB_vertex_buffer_object
  113. FF_PFNGLDELETEBUFFERSPROC glDeleteBuffers; //Require GL_ARB_vertex_buffer_object
  114. FF_PFNGLBUFFERDATAPROC glBufferData; //Require GL_ARB_vertex_buffer_object
  115. FF_PFNGLBINDBUFFERPROC glBindBuffer; //Require GL_ARB_vertex_buffer_object
  116. FF_PFNGLGETATTRIBLOCATIONPROC glGetAttribLocation; //Require GL_ARB_vertex_shader
  117. FF_PFNGLENABLEVERTEXATTRIBARRAYPROC glEnableVertexAttribArray; //Require GL_ARB_vertex_shader
  118. FF_PFNGLVERTEXATTRIBPOINTERPROC glVertexAttribPointer; //Require GL_ARB_vertex_shader
  119. FF_PFNGLGETUNIFORMLOCATIONPROC glGetUniformLocation; //Require GL_ARB_shader_objects
  120. FF_PFNGLUNIFORM1FPROC glUniform1f; //Require GL_ARB_shader_objects
  121. FF_PFNGLUNIFORM1IPROC glUniform1i; //Require GL_ARB_shader_objects
  122. FF_PFNGLUNIFORMMATRIX4FVPROC glUniformMatrix4fv; //Require GL_ARB_shader_objects
  123. FF_PFNGLCREATEPROGRAMPROC glCreateProgram; //Require GL_ARB_shader_objects
  124. FF_PFNGLDELETEPROGRAMPROC glDeleteProgram; //Require GL_ARB_shader_objects
  125. FF_PFNGLUSEPROGRAMPROC glUseProgram; //Require GL_ARB_shader_objects
  126. FF_PFNGLLINKPROGRAMPROC glLinkProgram; //Require GL_ARB_shader_objects
  127. FF_PFNGLGETPROGRAMIVPROC glGetProgramiv; //Require GL_ARB_shader_objects
  128. FF_PFNGLGETPROGRAMINFOLOGPROC glGetProgramInfoLog; //Require GL_ARB_shader_objects
  129. FF_PFNGLATTACHSHADERPROC glAttachShader; //Require GL_ARB_shader_objects
  130. FF_PFNGLCREATESHADERPROC glCreateShader; //Require GL_ARB_shader_objects
  131. FF_PFNGLDELETESHADERPROC glDeleteShader; //Require GL_ARB_shader_objects
  132. FF_PFNGLCOMPILESHADERPROC glCompileShader; //Require GL_ARB_shader_objects
  133. FF_PFNGLSHADERSOURCEPROC glShaderSource; //Require GL_ARB_shader_objects
  134. FF_PFNGLGETSHADERIVPROC glGetShaderiv; //Require GL_ARB_shader_objects
  135. FF_PFNGLGETSHADERINFOLOGPROC glGetShaderInfoLog; //Require GL_ARB_shader_objects
  136. } FFOpenGLFunctions;
  137. #define OPENGL_ERROR_CHECK(ctx) \
  138. {\
  139. GLenum err_code; \
  140. if ((err_code = glGetError()) != GL_NO_ERROR) { \
  141. av_log(ctx, AV_LOG_ERROR, "OpenGL error occurred in '%s', line %d: %d\n", __FUNCTION__, __LINE__, err_code); \
  142. goto fail; \
  143. } \
  144. }\
  145. typedef struct OpenGLVertexInfo
  146. {
  147. float x, y, z; ///<Position
  148. float s0, t0; ///<Texture coords
  149. } OpenGLVertexInfo;
  150. /* defines 2 triangles to display */
  151. static GLushort g_index[6] =
  152. {
  153. 0, 1, 2,
  154. 0, 3, 2,
  155. };
  156. typedef struct OpenGLContext {
  157. AVClass *class; ///< class for private options
  158. #if HAVE_SDL
  159. SDL_Surface *surface;
  160. #endif
  161. FFOpenGLFunctions glprocs;
  162. uint8_t background[4]; ///< Background color
  163. int no_window; ///< 0 for create default window
  164. char *window_title; ///< Title of the window
  165. /* OpenGL implementation limits */
  166. GLint max_texture_size; ///< Maximum texture size
  167. GLint max_viewport_width; ///< Maximum viewport size
  168. GLint max_viewport_height; ///< Maximum viewport size
  169. int non_pow_2_textures; ///< 1 when non power of 2 textures are supported
  170. /* Current OpenGL configuration */
  171. GLuint program; ///< Shader program
  172. GLuint vertex_shader; ///< Vertex shader
  173. GLuint fragment_shader; ///< Fragment shader for current pix_pmt
  174. GLuint texture_name[4]; ///< Textures' IDs
  175. GLuint index_buffer; ///< Index buffer
  176. GLuint vertex_buffer; ///< Vertex buffer
  177. OpenGLVertexInfo vertex[4]; ///< VBO
  178. GLint projection_matrix_location; ///< Uniforms' locations
  179. GLint model_view_matrix_location;
  180. GLint color_map_location;
  181. GLint chroma_div_w_location;
  182. GLint chroma_div_h_location;
  183. GLint texture_location[4];
  184. GLint position_attrib; ///< Attibutes' locations
  185. GLint texture_coords_attrib;
  186. GLfloat projection_matrix[16]; ///< Projection matrix
  187. GLfloat model_view_matrix[16]; ///< Modev view matrix
  188. GLfloat color_map[16]; ///< RGBA color map matrix
  189. GLfloat chroma_div_w; ///< Chroma subsampling w ratio
  190. GLfloat chroma_div_h; ///< Chroma subsampling h ratio
  191. /* Stream information */
  192. GLenum format;
  193. GLenum type;
  194. int width; ///< Stream width
  195. int height; ///< Stream height
  196. enum AVPixelFormat pix_fmt; ///< Stream pixel format
  197. int picture_width; ///< Rendered width
  198. int picture_height; ///< Rendered height
  199. int window_width;
  200. int window_height;
  201. } OpenGLContext;
  202. static const struct OpenGLFormatDesc {
  203. enum AVPixelFormat fixel_format;
  204. const char * const * fragment_shader;
  205. GLenum format;
  206. GLenum type;
  207. } opengl_format_desc[] = {
  208. { AV_PIX_FMT_YUV420P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  209. { AV_PIX_FMT_YUV444P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  210. { AV_PIX_FMT_YUV422P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  211. { AV_PIX_FMT_YUV410P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  212. { AV_PIX_FMT_YUV411P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  213. { AV_PIX_FMT_YUV440P, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  214. { AV_PIX_FMT_YUV420P16, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  215. { AV_PIX_FMT_YUV422P16, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  216. { AV_PIX_FMT_YUV444P16, &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  217. { AV_PIX_FMT_YUVA420P, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  218. { AV_PIX_FMT_YUVA444P, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  219. { AV_PIX_FMT_YUVA422P, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  220. { AV_PIX_FMT_YUVA420P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  221. { AV_PIX_FMT_YUVA422P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  222. { AV_PIX_FMT_YUVA444P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  223. { AV_PIX_FMT_RGB24, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, GL_UNSIGNED_BYTE },
  224. { AV_PIX_FMT_BGR24, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, GL_UNSIGNED_BYTE },
  225. { AV_PIX_FMT_0RGB, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  226. { AV_PIX_FMT_RGB0, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  227. { AV_PIX_FMT_0BGR, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  228. { AV_PIX_FMT_BGR0, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  229. { AV_PIX_FMT_RGB565, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, GL_UNSIGNED_SHORT_5_6_5 },
  230. { AV_PIX_FMT_BGR565, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, GL_UNSIGNED_SHORT_5_6_5 },
  231. { AV_PIX_FMT_RGB555, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, FF_GL_UNSIGNED_SHORT_1_5_5_5_REV },
  232. { AV_PIX_FMT_BGR555, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGBA, FF_GL_UNSIGNED_SHORT_1_5_5_5_REV },
  233. { AV_PIX_FMT_RGB8, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, FF_GL_UNSIGNED_BYTE_3_3_2 },
  234. { AV_PIX_FMT_BGR8, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, FF_GL_UNSIGNED_BYTE_2_3_3_REV },
  235. { AV_PIX_FMT_RGB48, &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET, GL_RGB, GL_UNSIGNED_SHORT },
  236. { AV_PIX_FMT_ARGB, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  237. { AV_PIX_FMT_RGBA, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  238. { AV_PIX_FMT_ABGR, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  239. { AV_PIX_FMT_BGRA, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },
  240. { AV_PIX_FMT_RGBA64, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_SHORT },
  241. { AV_PIX_FMT_BGRA64, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_SHORT },
  242. { AV_PIX_FMT_GBRP, &FF_OPENGL_FRAGMENT_SHADER_RGB_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  243. { AV_PIX_FMT_GBRP16, &FF_OPENGL_FRAGMENT_SHADER_RGB_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  244. { AV_PIX_FMT_GBRAP, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },
  245. { AV_PIX_FMT_GBRAP16, &FF_OPENGL_FRAGMENT_SHADER_RGBA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },
  246. { AV_PIX_FMT_NONE, NULL }
  247. };
  248. static av_cold int opengl_prepare_vertex(AVFormatContext *s);
  249. static int opengl_draw(AVFormatContext *h, AVPacket *pkt, int repaint);
  250. static av_cold int opengl_init_context(OpenGLContext *opengl);
  251. static av_cold void opengl_deinit_context(OpenGLContext *opengl)
  252. {
  253. glDeleteTextures(4, opengl->texture_name);
  254. opengl->texture_name[0] = opengl->texture_name[1] =
  255. opengl->texture_name[2] = opengl->texture_name[3] = 0;
  256. if (opengl->glprocs.glUseProgram)
  257. opengl->glprocs.glUseProgram(0);
  258. if (opengl->glprocs.glDeleteProgram) {
  259. opengl->glprocs.glDeleteProgram(opengl->program);
  260. opengl->program = 0;
  261. }
  262. if (opengl->glprocs.glDeleteShader) {
  263. opengl->glprocs.glDeleteShader(opengl->vertex_shader);
  264. opengl->glprocs.glDeleteShader(opengl->fragment_shader);
  265. opengl->vertex_shader = opengl->fragment_shader = 0;
  266. }
  267. if (opengl->glprocs.glBindBuffer) {
  268. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, 0);
  269. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, 0);
  270. }
  271. if (opengl->glprocs.glDeleteBuffers) {
  272. opengl->glprocs.glDeleteBuffers(2, &opengl->index_buffer);
  273. opengl->vertex_buffer = opengl->index_buffer = 0;
  274. }
  275. }
  276. static int opengl_resize(AVFormatContext *h, int width, int height)
  277. {
  278. int ret = 0;
  279. OpenGLContext *opengl = h->priv_data;
  280. opengl->window_width = width;
  281. opengl->window_height = height;
  282. /* max_viewport_width == 0 means write_header was not called yet. */
  283. if (opengl->max_viewport_width) {
  284. if (opengl->no_window &&
  285. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {
  286. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  287. goto end;
  288. }
  289. if ((ret = opengl_prepare_vertex(h)) < 0)
  290. goto end;
  291. ret = opengl_draw(h, NULL, 1);
  292. }
  293. end:
  294. return ret;
  295. }
  296. static int opengl_control_message(AVFormatContext *h, int type, void *data, size_t data_size)
  297. {
  298. OpenGLContext *opengl = h->priv_data;
  299. switch(type) {
  300. case AV_APP_TO_DEV_WINDOW_SIZE:
  301. if (data) {
  302. AVDeviceRect *message = data;
  303. return opengl_resize(h, message->width, message->height);
  304. }
  305. return AVERROR(EINVAL);
  306. case AV_APP_TO_DEV_WINDOW_REPAINT:
  307. return opengl_resize(h, opengl->window_width, opengl->window_height);
  308. }
  309. return AVERROR(ENOSYS);
  310. }
  311. #if HAVE_SDL
  312. static int opengl_sdl_recreate_window(OpenGLContext *opengl, int width, int height)
  313. {
  314. opengl->surface = SDL_SetVideoMode(width, height,
  315. 32, SDL_OPENGL | SDL_RESIZABLE);
  316. if (!opengl->surface) {
  317. av_log(opengl, AV_LOG_ERROR, "Unable to set video mode: %s\n", SDL_GetError());
  318. return AVERROR_EXTERNAL;
  319. }
  320. SDL_GL_SetAttribute(SDL_GL_RED_SIZE, 8);
  321. SDL_GL_SetAttribute(SDL_GL_GREEN_SIZE, 8);
  322. SDL_GL_SetAttribute(SDL_GL_BLUE_SIZE, 8);
  323. SDL_GL_SetAttribute(SDL_GL_ALPHA_SIZE, 8);
  324. SDL_GL_SetAttribute(SDL_GL_DOUBLEBUFFER, 1);
  325. return 0;
  326. }
  327. static int opengl_sdl_process_events(AVFormatContext *h)
  328. {
  329. int ret;
  330. OpenGLContext *opengl = h->priv_data;
  331. SDL_Event event;
  332. SDL_PumpEvents();
  333. while (SDL_PeepEvents(&event, 1, SDL_GETEVENT, SDL_ALLEVENTS) > 0) {
  334. switch (event.type) {
  335. case SDL_QUIT:
  336. return AVERROR(EIO);
  337. case SDL_KEYDOWN:
  338. switch (event.key.keysym.sym) {
  339. case SDLK_ESCAPE:
  340. case SDLK_q:
  341. return AVERROR(EIO);
  342. }
  343. return 0;
  344. case SDL_VIDEORESIZE: {
  345. char buffer[100];
  346. int reinit;
  347. AVDeviceRect message;
  348. /* clean up old context because SDL_SetVideoMode may lose its state. */
  349. SDL_VideoDriverName(buffer, sizeof(buffer));
  350. reinit = !av_strncasecmp(buffer, "quartz", sizeof(buffer));
  351. if (reinit) {
  352. opengl_deinit_context(opengl);
  353. }
  354. if ((ret = opengl_sdl_recreate_window(opengl, event.resize.w, event.resize.h)) < 0)
  355. return ret;
  356. if (reinit && (ret = opengl_init_context(opengl)) < 0)
  357. return ret;
  358. message.width = opengl->surface->w;
  359. message.height = opengl->surface->h;
  360. return opengl_control_message(h, AV_APP_TO_DEV_WINDOW_SIZE, &message, sizeof(AVDeviceRect));
  361. }
  362. }
  363. }
  364. return 0;
  365. }
  366. static int av_cold opengl_sdl_create_window(AVFormatContext *h)
  367. {
  368. int ret;
  369. char buffer[100];
  370. OpenGLContext *opengl = h->priv_data;
  371. AVDeviceRect message;
  372. if (SDL_Init(SDL_INIT_VIDEO)) {
  373. av_log(opengl, AV_LOG_ERROR, "Unable to initialize SDL: %s\n", SDL_GetError());
  374. return AVERROR_EXTERNAL;
  375. }
  376. if ((ret = opengl_sdl_recreate_window(opengl, opengl->width, opengl->height)) < 0)
  377. return ret;
  378. av_log(opengl, AV_LOG_INFO, "SDL driver: '%s'.\n", SDL_VideoDriverName(buffer, sizeof(buffer)));
  379. message.width = opengl->surface->w;
  380. message.height = opengl->surface->h;
  381. SDL_WM_SetCaption(opengl->window_title, NULL);
  382. opengl_control_message(h, AV_APP_TO_DEV_WINDOW_SIZE, &message, sizeof(AVDeviceRect));
  383. return 0;
  384. }
  385. static int av_cold opengl_sdl_load_procedures(OpenGLContext *opengl)
  386. {
  387. FFOpenGLFunctions *procs = &opengl->glprocs;
  388. #define LOAD_OPENGL_FUN(name, type) \
  389. procs->name = (type)SDL_GL_GetProcAddress(#name); \
  390. if (!procs->name) { \
  391. av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \
  392. return AVERROR(ENOSYS); \
  393. }
  394. LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)
  395. LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)
  396. LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)
  397. LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)
  398. LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)
  399. LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)
  400. LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)
  401. LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)
  402. LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)
  403. LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)
  404. LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)
  405. LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)
  406. LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)
  407. LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)
  408. LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)
  409. LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)
  410. LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)
  411. LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)
  412. LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)
  413. LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)
  414. LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)
  415. LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)
  416. LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)
  417. LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)
  418. LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)
  419. return 0;
  420. #undef LOAD_OPENGL_FUN
  421. }
  422. #endif /* HAVE_SDL */
  423. #if defined(__APPLE__)
  424. static int av_cold opengl_load_procedures(OpenGLContext *opengl)
  425. {
  426. FFOpenGLFunctions *procs = &opengl->glprocs;
  427. #if HAVE_SDL
  428. if (!opengl->no_window)
  429. return opengl_sdl_load_procedures(opengl);
  430. #endif
  431. procs->glActiveTexture = glActiveTexture;
  432. procs->glGenBuffers = glGenBuffers;
  433. procs->glDeleteBuffers = glDeleteBuffers;
  434. procs->glBufferData = glBufferData;
  435. procs->glBindBuffer = glBindBuffer;
  436. procs->glGetAttribLocation = glGetAttribLocation;
  437. procs->glGetUniformLocation = glGetUniformLocation;
  438. procs->glUniform1f = glUniform1f;
  439. procs->glUniform1i = glUniform1i;
  440. procs->glUniformMatrix4fv = glUniformMatrix4fv;
  441. procs->glCreateProgram = glCreateProgram;
  442. procs->glDeleteProgram = glDeleteProgram;
  443. procs->glUseProgram = glUseProgram;
  444. procs->glLinkProgram = glLinkProgram;
  445. procs->glGetProgramiv = glGetProgramiv;
  446. procs->glGetProgramInfoLog = glGetProgramInfoLog;
  447. procs->glAttachShader = glAttachShader;
  448. procs->glCreateShader = glCreateShader;
  449. procs->glDeleteShader = glDeleteShader;
  450. procs->glCompileShader = glCompileShader;
  451. procs->glShaderSource = glShaderSource;
  452. procs->glGetShaderiv = glGetShaderiv;
  453. procs->glGetShaderInfoLog = glGetShaderInfoLog;
  454. procs->glEnableVertexAttribArray = glEnableVertexAttribArray;
  455. procs->glVertexAttribPointer = (FF_PFNGLVERTEXATTRIBPOINTERPROC) glVertexAttribPointer;
  456. return 0;
  457. }
  458. #else
  459. static int av_cold opengl_load_procedures(OpenGLContext *opengl)
  460. {
  461. FFOpenGLFunctions *procs = &opengl->glprocs;
  462. #if HAVE_GLXGETPROCADDRESS
  463. #define SelectedGetProcAddress glXGetProcAddress
  464. #elif HAVE_WGLGETPROCADDRESS
  465. #define SelectedGetProcAddress wglGetProcAddress
  466. #endif
  467. #define LOAD_OPENGL_FUN(name, type) \
  468. procs->name = (type)SelectedGetProcAddress(#name); \
  469. if (!procs->name) { \
  470. av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \
  471. return AVERROR(ENOSYS); \
  472. }
  473. #if HAVE_SDL
  474. if (!opengl->no_window)
  475. return opengl_sdl_load_procedures(opengl);
  476. #endif
  477. LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)
  478. LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)
  479. LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)
  480. LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)
  481. LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)
  482. LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)
  483. LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)
  484. LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)
  485. LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)
  486. LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)
  487. LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)
  488. LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)
  489. LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)
  490. LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)
  491. LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)
  492. LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)
  493. LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)
  494. LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)
  495. LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)
  496. LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)
  497. LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)
  498. LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)
  499. LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)
  500. LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)
  501. LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)
  502. return 0;
  503. #undef SelectedGetProcAddress
  504. #undef LOAD_OPENGL_FUN
  505. }
  506. #endif
  507. static void opengl_make_identity(float matrix[16])
  508. {
  509. memset(matrix, 0, 16 * sizeof(float));
  510. matrix[0] = matrix[5] = matrix[10] = matrix[15] = 1.0f;
  511. }
  512. static void opengl_make_ortho(float matrix[16], float left, float right,
  513. float bottom, float top, float nearZ, float farZ)
  514. {
  515. float ral = right + left;
  516. float rsl = right - left;
  517. float tab = top + bottom;
  518. float tsb = top - bottom;
  519. float fan = farZ + nearZ;
  520. float fsn = farZ - nearZ;
  521. memset(matrix, 0, 16 * sizeof(float));
  522. matrix[0] = 2.0f / rsl;
  523. matrix[5] = 2.0f / tsb;
  524. matrix[10] = -2.0f / fsn;
  525. matrix[12] = -ral / rsl;
  526. matrix[13] = -tab / tsb;
  527. matrix[14] = -fan / fsn;
  528. matrix[15] = 1.0f;
  529. }
  530. static av_cold int opengl_read_limits(OpenGLContext *opengl)
  531. {
  532. static const struct{
  533. const char *extention;
  534. int major;
  535. int minor;
  536. } required_extensions[] = {
  537. { "GL_ARB_multitexture", 1, 3 },
  538. { "GL_ARB_vertex_buffer_object", 1, 5 }, //GLX_ARB_vertex_buffer_object
  539. { "GL_ARB_vertex_shader", 2, 0 },
  540. { "GL_ARB_fragment_shader", 2, 0 },
  541. { "GL_ARB_shader_objects", 2, 0 },
  542. { NULL, 0, 0 }
  543. };
  544. int i, major, minor;
  545. const char *extensions, *version;
  546. version = glGetString(GL_VERSION);
  547. extensions = glGetString(GL_EXTENSIONS);
  548. av_log(opengl, AV_LOG_DEBUG, "OpenGL version: %s\n", version);
  549. sscanf(version, "%d.%d", &major, &minor);
  550. for (i = 0; required_extensions[i].extention; i++) {
  551. if (major < required_extensions[i].major &&
  552. (major == required_extensions[i].major && minor < required_extensions[i].minor) &&
  553. !strstr(extensions, required_extensions[i].extention)) {
  554. av_log(opengl, AV_LOG_ERROR, "Required extension %s is not supported.\n",
  555. required_extensions[i].extention);
  556. av_log(opengl, AV_LOG_DEBUG, "Supported extensions are: %s\n", extensions);
  557. return AVERROR(ENOSYS);
  558. }
  559. }
  560. glGetIntegerv(GL_MAX_TEXTURE_SIZE, &opengl->max_texture_size);
  561. glGetIntegerv(GL_MAX_VIEWPORT_DIMS, &opengl->max_viewport_width);
  562. opengl->non_pow_2_textures = major >= 2 || strstr(extensions, "GL_ARB_texture_non_power_of_two");
  563. av_log(opengl, AV_LOG_DEBUG, "Non Power of 2 textures support: %s\n", opengl->non_pow_2_textures ? "Yes" : "No");
  564. av_log(opengl, AV_LOG_DEBUG, "Max texture size: %dx%d\n", opengl->max_texture_size, opengl->max_texture_size);
  565. av_log(opengl, AV_LOG_DEBUG, "Max viewport size: %dx%d\n",
  566. opengl->max_viewport_width, opengl->max_viewport_height);
  567. OPENGL_ERROR_CHECK(opengl);
  568. return 0;
  569. fail:
  570. return AVERROR_EXTERNAL;
  571. }
  572. static const char* opengl_get_fragment_shader_code(enum AVPixelFormat format)
  573. {
  574. int i;
  575. for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {
  576. if (opengl_format_desc[i].fixel_format == format)
  577. return *opengl_format_desc[i].fragment_shader;
  578. }
  579. return NULL;
  580. }
  581. static int opengl_type_size(GLenum type)
  582. {
  583. switch(type) {
  584. case GL_UNSIGNED_SHORT:
  585. case FF_GL_UNSIGNED_SHORT_1_5_5_5_REV:
  586. case GL_UNSIGNED_SHORT_5_6_5:
  587. return 2;
  588. case GL_UNSIGNED_BYTE:
  589. case FF_GL_UNSIGNED_BYTE_3_3_2:
  590. case FF_GL_UNSIGNED_BYTE_2_3_3_REV:
  591. default:
  592. break;
  593. }
  594. return 1;
  595. }
  596. static av_cold void opengl_get_texture_params(OpenGLContext *opengl)
  597. {
  598. int i;
  599. for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {
  600. if (opengl_format_desc[i].fixel_format == opengl->pix_fmt) {
  601. opengl->format = opengl_format_desc[i].format;
  602. opengl->type = opengl_format_desc[i].type;
  603. break;
  604. }
  605. }
  606. }
  607. static void opengl_compute_display_area(AVFormatContext *s)
  608. {
  609. AVRational sar, dar; /* sample and display aspect ratios */
  610. OpenGLContext *opengl = s->priv_data;
  611. AVStream *st = s->streams[0];
  612. AVCodecContext *encctx = st->codec;
  613. /* compute overlay width and height from the codec context information */
  614. sar = st->sample_aspect_ratio.num ? st->sample_aspect_ratio : (AVRational){ 1, 1 };
  615. dar = av_mul_q(sar, (AVRational){ encctx->width, encctx->height });
  616. /* we suppose the screen has a 1/1 sample aspect ratio */
  617. /* fit in the window */
  618. if (av_cmp_q(dar, (AVRational){ opengl->window_width, opengl->window_height }) > 0) {
  619. /* fit in width */
  620. opengl->picture_width = opengl->window_width;
  621. opengl->picture_height = av_rescale(opengl->picture_width, dar.den, dar.num);
  622. } else {
  623. /* fit in height */
  624. opengl->picture_height = opengl->window_height;
  625. opengl->picture_width = av_rescale(opengl->picture_height, dar.num, dar.den);
  626. }
  627. }
  628. static av_cold void opengl_get_texture_size(OpenGLContext *opengl, int in_width, int in_height,
  629. int *out_width, int *out_height)
  630. {
  631. if (opengl->non_pow_2_textures) {
  632. *out_width = in_width;
  633. *out_height = in_height;
  634. } else {
  635. int max = FFMIN(FFMAX(in_width, in_height), opengl->max_texture_size);
  636. unsigned power_of_2 = 1;
  637. while (power_of_2 < max)
  638. power_of_2 *= 2;
  639. *out_height = power_of_2;
  640. *out_width = power_of_2;
  641. av_log(opengl, AV_LOG_DEBUG, "Texture size calculated from %dx%d into %dx%d\n",
  642. in_width, in_height, *out_width, *out_height);
  643. }
  644. }
  645. static av_cold void opengl_fill_color_map(OpenGLContext *opengl)
  646. {
  647. const AVPixFmtDescriptor *desc;
  648. int shift;
  649. enum AVPixelFormat pix_fmt = opengl->pix_fmt;
  650. /* We need order of components, not exact position, some minor HACKs here */
  651. if (pix_fmt == AV_PIX_FMT_RGB565 || pix_fmt == AV_PIX_FMT_BGR555 ||
  652. pix_fmt == AV_PIX_FMT_BGR8 || pix_fmt == AV_PIX_FMT_RGB8)
  653. pix_fmt = AV_PIX_FMT_RGB24;
  654. else if (pix_fmt == AV_PIX_FMT_BGR565 || pix_fmt == AV_PIX_FMT_RGB555)
  655. pix_fmt = AV_PIX_FMT_BGR24;
  656. desc = av_pix_fmt_desc_get(pix_fmt);
  657. if (!(desc->flags & AV_PIX_FMT_FLAG_RGB))
  658. return;
  659. #define FILL_COMPONENT(i) { \
  660. shift = desc->comp[i].depth_minus1 >> 3; \
  661. opengl->color_map[(i << 2) + ((desc->comp[i].offset_plus1 - 1) >> shift)] = 1.0; \
  662. }
  663. memset(opengl->color_map, 0, sizeof(opengl->color_map));
  664. FILL_COMPONENT(0);
  665. FILL_COMPONENT(1);
  666. FILL_COMPONENT(2);
  667. if (desc->flags & AV_PIX_FMT_FLAG_ALPHA)
  668. FILL_COMPONENT(3);
  669. #undef FILL_COMPONENT
  670. }
  671. static av_cold GLuint opengl_load_shader(OpenGLContext *opengl, GLenum type, const char *source)
  672. {
  673. GLuint shader = opengl->glprocs.glCreateShader(type);
  674. GLint result;
  675. if (!shader) {
  676. av_log(opengl, AV_LOG_ERROR, "glCreateShader() failed\n");
  677. return 0;
  678. }
  679. opengl->glprocs.glShaderSource(shader, 1, &source, NULL);
  680. opengl->glprocs.glCompileShader(shader);
  681. opengl->glprocs.glGetShaderiv(shader, FF_GL_COMPILE_STATUS, &result);
  682. if (!result) {
  683. char *log;
  684. opengl->glprocs.glGetShaderiv(shader, FF_GL_INFO_LOG_LENGTH, &result);
  685. if (result) {
  686. if ((log = av_malloc(result))) {
  687. opengl->glprocs.glGetShaderInfoLog(shader, result, NULL, log);
  688. av_log(opengl, AV_LOG_ERROR, "Compile error: %s\n", log);
  689. av_free(log);
  690. }
  691. }
  692. goto fail;
  693. }
  694. OPENGL_ERROR_CHECK(opengl);
  695. return shader;
  696. fail:
  697. opengl->glprocs.glDeleteShader(shader);
  698. return 0;
  699. }
  700. static av_cold int opengl_compile_shaders(OpenGLContext *opengl, enum AVPixelFormat pix_fmt)
  701. {
  702. GLint result;
  703. const char *fragment_shader_code = opengl_get_fragment_shader_code(pix_fmt);
  704. if (!fragment_shader_code) {
  705. av_log(opengl, AV_LOG_ERROR, "Provided pixel format '%s' is not supported\n",
  706. av_get_pix_fmt_name(pix_fmt));
  707. return AVERROR(EINVAL);
  708. }
  709. opengl->vertex_shader = opengl_load_shader(opengl, FF_GL_VERTEX_SHADER,
  710. FF_OPENGL_VERTEX_SHADER);
  711. if (!opengl->vertex_shader) {
  712. av_log(opengl, AV_LOG_ERROR, "Vertex shader loading failed.\n");
  713. goto fail;
  714. }
  715. opengl->fragment_shader = opengl_load_shader(opengl, FF_GL_FRAGMENT_SHADER,
  716. fragment_shader_code);
  717. if (!opengl->fragment_shader) {
  718. av_log(opengl, AV_LOG_ERROR, "Fragment shader loading failed.\n");
  719. goto fail;
  720. }
  721. opengl->program = opengl->glprocs.glCreateProgram();
  722. if (!opengl->program)
  723. goto fail;
  724. opengl->glprocs.glAttachShader(opengl->program, opengl->vertex_shader);
  725. opengl->glprocs.glAttachShader(opengl->program, opengl->fragment_shader);
  726. opengl->glprocs.glLinkProgram(opengl->program);
  727. opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_LINK_STATUS, &result);
  728. if (!result) {
  729. char *log;
  730. opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_INFO_LOG_LENGTH, &result);
  731. if (result) {
  732. log = av_malloc(result);
  733. if (!log)
  734. goto fail;
  735. opengl->glprocs.glGetProgramInfoLog(opengl->program, result, NULL, log);
  736. av_log(opengl, AV_LOG_ERROR, "Link error: %s\n", log);
  737. av_free(log);
  738. }
  739. goto fail;
  740. }
  741. opengl->position_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_position");
  742. opengl->texture_coords_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_textureCoords");
  743. opengl->projection_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_projectionMatrix");
  744. opengl->model_view_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_modelViewMatrix");
  745. opengl->color_map_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_colorMap");
  746. opengl->texture_location[0] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture0");
  747. opengl->texture_location[1] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture1");
  748. opengl->texture_location[2] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture2");
  749. opengl->texture_location[3] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture3");
  750. opengl->chroma_div_w_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_w");
  751. opengl->chroma_div_h_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_h");
  752. OPENGL_ERROR_CHECK(opengl);
  753. return 0;
  754. fail:
  755. opengl->glprocs.glDeleteShader(opengl->vertex_shader);
  756. opengl->glprocs.glDeleteShader(opengl->fragment_shader);
  757. opengl->glprocs.glDeleteProgram(opengl->program);
  758. opengl->fragment_shader = opengl->vertex_shader = opengl->program = 0;
  759. return AVERROR_EXTERNAL;
  760. }
  761. static av_cold int opengl_configure_texture(OpenGLContext *opengl, GLuint texture,
  762. GLsizei width, GLsizei height)
  763. {
  764. if (texture) {
  765. int new_width, new_height;
  766. opengl_get_texture_size(opengl, width, height, &new_width, &new_height);
  767. glBindTexture(GL_TEXTURE_2D, texture);
  768. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  769. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
  770. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
  771. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
  772. glTexImage2D(GL_TEXTURE_2D, 0, opengl->format, new_width, new_height, 0,
  773. opengl->format, opengl->type, NULL);
  774. OPENGL_ERROR_CHECK(NULL);
  775. }
  776. return 0;
  777. fail:
  778. return AVERROR_EXTERNAL;
  779. }
  780. static av_cold int opengl_prepare_vertex(AVFormatContext *s)
  781. {
  782. OpenGLContext *opengl = s->priv_data;
  783. int tex_w, tex_h;
  784. if (opengl->window_width > opengl->max_viewport_width || opengl->window_height > opengl->max_viewport_height) {
  785. opengl->window_width = FFMAX(opengl->window_width, opengl->max_viewport_width);
  786. opengl->window_height = FFMAX(opengl->window_height, opengl->max_viewport_height);
  787. av_log(opengl, AV_LOG_WARNING, "Too big viewport requested, limited to %dx%d", opengl->window_width, opengl->window_height);
  788. }
  789. glViewport(0, 0, opengl->window_width, opengl->window_height);
  790. opengl_make_ortho(opengl->projection_matrix,
  791. - (float)opengl->window_width / 2.0f, (float)opengl->window_width / 2.0f,
  792. - (float)opengl->window_height / 2.0f, (float)opengl->window_height / 2.0f,
  793. 1.0f, -1.0f);
  794. opengl_make_identity(opengl->model_view_matrix);
  795. opengl_compute_display_area(s);
  796. opengl->vertex[0].z = opengl->vertex[1].z = opengl->vertex[2].z = opengl->vertex[3].z = 0.0f;
  797. opengl->vertex[0].x = opengl->vertex[1].x = - (float)opengl->picture_width / 2.0f;
  798. opengl->vertex[2].x = opengl->vertex[3].x = (float)opengl->picture_width / 2.0f;
  799. opengl->vertex[1].y = opengl->vertex[2].y = - (float)opengl->picture_height / 2.0f;
  800. opengl->vertex[0].y = opengl->vertex[3].y = (float)opengl->picture_height / 2.0f;
  801. opengl_get_texture_size(opengl, opengl->width, opengl->height, &tex_w, &tex_h);
  802. opengl->vertex[0].s0 = 0.0f;
  803. opengl->vertex[0].t0 = 0.0f;
  804. opengl->vertex[1].s0 = 0.0f;
  805. opengl->vertex[1].t0 = (float)opengl->height / (float)tex_h;
  806. opengl->vertex[2].s0 = (float)opengl->width / (float)tex_w;
  807. opengl->vertex[2].t0 = (float)opengl->height / (float)tex_h;
  808. opengl->vertex[3].s0 = (float)opengl->width / (float)tex_w;
  809. opengl->vertex[3].t0 = 0.0f;
  810. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);
  811. opengl->glprocs.glBufferData(FF_GL_ARRAY_BUFFER, sizeof(opengl->vertex), opengl->vertex, FF_GL_STATIC_DRAW);
  812. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, 0);
  813. OPENGL_ERROR_CHECK(opengl);
  814. return 0;
  815. fail:
  816. return AVERROR_EXTERNAL;
  817. }
  818. static int opengl_prepare(OpenGLContext *opengl)
  819. {
  820. int i;
  821. opengl->glprocs.glUseProgram(opengl->program);
  822. opengl->glprocs.glUniformMatrix4fv(opengl->projection_matrix_location, 1, GL_FALSE, opengl->projection_matrix);
  823. opengl->glprocs.glUniformMatrix4fv(opengl->model_view_matrix_location, 1, GL_FALSE, opengl->model_view_matrix);
  824. for (i = 0; i < 4; i++)
  825. if (opengl->texture_location[i] != -1) {
  826. opengl->glprocs.glActiveTexture(GL_TEXTURE0 + i);
  827. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[i]);
  828. opengl->glprocs.glUniform1i(opengl->texture_location[i], i);
  829. }
  830. if (opengl->color_map_location != -1)
  831. opengl->glprocs.glUniformMatrix4fv(opengl->color_map_location, 1, GL_FALSE, opengl->color_map);
  832. if (opengl->chroma_div_h_location != -1)
  833. opengl->glprocs.glUniform1f(opengl->chroma_div_h_location, opengl->chroma_div_h);
  834. if (opengl->chroma_div_w_location != -1)
  835. opengl->glprocs.glUniform1f(opengl->chroma_div_w_location, opengl->chroma_div_w);
  836. OPENGL_ERROR_CHECK(opengl);
  837. return 0;
  838. fail:
  839. return AVERROR_EXTERNAL;
  840. }
  841. static int opengl_create_window(AVFormatContext *h)
  842. {
  843. OpenGLContext *opengl = h->priv_data;
  844. int ret;
  845. if (!opengl->no_window) {
  846. #if HAVE_SDL
  847. if ((ret = opengl_sdl_create_window(h)) < 0) {
  848. av_log(opengl, AV_LOG_ERROR, "Cannot create default SDL window.\n");
  849. return ret;
  850. }
  851. #else
  852. av_log(opengl, AV_LOG_ERROR, "FFmpeg is compiled without SDL. Cannot create default window.\n");
  853. return AVERROR(ENOSYS);
  854. #endif
  855. } else {
  856. if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_CREATE_WINDOW_BUFFER, NULL , 0)) < 0) {
  857. av_log(opengl, AV_LOG_ERROR, "Application failed to create window buffer.\n");
  858. return ret;
  859. }
  860. if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {
  861. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  862. return ret;
  863. }
  864. }
  865. return 0;
  866. }
  867. static int opengl_release_window(AVFormatContext *h)
  868. {
  869. int ret;
  870. OpenGLContext *opengl = h->priv_data;
  871. if (!opengl->no_window) {
  872. #if HAVE_SDL
  873. SDL_Quit();
  874. #endif
  875. } else if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DESTROY_WINDOW_BUFFER, NULL , 0)) < 0) {
  876. av_log(opengl, AV_LOG_ERROR, "Application failed to release window buffer.\n");
  877. return ret;
  878. }
  879. return 0;
  880. }
  881. static av_cold int opengl_write_trailer(AVFormatContext *h)
  882. {
  883. OpenGLContext *opengl = h->priv_data;
  884. if (opengl->no_window &&
  885. avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0) < 0)
  886. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  887. opengl_deinit_context(opengl);
  888. opengl_release_window(h);
  889. return 0;
  890. }
  891. static av_cold int opengl_init_context(OpenGLContext *opengl)
  892. {
  893. int i, ret;
  894. const AVPixFmtDescriptor *desc;
  895. if ((ret = opengl_compile_shaders(opengl, opengl->pix_fmt)) < 0)
  896. goto fail;
  897. desc = av_pix_fmt_desc_get(opengl->pix_fmt);
  898. av_assert0(desc->nb_components > 0 && desc->nb_components <= 4);
  899. glGenTextures(desc->nb_components, opengl->texture_name);
  900. opengl->glprocs.glGenBuffers(2, &opengl->index_buffer);
  901. if (!opengl->index_buffer || !opengl->vertex_buffer) {
  902. av_log(opengl, AV_LOG_ERROR, "Buffer generation failed.\n");
  903. ret = AVERROR_EXTERNAL;
  904. goto fail;
  905. }
  906. opengl_configure_texture(opengl, opengl->texture_name[0], opengl->width, opengl->height);
  907. if (desc->nb_components > 1) {
  908. int has_alpha = desc->flags & AV_PIX_FMT_FLAG_ALPHA;
  909. int num_planes = desc->nb_components - (has_alpha ? 1 : 0);
  910. if (opengl->non_pow_2_textures) {
  911. opengl->chroma_div_w = 1.0f;
  912. opengl->chroma_div_h = 1.0f;
  913. } else {
  914. opengl->chroma_div_w = 1 << desc->log2_chroma_w;
  915. opengl->chroma_div_h = 1 << desc->log2_chroma_h;
  916. }
  917. for (i = 1; i < num_planes; i++)
  918. if (opengl->non_pow_2_textures)
  919. opengl_configure_texture(opengl, opengl->texture_name[i],
  920. FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w),
  921. FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h));
  922. else
  923. opengl_configure_texture(opengl, opengl->texture_name[i], opengl->width, opengl->height);
  924. if (has_alpha)
  925. opengl_configure_texture(opengl, opengl->texture_name[3], opengl->width, opengl->height);
  926. }
  927. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);
  928. opengl->glprocs.glBufferData(FF_GL_ELEMENT_ARRAY_BUFFER, sizeof(g_index), g_index, FF_GL_STATIC_DRAW);
  929. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, 0);
  930. glEnable(GL_BLEND);
  931. glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
  932. glClearColor((float)opengl->background[0] / 255.0f, (float)opengl->background[1] / 255.0f,
  933. (float)opengl->background[2] / 255.0f, 1.0f);
  934. ret = AVERROR_EXTERNAL;
  935. OPENGL_ERROR_CHECK(opengl);
  936. return 0;
  937. fail:
  938. return ret;
  939. }
  940. static av_cold int opengl_write_header(AVFormatContext *h)
  941. {
  942. OpenGLContext *opengl = h->priv_data;
  943. AVStream *st;
  944. int ret;
  945. if (h->nb_streams != 1 ||
  946. h->streams[0]->codec->codec_type != AVMEDIA_TYPE_VIDEO ||
  947. h->streams[0]->codec->codec_id != AV_CODEC_ID_RAWVIDEO) {
  948. av_log(opengl, AV_LOG_ERROR, "Only a single video stream is supported.\n");
  949. return AVERROR(EINVAL);
  950. }
  951. st = h->streams[0];
  952. opengl->width = st->codec->width;
  953. opengl->height = st->codec->height;
  954. opengl->pix_fmt = st->codec->pix_fmt;
  955. if (!opengl->window_title && !opengl->no_window)
  956. opengl->window_title = av_strdup(h->filename);
  957. if ((ret = opengl_create_window(h)))
  958. goto fail;
  959. if ((ret = opengl_read_limits(opengl)) < 0)
  960. goto fail;
  961. if (opengl->width > opengl->max_texture_size || opengl->height > opengl->max_texture_size) {
  962. av_log(opengl, AV_LOG_ERROR, "Too big picture %dx%d, max supported size is %dx%d\n",
  963. opengl->width, opengl->height, opengl->max_texture_size, opengl->max_texture_size);
  964. ret = AVERROR(EINVAL);
  965. goto fail;
  966. }
  967. if ((ret = opengl_load_procedures(opengl)) < 0)
  968. goto fail;
  969. opengl_fill_color_map(opengl);
  970. opengl_get_texture_params(opengl);
  971. if ((ret = opengl_init_context(opengl)) < 0)
  972. goto fail;
  973. if ((ret = opengl_prepare_vertex(h)) < 0)
  974. goto fail;
  975. glClear(GL_COLOR_BUFFER_BIT);
  976. #if HAVE_SDL
  977. if (!opengl->no_window)
  978. SDL_GL_SwapBuffers();
  979. #endif
  980. if (opengl->no_window &&
  981. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {
  982. av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");
  983. goto fail;
  984. }
  985. ret = AVERROR_EXTERNAL;
  986. OPENGL_ERROR_CHECK(opengl);
  987. return 0;
  988. fail:
  989. opengl_write_trailer(h);
  990. return ret;
  991. }
  992. static uint8_t* opengl_get_plane_pointer(OpenGLContext *opengl, AVPacket *pkt, int comp_index,
  993. const AVPixFmtDescriptor *desc)
  994. {
  995. uint8_t *data = pkt->data;
  996. int wordsize = opengl_type_size(opengl->type);
  997. int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);
  998. int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);
  999. int plane = desc->comp[comp_index].plane;
  1000. switch(plane) {
  1001. case 0:
  1002. break;
  1003. case 1:
  1004. data += opengl->width * opengl->height * wordsize;
  1005. break;
  1006. case 2:
  1007. data += opengl->width * opengl->height * wordsize;
  1008. data += width_chroma * height_chroma * wordsize;
  1009. break;
  1010. case 3:
  1011. data += opengl->width * opengl->height * wordsize;
  1012. data += 2 * width_chroma * height_chroma * wordsize;
  1013. break;
  1014. default:
  1015. return NULL;
  1016. }
  1017. return data;
  1018. }
  1019. static int opengl_draw(AVFormatContext *h, AVPacket *pkt, int repaint)
  1020. {
  1021. OpenGLContext *opengl = h->priv_data;
  1022. enum AVPixelFormat pix_fmt = h->streams[0]->codec->pix_fmt;
  1023. const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
  1024. int ret;
  1025. #if HAVE_SDL
  1026. if (!opengl->no_window && (ret = opengl_sdl_process_events(h)) < 0)
  1027. goto fail;
  1028. #endif
  1029. if (opengl->no_window &&
  1030. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {
  1031. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  1032. goto fail;
  1033. }
  1034. glClear(GL_COLOR_BUFFER_BIT);
  1035. if (!repaint) {
  1036. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[0]);
  1037. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,
  1038. opengl_get_plane_pointer(opengl, pkt, 0, desc));
  1039. if (desc->flags & AV_PIX_FMT_FLAG_PLANAR) {
  1040. int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);
  1041. int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);
  1042. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[1]);
  1043. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,
  1044. opengl_get_plane_pointer(opengl, pkt, 1, desc));
  1045. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[2]);
  1046. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,
  1047. opengl_get_plane_pointer(opengl, pkt, 2, desc));
  1048. if (desc->flags & AV_PIX_FMT_FLAG_ALPHA) {
  1049. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[3]);
  1050. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,
  1051. opengl_get_plane_pointer(opengl, pkt, 3, desc));
  1052. }
  1053. }
  1054. }
  1055. ret = AVERROR_EXTERNAL;
  1056. OPENGL_ERROR_CHECK(opengl);
  1057. if ((ret = opengl_prepare(opengl)) < 0)
  1058. goto fail;
  1059. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);
  1060. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);
  1061. opengl->glprocs.glVertexAttribPointer(opengl->position_attrib, 3, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 0);
  1062. opengl->glprocs.glEnableVertexAttribArray(opengl->position_attrib);
  1063. opengl->glprocs.glVertexAttribPointer(opengl->texture_coords_attrib, 2, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 12);
  1064. opengl->glprocs.glEnableVertexAttribArray(opengl->texture_coords_attrib);
  1065. glDrawElements(GL_TRIANGLES, FF_ARRAY_ELEMS(g_index), GL_UNSIGNED_SHORT, 0);
  1066. ret = AVERROR_EXTERNAL;
  1067. OPENGL_ERROR_CHECK(opengl);
  1068. #if HAVE_SDL
  1069. if (!opengl->no_window)
  1070. SDL_GL_SwapBuffers();
  1071. #endif
  1072. if (opengl->no_window &&
  1073. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {
  1074. av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");
  1075. goto fail;
  1076. }
  1077. return 0;
  1078. fail:
  1079. return ret;
  1080. }
  1081. static int opengl_write_packet(AVFormatContext *h, AVPacket *pkt)
  1082. {
  1083. return opengl_draw(h, pkt, 0);
  1084. }
  1085. #define OFFSET(x) offsetof(OpenGLContext, x)
  1086. #define ENC AV_OPT_FLAG_ENCODING_PARAM
  1087. static const AVOption options[] = {
  1088. { "background", "set background color", OFFSET(background), AV_OPT_TYPE_COLOR, {.str = "black"}, CHAR_MIN, CHAR_MAX, ENC },
  1089. { "no_window", "disable default window", OFFSET(no_window), AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, ENC },
  1090. { "window_title", "set window title", OFFSET(window_title), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, ENC },
  1091. { NULL }
  1092. };
  1093. static const AVClass opengl_class = {
  1094. .class_name = "opengl outdev",
  1095. .item_name = av_default_item_name,
  1096. .option = options,
  1097. .version = LIBAVUTIL_VERSION_INT,
  1098. };
  1099. AVOutputFormat ff_opengl_muxer = {
  1100. .name = "opengl",
  1101. .long_name = NULL_IF_CONFIG_SMALL("OpenGL output"),
  1102. .priv_data_size = sizeof(OpenGLContext),
  1103. .audio_codec = AV_CODEC_ID_NONE,
  1104. .video_codec = AV_CODEC_ID_RAWVIDEO,
  1105. .write_header = opengl_write_header,
  1106. .write_packet = opengl_write_packet,
  1107. .write_trailer = opengl_write_trailer,
  1108. .control_message = opengl_control_message,
  1109. .flags = AVFMT_NOFILE | AVFMT_VARIABLE_FPS | AVFMT_NOTIMESTAMPS,
  1110. .priv_class = &opengl_class,
  1111. };