cc: Implement UnionTraits for DrawQuads

cc/ipc/quads_struct_traits.h

 // Copyright 2016 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_IPC_QUADS_STRUCT_TRAITS_H_
 #define CC_IPC_QUADS_STRUCT_TRAITS_H_
 
 #include "cc/ipc/filter_operation_struct_traits.h"
 #include "cc/ipc/filter_operations_struct_traits.h"
 #include "cc/ipc/quads.mojom.h"
 #include "cc/ipc/render_pass_id_struct_traits.h"
 #include "cc/ipc/shared_quad_state_struct_traits.h"
 #include "cc/ipc/surface_id_struct_traits.h"
 #include "cc/quads/debug_border_draw_quad.h"
 #include "cc/quads/picture_draw_quad.h"
 #include "cc/quads/render_pass_draw_quad.h"
 #include "cc/quads/solid_color_draw_quad.h"
 #include "cc/quads/stream_video_draw_quad.h"
 #include "cc/quads/surface_draw_quad.h"
 #include "cc/quads/texture_draw_quad.h"
 #include "cc/quads/tile_draw_quad.h"
 #include "cc/quads/yuv_video_draw_quad.h"
 #include "ui/gfx/geometry/mojo/geometry_struct_traits.h"
 
 namespace mojo {
 
-cc::DrawQuad* AllocateAndConstruct(cc::mojom::Material material,
-                                   cc::QuadList* list);
+cc::DrawQuad* AllocateAndConstruct(
+    cc::mojom::DrawQuadState::DataView::Tag material,
+    cc::QuadList* list);
+
 template <>
-struct EnumTraits<cc::mojom::Material, cc::DrawQuad::Material> {
-  static cc::mojom::Material ToMojom(cc::DrawQuad::Material material);
-  static bool FromMojom(cc::mojom::Material input, cc::DrawQuad::Material* out);
+struct UnionTraits<cc::mojom::DrawQuadState, cc::DrawQuad> {
+  static cc::mojom::DrawQuadState::DataView::Tag GetTag(
+      const cc::DrawQuad& quad) {
+    switch (quad.material) {
+      case cc::DrawQuad::INVALID:
+        break;
+      case cc::DrawQuad::DEBUG_BORDER:
+        return cc::mojom::DrawQuadState::DataView::Tag::DEBUG_BORDER_QUAD_STATE;
+      case cc::DrawQuad::PICTURE_CONTENT:
+        break;
+      case cc::DrawQuad::RENDER_PASS:
+        return cc::mojom::DrawQuadState::DataView::Tag::RENDER_PASS_QUAD_STATE;
+      case cc::DrawQuad::SOLID_COLOR:
+        return cc::mojom::DrawQuadState::DataView::Tag::SOLID_COLOR_QUAD_STATE;
+      case cc::DrawQuad::STREAM_VIDEO_CONTENT:
+        return cc::mojom::DrawQuadState::DataView::Tag::STREAM_VIDEO_QUAD_STATE;
+      case cc::DrawQuad::SURFACE_CONTENT:
+        return cc::mojom::DrawQuadState::DataView::Tag::SURFACE_QUAD_STATE;
+      case cc::DrawQuad::TEXTURE_CONTENT:
+        return cc::mojom::DrawQuadState::DataView::Tag::TEXTURE_QUAD_STATE;
+      case cc::DrawQuad::TILED_CONTENT:
+        return cc::mojom::DrawQuadState::DataView::Tag::TILE_QUAD_STATE;
+      case cc::DrawQuad::YUV_VIDEO_CONTENT:
+        return cc::mojom::DrawQuadState::DataView::Tag::YUV_VIDEO_QUAD_STATE;
+    }
+    NOTREACHED();
+    return cc::mojom::DrawQuadState::DataView::Tag::DEBUG_BORDER_QUAD_STATE;
+  }
+
+  static const cc::DrawQuad& debug_border_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& render_pass_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& solid_color_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& surface_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& texture_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& tile_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& stream_video_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static const cc::DrawQuad& yuv_video_quad_state(const cc::DrawQuad& quad) {
+    return quad;
+  }
+
+  static bool Read(cc::mojom::DrawQuadState::DataView data, cc::DrawQuad* out) {
+    switch (data.tag()) {
+      case cc::mojom::DrawQuadState::DataView::Tag::DEBUG_BORDER_QUAD_STATE:
+        return data.ReadDebugBorderQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::RENDER_PASS_QUAD_STATE:
+        return data.ReadRenderPassQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::SOLID_COLOR_QUAD_STATE:
+        return data.ReadSolidColorQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::SURFACE_QUAD_STATE:
+        return data.ReadSurfaceQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::TEXTURE_QUAD_STATE:
+        return data.ReadTextureQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::TILE_QUAD_STATE:
+        return data.ReadTileQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::STREAM_VIDEO_QUAD_STATE:
+        return data.ReadStreamVideoQuadState(out);
+      case cc::mojom::DrawQuadState::DataView::Tag::YUV_VIDEO_QUAD_STATE:
+        return data.ReadYuvVideoQuadState(out);
+    }
+    NOTREACHED();
+    return false;
+  }
 };
 
 template <>
 struct StructTraits<cc::mojom::DebugBorderQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::DEBUG_BORDER;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // DebugBorderDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::DEBUG_BORDER, output->material);
-  }
-
   static uint32_t color(const cc::DrawQuad& input) {
     const cc::DebugBorderDrawQuad* quad =
         cc::DebugBorderDrawQuad::MaterialCast(&input);
     return quad->color;
   }
 
   static int32_t width(const cc::DrawQuad& input) {
     const cc::DebugBorderDrawQuad* quad =
         cc::DebugBorderDrawQuad::MaterialCast(&input);
     return quad->width;
   }
 
   static bool Read(cc::mojom::DebugBorderQuadStateDataView data,
                    cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::RenderPassQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& quad) {
-    return quad.material != cc::DrawQuad::RENDER_PASS;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // RenderPassDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::RENDER_PASS, output->material);
-  }
-
   static const cc::RenderPassId& render_pass_id(const cc::DrawQuad& input) {
     const cc::RenderPassDrawQuad* quad =
         cc::RenderPassDrawQuad::MaterialCast(&input);
     return quad->render_pass_id;
   }
 
   static uint32_t mask_resource_id(const cc::DrawQuad& input) {
     const cc::RenderPassDrawQuad* quad =
         cc::RenderPassDrawQuad::MaterialCast(&input);
     return quad->mask_resource_id();
@@ skipping 29 matching lines @@
         cc::RenderPassDrawQuad::MaterialCast(&input);
     return quad->background_filters;
   }
 
   static bool Read(cc::mojom::RenderPassQuadStateDataView data,
                    cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::SolidColorQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::SOLID_COLOR;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // SolidColorDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::SOLID_COLOR, output->material);
-  }
-
   static uint32_t color(const cc::DrawQuad& input) {
     const cc::SolidColorDrawQuad* quad =
         cc::SolidColorDrawQuad::MaterialCast(&input);
     return quad->color;
   }
 
   static bool force_anti_aliasing_off(const cc::DrawQuad& input) {
     const cc::SolidColorDrawQuad* quad =
         cc::SolidColorDrawQuad::MaterialCast(&input);
     return quad->force_anti_aliasing_off;
   }
 
   static bool Read(cc::mojom::SolidColorQuadStateDataView data,
                    cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::StreamVideoQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::STREAM_VIDEO_CONTENT;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // StreamVideoDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::STREAM_VIDEO_CONTENT, output->material);
-  }
-
   static uint32_t resource_id(const cc::DrawQuad& input) {
     const cc::StreamVideoDrawQuad* quad =
         cc::StreamVideoDrawQuad::MaterialCast(&input);
     return quad->resources.ids[cc::StreamVideoDrawQuad::kResourceIdIndex];
   }
 
   static const gfx::Size& resource_size_in_pixels(const cc::DrawQuad& input) {
     const cc::StreamVideoDrawQuad* quad =
         cc::StreamVideoDrawQuad::MaterialCast(&input);
     return quad->overlay_resources
         .size_in_pixels[cc::StreamVideoDrawQuad::kResourceIdIndex];
   }
 
   static const gfx::Transform& matrix(const cc::DrawQuad& input) {
     const cc::StreamVideoDrawQuad* quad =
         cc::StreamVideoDrawQuad::MaterialCast(&input);
     return quad->matrix;
   }
 
   static bool Read(cc::mojom::StreamVideoQuadStateDataView data,
                    cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::SurfaceQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::SURFACE_CONTENT;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // SurfaceDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::SURFACE_CONTENT, output->material);
-  }
-
   static const cc::SurfaceId& surface(const cc::DrawQuad& input) {
     const cc::SurfaceDrawQuad* quad = cc::SurfaceDrawQuad::MaterialCast(&input);
     return quad->surface_id;
   }
 
   static bool Read(cc::mojom::SurfaceQuadStateDataView data, cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::TextureQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::TEXTURE_CONTENT;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // TextureContentDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::TEXTURE_CONTENT, output->material);
-  }
-
   static uint32_t resource_id(const cc::DrawQuad& input) {
     const cc::TextureDrawQuad* quad = cc::TextureDrawQuad::MaterialCast(&input);
     return quad->resource_id();
   }
 
   static bool premultiplied_alpha(const cc::DrawQuad& input) {
     const cc::TextureDrawQuad* quad = cc::TextureDrawQuad::MaterialCast(&input);
     return quad->premultiplied_alpha;
   }
 
@@ skipping 30 matching lines @@
   static bool secure_output_only(const cc::DrawQuad& input) {
     const cc::TextureDrawQuad* quad = cc::TextureDrawQuad::MaterialCast(&input);
     return quad->secure_output_only;
   }
 
   static bool Read(cc::mojom::TextureQuadStateDataView data, cc::DrawQuad* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::TileQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::TILED_CONTENT;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // TileDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::TILED_CONTENT, output->material);
-  }
-
   static const gfx::RectF& tex_coord_rect(const cc::DrawQuad& input) {
     const cc::TileDrawQuad* quad = cc::TileDrawQuad::MaterialCast(&input);
     return quad->tex_coord_rect;
   }
 
   static const gfx::Size& texture_size(const cc::DrawQuad& input) {
     const cc::TileDrawQuad* quad = cc::TileDrawQuad::MaterialCast(&input);
     return quad->texture_size;
   }
 
@@ skipping 18 matching lines @@
 template <>
 struct EnumTraits<cc::mojom::YUVColorSpace, cc::YUVVideoDrawQuad::ColorSpace> {
   static cc::mojom::YUVColorSpace ToMojom(
       cc::YUVVideoDrawQuad::ColorSpace color_space);
   static bool FromMojom(cc::mojom::YUVColorSpace input,
                         cc::YUVVideoDrawQuad::ColorSpace* out);
 };
 
 template <>
 struct StructTraits<cc::mojom::YUVVideoQuadState, cc::DrawQuad> {
-  static bool IsNull(const cc::DrawQuad& input) {
-    return input.material != cc::DrawQuad::YUV_VIDEO_CONTENT;
-  }
-
-  static void SetToNull(cc::DrawQuad* output) {
-    // There is nothing to deserialize here if the DrawQuad is not a
-    // YUVVideoDrawQuad. If it is, then this should not be called.
-    DCHECK_NE(cc::DrawQuad::YUV_VIDEO_CONTENT, output->material);
-  }
-
   static const gfx::RectF& ya_tex_coord_rect(const cc::DrawQuad& input) {
     const cc::YUVVideoDrawQuad* quad =
         cc::YUVVideoDrawQuad::MaterialCast(&input);
     return quad->ya_tex_coord_rect;
   }
 
   static const gfx::RectF& uv_tex_coord_rect(const cc::DrawQuad& input) {
     const cc::YUVVideoDrawQuad* quad =
         cc::YUVVideoDrawQuad::MaterialCast(&input);
     return quad->uv_tex_coord_rect;
@@ skipping 64 matching lines @@
                    cc::DrawQuad* out);
 };
 
 struct DrawQuadWithSharedQuadState {
   const cc::DrawQuad* quad;
   const cc::SharedQuadState* shared_quad_state;
 };
 
 template <>
 struct StructTraits<cc::mojom::DrawQuad, DrawQuadWithSharedQuadState> {
-  static cc::DrawQuad::Material material(
-      const DrawQuadWithSharedQuadState& input) {
-    return input.quad->material;
-  }
-
   static const gfx::Rect& rect(const DrawQuadWithSharedQuadState& input) {
     return input.quad->rect;
   }
 
   static const gfx::Rect& opaque_rect(
       const DrawQuadWithSharedQuadState& input) {
     return input.quad->opaque_rect;
   }
 
   static const gfx::Rect& visible_rect(
       const DrawQuadWithSharedQuadState& input) {
     return input.quad->visible_rect;
   }
 
   static bool needs_blending(const DrawQuadWithSharedQuadState& input) {
     return input.quad->needs_blending;
   }
 
-  static const cc::DrawQuad& debug_border_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& render_pass_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& solid_color_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& surface_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& texture_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& tile_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& stream_video_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
-  static const cc::DrawQuad& yuv_video_quad_state(
-      const DrawQuadWithSharedQuadState& input) {
-    return *input.quad;
-  }
-
   static OptSharedQuadState sqs(const DrawQuadWithSharedQuadState& input) {
     return {input.shared_quad_state};
   }
+
+  static const cc::DrawQuad& draw_quad_state(
+      const DrawQuadWithSharedQuadState& input) {
+    return *input.quad;
+  }
 };
 
 // This StructTraits is only used for deserialization within RenderPasses.
 template <>
 struct StructTraits<cc::mojom::DrawQuad, cc::DrawQuad> {
   static bool Read(cc::mojom::DrawQuadDataView data, cc::DrawQuad* out);
 };
 
 template <>
 struct ArrayTraits<cc::QuadList> {
@@ skipping 24 matching lines @@
       dq.shared_quad_state = current_sqs;
     return dq;
   }
 
   static size_t GetSize(const cc::QuadList& input) { return input.size(); }
 };
 
 }  // namespace mojo
 
 #endif  // CC_IPC_QUADS_STRUCT_TRAITS_H_