The great shader refactor: Merge YUV shader class

cc/output/program_binding.h

 // Copyright 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CC_OUTPUT_PROGRAM_BINDING_H_
 #define CC_OUTPUT_PROGRAM_BINDING_H_
 
 #include <string>
 
 #include "base/logging.h"
@@ skipping 43 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(ProgramBindingBase);
 };
 
 enum ProgramType {
   PROGRAM_TYPE_DEBUG_BORDER,
   PROGRAM_TYPE_SOLID_COLOR,
   PROGRAM_TYPE_TILE,
   PROGRAM_TYPE_TEXTURE,
   PROGRAM_TYPE_RENDER_PASS,
   PROGRAM_TYPE_VIDEO_STREAM,
+  PROGRAM_TYPE_YUV_VIDEO,
 };
 
 class CC_EXPORT ProgramKey {
  public:
   ProgramKey(const ProgramKey& other);
   ~ProgramKey();
 
   static ProgramKey DebugBorder();
   static ProgramKey SolidColor(AAMode aa_mode);
   static ProgramKey Tile(TexCoordPrecision precision,
                          SamplerType sampler,
                          AAMode aa_mode,
                          SwizzleMode swizzle_mode,
                          bool is_opaque);
   static ProgramKey Texture(TexCoordPrecision precision,
                             SamplerType sampler,
                             PremultipliedAlphaMode premultiplied_alpha,
                             bool has_background_color);
 
   // TODO(ccameron): Merge |mask_for_background| into MaskMode.
   static ProgramKey RenderPass(TexCoordPrecision precision,
                                SamplerType sampler,
                                BlendMode blend_mode,
                                AAMode aa_mode,
                                MaskMode mask_mode,
                                bool mask_for_background,
                                bool has_color_matrix);
   static ProgramKey VideoStream(TexCoordPrecision precision);
+  static ProgramKey YUVVideo(TexCoordPrecision precision,
+                             SamplerType sampler,
+                             bool use_alpha_texture,
+                             bool use_nv12,
+                             bool use_color_lut);
 
   bool operator==(const ProgramKey& other) const;
 
  private:
   ProgramKey();
   friend struct ProgramKeyHash;
-  template <class VertexShader, class FragmentShader>
-  friend class ProgramBinding;
+  friend class Program;
 
   ProgramType type_ = PROGRAM_TYPE_DEBUG_BORDER;
   TexCoordPrecision precision_ = TEX_COORD_PRECISION_NA;
   SamplerType sampler_ = SAMPLER_TYPE_NA;
   BlendMode blend_mode_ = BLEND_MODE_NONE;
   AAMode aa_mode_ = NO_AA;
   SwizzleMode swizzle_mode_ = NO_SWIZZLE;
   bool is_opaque_ = false;
 
   PremultipliedAlphaMode premultiplied_alpha_ = PREMULTIPLIED_ALPHA;
   bool has_background_color_ = false;
 
   MaskMode mask_mode_ = NO_MASK;
   bool mask_for_background_ = false;
   bool has_color_matrix_ = false;
+
+  bool use_alpha_texture_ = false;
+  bool use_nv12_ = false;
+  bool use_color_lut_ = false;
 };
 
 struct ProgramKeyHash {
   size_t operator()(const ProgramKey& key) const {
     return (static_cast<size_t>(key.type_) << 0) ^
            (static_cast<size_t>(key.precision_) << 3) ^
            (static_cast<size_t>(key.sampler_) << 6) ^
            (static_cast<size_t>(key.blend_mode_) << 9) ^
            (static_cast<size_t>(key.aa_mode_) << 15) ^
            (static_cast<size_t>(key.swizzle_mode_) << 16) ^
            (static_cast<size_t>(key.is_opaque_) << 17) ^
            (static_cast<size_t>(key.premultiplied_alpha_) << 19) ^
            (static_cast<size_t>(key.has_background_color_) << 20) ^
            (static_cast<size_t>(key.mask_mode_) << 21) ^
            (static_cast<size_t>(key.mask_for_background_) << 22) ^
-           (static_cast<size_t>(key.has_color_matrix_) << 23);
+           (static_cast<size_t>(key.has_color_matrix_) << 23) ^
+           (static_cast<size_t>(key.use_alpha_texture_) << 24) ^
+           (static_cast<size_t>(key.use_nv12_) << 25) ^
+           (static_cast<size_t>(key.use_color_lut_) << 26);
   }
 };
 
-template <class VertexShader, class FragmentShader>
-class ProgramBinding : public ProgramBindingBase {
+class Program : public ProgramBindingBase {
  public:
-  ProgramBinding() {}
+  Program() {}
 
   void Initialize(ContextProvider* context_provider, const ProgramKey& key) {
     // Set parameters that are common to all sub-classes.
     vertex_shader_.aa_mode_ = key.aa_mode_;
     fragment_shader_.aa_mode_ = key.aa_mode_;
     fragment_shader_.blend_mode_ = key.blend_mode_;
     fragment_shader_.tex_coord_precision_ = key.precision_;
     fragment_shader_.sampler_type_ = key.sampler_;
     fragment_shader_.swizzle_mode_ = key.swizzle_mode_;
     fragment_shader_.premultiply_alpha_mode_ = key.premultiplied_alpha_;
@@ skipping 12 matching lines @@
         break;
       case PROGRAM_TYPE_TEXTURE:
         InitializeTextureProgram(key);
         break;
       case PROGRAM_TYPE_RENDER_PASS:
         InitializeRenderPassProgram(key);
         break;
       case PROGRAM_TYPE_VIDEO_STREAM:
         InitializeVideoStreamProgram(key);
         break;
+      case PROGRAM_TYPE_YUV_VIDEO:
+        InitializeYUVVideo(key);
+        break;
     }
     InitializeInternal(context_provider);
   }
 
-  void InitializeVideoYUV(ContextProvider* context_provider,
-                          TexCoordPrecision precision,
-                          SamplerType sampler,
-                          bool use_alpha_plane,
-                          bool use_nv12,
-                          bool use_color_lut) {
-    vertex_shader_.tex_coord_source_ = TEX_COORD_SOURCE_ATTRIBUTE;
-    vertex_shader_.has_matrix_ = true;
-    vertex_shader_.is_ya_uv_ = true;
-
-    fragment_shader_.use_alpha_texture_ = use_alpha_plane;
-    fragment_shader_.use_nv12_ = use_nv12;
-    fragment_shader_.use_color_lut_ = use_color_lut;
-    fragment_shader_.tex_coord_precision_ = precision;
-    fragment_shader_.sampler_type_ = sampler;
-
-    InitializeInternal(context_provider);
-  }
-
   const VertexShader& vertex_shader() const { return vertex_shader_; }
   const FragmentShader& fragment_shader() const { return fragment_shader_; }
 
   // Functions for querying uniform locations.
   int vertex_tex_transform_location() const {
     return vertex_shader_.vertex_tex_transform_location_;
   }
   int tex_matrix_location() const {
     return vertex_shader_.tex_matrix_location_;
   }
@@ skipping 41 matching lines @@
   }
   int mask_tex_coord_offset_location() const {
     return fragment_shader_.mask_tex_coord_offset_location_;
   }
   int color_matrix_location() const {
     return fragment_shader_.color_matrix_location_;
   }
   int color_offset_location() const {
     return fragment_shader_.color_offset_location_;
   }
+  int y_texture_location() const {
+    return fragment_shader_.y_texture_location_;
+  }
+  int u_texture_location() const {
+    return fragment_shader_.u_texture_location_;
+  }
+  int v_texture_location() const {
+    return fragment_shader_.v_texture_location_;
+  }
+  int uv_texture_location() const {
+    return fragment_shader_.uv_texture_location_;
+  }
+  int a_texture_location() const {
+    return fragment_shader_.a_texture_location_;
+  }
+  int lut_texture_location() const {
+    return fragment_shader_.lut_texture_location_;
+  }
+  int yuv_matrix_location() const {
+    return fragment_shader_.yuv_matrix_location_;
+  }
+  int yuv_adj_location() const { return fragment_shader_.yuv_adj_location_; }
+  int ya_clamp_rect_location() const {
+    return fragment_shader_.ya_clamp_rect_location_;
+  }
+  int uv_clamp_rect_location() const {
+    return fragment_shader_.uv_clamp_rect_location_;
+  }
+  int resource_multiplier_location() const {
+    return fragment_shader_.resource_multiplier_location_;
+  }
+  int resource_offset_location() const {
+    return fragment_shader_.resource_offset_location_;
+  }
 
  private:
   void InitializeDebugBorderProgram() {
     // Initialize vertex program.
     vertex_shader_.has_matrix_ = true;
 
     // Initialize fragment program.
     fragment_shader_.input_color_type_ = INPUT_COLOR_SOURCE_UNIFORM;
     fragment_shader_.frag_color_mode_ = FRAG_COLOR_MODE_DEFAULT;
   }
@@ skipping 75 matching lines @@
       fragment_shader_.ignore_sampler_type_ = true;
     } else {
       DCHECK(!key.mask_for_background_);
     }
   }
 
   void InitializeVideoStreamProgram(const ProgramKey& key) {
     vertex_shader_.tex_coord_source_ = TEX_COORD_SOURCE_ATTRIBUTE;
     vertex_shader_.tex_coord_transform_ = TEX_COORD_TRANSFORM_MATRIX;
     vertex_shader_.has_matrix_ = true;
+    DCHECK_EQ(key.sampler_, SAMPLER_TYPE_EXTERNAL_OES);
+  }
 
-    fragment_shader_.tex_coord_precision_ = key.precision_;
-    fragment_shader_.sampler_type_ = SAMPLER_TYPE_EXTERNAL_OES;
-    fragment_shader_.frag_color_mode_ = FRAG_COLOR_MODE_DEFAULT;
+  void InitializeYUVVideo(const ProgramKey& key) {
+    vertex_shader_.tex_coord_source_ = TEX_COORD_SOURCE_ATTRIBUTE;
+    vertex_shader_.has_matrix_ = true;
+    vertex_shader_.is_ya_uv_ = true;
+
+    fragment_shader_.input_color_type_ = INPUT_COLOR_SOURCE_YUV_TEXTURES;
+    fragment_shader_.use_alpha_texture_ = key.use_alpha_texture_;
+    fragment_shader_.use_nv12_ = key.use_nv12_;
+    fragment_shader_.use_color_lut_ = key.use_color_lut_;
   }
 
   void InitializeInternal(ContextProvider* context_provider) {
     DCHECK(context_provider);
     DCHECK(!initialized_);
 
     if (IsContextLost(context_provider->ContextGL()))
       return;
 
     if (!ProgramBindingBase::Init(context_provider->ContextGL(),
@@ skipping 14 matching lines @@
       DCHECK(IsContextLost(context_provider->ContextGL()));
       return;
     }
 
     initialized_ = true;
   }
 
   VertexShader vertex_shader_;
   FragmentShader fragment_shader_;
 
-  DISALLOW_COPY_AND_ASSIGN(ProgramBinding);
+  DISALLOW_COPY_AND_ASSIGN(Program);
 };
 
-typedef ProgramBinding<VertexShaderBase, FragmentShaderBase> Program;
-
 }  // namespace cc
 
 #endif  // CC_OUTPUT_PROGRAM_BINDING_H_