git.sesse.net Git - movit/blob - flat_input.cpp

   1 #include <string.h>
   2 #include <assert.h>
   3 #include <epoxy/gl.h>
   4
   5 #include "effect_util.h"
   6 #include "flat_input.h"
   7 #include "resource_pool.h"
   8 #include "util.h"
   9
  10 using namespace std;
  11
  12 namespace movit {
  13
  14 FlatInput::FlatInput(ImageFormat image_format, MovitPixelFormat pixel_format, GLenum type, unsigned width, unsigned height)
  15         : image_format(image_format),
  16           pixel_format(pixel_format),
  17           type(type),
  18           pbo(0),
  19           texture_num(0),
  20           output_linear_gamma(false),
  21           needs_mipmaps(false),
  22           width(width),
  23           height(height),
  24           pitch(width),
  25           pixel_data(NULL)
  26 {
  27         assert(type == GL_FLOAT || type == GL_HALF_FLOAT || type == GL_UNSIGNED_SHORT || type == GL_UNSIGNED_BYTE);
  28         register_int("output_linear_gamma", &output_linear_gamma);
  29         register_int("needs_mipmaps", &needs_mipmaps);
  30 }
  31
  32 FlatInput::~FlatInput()
  33 {
  34         if (texture_num != 0) {
  35                 resource_pool->release_2d_texture(texture_num);
  36         }
  37 }
  38
  39 void FlatInput::set_gl_state(GLuint glsl_program_num, const string& prefix, unsigned *sampler_num)
  40 {
  41         glActiveTexture(GL_TEXTURE0 + *sampler_num);
  42         check_error();
  43
  44         if (texture_num == 0) {
  45                 // Translate the input format to OpenGL's enums.
  46                 GLint internal_format;
  47                 GLenum format;
  48                 if (type == GL_FLOAT) {
  49                         if (pixel_format == FORMAT_R) {
  50                                 internal_format = GL_R32F;
  51                         } else if (pixel_format == FORMAT_RG) {
  52                                 internal_format = GL_RG32F;
  53                         } else if (pixel_format == FORMAT_RGB) {
  54                                 internal_format = GL_RGB32F;
  55                         } else {
  56                                 internal_format = GL_RGBA32F;
  57                         }
  58                 } else if (type == GL_HALF_FLOAT) {
  59                         if (pixel_format == FORMAT_R) {
  60                                 internal_format = GL_R16F;
  61                         } else if (pixel_format == FORMAT_RG) {
  62                                 internal_format = GL_RG16F;
  63                         } else if (pixel_format == FORMAT_RGB) {
  64                                 internal_format = GL_RGB16F;
  65                         } else {
  66                                 internal_format = GL_RGBA16F;
  67                         }
  68                 } else if (type == GL_UNSIGNED_SHORT) {
  69                         if (pixel_format == FORMAT_R) {
  70                                 internal_format = GL_R16;
  71                         } else if (pixel_format == FORMAT_RG) {
  72                                 internal_format = GL_RG16;
  73                         } else if (pixel_format == FORMAT_RGB) {
  74                                 internal_format = GL_RGB16;
  75                         } else {
  76                                 internal_format = GL_RGBA16;
  77                         }
  78                 } else if (output_linear_gamma) {
  79                         assert(type == GL_UNSIGNED_BYTE);
  80                         internal_format = GL_SRGB8_ALPHA8;
  81                 } else {
  82                         assert(type == GL_UNSIGNED_BYTE);
  83                         if (pixel_format == FORMAT_R) {
  84                                 internal_format = GL_R8;
  85                         } else if (pixel_format == FORMAT_RG) {
  86                                 internal_format = GL_RG8;
  87                         } else if (pixel_format == FORMAT_RGB) {
  88                                 internal_format = GL_RGB8;
  89                         } else {
  90                                 internal_format = GL_RGBA8;
  91                         }
  92                 }
  93                 if (pixel_format == FORMAT_RGB) {
  94                         format = GL_RGB;
  95                 } else if (pixel_format == FORMAT_RGBA_PREMULTIPLIED_ALPHA ||
  96                            pixel_format == FORMAT_RGBA_POSTMULTIPLIED_ALPHA) {
  97                         format = GL_RGBA;
  98                 } else if (pixel_format == FORMAT_BGR) {
  99                         format = GL_BGR;
 100                 } else if (pixel_format == FORMAT_BGRA_PREMULTIPLIED_ALPHA ||
 101                            pixel_format == FORMAT_BGRA_POSTMULTIPLIED_ALPHA) {
 102                         format = GL_BGRA;
 103                 } else if (pixel_format == FORMAT_GRAYSCALE) {
 104                         format = GL_LUMINANCE;
 105                 } else if (pixel_format == FORMAT_RG) {
 106                         format = GL_RG;
 107                 } else {
 108                         assert(false);
 109                 }
 110
 111                 // (Re-)upload the texture.
 112                 texture_num = resource_pool->create_2d_texture(internal_format, width, height);
 113                 glBindTexture(GL_TEXTURE_2D, texture_num);
 114                 check_error();
 115                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, pbo);
 116                 check_error();
 117                 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, needs_mipmaps ? GL_LINEAR_MIPMAP_NEAREST : GL_LINEAR);
 118                 check_error();
 119                 glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
 120                 check_error();
 121                 glPixelStorei(GL_UNPACK_ROW_LENGTH, pitch);
 122                 check_error();
 123                 glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, width, height, format, type, pixel_data);
 124                 check_error();
 125                 glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
 126                 check_error();
 127                 if (needs_mipmaps) {
 128                         glGenerateMipmap(GL_TEXTURE_2D);
 129                         check_error();
 130                 }
 131                 glPixelStorei(GL_UNPACK_ROW_LENGTH, 0);
 132                 check_error();
 133                 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 134                 check_error();
 135                 glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 136                 check_error();
 137                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, 0);
 138                 check_error();
 139         } else {
 140                 glBindTexture(GL_TEXTURE_2D, texture_num);
 141                 check_error();
 142         }
 143
 144         // Bind it to a sampler.
 145         set_uniform_int(glsl_program_num, prefix, "tex", *sampler_num);
 146         ++*sampler_num;
 147 }
 148
 149 string FlatInput::output_fragment_shader()
 150 {
 151         return read_file("flat_input.frag");
 152 }
 153
 154 void FlatInput::invalidate_pixel_data()
 155 {
 156         if (texture_num != 0) {
 157                 resource_pool->release_2d_texture(texture_num);
 158                 texture_num = 0;
 159         }
 160 }
 161
 162 }  // namespace movit