opengl_enc.c 56 KB

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