The great shader refactor: Merge YUV shader class

cc/output/shader.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_SHADER_H_
 #define CC_OUTPUT_SHADER_H_
 
 #include <string>
 
 #include "base/logging.h"
@@ skipping 10 matching lines @@
 class GLES2Interface;
 }
 }
 
 namespace cc {
 
 enum TexCoordPrecision {
   TEX_COORD_PRECISION_NA = 0,
   TEX_COORD_PRECISION_MEDIUM = 1,
   TEX_COORD_PRECISION_HIGH = 2,
-  LAST_TEX_COORD_PRECISION = 2
 };
 
 // Texture coordinate sources for the vertex shader.
 enum TexCoordSource {
   // Vertex shader does not populate a texture coordinate.
   TEX_COORD_SOURCE_NONE,
   // Texture coordinate is set to the untransformed position.
   TEX_COORD_SOURCE_POSITION,
   // Texture coordinate has its own attribute.
   TEX_COORD_SOURCE_ATTRIBUTE,
@@ skipping 34 matching lines @@
 enum PremultipliedAlphaMode {
   PREMULTIPLIED_ALPHA = 0,
   NON_PREMULTIPLIED_ALPHA = 1,
 };
 
 enum SamplerType {
   SAMPLER_TYPE_NA = 0,
   SAMPLER_TYPE_2D = 1,
   SAMPLER_TYPE_2D_RECT = 2,
   SAMPLER_TYPE_EXTERNAL_OES = 3,
-  LAST_SAMPLER_TYPE = 3
 };
 
 enum BlendMode {
   BLEND_MODE_NONE,
   BLEND_MODE_NORMAL,
   BLEND_MODE_SCREEN,
   BLEND_MODE_OVERLAY,
   BLEND_MODE_DARKEN,
   BLEND_MODE_LIGHTEN,
   BLEND_MODE_COLOR_DODGE,
   BLEND_MODE_COLOR_BURN,
   BLEND_MODE_HARD_LIGHT,
   BLEND_MODE_SOFT_LIGHT,
   BLEND_MODE_DIFFERENCE,
   BLEND_MODE_EXCLUSION,
   BLEND_MODE_MULTIPLY,
   BLEND_MODE_HUE,
   BLEND_MODE_SATURATION,
   BLEND_MODE_COLOR,
   BLEND_MODE_LUMINOSITY,
   LAST_BLEND_MODE = BLEND_MODE_LUMINOSITY
 };
 
 enum InputColorSource {
+  // This includes RGB and RGBA textures.
   INPUT_COLOR_SOURCE_RGBA_TEXTURE,
+  // This includes Y and either UV or U-and-V textures.
+  INPUT_COLOR_SOURCE_YUV_TEXTURES,
+  // A solid color specified as a uniform value.
   INPUT_COLOR_SOURCE_UNIFORM,
 };
 
 // TODO(ccameron): Merge this with BlendMode.
 enum FragColorMode {
   FRAG_COLOR_MODE_DEFAULT,
   FRAG_COLOR_MODE_OPAQUE,
   FRAG_COLOR_MODE_APPLY_BLEND_MODE,
 };
 
 enum MaskMode {
   NO_MASK = 0,
   HAS_MASK = 1,
-  LAST_MASK_VALUE = HAS_MASK
 };
 
 // Note: The highp_threshold_cache must be provided by the caller to make
 // the caching multi-thread/context safe in an easy low-overhead manner.
 // The caller must make sure to clear highp_threshold_cache to 0, so it can be
 // reinitialized, if a new or different context is used.
 CC_EXPORT TexCoordPrecision
     TexCoordPrecisionRequired(gpu::gles2::GLES2Interface* context,
                               int* highp_threshold_cache,
                               int highp_threshold_min,
                               const gfx::Point& max_coordinate);
 
 CC_EXPORT TexCoordPrecision TexCoordPrecisionRequired(
     gpu::gles2::GLES2Interface* context,
     int *highp_threshold_cache,
     int highp_threshold_min,
     const gfx::Size& max_size);
 
-class VertexShaderBase {
+class VertexShader {
  public:
-  VertexShaderBase();
+  VertexShader();
   void Init(gpu::gles2::GLES2Interface* context,
             unsigned program,
             int* base_uniform_index);
   std::string GetShaderString() const;
 
  protected:
-  template <class VertexShader, class FragmentShader>
-  friend class ProgramBinding;
+  friend class Program;
 
   // Use arrays of uniforms for matrix, texTransform, and opacity.
   bool use_uniform_arrays_ = false;
 
   PositionSource position_source_ = POSITION_SOURCE_ATTRIBUTE;
   TexCoordSource tex_coord_source_ = TEX_COORD_SOURCE_NONE;
   TexCoordTransform tex_coord_transform_ = TEX_COORD_TRANSFORM_NONE;
 
   // Used only with TEX_COORD_TRANSFORM_VEC4.
   int vertex_tex_transform_location_ = -1;
@@ skipping 21 matching lines @@
   bool has_dummy_variables_ = false;
 
   bool has_vertex_opacity_ = false;
   int vertex_opacity_location_ = -1;
 
   AAMode aa_mode_ = NO_AA;
   int viewport_location_ = -1;
   int edge_location_ = -1;
 };
 
-class FragmentShaderBase {
+class FragmentShader {
  public:
   virtual void Init(gpu::gles2::GLES2Interface* context,
                     unsigned program,
                     int* base_uniform_index);
   std::string GetShaderString() const;
 
  protected:
-  FragmentShaderBase();
+  FragmentShader();
   virtual std::string GetShaderSource() const;
   bool has_blend_mode() const { return blend_mode_ != BLEND_MODE_NONE; }
 
   std::string SetBlendModeFunctions(const std::string& shader_string) const;
 
   // Settings that are modified by sub-classes.
   AAMode aa_mode_ = NO_AA;
   bool has_varying_alpha_ = false;
   SwizzleMode swizzle_mode_ = NO_SWIZZLE;
   PremultipliedAlphaMode premultiply_alpha_mode_ = PREMULTIPLIED_ALPHA;
@@ skipping 33 matching lines @@
 
   bool has_background_color_ = false;
   int background_color_location_ = -1;
 
   TexCoordPrecision tex_coord_precision_ = TEX_COORD_PRECISION_NA;
   SamplerType sampler_type_ = SAMPLER_TYPE_NA;
 
   BlendMode blend_mode_ = BLEND_MODE_NONE;
   bool mask_for_background_ = false;
 
- private:
-  template <class VertexShader, class FragmentShader>
-  friend class ProgramBinding;
-
-  std::string GetHelperFunctions() const;
-  std::string GetBlendFunction() const;
-  std::string GetBlendFunctionBodyForRGB() const;
-
-  DISALLOW_COPY_AND_ASSIGN(FragmentShaderBase);
-};
-
-class FragmentShaderYUVVideo : public FragmentShaderBase {
- public:
-  FragmentShaderYUVVideo();
-
-  void CheckSubclassProperties() {}
-
-  void Init(gpu::gles2::GLES2Interface* context,
-            unsigned program,
-            int* base_uniform_index) override;
-  int y_texture_location() const { return y_texture_location_; }
-  int u_texture_location() const { return u_texture_location_; }
-  int v_texture_location() const { return v_texture_location_; }
-  int uv_texture_location() const { return uv_texture_location_; }
-  int a_texture_location() const { return a_texture_location_; }
-  int lut_texture_location() const { return lut_texture_location_; }
-  int alpha_location() const { return alpha_location_; }
-  int yuv_matrix_location() const { return yuv_matrix_location_; }
-  int yuv_adj_location() const { return yuv_adj_location_; }
-  int ya_clamp_rect_location() const { return ya_clamp_rect_location_; }
-  int uv_clamp_rect_location() const { return uv_clamp_rect_location_; }
-  int resource_multiplier_location() const {
-    return resource_multiplier_location_;
-  }
-  int resource_offset_location() const { return resource_offset_location_; }
-
- private:
-  template <class VertexShader, class FragmentShader>
-  friend class ProgramBinding;
-
-  std::string GetShaderSource() const override;
-
+  // YUV-only parameters.
   bool use_alpha_texture_ = false;
   bool use_nv12_ = false;
   bool use_color_lut_ = false;
 
+  // YUV uniform locations.
   int y_texture_location_ = -1;
   int u_texture_location_ = -1;
   int v_texture_location_ = -1;
   int uv_texture_location_ = -1;
   int a_texture_location_ = -1;
   int lut_texture_location_ = -1;
-  int alpha_location_ = -1;
   int yuv_matrix_location_ = -1;
   int yuv_adj_location_ = -1;
   int ya_clamp_rect_location_ = -1;
   int uv_clamp_rect_location_ = -1;
   int resource_multiplier_location_ = -1;
   int resource_offset_location_ = -1;
+
+ private:
+  friend class Program;
+
+  // Functions specific to YUV video.
+  std::string GetShaderStringYUVVideo() const;
+  void InitYUVVideo(gpu::gles2::GLES2Interface* context,
+                    unsigned program,
+                    int* base_uniform_index);
+
+  std::string GetHelperFunctions() const;
+  std::string GetBlendFunction() const;
+  std::string GetBlendFunctionBodyForRGB() const;
+
+  DISALLOW_COPY_AND_ASSIGN(FragmentShader);
 };
 
 }  // namespace cc
 
 #endif  // CC_OUTPUT_SHADER_H_