git.sesse.net Git - ffmpeg/blob - libavdevice/opengl_enc.c

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