You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1240 lines
51KB

  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. opengl_control_message(h, AV_APP_TO_DEV_WINDOW_SIZE, &message, sizeof(AVDeviceRect));
  382. return 0;
  383. }
  384. static int av_cold opengl_sdl_load_procedures(OpenGLContext *opengl)
  385. {
  386. FFOpenGLFunctions *procs = &opengl->glprocs;
  387. #define LOAD_OPENGL_FUN(name, type) \
  388. procs->name = (type)SDL_GL_GetProcAddress(#name); \
  389. if (!procs->name) { \
  390. av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \
  391. return AVERROR(ENOSYS); \
  392. }
  393. LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)
  394. LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)
  395. LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)
  396. LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)
  397. LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)
  398. LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)
  399. LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)
  400. LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)
  401. LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)
  402. LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)
  403. LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)
  404. LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)
  405. LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)
  406. LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)
  407. LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)
  408. LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)
  409. LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)
  410. LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)
  411. LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)
  412. LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)
  413. LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)
  414. LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)
  415. LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)
  416. LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)
  417. LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)
  418. return 0;
  419. #undef LOAD_OPENGL_FUN
  420. }
  421. #endif /* HAVE_SDL */
  422. #if defined(__APPLE__)
  423. static int av_cold opengl_load_procedures(OpenGLContext *opengl)
  424. {
  425. FFOpenGLFunctions *procs = &opengl->glprocs;
  426. procs->glActiveTexture = glActiveTexture;
  427. procs->glGenBuffers = glGenBuffers;
  428. procs->glDeleteBuffers = glDeleteBuffers;
  429. procs->glBufferData = glBufferData;
  430. procs->glBindBuffer = glBindBuffer;
  431. procs->glGetAttribLocation = glGetAttribLocation;
  432. procs->glGetUniformLocation = glGetUniformLocation;
  433. procs->glUniform1f = glUniform1f;
  434. procs->glUniform1i = glUniform1i;
  435. procs->glUniformMatrix4fv = glUniformMatrix4fv;
  436. procs->glCreateProgram = glCreateProgram;
  437. procs->glDeleteProgram = glDeleteProgram;
  438. procs->glUseProgram = glUseProgram;
  439. procs->glLinkProgram = glLinkProgram;
  440. procs->glGetProgramiv = glGetProgramiv;
  441. procs->glGetProgramInfoLog = glGetProgramInfoLog;
  442. procs->glAttachShader = glAttachShader;
  443. procs->glCreateShader = glCreateShader;
  444. procs->glDeleteShader = glDeleteShader;
  445. procs->glCompileShader = glCompileShader;
  446. procs->glShaderSource = glShaderSource;
  447. procs->glGetShaderiv = glGetShaderiv;
  448. procs->glGetShaderInfoLog = glGetShaderInfoLog;
  449. procs->glEnableVertexAttribArray = glEnableVertexAttribArray;
  450. procs->glVertexAttribPointer = (FF_PFNGLVERTEXATTRIBPOINTERPROC) glVertexAttribPointer;
  451. return 0;
  452. }
  453. #else
  454. static int av_cold opengl_load_procedures(OpenGLContext *opengl)
  455. {
  456. FFOpenGLFunctions *procs = &opengl->glprocs;
  457. #if HAVE_GLXGETPROCADDRESS
  458. #define SelectedGetProcAddress glXGetProcAddress
  459. #elif HAVE_WGLGETPROCADDRESS
  460. #define SelectedGetProcAddress wglGetProcAddress
  461. #endif
  462. #define LOAD_OPENGL_FUN(name, type) \
  463. procs->name = (type)SelectedGetProcAddress(#name); \
  464. if (!procs->name) { \
  465. av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \
  466. return AVERROR(ENOSYS); \
  467. }
  468. LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)
  469. LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)
  470. LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)
  471. LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)
  472. LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)
  473. LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)
  474. LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)
  475. LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)
  476. LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)
  477. LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)
  478. LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)
  479. LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)
  480. LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)
  481. LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)
  482. LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)
  483. LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)
  484. LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)
  485. LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)
  486. LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)
  487. LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)
  488. LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)
  489. LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)
  490. LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)
  491. LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)
  492. LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)
  493. return 0;
  494. #undef SelectedGetProcAddress
  495. #undef LOAD_OPENGL_FUN
  496. }
  497. #endif
  498. static av_always_inline void opengl_make_identity(float matrix[16])
  499. {
  500. memset(matrix, 0, 16 * sizeof(float));
  501. matrix[0] = matrix[5] = matrix[10] = matrix[15] = 1.0f;
  502. }
  503. static av_always_inline void opengl_make_ortho(float matrix[16],
  504. float left, float right,
  505. float bottom, float top,
  506. float nearZ, float farZ)
  507. {
  508. float ral = right + left;
  509. float rsl = right - left;
  510. float tab = top + bottom;
  511. float tsb = top - bottom;
  512. float fan = farZ + nearZ;
  513. float fsn = farZ - nearZ;
  514. memset(matrix, 0, 16 * sizeof(float));
  515. matrix[0] = 2.0f / rsl;
  516. matrix[5] = 2.0f / tsb;
  517. matrix[10] = -2.0f / fsn;
  518. matrix[12] = -ral / rsl;
  519. matrix[13] = -tab / tsb;
  520. matrix[14] = -fan / fsn;
  521. matrix[15] = 1.0f;
  522. }
  523. static av_cold int opengl_read_limits(OpenGLContext *opengl)
  524. {
  525. static const struct{
  526. const char *extention;
  527. int major;
  528. int minor;
  529. } required_extensions[] = {
  530. { "GL_ARB_multitexture", 1, 3 },
  531. { "GL_ARB_vertex_buffer_object", 1, 5 }, //GLX_ARB_vertex_buffer_object
  532. { "GL_ARB_vertex_shader", 2, 0 },
  533. { "GL_ARB_fragment_shader", 2, 0 },
  534. { "GL_ARB_shader_objects", 2, 0 },
  535. { NULL, 0, 0 }
  536. };
  537. int i, major, minor;
  538. const char *extensions, *version;
  539. version = glGetString(GL_VERSION);
  540. extensions = glGetString(GL_EXTENSIONS);
  541. av_log(opengl, AV_LOG_DEBUG, "OpenGL version: %s\n", version);
  542. sscanf(version, "%d.%d", &major, &minor);
  543. for (i = 0; required_extensions[i].extention; i++) {
  544. if (major < required_extensions[i].major &&
  545. (major == required_extensions[i].major && minor < required_extensions[i].minor) &&
  546. !strstr(extensions, required_extensions[i].extention)) {
  547. av_log(opengl, AV_LOG_ERROR, "Required extension %s is not supported.\n",
  548. required_extensions[i].extention);
  549. av_log(opengl, AV_LOG_DEBUG, "Supported extensions are: %s\n", extensions);
  550. return AVERROR(ENOSYS);
  551. }
  552. }
  553. glGetIntegerv(GL_MAX_TEXTURE_SIZE, &opengl->max_texture_size);
  554. glGetIntegerv(GL_MAX_VIEWPORT_DIMS, &opengl->max_viewport_width);
  555. opengl->non_pow_2_textures = major >= 2 || strstr(extensions, "GL_ARB_texture_non_power_of_two");
  556. av_log(opengl, AV_LOG_DEBUG, "Non Power of 2 textures support: %s\n", opengl->non_pow_2_textures ? "Yes" : "No");
  557. av_log(opengl, AV_LOG_DEBUG, "Max texture size: %dx%d\n", opengl->max_texture_size, opengl->max_texture_size);
  558. av_log(opengl, AV_LOG_DEBUG, "Max viewport size: %dx%d\n",
  559. opengl->max_viewport_width, opengl->max_viewport_height);
  560. OPENGL_ERROR_CHECK(opengl);
  561. return 0;
  562. fail:
  563. return AVERROR_EXTERNAL;
  564. }
  565. static const char* opengl_get_fragment_shader_code(enum AVPixelFormat format)
  566. {
  567. int i;
  568. for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {
  569. if (opengl_format_desc[i].fixel_format == format)
  570. return *opengl_format_desc[i].fragment_shader;
  571. }
  572. return NULL;
  573. }
  574. static av_always_inline int opengl_type_size(GLenum type)
  575. {
  576. switch(type) {
  577. case GL_UNSIGNED_SHORT:
  578. case FF_GL_UNSIGNED_SHORT_1_5_5_5_REV:
  579. case GL_UNSIGNED_SHORT_5_6_5:
  580. return 2;
  581. case GL_UNSIGNED_BYTE:
  582. case FF_GL_UNSIGNED_BYTE_3_3_2:
  583. case FF_GL_UNSIGNED_BYTE_2_3_3_REV:
  584. default:
  585. break;
  586. }
  587. return 1;
  588. }
  589. static av_cold void opengl_get_texture_params(OpenGLContext *opengl)
  590. {
  591. int i;
  592. for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {
  593. if (opengl_format_desc[i].fixel_format == opengl->pix_fmt) {
  594. opengl->format = opengl_format_desc[i].format;
  595. opengl->type = opengl_format_desc[i].type;
  596. break;
  597. }
  598. }
  599. }
  600. static void opengl_compute_display_area(AVFormatContext *s)
  601. {
  602. AVRational sar, dar; /* sample and display aspect ratios */
  603. OpenGLContext *opengl = s->priv_data;
  604. AVStream *st = s->streams[0];
  605. AVCodecContext *encctx = st->codec;
  606. /* compute overlay width and height from the codec context information */
  607. sar = st->sample_aspect_ratio.num ? st->sample_aspect_ratio : (AVRational){ 1, 1 };
  608. dar = av_mul_q(sar, (AVRational){ encctx->width, encctx->height });
  609. /* we suppose the screen has a 1/1 sample aspect ratio */
  610. /* fit in the window */
  611. if (av_cmp_q(dar, (AVRational){ opengl->window_width, opengl->window_height }) > 0) {
  612. /* fit in width */
  613. opengl->picture_width = opengl->window_width;
  614. opengl->picture_height = av_rescale(opengl->picture_width, dar.den, dar.num);
  615. } else {
  616. /* fit in height */
  617. opengl->picture_height = opengl->window_height;
  618. opengl->picture_width = av_rescale(opengl->picture_height, dar.num, dar.den);
  619. }
  620. }
  621. static av_cold void opengl_get_texture_size(OpenGLContext *opengl, int in_width, int in_height,
  622. int *out_width, int *out_height)
  623. {
  624. if (opengl->non_pow_2_textures) {
  625. *out_width = in_width;
  626. *out_height = in_height;
  627. } else {
  628. int max = FFMIN(FFMAX(in_width, in_height), opengl->max_texture_size);
  629. unsigned power_of_2 = 1;
  630. while (power_of_2 < max)
  631. power_of_2 *= 2;
  632. *out_height = power_of_2;
  633. *out_width = power_of_2;
  634. av_log(opengl, AV_LOG_DEBUG, "Texture size calculated from %dx%d into %dx%d\n",
  635. in_width, in_height, *out_width, *out_height);
  636. }
  637. }
  638. static av_cold void opengl_fill_color_map(OpenGLContext *opengl)
  639. {
  640. const AVPixFmtDescriptor *desc;
  641. int shift;
  642. enum AVPixelFormat pix_fmt = opengl->pix_fmt;
  643. /* We need order of components, not exact position, some minor HACKs here */
  644. if (pix_fmt == AV_PIX_FMT_RGB565 || pix_fmt == AV_PIX_FMT_BGR555 ||
  645. pix_fmt == AV_PIX_FMT_BGR8 || pix_fmt == AV_PIX_FMT_RGB8)
  646. pix_fmt = AV_PIX_FMT_RGB24;
  647. else if (pix_fmt == AV_PIX_FMT_BGR565 || pix_fmt == AV_PIX_FMT_RGB555)
  648. pix_fmt = AV_PIX_FMT_BGR24;
  649. desc = av_pix_fmt_desc_get(pix_fmt);
  650. if (!(desc->flags & AV_PIX_FMT_FLAG_RGB))
  651. return;
  652. #define FILL_COMPONENT(i) { \
  653. shift = desc->comp[i].depth_minus1 >> 3; \
  654. opengl->color_map[(i << 2) + ((desc->comp[i].offset_plus1 - 1) >> shift)] = 1.0; \
  655. }
  656. memset(opengl->color_map, 0, sizeof(opengl->color_map));
  657. FILL_COMPONENT(0);
  658. FILL_COMPONENT(1);
  659. FILL_COMPONENT(2);
  660. if (desc->flags & AV_PIX_FMT_FLAG_ALPHA)
  661. FILL_COMPONENT(3);
  662. #undef FILL_COMPONENT
  663. }
  664. static av_cold GLuint opengl_load_shader(OpenGLContext *opengl, GLenum type, const char *source)
  665. {
  666. GLuint shader = opengl->glprocs.glCreateShader(type);
  667. GLint result;
  668. if (!shader) {
  669. av_log(opengl, AV_LOG_ERROR, "glCreateShader() failed\n");
  670. return 0;
  671. }
  672. opengl->glprocs.glShaderSource(shader, 1, &source, NULL);
  673. opengl->glprocs.glCompileShader(shader);
  674. opengl->glprocs.glGetShaderiv(shader, FF_GL_COMPILE_STATUS, &result);
  675. if (!result) {
  676. char *log;
  677. opengl->glprocs.glGetShaderiv(shader, FF_GL_INFO_LOG_LENGTH, &result);
  678. if (result) {
  679. if ((log = av_malloc(result))) {
  680. opengl->glprocs.glGetShaderInfoLog(shader, result, NULL, log);
  681. av_log(opengl, AV_LOG_ERROR, "Compile error: %s\n", log);
  682. av_free(log);
  683. }
  684. }
  685. goto fail;
  686. }
  687. OPENGL_ERROR_CHECK(opengl);
  688. return shader;
  689. fail:
  690. opengl->glprocs.glDeleteShader(shader);
  691. return 0;
  692. }
  693. static av_cold int opengl_compile_shaders(OpenGLContext *opengl, enum AVPixelFormat pix_fmt)
  694. {
  695. GLint result;
  696. const char *fragment_shader_code = opengl_get_fragment_shader_code(pix_fmt);
  697. if (!fragment_shader_code) {
  698. av_log(opengl, AV_LOG_ERROR, "Provided pixel format '%s' is not supported\n",
  699. av_get_pix_fmt_name(pix_fmt));
  700. return AVERROR(EINVAL);
  701. }
  702. opengl->vertex_shader = opengl_load_shader(opengl, FF_GL_VERTEX_SHADER,
  703. FF_OPENGL_VERTEX_SHADER);
  704. if (!opengl->vertex_shader) {
  705. av_log(opengl, AV_LOG_ERROR, "Vertex shader loading failed.\n");
  706. goto fail;
  707. }
  708. opengl->fragment_shader = opengl_load_shader(opengl, FF_GL_FRAGMENT_SHADER,
  709. fragment_shader_code);
  710. if (!opengl->fragment_shader) {
  711. av_log(opengl, AV_LOG_ERROR, "Fragment shader loading failed.\n");
  712. goto fail;
  713. }
  714. opengl->program = opengl->glprocs.glCreateProgram();
  715. if (!opengl->program)
  716. goto fail;
  717. opengl->glprocs.glAttachShader(opengl->program, opengl->vertex_shader);
  718. opengl->glprocs.glAttachShader(opengl->program, opengl->fragment_shader);
  719. opengl->glprocs.glLinkProgram(opengl->program);
  720. opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_LINK_STATUS, &result);
  721. if (!result) {
  722. char *log;
  723. opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_INFO_LOG_LENGTH, &result);
  724. if (result) {
  725. log = av_malloc(result);
  726. if (!log)
  727. goto fail;
  728. opengl->glprocs.glGetProgramInfoLog(opengl->program, result, NULL, log);
  729. av_log(opengl, AV_LOG_ERROR, "Link error: %s\n", log);
  730. av_free(log);
  731. }
  732. goto fail;
  733. }
  734. opengl->position_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_position");
  735. opengl->texture_coords_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_textureCoords");
  736. opengl->projection_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_projectionMatrix");
  737. opengl->model_view_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_modelViewMatrix");
  738. opengl->color_map_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_colorMap");
  739. opengl->texture_location[0] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture0");
  740. opengl->texture_location[1] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture1");
  741. opengl->texture_location[2] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture2");
  742. opengl->texture_location[3] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture3");
  743. opengl->chroma_div_w_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_w");
  744. opengl->chroma_div_h_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_h");
  745. OPENGL_ERROR_CHECK(opengl);
  746. return 0;
  747. fail:
  748. opengl->glprocs.glDeleteShader(opengl->vertex_shader);
  749. opengl->glprocs.glDeleteShader(opengl->fragment_shader);
  750. opengl->glprocs.glDeleteProgram(opengl->program);
  751. opengl->fragment_shader = opengl->vertex_shader = opengl->program = 0;
  752. return AVERROR_EXTERNAL;
  753. }
  754. static av_cold int opengl_configure_texture(OpenGLContext *opengl, GLuint texture,
  755. GLsizei width, GLsizei height)
  756. {
  757. if (texture) {
  758. int new_width, new_height;
  759. opengl_get_texture_size(opengl, width, height, &new_width, &new_height);
  760. glBindTexture(GL_TEXTURE_2D, texture);
  761. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  762. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
  763. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
  764. glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
  765. glTexImage2D(GL_TEXTURE_2D, 0, opengl->format, new_width, new_height, 0,
  766. opengl->format, opengl->type, NULL);
  767. OPENGL_ERROR_CHECK(NULL);
  768. }
  769. return 0;
  770. fail:
  771. return AVERROR_EXTERNAL;
  772. }
  773. static av_cold int opengl_prepare_vertex(AVFormatContext *s)
  774. {
  775. OpenGLContext *opengl = s->priv_data;
  776. int tex_w, tex_h;
  777. if (opengl->window_width > opengl->max_viewport_width || opengl->window_height > opengl->max_viewport_height) {
  778. opengl->window_width = FFMAX(opengl->window_width, opengl->max_viewport_width);
  779. opengl->window_height = FFMAX(opengl->window_height, opengl->max_viewport_height);
  780. av_log(opengl, AV_LOG_WARNING, "Too big viewport requested, limited to %dx%d", opengl->window_width, opengl->window_height);
  781. }
  782. glViewport(0, 0, opengl->window_width, opengl->window_height);
  783. opengl_make_ortho(opengl->projection_matrix,
  784. - (float)opengl->window_width / 2.0f, (float)opengl->window_width / 2.0f,
  785. - (float)opengl->window_height / 2.0f, (float)opengl->window_height / 2.0f,
  786. 1.0f, -1.0f);
  787. opengl_make_identity(opengl->model_view_matrix);
  788. opengl_compute_display_area(s);
  789. opengl->vertex[0].z = opengl->vertex[1].z = opengl->vertex[2].z = opengl->vertex[3].z = 0.0f;
  790. opengl->vertex[0].x = opengl->vertex[1].x = - (float)opengl->picture_width / 2.0f;
  791. opengl->vertex[2].x = opengl->vertex[3].x = (float)opengl->picture_width / 2.0f;
  792. opengl->vertex[1].y = opengl->vertex[2].y = - (float)opengl->picture_height / 2.0f;
  793. opengl->vertex[0].y = opengl->vertex[3].y = (float)opengl->picture_height / 2.0f;
  794. opengl_get_texture_size(opengl, opengl->width, opengl->height, &tex_w, &tex_h);
  795. opengl->vertex[0].s0 = 0.0f;
  796. opengl->vertex[0].t0 = 0.0f;
  797. opengl->vertex[1].s0 = 0.0f;
  798. opengl->vertex[1].t0 = (float)opengl->height / (float)tex_h;
  799. opengl->vertex[2].s0 = (float)opengl->width / (float)tex_w;
  800. opengl->vertex[2].t0 = (float)opengl->height / (float)tex_h;
  801. opengl->vertex[3].s0 = (float)opengl->width / (float)tex_w;
  802. opengl->vertex[3].t0 = 0.0f;
  803. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);
  804. opengl->glprocs.glBufferData(FF_GL_ARRAY_BUFFER, sizeof(opengl->vertex), opengl->vertex, FF_GL_STATIC_DRAW);
  805. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, 0);
  806. OPENGL_ERROR_CHECK(opengl);
  807. return 0;
  808. fail:
  809. return AVERROR_EXTERNAL;
  810. }
  811. static int opengl_prepare(OpenGLContext *opengl)
  812. {
  813. int i;
  814. opengl->glprocs.glUseProgram(opengl->program);
  815. opengl->glprocs.glUniformMatrix4fv(opengl->projection_matrix_location, 1, GL_FALSE, opengl->projection_matrix);
  816. opengl->glprocs.glUniformMatrix4fv(opengl->model_view_matrix_location, 1, GL_FALSE, opengl->model_view_matrix);
  817. for (i = 0; i < 4; i++)
  818. if (opengl->texture_location[i] != -1) {
  819. opengl->glprocs.glActiveTexture(GL_TEXTURE0 + i);
  820. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[i]);
  821. opengl->glprocs.glUniform1i(opengl->texture_location[i], i);
  822. }
  823. if (opengl->color_map_location != -1)
  824. opengl->glprocs.glUniformMatrix4fv(opengl->color_map_location, 1, GL_FALSE, opengl->color_map);
  825. if (opengl->chroma_div_h_location != -1)
  826. opengl->glprocs.glUniform1f(opengl->chroma_div_h_location, opengl->chroma_div_h);
  827. if (opengl->chroma_div_w_location != -1)
  828. opengl->glprocs.glUniform1f(opengl->chroma_div_w_location, opengl->chroma_div_w);
  829. OPENGL_ERROR_CHECK(opengl);
  830. return 0;
  831. fail:
  832. return AVERROR_EXTERNAL;
  833. }
  834. static int opengl_create_window(AVFormatContext *h)
  835. {
  836. OpenGLContext *opengl = h->priv_data;
  837. int ret;
  838. if (!opengl->no_window) {
  839. #if HAVE_SDL
  840. if ((ret = opengl_sdl_create_window(h)) < 0) {
  841. av_log(opengl, AV_LOG_ERROR, "Cannot create default SDL window.\n");
  842. return ret;
  843. }
  844. #else
  845. av_log(opengl, AV_LOG_ERROR, "FFmpeg is compiled without SDL. Cannot create default window.\n");
  846. return AVERROR(ENOSYS);
  847. #endif
  848. } else {
  849. if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_CREATE_WINDOW_BUFFER, NULL , 0)) < 0) {
  850. av_log(opengl, AV_LOG_ERROR, "Application failed to create window buffer.\n");
  851. return ret;
  852. }
  853. if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {
  854. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  855. return ret;
  856. }
  857. }
  858. return 0;
  859. }
  860. static int opengl_release_window(AVFormatContext *h)
  861. {
  862. int ret;
  863. OpenGLContext *opengl = h->priv_data;
  864. if (!opengl->no_window) {
  865. #if HAVE_SDL
  866. SDL_Quit();
  867. #endif
  868. } else if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DESTROY_WINDOW_BUFFER, NULL , 0) < 0)) {
  869. av_log(opengl, AV_LOG_ERROR, "Application failed to release window buffer.\n");
  870. return ret;
  871. }
  872. return 0;
  873. }
  874. static av_cold int opengl_write_trailer(AVFormatContext *h)
  875. {
  876. OpenGLContext *opengl = h->priv_data;
  877. if (opengl->no_window &&
  878. avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0) < 0)
  879. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  880. opengl_deinit_context(opengl);
  881. opengl_release_window(h);
  882. return 0;
  883. }
  884. static av_cold int opengl_init_context(OpenGLContext *opengl)
  885. {
  886. int i, ret;
  887. const AVPixFmtDescriptor *desc;
  888. if ((ret = opengl_compile_shaders(opengl, opengl->pix_fmt)) < 0)
  889. goto fail;
  890. desc = av_pix_fmt_desc_get(opengl->pix_fmt);
  891. av_assert0(desc->nb_components > 0 && desc->nb_components <= 4);
  892. glGenTextures(desc->nb_components, opengl->texture_name);
  893. opengl->glprocs.glGenBuffers(2, &opengl->index_buffer);
  894. if (!opengl->index_buffer || !opengl->vertex_buffer) {
  895. av_log(opengl, AV_LOG_ERROR, "Buffer generation failed.\n");
  896. ret = AVERROR_EXTERNAL;
  897. goto fail;
  898. }
  899. opengl_configure_texture(opengl, opengl->texture_name[0], opengl->width, opengl->height);
  900. if (desc->nb_components > 1) {
  901. int has_alpha = desc->flags & AV_PIX_FMT_FLAG_ALPHA;
  902. int num_planes = desc->nb_components - (has_alpha ? 1 : 0);
  903. if (opengl->non_pow_2_textures) {
  904. opengl->chroma_div_w = 1.0f;
  905. opengl->chroma_div_h = 1.0f;
  906. } else {
  907. opengl->chroma_div_w = 1 << desc->log2_chroma_w;
  908. opengl->chroma_div_h = 1 << desc->log2_chroma_h;
  909. }
  910. for (i = 1; i < num_planes; i++)
  911. if (opengl->non_pow_2_textures)
  912. opengl_configure_texture(opengl, opengl->texture_name[i],
  913. FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w),
  914. FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h));
  915. else
  916. opengl_configure_texture(opengl, opengl->texture_name[i], opengl->width, opengl->height);
  917. if (has_alpha)
  918. opengl_configure_texture(opengl, opengl->texture_name[3], opengl->width, opengl->height);
  919. }
  920. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);
  921. opengl->glprocs.glBufferData(FF_GL_ELEMENT_ARRAY_BUFFER, sizeof(g_index), g_index, FF_GL_STATIC_DRAW);
  922. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, 0);
  923. glEnable(GL_BLEND);
  924. glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
  925. glClearColor((float)opengl->background[0] / 255.0f, (float)opengl->background[1] / 255.0f,
  926. (float)opengl->background[2] / 255.0f, 1.0f);
  927. ret = AVERROR_EXTERNAL;
  928. OPENGL_ERROR_CHECK(opengl);
  929. return 0;
  930. fail:
  931. return ret;
  932. }
  933. static av_cold int opengl_write_header(AVFormatContext *h)
  934. {
  935. OpenGLContext *opengl = h->priv_data;
  936. AVStream *st;
  937. int ret;
  938. if (h->nb_streams != 1 ||
  939. h->streams[0]->codec->codec_type != AVMEDIA_TYPE_VIDEO ||
  940. h->streams[0]->codec->codec_id != AV_CODEC_ID_RAWVIDEO) {
  941. av_log(opengl, AV_LOG_ERROR, "Only a single video stream is supported.\n");
  942. return AVERROR(EINVAL);
  943. }
  944. st = h->streams[0];
  945. opengl->width = st->codec->width;
  946. opengl->height = st->codec->height;
  947. opengl->pix_fmt = st->codec->pix_fmt;
  948. if (!opengl->window_title && !opengl->no_window)
  949. opengl->window_title = av_strdup(h->filename);
  950. if ((ret = opengl_create_window(h)))
  951. goto fail;
  952. if ((ret = opengl_read_limits(opengl)) < 0)
  953. goto fail;
  954. if (opengl->width > opengl->max_texture_size || opengl->height > opengl->max_texture_size) {
  955. av_log(opengl, AV_LOG_ERROR, "Too big picture %dx%d, max supported size is %dx%d\n",
  956. opengl->width, opengl->height, opengl->max_texture_size, opengl->max_texture_size);
  957. ret = AVERROR(EINVAL);
  958. goto fail;
  959. }
  960. if (!opengl->no_window) {
  961. #if HAVE_SDL
  962. if ((ret = opengl_sdl_load_procedures(opengl)) < 0)
  963. goto fail;
  964. #endif
  965. } else if ((ret = opengl_load_procedures(opengl)) < 0)
  966. goto fail;
  967. opengl_fill_color_map(opengl);
  968. opengl_get_texture_params(opengl);
  969. if ((ret = opengl_init_context(opengl)) < 0)
  970. goto fail;
  971. if ((ret = opengl_prepare_vertex(h)) < 0)
  972. goto fail;
  973. glClear(GL_COLOR_BUFFER_BIT);
  974. #if HAVE_SDL
  975. if (!opengl->no_window)
  976. SDL_GL_SwapBuffers();
  977. #endif
  978. if (opengl->no_window &&
  979. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {
  980. av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");
  981. goto fail;
  982. }
  983. ret = AVERROR_EXTERNAL;
  984. OPENGL_ERROR_CHECK(opengl);
  985. return 0;
  986. fail:
  987. opengl_write_trailer(h);
  988. return ret;
  989. }
  990. static uint8_t* opengl_get_plane_pointer(OpenGLContext *opengl, AVPacket *pkt, int comp_index,
  991. const AVPixFmtDescriptor *desc)
  992. {
  993. uint8_t *data = pkt->data;
  994. int wordsize = opengl_type_size(opengl->type);
  995. int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);
  996. int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);
  997. int plane = desc->comp[comp_index].plane;
  998. switch(plane) {
  999. case 0:
  1000. break;
  1001. case 1:
  1002. data += opengl->width * opengl->height * wordsize;
  1003. break;
  1004. case 2:
  1005. data += opengl->width * opengl->height * wordsize;
  1006. data += width_chroma * height_chroma * wordsize;
  1007. break;
  1008. case 3:
  1009. data += opengl->width * opengl->height * wordsize;
  1010. data += 2 * width_chroma * height_chroma * wordsize;
  1011. break;
  1012. default:
  1013. return NULL;
  1014. }
  1015. return data;
  1016. }
  1017. static int opengl_draw(AVFormatContext *h, AVPacket *pkt, int repaint)
  1018. {
  1019. OpenGLContext *opengl = h->priv_data;
  1020. enum AVPixelFormat pix_fmt = h->streams[0]->codec->pix_fmt;
  1021. const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);
  1022. int ret;
  1023. #if HAVE_SDL
  1024. if (!opengl->no_window && (ret = opengl_sdl_process_events(h)) < 0)
  1025. goto fail;
  1026. #endif
  1027. if (opengl->no_window &&
  1028. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {
  1029. av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");
  1030. goto fail;
  1031. }
  1032. glClear(GL_COLOR_BUFFER_BIT);
  1033. if (!repaint) {
  1034. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[0]);
  1035. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,
  1036. opengl_get_plane_pointer(opengl, pkt, 0, desc));
  1037. if (desc->flags & AV_PIX_FMT_FLAG_PLANAR) {
  1038. int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);
  1039. int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);
  1040. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[1]);
  1041. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,
  1042. opengl_get_plane_pointer(opengl, pkt, 1, desc));
  1043. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[2]);
  1044. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,
  1045. opengl_get_plane_pointer(opengl, pkt, 2, desc));
  1046. if (desc->flags & AV_PIX_FMT_FLAG_ALPHA) {
  1047. glBindTexture(GL_TEXTURE_2D, opengl->texture_name[3]);
  1048. glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,
  1049. opengl_get_plane_pointer(opengl, pkt, 3, desc));
  1050. }
  1051. }
  1052. }
  1053. ret = AVERROR_EXTERNAL;
  1054. OPENGL_ERROR_CHECK(opengl);
  1055. if ((ret = opengl_prepare(opengl)) < 0)
  1056. goto fail;
  1057. opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);
  1058. opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);
  1059. opengl->glprocs.glVertexAttribPointer(opengl->position_attrib, 3, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 0);
  1060. opengl->glprocs.glEnableVertexAttribArray(opengl->position_attrib);
  1061. opengl->glprocs.glVertexAttribPointer(opengl->texture_coords_attrib, 2, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 12);
  1062. opengl->glprocs.glEnableVertexAttribArray(opengl->texture_coords_attrib);
  1063. glDrawElements(GL_TRIANGLES, FF_ARRAY_ELEMS(g_index), GL_UNSIGNED_SHORT, 0);
  1064. ret = AVERROR_EXTERNAL;
  1065. OPENGL_ERROR_CHECK(opengl);
  1066. #if HAVE_SDL
  1067. if (!opengl->no_window)
  1068. SDL_GL_SwapBuffers();
  1069. #endif
  1070. if (opengl->no_window &&
  1071. (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {
  1072. av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");
  1073. goto fail;
  1074. }
  1075. return 0;
  1076. fail:
  1077. return ret;
  1078. }
  1079. static int opengl_write_packet(AVFormatContext *h, AVPacket *pkt)
  1080. {
  1081. return opengl_draw(h, pkt, 0);
  1082. }
  1083. #define OFFSET(x) offsetof(OpenGLContext, x)
  1084. #define ENC AV_OPT_FLAG_ENCODING_PARAM
  1085. static const AVOption options[] = {
  1086. { "background", "set background color", OFFSET(background), AV_OPT_TYPE_COLOR, {.str = "black"}, CHAR_MIN, CHAR_MAX, ENC },
  1087. { "no_window", "disable default window", OFFSET(no_window), AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, ENC },
  1088. { "window_title", "set window title", OFFSET(window_title), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, ENC },
  1089. { NULL }
  1090. };
  1091. static const AVClass opengl_class = {
  1092. .class_name = "opengl outdev",
  1093. .item_name = av_default_item_name,
  1094. .option = options,
  1095. .version = LIBAVUTIL_VERSION_INT,
  1096. };
  1097. AVOutputFormat ff_opengl_muxer = {
  1098. .name = "opengl",
  1099. .long_name = NULL_IF_CONFIG_SMALL("OpenGL output"),
  1100. .priv_data_size = sizeof(OpenGLContext),
  1101. .audio_codec = AV_CODEC_ID_NONE,
  1102. .video_codec = AV_CODEC_ID_RAWVIDEO,
  1103. .write_header = opengl_write_header,
  1104. .write_packet = opengl_write_packet,
  1105. .write_trailer = opengl_write_trailer,
  1106. .control_message = opengl_control_message,
  1107. .flags = AVFMT_NOFILE | AVFMT_VARIABLE_FPS | AVFMT_NOTIMESTAMPS,
  1108. .priv_class = &opengl_class,
  1109. };