gpu/command_buffer/service/gles2_cmd_copy_texture_chromium.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "gpu/command_buffer/service/gles2_cmd_copy_texture_chromium.h"
   6
   7 #include <algorithm>
   8
   9 #include "base/basictypes.h"
  10 #include "gpu/command_buffer/service/gl_utils.h"
  11 #include "gpu/command_buffer/service/gles2_cmd_decoder.h"
  12
  13 #define SHADER(src)                     \
  14   "#ifdef GL_ES\n"                      \
  15   "precision mediump float;\n"          \
  16   "#define TexCoordPrecision mediump\n" \
  17   "#else\n"                             \
  18   "#define TexCoordPrecision\n"         \
  19   "#endif\n" #src
  20 #define SHADER_2D(src)              \
  21   "#define SamplerType sampler2D\n" \
  22   "#define TextureLookup texture2D\n" SHADER(src)
  23 #define SHADER_RECTANGLE_ARB(src)     \
  24   "#define SamplerType samplerRect\n" \
  25   "#define TextureLookup textureRect\n" SHADER(src)
  26 #define SHADER_EXTERNAL_OES(src)                     \
  27   "#extension GL_OES_EGL_image_external : require\n" \
  28   "#define SamplerType samplerExternalOES\n"         \
  29   "#define TextureLookup texture2D\n" SHADER(src)
  30 #define FRAGMENT_SHADERS(src) \
  31   SHADER_2D(src), SHADER_RECTANGLE_ARB(src), SHADER_EXTERNAL_OES(src)
  32
  33 namespace {
  34
  35 enum VertexShaderId {
  36   VERTEX_SHADER_COPY_TEXTURE,
  37   VERTEX_SHADER_COPY_TEXTURE_FLIP_Y,
  38   NUM_VERTEX_SHADERS,
  39 };
  40
  41 enum FragmentShaderId {
  42   FRAGMENT_SHADER_COPY_TEXTURE_2D,
  43   FRAGMENT_SHADER_COPY_TEXTURE_RECTANGLE_ARB,
  44   FRAGMENT_SHADER_COPY_TEXTURE_EXTERNAL_OES,
  45   FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_2D,
  46   FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_RECTANGLE_ARB,
  47   FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_EXTERNAL_OES,
  48   FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_2D,
  49   FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_RECTANGLE_ARB,
  50   FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_EXTERNAL_OES,
  51   NUM_FRAGMENT_SHADERS,
  52 };
  53
  54 const char* vertex_shader_source[NUM_VERTEX_SHADERS] = {
  55   // VERTEX_SHADER_COPY_TEXTURE
  56   SHADER(
  57     uniform mat4 u_matrix;
  58     uniform vec2 u_half_size;
  59     attribute vec4 a_position;
  60     varying TexCoordPrecision vec2 v_uv;
  61     void main(void) {
  62       gl_Position = u_matrix * a_position;
  63       v_uv = a_position.xy * vec2(u_half_size.s, u_half_size.t) +
  64              vec2(u_half_size.s, u_half_size.t);
  65     }),
  66   // VERTEX_SHADER_COPY_TEXTURE_FLIP_Y
  67   SHADER(
  68     uniform mat4 u_matrix;
  69     uniform vec2 u_half_size;
  70     attribute vec4 a_position;
  71     varying TexCoordPrecision vec2 v_uv;
  72     void main(void) {
  73       gl_Position = u_matrix * a_position;
  74       v_uv = a_position.xy * vec2(u_half_size.s, -u_half_size.t) +
  75              vec2(u_half_size.s, u_half_size.t);
  76     }),
  77 };
  78
  79 const char* fragment_shader_source[NUM_FRAGMENT_SHADERS] = {
  80   // FRAGMENT_SHADER_COPY_TEXTURE_*
  81   FRAGMENT_SHADERS(
  82     uniform SamplerType u_sampler;
  83     varying TexCoordPrecision vec2 v_uv;
  84     void main(void) {
  85       gl_FragColor = TextureLookup(u_sampler, v_uv.st);
  86     }),
  87   // FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_*
  88   FRAGMENT_SHADERS(
  89     uniform SamplerType u_sampler;
  90     varying TexCoordPrecision vec2 v_uv;
  91     void main(void) {
  92       gl_FragColor = TextureLookup(u_sampler, v_uv.st);
  93       gl_FragColor.rgb *= gl_FragColor.a;
  94     }),
  95   // FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_*
  96   FRAGMENT_SHADERS(
  97     uniform SamplerType u_sampler;
  98     varying TexCoordPrecision vec2 v_uv;
  99     void main(void) {
 100       gl_FragColor = TextureLookup(u_sampler, v_uv.st);
 101       if (gl_FragColor.a > 0.0)
 102         gl_FragColor.rgb /= gl_FragColor.a;
 103     }),
 104 };
 105
 106 // Returns the correct vertex shader id to evaluate the copy operation for
 107 // the CHROMIUM_flipy setting.
 108 VertexShaderId GetVertexShaderId(bool flip_y) {
 109   // bit 0: flip y
 110   static VertexShaderId shader_ids[] = {
 111       VERTEX_SHADER_COPY_TEXTURE,
 112       VERTEX_SHADER_COPY_TEXTURE_FLIP_Y,
 113   };
 114
 115   unsigned index = flip_y ? 1 : 0;
 116   return shader_ids[index];
 117 }
 118
 119 // Returns the correct fragment shader id to evaluate the copy operation for
 120 // the premultiply alpha pixel store settings and target.
 121 FragmentShaderId GetFragmentShaderId(bool premultiply_alpha,
 122                                      bool unpremultiply_alpha,
 123                                      GLenum target) {
 124   enum {
 125     SAMPLER_2D,
 126     SAMPLER_RECTANGLE_ARB,
 127     SAMPLER_EXTERNAL_OES,
 128     NUM_SAMPLERS
 129   };
 130
 131   // bit 0: premultiply alpha
 132   // bit 1: unpremultiply alpha
 133   static FragmentShaderId shader_ids[][NUM_SAMPLERS] = {
 134       {
 135        FRAGMENT_SHADER_COPY_TEXTURE_2D,
 136        FRAGMENT_SHADER_COPY_TEXTURE_RECTANGLE_ARB,
 137        FRAGMENT_SHADER_COPY_TEXTURE_EXTERNAL_OES,
 138       },
 139       {
 140        FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_2D,
 141        FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_RECTANGLE_ARB,
 142        FRAGMENT_SHADER_COPY_TEXTURE_PREMULTIPLY_ALPHA_EXTERNAL_OES,
 143       },
 144       {
 145        FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_2D,
 146        FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_RECTANGLE_ARB,
 147        FRAGMENT_SHADER_COPY_TEXTURE_UNPREMULTIPLY_ALPHA_EXTERNAL_OES,
 148       },
 149       {
 150        FRAGMENT_SHADER_COPY_TEXTURE_2D,
 151        FRAGMENT_SHADER_COPY_TEXTURE_RECTANGLE_ARB,
 152        FRAGMENT_SHADER_COPY_TEXTURE_EXTERNAL_OES,
 153       }};
 154
 155   unsigned index = (premultiply_alpha   ? (1 << 0) : 0) |
 156                    (unpremultiply_alpha ? (1 << 1) : 0);
 157
 158   switch (target) {
 159     case GL_TEXTURE_2D:
 160       return shader_ids[index][SAMPLER_2D];
 161     case GL_TEXTURE_RECTANGLE_ARB:
 162       return shader_ids[index][SAMPLER_RECTANGLE_ARB];
 163     case GL_TEXTURE_EXTERNAL_OES:
 164       return shader_ids[index][SAMPLER_EXTERNAL_OES];
 165     default:
 166       break;
 167   }
 168
 169   NOTREACHED();
 170   return shader_ids[index][SAMPLER_2D];
 171 }
 172
 173 void CompileShader(GLuint shader, const char* shader_source) {
 174   glShaderSource(shader, 1, &shader_source, 0);
 175   glCompileShader(shader);
 176 #ifndef NDEBUG
 177   GLint compile_status;
 178   glGetShaderiv(shader, GL_COMPILE_STATUS, &compile_status);
 179   if (GL_TRUE != compile_status)
 180     DLOG(ERROR) << "CopyTextureCHROMIUM: shader compilation failure.";
 181 #endif
 182 }
 183
 184 void DeleteShader(GLuint shader) {
 185   if (shader)
 186     glDeleteShader(shader);
 187 }
 188
 189 }  // namespace
 190
 191 namespace gpu {
 192
 193 CopyTextureCHROMIUMResourceManager::CopyTextureCHROMIUMResourceManager()
 194     : initialized_(false),
 195       vertex_shaders_(NUM_VERTEX_SHADERS, 0u),
 196       fragment_shaders_(NUM_FRAGMENT_SHADERS, 0u),
 197       buffer_id_(0u),
 198       framebuffer_(0u) {}
 199
 200 CopyTextureCHROMIUMResourceManager::~CopyTextureCHROMIUMResourceManager() {}
 201
 202 void CopyTextureCHROMIUMResourceManager::Initialize(
 203     const gles2::GLES2Decoder* decoder) {
 204   COMPILE_ASSERT(
 205       kVertexPositionAttrib == 0u,
 206       Position_attribs_must_be_0);
 207
 208   // Initialize all of the GPU resources required to perform the copy.
 209   glGenBuffersARB(1, &buffer_id_);
 210   glBindBuffer(GL_ARRAY_BUFFER, buffer_id_);
 211   const GLfloat kQuadVertices[] = {-1.0f, -1.0f,
 212                                     1.0f, -1.0f,
 213                                     1.0f,  1.0f,
 214                                    -1.0f,  1.0f};
 215   glBufferData(
 216       GL_ARRAY_BUFFER, sizeof(kQuadVertices), kQuadVertices, GL_STATIC_DRAW);
 217
 218   glGenFramebuffersEXT(1, &framebuffer_);
 219
 220   decoder->RestoreBufferBindings();
 221
 222   initialized_ = true;
 223 }
 224
 225 void CopyTextureCHROMIUMResourceManager::Destroy() {
 226   if (!initialized_)
 227     return;
 228
 229   glDeleteFramebuffersEXT(1, &framebuffer_);
 230
 231   std::for_each(vertex_shaders_.begin(), vertex_shaders_.end(), DeleteShader);
 232   std::for_each(
 233       fragment_shaders_.begin(), fragment_shaders_.end(), DeleteShader);
 234
 235   for (ProgramMap::const_iterator it = programs_.begin(); it != programs_.end();
 236        ++it) {
 237     const ProgramInfo& info = it->second;
 238     glDeleteProgram(info.program);
 239   }
 240
 241   glDeleteBuffersARB(1, &buffer_id_);
 242 }
 243
 244 void CopyTextureCHROMIUMResourceManager::DoCopyTexture(
 245     const gles2::GLES2Decoder* decoder,
 246     GLenum source_target,
 247     GLenum dest_target,
 248     GLuint source_id,
 249     GLuint dest_id,
 250     GLint level,
 251     GLsizei width,
 252     GLsizei height,
 253     bool flip_y,
 254     bool premultiply_alpha,
 255     bool unpremultiply_alpha) {
 256   // Use default transform matrix if no transform passed in.
 257   const static GLfloat default_matrix[16] = {1.0f, 0.0f, 0.0f, 0.0f,
 258                                              0.0f, 1.0f, 0.0f, 0.0f,
 259                                              0.0f, 0.0f, 1.0f, 0.0f,
 260                                              0.0f, 0.0f, 0.0f, 1.0f};
 261   DoCopyTextureWithTransform(decoder, source_target, dest_target, source_id,
 262       dest_id, level, width, height, flip_y, premultiply_alpha,
 263       unpremultiply_alpha, default_matrix);
 264 }
 265
 266 void CopyTextureCHROMIUMResourceManager::DoCopyTextureWithTransform(
 267     const gles2::GLES2Decoder* decoder,
 268     GLenum source_target,
 269     GLenum dest_target,
 270     GLuint source_id,
 271     GLuint dest_id,
 272     GLint level,
 273     GLsizei width,
 274     GLsizei height,
 275     bool flip_y,
 276     bool premultiply_alpha,
 277     bool unpremultiply_alpha,
 278     const GLfloat transform_matrix[16]) {
 279   DCHECK(source_target == GL_TEXTURE_2D ||
 280          source_target == GL_TEXTURE_RECTANGLE_ARB ||
 281          source_target == GL_TEXTURE_EXTERNAL_OES);
 282   if (!initialized_) {
 283     DLOG(ERROR) << "CopyTextureCHROMIUM: Uninitialized manager.";
 284     return;
 285   }
 286
 287   VertexShaderId vertex_shader_id = GetVertexShaderId(flip_y);
 288   DCHECK_LT(static_cast<size_t>(vertex_shader_id), vertex_shaders_.size());
 289   GLuint* vertex_shader = &vertex_shaders_[vertex_shader_id];
 290   if (!*vertex_shader) {
 291     *vertex_shader = glCreateShader(GL_VERTEX_SHADER);
 292     CompileShader(*vertex_shader, vertex_shader_source[vertex_shader_id]);
 293   }
 294
 295   FragmentShaderId fragment_shader_id = GetFragmentShaderId(
 296       premultiply_alpha, unpremultiply_alpha, source_target);
 297   DCHECK_LT(static_cast<size_t>(fragment_shader_id), fragment_shaders_.size());
 298   GLuint* fragment_shader = &fragment_shaders_[fragment_shader_id];
 299   if (!*fragment_shader) {
 300     *fragment_shader = glCreateShader(GL_FRAGMENT_SHADER);
 301     CompileShader(*fragment_shader, fragment_shader_source[fragment_shader_id]);
 302   }
 303
 304   ProgramMapKey key(vertex_shader_id, fragment_shader_id);
 305   ProgramInfo* info = &programs_[key];
 306   // Create program if necessary.
 307   if (!info->program) {
 308     info->program = glCreateProgram();
 309     glAttachShader(info->program, *vertex_shader);
 310     glAttachShader(info->program, *fragment_shader);
 311     glBindAttribLocation(info->program, kVertexPositionAttrib, "a_position");
 312     glLinkProgram(info->program);
 313 #ifndef NDEBUG
 314     GLint linked;
 315     glGetProgramiv(info->program, GL_LINK_STATUS, &linked);
 316     if (!linked)
 317       DLOG(ERROR) << "CopyTextureCHROMIUM: program link failure.";
 318 #endif
 319     info->matrix_handle = glGetUniformLocation(info->program, "u_matrix");
 320     info->half_size_handle = glGetUniformLocation(info->program, "u_half_size");
 321     info->sampler_handle = glGetUniformLocation(info->program, "u_sampler");
 322   }
 323   glUseProgram(info->program);
 324
 325 #ifndef NDEBUG
 326   glValidateProgram(info->program);
 327   GLint validation_status;
 328   glGetProgramiv(info->program, GL_VALIDATE_STATUS, &validation_status);
 329   if (GL_TRUE != validation_status) {
 330     DLOG(ERROR) << "CopyTextureCHROMIUM: Invalid shader.";
 331     return;
 332   }
 333 #endif
 334
 335   glUniformMatrix4fv(info->matrix_handle, 1, GL_FALSE, transform_matrix);
 336   if (source_target == GL_TEXTURE_RECTANGLE_ARB)
 337     glUniform2f(info->half_size_handle, width / 2.0f, height / 2.0f);
 338   else
 339     glUniform2f(info->half_size_handle, 0.5f, 0.5f);
 340   glActiveTexture(GL_TEXTURE0);
 341   glBindTexture(GL_TEXTURE_2D, dest_id);
 342   // NVidia drivers require texture settings to be a certain way
 343   // or they won't report FRAMEBUFFER_COMPLETE.
 344   glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 345   glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 346   glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
 347   glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
 348   glBindFramebufferEXT(GL_FRAMEBUFFER_EXT, framebuffer_);
 349   glFramebufferTexture2DEXT(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, dest_target,
 350                             dest_id, level);
 351
 352 #ifndef NDEBUG
 353   GLenum fb_status = glCheckFramebufferStatusEXT(GL_FRAMEBUFFER);
 354   if (GL_FRAMEBUFFER_COMPLETE != fb_status) {
 355     DLOG(ERROR) << "CopyTextureCHROMIUM: Incomplete framebuffer.";
 356   } else
 357 #endif
 358   {
 359     decoder->ClearAllAttributes();
 360     glEnableVertexAttribArray(kVertexPositionAttrib);
 361
 362     glBindBuffer(GL_ARRAY_BUFFER, buffer_id_);
 363     glVertexAttribPointer(kVertexPositionAttrib, 2, GL_FLOAT, GL_FALSE, 0, 0);
 364
 365     glUniform1i(info->sampler_handle, 0);
 366
 367     glBindTexture(source_target, source_id);
 368     glTexParameterf(source_target, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE);
 369     glTexParameterf(source_target, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE);
 370     glTexParameteri(source_target, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
 371     glTexParameteri(source_target, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
 372
 373     glDisable(GL_DEPTH_TEST);
 374     glDisable(GL_SCISSOR_TEST);
 375     glDisable(GL_STENCIL_TEST);
 376     glDisable(GL_CULL_FACE);
 377     glColorMask(GL_TRUE, GL_TRUE, GL_TRUE, GL_TRUE);
 378     glDepthMask(GL_FALSE);
 379     glDisable(GL_BLEND);
 380
 381     glViewport(0, 0, width, height);
 382     glDrawArrays(GL_TRIANGLE_FAN, 0, 4);
 383   }
 384
 385   decoder->RestoreAllAttributes();
 386   decoder->RestoreTextureState(source_id);
 387   decoder->RestoreTextureState(dest_id);
 388   decoder->RestoreTextureUnitBindings(0);
 389   decoder->RestoreActiveTexture();
 390   decoder->RestoreProgramBindings();
 391   decoder->RestoreBufferBindings();
 392   decoder->RestoreFramebufferBindings();
 393   decoder->RestoreGlobalState();
 394 }
 395
 396 }  // namespace gpu