falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
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.
60291 Commits
32 Branches
552MB
Tree: 69b19f2642
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '69b19f2642'
${ noResults }
FFmpeg/libavdevice/opengl_enc.c

1246 lines
51KB
Raw Blame History 

  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. 

  21. //TODO: support for more formats

  22. //TODO: support for more systems.

  23. //TODO: implement X11, Windows, Mac OS native default window. SDL 1.2 doesn't allow to render to custom thread.

  24. 

  25. #include <stdio.h>

  26. #include <stdlib.h>

  27. #include <string.h>

  28. #include <unistd.h>

  29. #include <stddef.h>

  30. 

  31. #include "config.h"

  32. 

  33. #if HAVE_OPENGL_GL3_H

  34. #include <OpenGL/gl3.h>

  35. #elif HAVE_ES2_GL_H

  36. #include <ES2/gl.h>

  37. #else

  38. #include <GL/gl.h>

  39. #include <GL/glext.h>

  40. #endif

  41. #if HAVE_GLXGETPROCADDRESS

  42. #include <GL/glx.h>

  43. #endif

  44. #if HAVE_WINDOWS_H

  45. #include <windows.h>

  46. #endif

  47. 

  48. #if HAVE_SDL

  49. #include <SDL.h>

  50. #endif

  51. 

  52. #include "libavutil/common.h"

  53. #include "libavutil/pixdesc.h"

  54. #include "libavutil/log.h"

  55. #include "libavutil/opt.h"

  56. #include "libavutil/avassert.h"

  57. #include "libavutil/avstring.h"

  58. #include "libavformat/avformat.h"

  59. #include "libavdevice/avdevice.h"

  60. #include "opengl_enc_shaders.h"

  61. 

  62. #ifndef APIENTRY

  63. #define APIENTRY

  64. #endif

  65. 

  66. /* FF_GL_RED_COMPONENT is used for plannar pixel types.

  67.  * Only red component is sampled in shaders.

  68.  * On some platforms GL_RED is not availabe and GL_LUMINANCE have to be used,

  69.  * but since OpenGL 3.0 GL_LUMINANCE is deprecated.

  70.  * GL_RED produces RGBA = value, 0, 0, 1.

  71.  * GL_LUMINANCE produces RGBA = value, value, value, 1.

  72.  * Note: GL_INTENSITY may also be used which produce RGBA = value, value, value, value. */

  73. #if defined(GL_RED)

  74. #define FF_GL_RED_COMPONENT GL_RED

  75. #elif defined(GL_LUMINANCE)

  76. #define FF_GL_RED_COMPONENT GL_LUMINANCE

  77. #else

  78. #define FF_GL_RED_COMPONENT 0x1903; //GL_RED

  79. #endif

  80. 

  81. /* Constants not defined for iOS */

  82. #define FF_GL_UNSIGNED_BYTE_3_3_2 0x8032

  83. #define FF_GL_UNSIGNED_BYTE_2_3_3_REV 0x8362

  84. #define FF_GL_UNSIGNED_SHORT_1_5_5_5_REV 0x8366

  85. 

  86. /* MinGW exposes only OpenGL 1.1 API */

  87. #define FF_GL_ARRAY_BUFFER                0x8892

  88. #define FF_GL_ELEMENT_ARRAY_BUFFER        0x8893

  89. #define FF_GL_STATIC_DRAW                 0x88E4

  90. #define FF_GL_FRAGMENT_SHADER             0x8B30

  91. #define FF_GL_VERTEX_SHADER               0x8B31

  92. #define FF_GL_COMPILE_STATUS              0x8B81

  93. #define FF_GL_LINK_STATUS                 0x8B82

  94. #define FF_GL_INFO_LOG_LENGTH             0x8B84

  95. typedef void   (APIENTRY *FF_PFNGLACTIVETEXTUREPROC) (GLenum texture);

  96. typedef void   (APIENTRY *FF_PFNGLGENBUFFERSPROC) (GLsizei n, GLuint *buffers);

  97. typedef void   (APIENTRY *FF_PFNGLDELETEBUFFERSPROC) (GLsizei n, const GLuint *buffers);

  98. typedef void   (APIENTRY *FF_PFNGLBUFFERDATAPROC) (GLenum target, ptrdiff_t size, const GLvoid *data, GLenum usage);

  99. typedef void   (APIENTRY *FF_PFNGLBINDBUFFERPROC) (GLenum target, GLuint buffer);

  100. typedef GLint  (APIENTRY *FF_PFNGLGETATTRIBLOCATIONPROC) (GLuint program, const char *name);

  101. typedef void   (APIENTRY *FF_PFNGLENABLEVERTEXATTRIBARRAYPROC) (GLuint index);

  102. typedef void   (APIENTRY *FF_PFNGLVERTEXATTRIBPOINTERPROC) (GLuint index, GLint size, GLenum type, GLboolean normalized, GLsizei stride, uintptr_t pointer);

  103. typedef GLint  (APIENTRY *FF_PFNGLGETUNIFORMLOCATIONPROC) (GLuint program, const char *name);

  104. typedef void   (APIENTRY *FF_PFNGLUNIFORM1FPROC) (GLint location, GLfloat v0);

  105. typedef void   (APIENTRY *FF_PFNGLUNIFORM1IPROC) (GLint location, GLint v0);

  106. typedef void   (APIENTRY *FF_PFNGLUNIFORMMATRIX4FVPROC) (GLint location, GLsizei count, GLboolean transpose, const GLfloat *value);

  107. typedef GLuint (APIENTRY *FF_PFNGLCREATEPROGRAMPROC) (void);

  108. typedef void   (APIENTRY *FF_PFNGLDELETEPROGRAMPROC) (GLuint program);

  109. typedef void   (APIENTRY *FF_PFNGLUSEPROGRAMPROC) (GLuint program);

  110. typedef void   (APIENTRY *FF_PFNGLLINKPROGRAMPROC) (GLuint program);

  111. typedef void   (APIENTRY *FF_PFNGLGETPROGRAMIVPROC) (GLuint program, GLenum pname, GLint *params);

  112. typedef void   (APIENTRY *FF_PFNGLGETPROGRAMINFOLOGPROC) (GLuint program, GLsizei bufSize, GLsizei *length, char *infoLog);

  113. typedef void   (APIENTRY *FF_PFNGLATTACHSHADERPROC) (GLuint program, GLuint shader);

  114. typedef GLuint (APIENTRY *FF_PFNGLCREATESHADERPROC) (GLenum type);

  115. typedef void   (APIENTRY *FF_PFNGLDELETESHADERPROC) (GLuint shader);

  116. typedef void   (APIENTRY *FF_PFNGLCOMPILESHADERPROC) (GLuint shader);

  117. typedef void   (APIENTRY *FF_PFNGLSHADERSOURCEPROC) (GLuint shader, GLsizei count, const char* *string, const GLint *length);

  118. typedef void   (APIENTRY *FF_PFNGLGETSHADERIVPROC) (GLuint shader, GLenum pname, GLint *params);

  119. typedef void   (APIENTRY *FF_PFNGLGETSHADERINFOLOGPROC) (GLuint shader, GLsizei bufSize, GLsizei *length, char *infoLog);

  120. 

  121. typedef struct FFOpenGLFunctions {

  122.     FF_PFNGLACTIVETEXTUREPROC glActiveTexture;                     //Require GL ARB multitexture

  123.     FF_PFNGLGENBUFFERSPROC glGenBuffers;                           //Require GL_ARB_vertex_buffer_object

  124.     FF_PFNGLDELETEBUFFERSPROC glDeleteBuffers;                     //Require GL_ARB_vertex_buffer_object

  125.     FF_PFNGLBUFFERDATAPROC glBufferData;                           //Require GL_ARB_vertex_buffer_object

  126.     FF_PFNGLBINDBUFFERPROC glBindBuffer;                           //Require GL_ARB_vertex_buffer_object

  127.     FF_PFNGLGETATTRIBLOCATIONPROC glGetAttribLocation;             //Require GL_ARB_vertex_shader

  128.     FF_PFNGLENABLEVERTEXATTRIBARRAYPROC glEnableVertexAttribArray; //Require GL_ARB_vertex_shader

  129.     FF_PFNGLVERTEXATTRIBPOINTERPROC glVertexAttribPointer;         //Require GL_ARB_vertex_shader

  130.     FF_PFNGLGETUNIFORMLOCATIONPROC glGetUniformLocation;           //Require GL_ARB_shader_objects

  131.     FF_PFNGLUNIFORM1FPROC glUniform1f;                             //Require GL_ARB_shader_objects

  132.     FF_PFNGLUNIFORM1IPROC glUniform1i;                             //Require GL_ARB_shader_objects

  133.     FF_PFNGLUNIFORMMATRIX4FVPROC glUniformMatrix4fv;               //Require GL_ARB_shader_objects

  134.     FF_PFNGLCREATEPROGRAMPROC glCreateProgram;                     //Require GL_ARB_shader_objects

  135.     FF_PFNGLDELETEPROGRAMPROC glDeleteProgram;                     //Require GL_ARB_shader_objects

  136.     FF_PFNGLUSEPROGRAMPROC glUseProgram;                           //Require GL_ARB_shader_objects

  137.     FF_PFNGLLINKPROGRAMPROC glLinkProgram;                         //Require GL_ARB_shader_objects

  138.     FF_PFNGLGETPROGRAMIVPROC glGetProgramiv;                       //Require GL_ARB_shader_objects

  139.     FF_PFNGLGETPROGRAMINFOLOGPROC glGetProgramInfoLog;             //Require GL_ARB_shader_objects

  140.     FF_PFNGLATTACHSHADERPROC glAttachShader;                       //Require GL_ARB_shader_objects

  141.     FF_PFNGLCREATESHADERPROC glCreateShader;                       //Require GL_ARB_shader_objects

  142.     FF_PFNGLDELETESHADERPROC glDeleteShader;                       //Require GL_ARB_shader_objects

  143.     FF_PFNGLCOMPILESHADERPROC glCompileShader;                     //Require GL_ARB_shader_objects

  144.     FF_PFNGLSHADERSOURCEPROC glShaderSource;                       //Require GL_ARB_shader_objects

  145.     FF_PFNGLGETSHADERIVPROC glGetShaderiv;                         //Require GL_ARB_shader_objects

  146.     FF_PFNGLGETSHADERINFOLOGPROC glGetShaderInfoLog;               //Require GL_ARB_shader_objects

  147. } FFOpenGLFunctions;

  148. 

  149. #define OPENGL_ERROR_CHECK(ctx) \

  150. {\

  151.     GLenum err_code; \

  152.     if ((err_code = glGetError()) != GL_NO_ERROR) { \

  153.         av_log(ctx, AV_LOG_ERROR, "OpenGL error occurred in '%s', line %d: %d\n", __FUNCTION__, __LINE__, err_code); \

  154.         goto fail; \

  155.     } \

  156. }\

  157. 

  158. typedef struct OpenGLVertexInfo

  159. {

  160.     float x, y, z;    ///<Position

  161.     float s0, t0;     ///<Texture coords

  162. } OpenGLVertexInfo;

  163. 

  164. /* defines 2 triangles to display */

  165. static GLushort g_index[6] =

  166. {

  167.     0, 1, 2,

  168.     0, 3, 2,

  169. };

  170. 

  171. typedef struct OpenGLContext {

  172.     AVClass *class;                    ///< class for private options

  173. 

  174. #if HAVE_SDL

  175.     SDL_Surface *surface;

  176. #endif

  177.     FFOpenGLFunctions glprocs;

  178. 

  179.     uint8_t background[4];             ///< Background color

  180.     int no_window;                     ///< 0 for create default window

  181.     char *window_title;                ///< Title of the window

  182. 

  183.     /* OpenGL implementation limits */

  184.     GLint max_texture_size;            ///< Maximum texture size

  185.     GLint max_viewport_width;          ///< Maximum viewport size

  186.     GLint max_viewport_height;         ///< Maximum viewport size

  187.     int non_pow_2_textures;            ///< 1 when non power of 2 textures are supported

  188. 

  189.     /* Current OpenGL configuration */

  190.     GLuint program;                    ///< Shader program

  191.     GLuint vertex_shader;              ///< Vertex shader

  192.     GLuint fragment_shader;            ///< Fragment shader for current pix_pmt

  193.     GLuint texture_name[4];            ///< Textures' IDs

  194.     GLuint index_buffer;               ///< Index buffer

  195.     GLuint vertex_buffer;              ///< Vertex buffer

  196.     OpenGLVertexInfo vertex[4];        ///< VBO

  197.     GLint projection_matrix_location;  ///< Uniforms' locations

  198.     GLint model_view_matrix_location;

  199.     GLint color_map_location;

  200.     GLint chroma_div_w_location;

  201.     GLint chroma_div_h_location;

  202.     GLint texture_location[4];

  203.     GLint position_attrib;             ///< Attibutes' locations

  204.     GLint texture_coords_attrib;

  205. 

  206.     GLfloat projection_matrix[16];     ///< Projection matrix

  207.     GLfloat model_view_matrix[16];     ///< Modev view matrix

  208.     GLfloat color_map[16];             ///< RGBA color map matrix

  209.     GLfloat chroma_div_w;              ///< Chroma subsampling w ratio

  210.     GLfloat chroma_div_h;              ///< Chroma subsampling h ratio

  211. 

  212.     /* Stream information */

  213.     GLenum format;

  214.     GLenum type;

  215.     int width;                         ///< Stream width

  216.     int height;                        ///< Stream height

  217.     enum AVPixelFormat pix_fmt;        ///< Stream pixel format

  218.     int picture_width;                 ///< Rendered width

  219.     int picture_height;                ///< Rendered height

  220.     int window_width;

  221.     int window_height;

  222. } OpenGLContext;

  223. 

  224. static const struct OpenGLFormatDesc {

  225.     enum AVPixelFormat fixel_format;

  226.     const char * const * fragment_shader;

  227.     GLenum format;

  228.     GLenum type;

  229. } opengl_format_desc[] = {

  230.     { AV_PIX_FMT_YUV420P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  231.     { AV_PIX_FMT_YUV444P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  232.     { AV_PIX_FMT_YUV422P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  233.     { AV_PIX_FMT_YUV410P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  234.     { AV_PIX_FMT_YUV411P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  235.     { AV_PIX_FMT_YUV440P,    &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  236.     { AV_PIX_FMT_YUV420P16,  &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  237.     { AV_PIX_FMT_YUV422P16,  &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  238.     { AV_PIX_FMT_YUV444P16,  &FF_OPENGL_FRAGMENT_SHADER_YUV_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  239.     { AV_PIX_FMT_YUVA420P,   &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  240.     { AV_PIX_FMT_YUVA444P,   &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  241.     { AV_PIX_FMT_YUVA422P,   &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  242.     { AV_PIX_FMT_YUVA420P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  243.     { AV_PIX_FMT_YUVA422P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  244.     { AV_PIX_FMT_YUVA444P16, &FF_OPENGL_FRAGMENT_SHADER_YUVA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  245.     { AV_PIX_FMT_RGB24,      &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, GL_UNSIGNED_BYTE },

  246.     { AV_PIX_FMT_BGR24,      &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, GL_UNSIGNED_BYTE },

  247.     { AV_PIX_FMT_0RGB,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, GL_UNSIGNED_BYTE },

  248.     { AV_PIX_FMT_RGB0,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, GL_UNSIGNED_BYTE },

  249.     { AV_PIX_FMT_0BGR,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, GL_UNSIGNED_BYTE },

  250.     { AV_PIX_FMT_BGR0,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, GL_UNSIGNED_BYTE },

  251.     { AV_PIX_FMT_RGB565,     &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, GL_UNSIGNED_SHORT_5_6_5 },

  252.     { AV_PIX_FMT_BGR565,     &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, GL_UNSIGNED_SHORT_5_6_5 },

  253.     { AV_PIX_FMT_RGB555,     &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, FF_GL_UNSIGNED_SHORT_1_5_5_5_REV },

  254.     { AV_PIX_FMT_BGR555,     &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGBA, FF_GL_UNSIGNED_SHORT_1_5_5_5_REV },

  255.     { AV_PIX_FMT_RGB8,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, FF_GL_UNSIGNED_BYTE_3_3_2 },

  256.     { AV_PIX_FMT_BGR8,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, FF_GL_UNSIGNED_BYTE_2_3_3_REV },

  257.     { AV_PIX_FMT_RGB48,      &FF_OPENGL_FRAGMENT_SHADER_RGB_PACKET,  GL_RGB, GL_UNSIGNED_SHORT },

  258.     { AV_PIX_FMT_ARGB,       &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },

  259.     { AV_PIX_FMT_RGBA,       &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },

  260.     { AV_PIX_FMT_ABGR,       &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },

  261.     { AV_PIX_FMT_BGRA,       &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_BYTE },

  262.     { AV_PIX_FMT_RGBA64,     &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_SHORT },

  263.     { AV_PIX_FMT_BGRA64,     &FF_OPENGL_FRAGMENT_SHADER_RGBA_PACKET, GL_RGBA, GL_UNSIGNED_SHORT },

  264.     { AV_PIX_FMT_GBRP,       &FF_OPENGL_FRAGMENT_SHADER_RGB_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  265.     { AV_PIX_FMT_GBRP16,     &FF_OPENGL_FRAGMENT_SHADER_RGB_PLANAR,  FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  266.     { AV_PIX_FMT_GBRAP,      &FF_OPENGL_FRAGMENT_SHADER_RGBA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_BYTE },

  267.     { AV_PIX_FMT_GBRAP16,    &FF_OPENGL_FRAGMENT_SHADER_RGBA_PLANAR, FF_GL_RED_COMPONENT, GL_UNSIGNED_SHORT },

  268.     { AV_PIX_FMT_NONE,       NULL }

  269. };

  270. 

  271. static av_cold int opengl_prepare_vertex(AVFormatContext *s);

  272. static int opengl_draw(AVFormatContext *h, AVPacket *pkt, int repaint);

  273. static av_cold int opengl_init_context(OpenGLContext *opengl);

  274. 

  275. static av_cold void opengl_deinit_context(OpenGLContext *opengl)

  276. {

  277.     glDeleteTextures(4, opengl->texture_name);

  278.     opengl->texture_name[0] = opengl->texture_name[1] =

  279.     opengl->texture_name[2] = opengl->texture_name[3] = 0;

  280.     if (opengl->glprocs.glUseProgram)

  281.         opengl->glprocs.glUseProgram(0);

  282.     if (opengl->glprocs.glDeleteProgram) {

  283.         opengl->glprocs.glDeleteProgram(opengl->program);

  284.         opengl->program = 0;

  285.     }

  286.     if (opengl->glprocs.glDeleteShader) {

  287.         opengl->glprocs.glDeleteShader(opengl->vertex_shader);

  288.         opengl->glprocs.glDeleteShader(opengl->fragment_shader);

  289.         opengl->vertex_shader = opengl->fragment_shader = 0;

  290.     }

  291.     if (opengl->glprocs.glBindBuffer) {

  292.         opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, 0);

  293.         opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, 0);

  294.     }

  295.     if (opengl->glprocs.glDeleteBuffers) {

  296.         opengl->glprocs.glDeleteBuffers(2, &opengl->index_buffer);

  297.         opengl->vertex_buffer = opengl->index_buffer = 0;

  298.     }

  299. }

  300. 

  301. static int opengl_resize(AVFormatContext *h, int width, int height)

  302. {

  303.     int ret = 0;

  304.     OpenGLContext *opengl = h->priv_data;

  305.     opengl->window_width = width;

  306.     opengl->window_height = height;

  307.     /* max_viewport_width == 0 means write_header was not called yet. */

  308.     if (opengl->max_viewport_width) {

  309.         if (opengl->no_window &&

  310.             (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {

  311.             av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");

  312.             goto end;

  313.         }

  314.         if ((ret = opengl_prepare_vertex(h)) < 0)

  315.             goto end;

  316.         ret = opengl_draw(h, NULL, 1);

  317.     }

  318.   end:

  319.     return ret;

  320. }

  321. 

  322. static int opengl_control_message(AVFormatContext *h, int type, void *data, size_t data_size)

  323. {

  324.     OpenGLContext *opengl = h->priv_data;

  325.     switch(type) {

  326.     case AV_APP_TO_DEV_WINDOW_SIZE:

  327.         if (data) {

  328.             AVDeviceRect *message = data;

  329.             return opengl_resize(h, message->width, message->height);

  330.         }

  331.         return AVERROR(EINVAL);

  332.     case AV_APP_TO_DEV_WINDOW_REPAINT:

  333.         return opengl_resize(h, opengl->window_width, opengl->window_height);

  334.     }

  335.     return AVERROR(ENOSYS);

  336. }

  337. 

  338. #if HAVE_SDL

  339. static int opengl_sdl_recreate_window(OpenGLContext *opengl, int width, int height)

  340. {

  341.     opengl->surface = SDL_SetVideoMode(width, height,

  342.                                        32, SDL_OPENGL | SDL_RESIZABLE);

  343.     if (!opengl->surface) {

  344.         av_log(opengl, AV_LOG_ERROR, "Unable to set video mode: %s\n", SDL_GetError());

  345.         return AVERROR_EXTERNAL;

  346.     }

  347.     SDL_GL_SetAttribute(SDL_GL_RED_SIZE, 8);

  348.     SDL_GL_SetAttribute(SDL_GL_GREEN_SIZE, 8);

  349.     SDL_GL_SetAttribute(SDL_GL_BLUE_SIZE, 8);

  350.     SDL_GL_SetAttribute(SDL_GL_ALPHA_SIZE, 8);

  351.     SDL_GL_SetAttribute(SDL_GL_DOUBLEBUFFER, 1);

  352.     return 0;

  353. }

  354. 

  355. static int opengl_sdl_process_events(AVFormatContext *h)

  356. {

  357.     int ret;

  358.     OpenGLContext *opengl = h->priv_data;

  359.     SDL_Event event;

  360.     SDL_PumpEvents();

  361.     while (SDL_PeepEvents(&event, 1, SDL_GETEVENT, SDL_ALLEVENTS) > 0) {

  362.         switch (event.type) {

  363.         case SDL_QUIT:

  364.             return AVERROR(EIO);

  365.         case SDL_KEYDOWN:

  366.             switch (event.key.keysym.sym) {

  367.             case SDLK_ESCAPE:

  368.             case SDLK_q:

  369.                 return AVERROR(EIO);

  370.             }

  371.             return 0;

  372.         case SDL_VIDEORESIZE: {

  373.             char buffer[100];

  374.             int reinit;

  375.             AVDeviceRect message;

  376.             /* clean up old context because SDL_SetVideoMode may lose its state. */

  377.             SDL_VideoDriverName(buffer, sizeof(buffer));

  378.             reinit = !av_strncasecmp(buffer, "quartz", sizeof(buffer));

  379.             if (reinit) {

  380.                 opengl_deinit_context(opengl);

  381.             }

  382.             if ((ret = opengl_sdl_recreate_window(opengl, event.resize.w, event.resize.h)) < 0)

  383.                 return ret;

  384.             if (reinit && (ret = opengl_init_context(opengl)) < 0)

  385.                 return ret;

  386.             message.width = opengl->surface->w;

  387.             message.height = opengl->surface->h;

  388.             return opengl_control_message(h, AV_APP_TO_DEV_WINDOW_SIZE, &message, sizeof(AVDeviceRect));

  389.             }

  390.         }

  391.     }

  392.     return 0;

  393. }

  394. 

  395. static int av_cold opengl_sdl_create_window(AVFormatContext *h)

  396. {

  397.     int ret;

  398.     char buffer[100];

  399.     OpenGLContext *opengl = h->priv_data;

  400.     AVDeviceRect message;

  401.     if (SDL_Init(SDL_INIT_VIDEO)) {

  402.         av_log(opengl, AV_LOG_ERROR, "Unable to initialize SDL: %s\n", SDL_GetError());

  403.         return AVERROR_EXTERNAL;

  404.     }

  405.     if ((ret = opengl_sdl_recreate_window(opengl, opengl->width, opengl->height)) < 0)

  406.         return ret;

  407.     av_log(opengl, AV_LOG_INFO, "SDL driver: '%s'.\n", SDL_VideoDriverName(buffer, sizeof(buffer)));

  408.     message.width = opengl->surface->w;

  409.     message.height = opengl->surface->h;

  410.     SDL_WM_SetCaption(opengl->window_title, NULL);

  411.     opengl_control_message(h, AV_APP_TO_DEV_WINDOW_SIZE, &message, sizeof(AVDeviceRect));

  412.     return 0;

  413. }

  414. 

  415. static int av_cold opengl_sdl_load_procedures(OpenGLContext *opengl)

  416. {

  417.     FFOpenGLFunctions *procs = &opengl->glprocs;

  418. 

  419. #define LOAD_OPENGL_FUN(name, type) \

  420.     procs->name = (type)SDL_GL_GetProcAddress(#name); \

  421.     if (!procs->name) { \

  422.         av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \

  423.         return AVERROR(ENOSYS); \

  424.     }

  425. 

  426.     LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)

  427.     LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)

  428.     LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)

  429.     LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)

  430.     LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)

  431.     LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)

  432.     LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)

  433.     LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)

  434.     LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)

  435.     LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)

  436.     LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)

  437.     LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)

  438.     LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)

  439.     LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)

  440.     LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)

  441.     LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)

  442.     LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)

  443.     LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)

  444.     LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)

  445.     LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)

  446.     LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)

  447.     LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)

  448.     LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)

  449.     LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)

  450.     LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)

  451. 

  452.     return 0;

  453. 

  454. #undef LOAD_OPENGL_FUN

  455. }

  456. #endif /* HAVE_SDL */

  457. 

  458. #if defined(__APPLE__)

  459. static int av_cold opengl_load_procedures(OpenGLContext *opengl)

  460. {

  461.     FFOpenGLFunctions *procs = &opengl->glprocs;

  462. 

  463. #if HAVE_SDL

  464.     if (!opengl->no_window)

  465.         return opengl_sdl_load_procedures(opengl);

  466. #endif

  467. 

  468.     procs->glActiveTexture = glActiveTexture;

  469.     procs->glGenBuffers = glGenBuffers;

  470.     procs->glDeleteBuffers = glDeleteBuffers;

  471.     procs->glBufferData = glBufferData;

  472.     procs->glBindBuffer = glBindBuffer;

  473.     procs->glGetAttribLocation = glGetAttribLocation;

  474.     procs->glGetUniformLocation = glGetUniformLocation;

  475.     procs->glUniform1f = glUniform1f;

  476.     procs->glUniform1i = glUniform1i;

  477.     procs->glUniformMatrix4fv = glUniformMatrix4fv;

  478.     procs->glCreateProgram = glCreateProgram;

  479.     procs->glDeleteProgram = glDeleteProgram;

  480.     procs->glUseProgram = glUseProgram;

  481.     procs->glLinkProgram = glLinkProgram;

  482.     procs->glGetProgramiv = glGetProgramiv;

  483.     procs->glGetProgramInfoLog = glGetProgramInfoLog;

  484.     procs->glAttachShader = glAttachShader;

  485.     procs->glCreateShader = glCreateShader;

  486.     procs->glDeleteShader = glDeleteShader;

  487.     procs->glCompileShader = glCompileShader;

  488.     procs->glShaderSource = glShaderSource;

  489.     procs->glGetShaderiv = glGetShaderiv;

  490.     procs->glGetShaderInfoLog = glGetShaderInfoLog;

  491.     procs->glEnableVertexAttribArray = glEnableVertexAttribArray;

  492.     procs->glVertexAttribPointer = (FF_PFNGLVERTEXATTRIBPOINTERPROC) glVertexAttribPointer;

  493.     return 0;

  494. }

  495. #else

  496. static int av_cold opengl_load_procedures(OpenGLContext *opengl)

  497. {

  498.     FFOpenGLFunctions *procs = &opengl->glprocs;

  499. 

  500. #if HAVE_GLXGETPROCADDRESS

  501. #define SelectedGetProcAddress glXGetProcAddress

  502. #elif HAVE_WGLGETPROCADDRESS

  503. #define SelectedGetProcAddress wglGetProcAddress

  504. #endif

  505. 

  506. #define LOAD_OPENGL_FUN(name, type) \

  507.     procs->name = (type)SelectedGetProcAddress(#name); \

  508.     if (!procs->name) { \

  509.         av_log(opengl, AV_LOG_ERROR, "Cannot load OpenGL function: '%s'\n", #name); \

  510.         return AVERROR(ENOSYS); \

  511.     }

  512. 

  513. #if HAVE_SDL

  514.     if (!opengl->no_window)

  515.         return opengl_sdl_load_procedures(opengl);

  516. #endif

  517. 

  518.     LOAD_OPENGL_FUN(glActiveTexture, FF_PFNGLACTIVETEXTUREPROC)

  519.     LOAD_OPENGL_FUN(glGenBuffers, FF_PFNGLGENBUFFERSPROC)

  520.     LOAD_OPENGL_FUN(glDeleteBuffers, FF_PFNGLDELETEBUFFERSPROC)

  521.     LOAD_OPENGL_FUN(glBufferData, FF_PFNGLBUFFERDATAPROC)

  522.     LOAD_OPENGL_FUN(glBindBuffer, FF_PFNGLBINDBUFFERPROC)

  523.     LOAD_OPENGL_FUN(glGetAttribLocation, FF_PFNGLGETATTRIBLOCATIONPROC)

  524.     LOAD_OPENGL_FUN(glGetUniformLocation, FF_PFNGLGETUNIFORMLOCATIONPROC)

  525.     LOAD_OPENGL_FUN(glUniform1f, FF_PFNGLUNIFORM1FPROC)

  526.     LOAD_OPENGL_FUN(glUniform1i, FF_PFNGLUNIFORM1IPROC)

  527.     LOAD_OPENGL_FUN(glUniformMatrix4fv, FF_PFNGLUNIFORMMATRIX4FVPROC)

  528.     LOAD_OPENGL_FUN(glCreateProgram, FF_PFNGLCREATEPROGRAMPROC)

  529.     LOAD_OPENGL_FUN(glDeleteProgram, FF_PFNGLDELETEPROGRAMPROC)

  530.     LOAD_OPENGL_FUN(glUseProgram, FF_PFNGLUSEPROGRAMPROC)

  531.     LOAD_OPENGL_FUN(glLinkProgram, FF_PFNGLLINKPROGRAMPROC)

  532.     LOAD_OPENGL_FUN(glGetProgramiv, FF_PFNGLGETPROGRAMIVPROC)

  533.     LOAD_OPENGL_FUN(glGetProgramInfoLog, FF_PFNGLGETPROGRAMINFOLOGPROC)

  534.     LOAD_OPENGL_FUN(glAttachShader, FF_PFNGLATTACHSHADERPROC)

  535.     LOAD_OPENGL_FUN(glCreateShader, FF_PFNGLCREATESHADERPROC)

  536.     LOAD_OPENGL_FUN(glDeleteShader, FF_PFNGLDELETESHADERPROC)

  537.     LOAD_OPENGL_FUN(glCompileShader, FF_PFNGLCOMPILESHADERPROC)

  538.     LOAD_OPENGL_FUN(glShaderSource, FF_PFNGLSHADERSOURCEPROC)

  539.     LOAD_OPENGL_FUN(glGetShaderiv, FF_PFNGLGETSHADERIVPROC)

  540.     LOAD_OPENGL_FUN(glGetShaderInfoLog, FF_PFNGLGETSHADERINFOLOGPROC)

  541.     LOAD_OPENGL_FUN(glEnableVertexAttribArray, FF_PFNGLENABLEVERTEXATTRIBARRAYPROC)

  542.     LOAD_OPENGL_FUN(glVertexAttribPointer, FF_PFNGLVERTEXATTRIBPOINTERPROC)

  543. 

  544.     return 0;

  545. 

  546. #undef SelectedGetProcAddress

  547. #undef LOAD_OPENGL_FUN

  548. }

  549. #endif

  550. 

  551. static av_always_inline void opengl_make_identity(float matrix[16])

  552. {

  553.     memset(matrix, 0, 16 * sizeof(float));

  554.     matrix[0] = matrix[5] = matrix[10] = matrix[15] = 1.0f;

  555. }

  556. 

  557. static av_always_inline void opengl_make_ortho(float matrix[16],

  558.                                                float left,   float right,

  559.                                                float bottom, float top,

  560.                                                float nearZ,  float farZ)

  561. {

  562.     float ral = right + left;

  563.     float rsl = right - left;

  564.     float tab = top + bottom;

  565.     float tsb = top - bottom;

  566.     float fan = farZ + nearZ;

  567.     float fsn = farZ - nearZ;

  568. 

  569.     memset(matrix, 0, 16 * sizeof(float));

  570.     matrix[0] = 2.0f / rsl;

  571.     matrix[5] = 2.0f / tsb;

  572.     matrix[10] = -2.0f / fsn;

  573.     matrix[12] = -ral / rsl;

  574.     matrix[13] = -tab / tsb;

  575.     matrix[14] = -fan / fsn;

  576.     matrix[15] = 1.0f;

  577. }

  578. 

  579. static av_cold int opengl_read_limits(OpenGLContext *opengl)

  580. {

  581.     static const struct{

  582.         const char *extention;

  583.         int major;

  584.         int minor;

  585.     } required_extensions[] = {

  586.         { "GL_ARB_multitexture",         1, 3 },

  587.         { "GL_ARB_vertex_buffer_object", 1, 5 }, //GLX_ARB_vertex_buffer_object

  588.         { "GL_ARB_vertex_shader",        2, 0 },

  589.         { "GL_ARB_fragment_shader",      2, 0 },

  590.         { "GL_ARB_shader_objects",       2, 0 },

  591.         { NULL,                          0, 0 }

  592.     };

  593.     int i, major, minor;

  594.     const char *extensions, *version;

  595. 

  596.     version = glGetString(GL_VERSION);

  597.     extensions = glGetString(GL_EXTENSIONS);

  598. 

  599.     av_log(opengl, AV_LOG_DEBUG, "OpenGL version: %s\n", version);

  600.     sscanf(version, "%d.%d", &major, &minor);

  601. 

  602.     for (i = 0; required_extensions[i].extention; i++) {

  603.         if (major < required_extensions[i].major &&

  604.             (major == required_extensions[i].major && minor < required_extensions[i].minor) &&

  605.             !strstr(extensions, required_extensions[i].extention)) {

  606.             av_log(opengl, AV_LOG_ERROR, "Required extension %s is not supported.\n",

  607.                    required_extensions[i].extention);

  608.             av_log(opengl, AV_LOG_DEBUG, "Supported extensions are: %s\n", extensions);

  609.             return AVERROR(ENOSYS);

  610.         }

  611.     }

  612.     glGetIntegerv(GL_MAX_TEXTURE_SIZE, &opengl->max_texture_size);

  613.     glGetIntegerv(GL_MAX_VIEWPORT_DIMS, &opengl->max_viewport_width);

  614.     opengl->non_pow_2_textures = major >= 2 || strstr(extensions, "GL_ARB_texture_non_power_of_two");

  615. 

  616.     av_log(opengl, AV_LOG_DEBUG, "Non Power of 2 textures support: %s\n", opengl->non_pow_2_textures ? "Yes" : "No");

  617.     av_log(opengl, AV_LOG_DEBUG, "Max texture size: %dx%d\n", opengl->max_texture_size, opengl->max_texture_size);

  618.     av_log(opengl, AV_LOG_DEBUG, "Max viewport size: %dx%d\n",

  619.            opengl->max_viewport_width, opengl->max_viewport_height);

  620. 

  621.     OPENGL_ERROR_CHECK(opengl);

  622.     return 0;

  623.   fail:

  624.     return AVERROR_EXTERNAL;

  625. }

  626. 

  627. static const char* opengl_get_fragment_shader_code(enum AVPixelFormat format)

  628. {

  629.     int i;

  630.     for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {

  631.         if (opengl_format_desc[i].fixel_format == format)

  632.             return *opengl_format_desc[i].fragment_shader;

  633.     }

  634.     return NULL;

  635. }

  636. 

  637. static av_always_inline int opengl_type_size(GLenum type)

  638. {

  639.     switch(type) {

  640.     case GL_UNSIGNED_SHORT:

  641.     case FF_GL_UNSIGNED_SHORT_1_5_5_5_REV:

  642.     case GL_UNSIGNED_SHORT_5_6_5:

  643.         return 2;

  644.     case GL_UNSIGNED_BYTE:

  645.     case FF_GL_UNSIGNED_BYTE_3_3_2:

  646.     case FF_GL_UNSIGNED_BYTE_2_3_3_REV:

  647.     default:

  648.         break;

  649.     }

  650.     return 1;

  651. }

  652. 

  653. static av_cold void opengl_get_texture_params(OpenGLContext *opengl)

  654. {

  655.     int i;

  656.     for (i = 0; i < FF_ARRAY_ELEMS(opengl_format_desc); i++) {

  657.         if (opengl_format_desc[i].fixel_format == opengl->pix_fmt) {

  658.             opengl->format = opengl_format_desc[i].format;

  659.             opengl->type = opengl_format_desc[i].type;

  660.             break;

  661.         }

  662.     }

  663. }

  664. 

  665. static void opengl_compute_display_area(AVFormatContext *s)

  666. {

  667.     AVRational sar, dar; /* sample and display aspect ratios */

  668.     OpenGLContext *opengl = s->priv_data;

  669.     AVStream *st = s->streams[0];

  670.     AVCodecContext *encctx = st->codec;

  671. 

  672.     /* compute overlay width and height from the codec context information */

  673.     sar = st->sample_aspect_ratio.num ? st->sample_aspect_ratio : (AVRational){ 1, 1 };

  674.     dar = av_mul_q(sar, (AVRational){ encctx->width, encctx->height });

  675. 

  676.     /* we suppose the screen has a 1/1 sample aspect ratio */

  677.     /* fit in the window */

  678.     if (av_cmp_q(dar, (AVRational){ opengl->window_width, opengl->window_height }) > 0) {

  679.         /* fit in width */

  680.         opengl->picture_width = opengl->window_width;

  681.         opengl->picture_height = av_rescale(opengl->picture_width, dar.den, dar.num);

  682.     } else {

  683.         /* fit in height */

  684.         opengl->picture_height = opengl->window_height;

  685.         opengl->picture_width = av_rescale(opengl->picture_height, dar.num, dar.den);

  686.     }

  687. }

  688. 

  689. static av_cold void opengl_get_texture_size(OpenGLContext *opengl, int in_width, int in_height,

  690.                                             int *out_width, int *out_height)

  691. {

  692.     if (opengl->non_pow_2_textures) {

  693.         *out_width = in_width;

  694.         *out_height = in_height;

  695.     } else {

  696.         int max = FFMIN(FFMAX(in_width, in_height), opengl->max_texture_size);

  697.         unsigned power_of_2 = 1;

  698.         while (power_of_2 < max)

  699.             power_of_2 *= 2;

  700.         *out_height = power_of_2;

  701.         *out_width = power_of_2;

  702.         av_log(opengl, AV_LOG_DEBUG, "Texture size calculated from %dx%d into %dx%d\n",

  703.                in_width, in_height, *out_width, *out_height);

  704.     }

  705. }

  706. 

  707. static av_cold void opengl_fill_color_map(OpenGLContext *opengl)

  708. {

  709.     const AVPixFmtDescriptor *desc;

  710.     int shift;

  711.     enum AVPixelFormat pix_fmt = opengl->pix_fmt;

  712. 

  713.     /* We need order of components, not exact position, some minor HACKs here */

  714.     if (pix_fmt == AV_PIX_FMT_RGB565 || pix_fmt == AV_PIX_FMT_BGR555 ||

  715.         pix_fmt == AV_PIX_FMT_BGR8   || pix_fmt == AV_PIX_FMT_RGB8)

  716.         pix_fmt = AV_PIX_FMT_RGB24;

  717.     else if (pix_fmt == AV_PIX_FMT_BGR565 || pix_fmt == AV_PIX_FMT_RGB555)

  718.         pix_fmt = AV_PIX_FMT_BGR24;

  719. 

  720.     desc = av_pix_fmt_desc_get(pix_fmt);

  721.     if (!(desc->flags & AV_PIX_FMT_FLAG_RGB))

  722.         return;

  723. 

  724. #define FILL_COMPONENT(i) { \

  725.         shift = desc->comp[i].depth_minus1 >> 3; \

  726.         opengl->color_map[(i << 2) + ((desc->comp[i].offset_plus1 - 1) >> shift)] = 1.0; \

  727.     }

  728. 

  729.     memset(opengl->color_map, 0, sizeof(opengl->color_map));

  730.     FILL_COMPONENT(0);

  731.     FILL_COMPONENT(1);

  732.     FILL_COMPONENT(2);

  733.     if (desc->flags & AV_PIX_FMT_FLAG_ALPHA)

  734.         FILL_COMPONENT(3);

  735. 

  736. #undef FILL_COMPONENT

  737. }

  738. 

  739. static av_cold GLuint opengl_load_shader(OpenGLContext *opengl, GLenum type, const char *source)

  740. {

  741.     GLuint shader = opengl->glprocs.glCreateShader(type);

  742.     GLint result;

  743.     if (!shader) {

  744.         av_log(opengl, AV_LOG_ERROR, "glCreateShader() failed\n");

  745.         return 0;

  746.     }

  747.     opengl->glprocs.glShaderSource(shader, 1, &source, NULL);

  748.     opengl->glprocs.glCompileShader(shader);

  749. 

  750.     opengl->glprocs.glGetShaderiv(shader, FF_GL_COMPILE_STATUS, &result);

  751.     if (!result) {

  752.         char *log;

  753.         opengl->glprocs.glGetShaderiv(shader, FF_GL_INFO_LOG_LENGTH, &result);

  754.         if (result) {

  755.             if ((log = av_malloc(result))) {

  756.                 opengl->glprocs.glGetShaderInfoLog(shader, result, NULL, log);

  757.                 av_log(opengl, AV_LOG_ERROR, "Compile error: %s\n", log);

  758.                 av_free(log);

  759.             }

  760.         }

  761.         goto fail;

  762.     }

  763.     OPENGL_ERROR_CHECK(opengl);

  764.     return shader;

  765.   fail:

  766.     opengl->glprocs.glDeleteShader(shader);

  767.     return 0;

  768. }

  769. 

  770. static av_cold int opengl_compile_shaders(OpenGLContext *opengl, enum AVPixelFormat pix_fmt)

  771. {

  772.     GLint result;

  773.     const char *fragment_shader_code = opengl_get_fragment_shader_code(pix_fmt);

  774. 

  775.     if (!fragment_shader_code) {

  776.         av_log(opengl, AV_LOG_ERROR, "Provided pixel format '%s' is not supported\n",

  777.                av_get_pix_fmt_name(pix_fmt));

  778.         return AVERROR(EINVAL);

  779.     }

  780. 

  781.     opengl->vertex_shader = opengl_load_shader(opengl, FF_GL_VERTEX_SHADER,

  782.                                                FF_OPENGL_VERTEX_SHADER);

  783.     if (!opengl->vertex_shader) {

  784.         av_log(opengl, AV_LOG_ERROR, "Vertex shader loading failed.\n");

  785.         goto fail;

  786.     }

  787.     opengl->fragment_shader = opengl_load_shader(opengl, FF_GL_FRAGMENT_SHADER,

  788.                                                  fragment_shader_code);

  789.     if (!opengl->fragment_shader) {

  790.         av_log(opengl, AV_LOG_ERROR, "Fragment shader loading failed.\n");

  791.         goto fail;

  792.     }

  793. 

  794.     opengl->program = opengl->glprocs.glCreateProgram();

  795.     if (!opengl->program)

  796.         goto fail;

  797. 

  798.     opengl->glprocs.glAttachShader(opengl->program, opengl->vertex_shader);

  799.     opengl->glprocs.glAttachShader(opengl->program, opengl->fragment_shader);

  800.     opengl->glprocs.glLinkProgram(opengl->program);

  801. 

  802.     opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_LINK_STATUS, &result);

  803.     if (!result) {

  804.         char *log;

  805.         opengl->glprocs.glGetProgramiv(opengl->program, FF_GL_INFO_LOG_LENGTH, &result);

  806.         if (result) {

  807.             log = av_malloc(result);

  808.             if (!log)

  809.                 goto fail;

  810.             opengl->glprocs.glGetProgramInfoLog(opengl->program, result, NULL, log);

  811.             av_log(opengl, AV_LOG_ERROR, "Link error: %s\n", log);

  812.             av_free(log);

  813.         }

  814.         goto fail;

  815.     }

  816. 

  817.     opengl->position_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_position");

  818.     opengl->texture_coords_attrib = opengl->glprocs.glGetAttribLocation(opengl->program, "a_textureCoords");

  819.     opengl->projection_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_projectionMatrix");

  820.     opengl->model_view_matrix_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_modelViewMatrix");

  821.     opengl->color_map_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_colorMap");

  822.     opengl->texture_location[0] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture0");

  823.     opengl->texture_location[1] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture1");

  824.     opengl->texture_location[2] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture2");

  825.     opengl->texture_location[3] = opengl->glprocs.glGetUniformLocation(opengl->program, "u_texture3");

  826.     opengl->chroma_div_w_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_w");

  827.     opengl->chroma_div_h_location = opengl->glprocs.glGetUniformLocation(opengl->program, "u_chroma_div_h");

  828. 

  829.     OPENGL_ERROR_CHECK(opengl);

  830.     return 0;

  831.   fail:

  832.     opengl->glprocs.glDeleteShader(opengl->vertex_shader);

  833.     opengl->glprocs.glDeleteShader(opengl->fragment_shader);

  834.     opengl->glprocs.glDeleteProgram(opengl->program);

  835.     opengl->fragment_shader = opengl->vertex_shader = opengl->program = 0;

  836.     return AVERROR_EXTERNAL;

  837. }

  838. 

  839. static av_cold int opengl_configure_texture(OpenGLContext *opengl, GLuint texture,

  840.                                             GLsizei width, GLsizei height)

  841. {

  842.     if (texture) {

  843.         int new_width, new_height;

  844.         opengl_get_texture_size(opengl, width, height, &new_width, &new_height);

  845.         glBindTexture(GL_TEXTURE_2D, texture);

  846.         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);

  847.         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

  848.         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);

  849.         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);

  850.         glTexImage2D(GL_TEXTURE_2D, 0, opengl->format, new_width, new_height, 0,

  851.                      opengl->format, opengl->type, NULL);

  852.         OPENGL_ERROR_CHECK(NULL);

  853.     }

  854.     return 0;

  855.   fail:

  856.     return AVERROR_EXTERNAL;

  857. }

  858. 

  859. static av_cold int opengl_prepare_vertex(AVFormatContext *s)

  860. {

  861.     OpenGLContext *opengl = s->priv_data;

  862.     int tex_w, tex_h;

  863. 

  864.     if (opengl->window_width > opengl->max_viewport_width || opengl->window_height > opengl->max_viewport_height) {

  865.         opengl->window_width = FFMAX(opengl->window_width, opengl->max_viewport_width);

  866.         opengl->window_height = FFMAX(opengl->window_height, opengl->max_viewport_height);

  867.         av_log(opengl, AV_LOG_WARNING, "Too big viewport requested, limited to %dx%d", opengl->window_width, opengl->window_height);

  868.     }

  869.     glViewport(0, 0, opengl->window_width, opengl->window_height);

  870.     opengl_make_ortho(opengl->projection_matrix,

  871.                       - (float)opengl->window_width  / 2.0f, (float)opengl->window_width  / 2.0f,

  872.                       - (float)opengl->window_height / 2.0f, (float)opengl->window_height / 2.0f,

  873.                       1.0f, -1.0f);

  874.     opengl_make_identity(opengl->model_view_matrix);

  875. 

  876.     opengl_compute_display_area(s);

  877. 

  878.     opengl->vertex[0].z = opengl->vertex[1].z = opengl->vertex[2].z = opengl->vertex[3].z = 0.0f;

  879.     opengl->vertex[0].x = opengl->vertex[1].x = - (float)opengl->picture_width / 2.0f;

  880.     opengl->vertex[2].x = opengl->vertex[3].x =   (float)opengl->picture_width / 2.0f;

  881.     opengl->vertex[1].y = opengl->vertex[2].y = - (float)opengl->picture_height / 2.0f;

  882.     opengl->vertex[0].y = opengl->vertex[3].y =   (float)opengl->picture_height / 2.0f;

  883. 

  884.     opengl_get_texture_size(opengl, opengl->width, opengl->height, &tex_w, &tex_h);

  885. 

  886.     opengl->vertex[0].s0 = 0.0f;

  887.     opengl->vertex[0].t0 = 0.0f;

  888.     opengl->vertex[1].s0 = 0.0f;

  889.     opengl->vertex[1].t0 = (float)opengl->height / (float)tex_h;

  890.     opengl->vertex[2].s0 = (float)opengl->width  / (float)tex_w;

  891.     opengl->vertex[2].t0 = (float)opengl->height / (float)tex_h;

  892.     opengl->vertex[3].s0 = (float)opengl->width  / (float)tex_w;

  893.     opengl->vertex[3].t0 = 0.0f;

  894. 

  895.     opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);

  896.     opengl->glprocs.glBufferData(FF_GL_ARRAY_BUFFER, sizeof(opengl->vertex), opengl->vertex, FF_GL_STATIC_DRAW);

  897.     opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, 0);

  898.     OPENGL_ERROR_CHECK(opengl);

  899.     return 0;

  900.   fail:

  901.     return AVERROR_EXTERNAL;

  902. }

  903. 

  904. static int opengl_prepare(OpenGLContext *opengl)

  905. {

  906.     int i;

  907.     opengl->glprocs.glUseProgram(opengl->program);

  908.     opengl->glprocs.glUniformMatrix4fv(opengl->projection_matrix_location, 1, GL_FALSE, opengl->projection_matrix);

  909.     opengl->glprocs.glUniformMatrix4fv(opengl->model_view_matrix_location, 1, GL_FALSE, opengl->model_view_matrix);

  910.     for (i = 0; i < 4; i++)

  911.         if (opengl->texture_location[i] != -1) {

  912.             opengl->glprocs.glActiveTexture(GL_TEXTURE0 + i);

  913.             glBindTexture(GL_TEXTURE_2D, opengl->texture_name[i]);

  914.             opengl->glprocs.glUniform1i(opengl->texture_location[i], i);

  915.         }

  916.     if (opengl->color_map_location != -1)

  917.         opengl->glprocs.glUniformMatrix4fv(opengl->color_map_location, 1, GL_FALSE, opengl->color_map);

  918.     if (opengl->chroma_div_h_location != -1)

  919.         opengl->glprocs.glUniform1f(opengl->chroma_div_h_location, opengl->chroma_div_h);

  920.     if (opengl->chroma_div_w_location != -1)

  921.         opengl->glprocs.glUniform1f(opengl->chroma_div_w_location, opengl->chroma_div_w);

  922. 

  923.     OPENGL_ERROR_CHECK(opengl);

  924.     return 0;

  925.   fail:

  926.     return AVERROR_EXTERNAL;

  927. }

  928. 

  929. static int opengl_create_window(AVFormatContext *h)

  930. {

  931.     OpenGLContext *opengl = h->priv_data;

  932.     int ret;

  933. 

  934.     if (!opengl->no_window) {

  935. #if HAVE_SDL

  936.         if ((ret = opengl_sdl_create_window(h)) < 0) {

  937.             av_log(opengl, AV_LOG_ERROR, "Cannot create default SDL window.\n");

  938.             return ret;

  939.         }

  940. #else

  941.         av_log(opengl, AV_LOG_ERROR, "FFmpeg is compiled without SDL. Cannot create default window.\n");

  942.         return AVERROR(ENOSYS);

  943. #endif

  944.     } else {

  945.         if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_CREATE_WINDOW_BUFFER, NULL , 0)) < 0) {

  946.             av_log(opengl, AV_LOG_ERROR, "Application failed to create window buffer.\n");

  947.             return ret;

  948.         }

  949.         if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {

  950.             av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");

  951.             return ret;

  952.         }

  953.     }

  954.     return 0;

  955. }

  956. 

  957. static int opengl_release_window(AVFormatContext *h)

  958. {

  959.     int ret;

  960.     OpenGLContext *opengl = h->priv_data;

  961.     if (!opengl->no_window) {

  962. #if HAVE_SDL

  963.         SDL_Quit();

  964. #endif

  965.     } else if ((ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DESTROY_WINDOW_BUFFER, NULL , 0) < 0)) {

  966.         av_log(opengl, AV_LOG_ERROR, "Application failed to release window buffer.\n");

  967.         return ret;

  968.     }

  969.     return 0;

  970. }

  971. 

  972. static av_cold int opengl_write_trailer(AVFormatContext *h)

  973. {

  974.     OpenGLContext *opengl = h->priv_data;

  975. 

  976.     if (opengl->no_window &&

  977.         avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0) < 0)

  978.         av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");

  979. 

  980.     opengl_deinit_context(opengl);

  981.     opengl_release_window(h);

  982. 

  983.     return 0;

  984. }

  985. 

  986. static av_cold int opengl_init_context(OpenGLContext *opengl)

  987. {

  988.     int i, ret;

  989.     const AVPixFmtDescriptor *desc;

  990. 

  991.     if ((ret = opengl_compile_shaders(opengl, opengl->pix_fmt)) < 0)

  992.         goto fail;

  993. 

  994.     desc = av_pix_fmt_desc_get(opengl->pix_fmt);

  995.     av_assert0(desc->nb_components > 0 && desc->nb_components <= 4);

  996.     glGenTextures(desc->nb_components, opengl->texture_name);

  997. 

  998.     opengl->glprocs.glGenBuffers(2, &opengl->index_buffer);

  999.     if (!opengl->index_buffer || !opengl->vertex_buffer) {

  1000.         av_log(opengl, AV_LOG_ERROR, "Buffer generation failed.\n");

  1001.         ret = AVERROR_EXTERNAL;

  1002.         goto fail;

  1003.     }

  1004. 

  1005.     opengl_configure_texture(opengl, opengl->texture_name[0], opengl->width, opengl->height);

  1006.     if (desc->nb_components > 1) {

  1007.         int has_alpha = desc->flags & AV_PIX_FMT_FLAG_ALPHA;

  1008.         int num_planes = desc->nb_components - (has_alpha ? 1 : 0);

  1009.         if (opengl->non_pow_2_textures) {

  1010.             opengl->chroma_div_w = 1.0f;

  1011.             opengl->chroma_div_h = 1.0f;

  1012.         } else {

  1013.             opengl->chroma_div_w = 1 << desc->log2_chroma_w;

  1014.             opengl->chroma_div_h = 1 << desc->log2_chroma_h;

  1015.         }

  1016.         for (i = 1; i < num_planes; i++)

  1017.             if (opengl->non_pow_2_textures)

  1018.                 opengl_configure_texture(opengl, opengl->texture_name[i],

  1019.                         FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w),

  1020.                         FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h));

  1021.             else

  1022.                 opengl_configure_texture(opengl, opengl->texture_name[i], opengl->width, opengl->height);

  1023.         if (has_alpha)

  1024.             opengl_configure_texture(opengl, opengl->texture_name[3], opengl->width, opengl->height);

  1025.     }

  1026. 

  1027.     opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);

  1028.     opengl->glprocs.glBufferData(FF_GL_ELEMENT_ARRAY_BUFFER, sizeof(g_index), g_index, FF_GL_STATIC_DRAW);

  1029.     opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, 0);

  1030. 

  1031.     glEnable(GL_BLEND);

  1032.     glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);

  1033. 

  1034.     glClearColor((float)opengl->background[0] / 255.0f, (float)opengl->background[1] / 255.0f,

  1035.                  (float)opengl->background[2] / 255.0f, 1.0f);

  1036. 

  1037.     ret = AVERROR_EXTERNAL;

  1038.     OPENGL_ERROR_CHECK(opengl);

  1039. 

  1040.     return 0;

  1041.   fail:

  1042.     return ret;

  1043. }

  1044. 

  1045. static av_cold int opengl_write_header(AVFormatContext *h)

  1046. {

  1047.     OpenGLContext *opengl = h->priv_data;

  1048.     AVStream *st;

  1049.     int ret;

  1050. 

  1051.     if (h->nb_streams != 1 ||

  1052.         h->streams[0]->codec->codec_type != AVMEDIA_TYPE_VIDEO ||

  1053.         h->streams[0]->codec->codec_id != AV_CODEC_ID_RAWVIDEO) {

  1054.         av_log(opengl, AV_LOG_ERROR, "Only a single video stream is supported.\n");

  1055.         return AVERROR(EINVAL);

  1056.     }

  1057.     st = h->streams[0];

  1058.     opengl->width = st->codec->width;

  1059.     opengl->height = st->codec->height;

  1060.     opengl->pix_fmt = st->codec->pix_fmt;

  1061. 

  1062.     if (!opengl->window_title && !opengl->no_window)

  1063.         opengl->window_title = av_strdup(h->filename);

  1064. 

  1065.     if ((ret = opengl_create_window(h)))

  1066.         goto fail;

  1067. 

  1068.     if ((ret = opengl_read_limits(opengl)) < 0)

  1069.         goto fail;

  1070. 

  1071.     if (opengl->width > opengl->max_texture_size || opengl->height > opengl->max_texture_size) {

  1072.         av_log(opengl, AV_LOG_ERROR, "Too big picture %dx%d, max supported size is %dx%d\n",

  1073.                opengl->width, opengl->height, opengl->max_texture_size, opengl->max_texture_size);

  1074.         ret = AVERROR(EINVAL);

  1075.         goto fail;

  1076.     }

  1077. 

  1078.     if ((ret = opengl_load_procedures(opengl)) < 0)

  1079.         goto fail;

  1080. 

  1081.     opengl_fill_color_map(opengl);

  1082.     opengl_get_texture_params(opengl);

  1083. 

  1084.     if ((ret = opengl_init_context(opengl)) < 0)

  1085.         goto fail;

  1086. 

  1087.     if ((ret = opengl_prepare_vertex(h)) < 0)

  1088.         goto fail;

  1089. 

  1090.     glClear(GL_COLOR_BUFFER_BIT);

  1091. 

  1092. #if HAVE_SDL

  1093.     if (!opengl->no_window)

  1094.         SDL_GL_SwapBuffers();

  1095. #endif

  1096.     if (opengl->no_window &&

  1097.         (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {

  1098.         av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");

  1099.         goto fail;

  1100.     }

  1101. 

  1102.     ret = AVERROR_EXTERNAL;

  1103.     OPENGL_ERROR_CHECK(opengl);

  1104.     return 0;

  1105. 

  1106.   fail:

  1107.     opengl_write_trailer(h);

  1108.     return ret;

  1109. }

  1110. 

  1111. static uint8_t* opengl_get_plane_pointer(OpenGLContext *opengl, AVPacket *pkt, int comp_index,

  1112.                                          const AVPixFmtDescriptor *desc)

  1113. {

  1114.     uint8_t *data = pkt->data;

  1115.     int wordsize = opengl_type_size(opengl->type);

  1116.     int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);

  1117.     int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);

  1118.     int plane = desc->comp[comp_index].plane;

  1119. 

  1120.     switch(plane) {

  1121.     case 0:

  1122.         break;

  1123.     case 1:

  1124.         data += opengl->width * opengl->height * wordsize;

  1125.         break;

  1126.     case 2:

  1127.         data += opengl->width * opengl->height * wordsize;

  1128.         data += width_chroma * height_chroma * wordsize;

  1129.         break;

  1130.     case 3:

  1131.         data += opengl->width * opengl->height * wordsize;

  1132.         data += 2 * width_chroma * height_chroma * wordsize;

  1133.         break;

  1134.     default:

  1135.         return NULL;

  1136.     }

  1137.     return data;

  1138. }

  1139. 

  1140. static int opengl_draw(AVFormatContext *h, AVPacket *pkt, int repaint)

  1141. {

  1142.     OpenGLContext *opengl = h->priv_data;

  1143.     enum AVPixelFormat pix_fmt = h->streams[0]->codec->pix_fmt;

  1144.     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(pix_fmt);

  1145.     int ret;

  1146. 

  1147. #if HAVE_SDL

  1148.     if (!opengl->no_window && (ret = opengl_sdl_process_events(h)) < 0)

  1149.         goto fail;

  1150. #endif

  1151.     if (opengl->no_window &&

  1152.         (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_PREPARE_WINDOW_BUFFER, NULL , 0)) < 0) {

  1153.         av_log(opengl, AV_LOG_ERROR, "Application failed to prepare window buffer.\n");

  1154.         goto fail;

  1155.     }

  1156. 

  1157.     glClear(GL_COLOR_BUFFER_BIT);

  1158. 

  1159.     if (!repaint) {

  1160.         glBindTexture(GL_TEXTURE_2D, opengl->texture_name[0]);

  1161.         glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,

  1162.                         opengl_get_plane_pointer(opengl, pkt, 0, desc));

  1163.         if (desc->flags & AV_PIX_FMT_FLAG_PLANAR) {

  1164.             int width_chroma = FF_CEIL_RSHIFT(opengl->width, desc->log2_chroma_w);

  1165.             int height_chroma = FF_CEIL_RSHIFT(opengl->height, desc->log2_chroma_h);

  1166.             glBindTexture(GL_TEXTURE_2D, opengl->texture_name[1]);

  1167.             glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,

  1168.                             opengl_get_plane_pointer(opengl, pkt, 1, desc));

  1169.             glBindTexture(GL_TEXTURE_2D, opengl->texture_name[2]);

  1170.             glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width_chroma, height_chroma, opengl->format, opengl->type,

  1171.                             opengl_get_plane_pointer(opengl, pkt, 2, desc));

  1172.             if (desc->flags & AV_PIX_FMT_FLAG_ALPHA) {

  1173.                 glBindTexture(GL_TEXTURE_2D, opengl->texture_name[3]);

  1174.                 glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, opengl->width, opengl->height, opengl->format, opengl->type,

  1175.                                 opengl_get_plane_pointer(opengl, pkt, 3, desc));

  1176.             }

  1177.         }

  1178.     }

  1179.     ret = AVERROR_EXTERNAL;

  1180.     OPENGL_ERROR_CHECK(opengl);

  1181. 

  1182.     if ((ret = opengl_prepare(opengl)) < 0)

  1183.         goto fail;

  1184. 

  1185.     opengl->glprocs.glBindBuffer(FF_GL_ARRAY_BUFFER, opengl->vertex_buffer);

  1186.     opengl->glprocs.glBindBuffer(FF_GL_ELEMENT_ARRAY_BUFFER, opengl->index_buffer);

  1187.     opengl->glprocs.glVertexAttribPointer(opengl->position_attrib, 3, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 0);

  1188.     opengl->glprocs.glEnableVertexAttribArray(opengl->position_attrib);

  1189.     opengl->glprocs.glVertexAttribPointer(opengl->texture_coords_attrib, 2, GL_FLOAT, GL_FALSE, sizeof(OpenGLVertexInfo), 12);

  1190.     opengl->glprocs.glEnableVertexAttribArray(opengl->texture_coords_attrib);

  1191. 

  1192.     glDrawElements(GL_TRIANGLES, FF_ARRAY_ELEMS(g_index), GL_UNSIGNED_SHORT, 0);

  1193. 

  1194.     ret = AVERROR_EXTERNAL;

  1195.     OPENGL_ERROR_CHECK(opengl);

  1196. 

  1197. #if HAVE_SDL

  1198.     if (!opengl->no_window)

  1199.         SDL_GL_SwapBuffers();

  1200. #endif

  1201.     if (opengl->no_window &&

  1202.         (ret = avdevice_dev_to_app_control_message(h, AV_DEV_TO_APP_DISPLAY_WINDOW_BUFFER, NULL , 0)) < 0) {

  1203.         av_log(opengl, AV_LOG_ERROR, "Application failed to display window buffer.\n");

  1204.         goto fail;

  1205.     }

  1206. 

  1207.     return 0;

  1208.   fail:

  1209.     return ret;

  1210. }

  1211. 

  1212. static int opengl_write_packet(AVFormatContext *h, AVPacket *pkt)

  1213. {

  1214.     return opengl_draw(h, pkt, 0);

  1215. }

  1216. 

  1217. #define OFFSET(x) offsetof(OpenGLContext, x)

  1218. #define ENC AV_OPT_FLAG_ENCODING_PARAM

  1219. static const AVOption options[] = {

  1220.     { "background",   "set background color",   OFFSET(background),   AV_OPT_TYPE_COLOR,  {.str = "black"}, CHAR_MIN, CHAR_MAX, ENC },

  1221.     { "no_window",    "disable default window", OFFSET(no_window),    AV_OPT_TYPE_INT,    {.i64 = 0}, INT_MIN, INT_MAX, ENC },

  1222.     { "window_title", "set window title",       OFFSET(window_title), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, ENC },

  1223.     { NULL }

  1224. };

  1225. 

  1226. static const AVClass opengl_class = {

  1227.     .class_name = "opengl outdev",

  1228.     .item_name  = av_default_item_name,

  1229.     .option     = options,

  1230.     .version    = LIBAVUTIL_VERSION_INT,

  1231. };

  1232. 

  1233. AVOutputFormat ff_opengl_muxer = {

  1234.     .name           = "opengl",

  1235.     .long_name      = NULL_IF_CONFIG_SMALL("OpenGL output"),

  1236.     .priv_data_size = sizeof(OpenGLContext),

  1237.     .audio_codec    = AV_CODEC_ID_NONE,

  1238.     .video_codec    = AV_CODEC_ID_RAWVIDEO,

  1239.     .write_header   = opengl_write_header,

  1240.     .write_packet   = opengl_write_packet,

  1241.     .write_trailer  = opengl_write_trailer,

  1242.     .control_message = opengl_control_message,

  1243.     .flags          = AVFMT_NOFILE | AVFMT_VARIABLE_FPS | AVFMT_NOTIMESTAMPS,

  1244.     .priv_class     = &opengl_class,

  1245. };