| Flutter Impeller |

yuv_to_rgb_filter_contents.cc
Go to the documentation of this file.
1 // Copyright 2013 The Flutter 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 "impeller/entity/contents/filters/yuv_to_rgb_filter_contents.h"
6
7 #include "impeller/core/formats.h"
8 #include "impeller/entity/contents/anonymous_contents.h"
9 #include "impeller/entity/contents/content_context.h"
10 #include "impeller/geometry/matrix.h"
11 #include "impeller/renderer/render_pass.h"
12 #include "impeller/renderer/vertex_buffer_builder.h"
13
14 namespace impeller {
15
16 // clang-format off
17 constexpr Matrix kMatrixBT601LimitedRange = {
18 1.164, 1.164, 1.164, 0.0,
19 0.0, -0.392, 2.017, 0.0,
20 1.596, -0.813, 0.0, 0.0,
21 0.0, 0.0, 0.0, 1.0
22 };
23
24 constexpr Matrix kMatrixBT601FullRange = {
25 1.0, 1.0, 1.0, 0.0,
26 0.0, -0.344, 1.772, 0.0,
27 1.402, -0.714, 0.0, 0.0,
28 0.0, 0.0, 0.0, 1.0
29 };
30 // clang-format on
31
32 YUVToRGBFilterContents::YUVToRGBFilterContents() = default;
33
34 YUVToRGBFilterContents::~YUVToRGBFilterContents() = default;
35
36 void YUVToRGBFilterContents::SetYUVColorSpace(YUVColorSpace yuv_color_space) {
37 yuv_color_space_ = yuv_color_space;
38 }
39
40 std::optional<Entity> YUVToRGBFilterContents::RenderFilter(
41 const FilterInput::Vector& inputs,
42 const ContentContext& renderer,
43 const Entity& entity,
44 const Matrix& effect_transform,
45 const Rect& coverage,
46 const std::optional<Rect>& coverage_hint) const {
47 if (inputs.size() < 2) {
48 return std::nullopt;
49 }
50
51 using VS = YUVToRGBFilterPipeline::VertexShader;
52 using FS = YUVToRGBFilterPipeline::FragmentShader;
53
54 auto y_input_snapshot =
55 inputs[0]->GetSnapshot("YUVToRGB(Y)", renderer, entity);
56 auto uv_input_snapshot =
57 inputs[1]->GetSnapshot("YUVToRGB(UV)", renderer, entity);
58 if (!y_input_snapshot.has_value() || !uv_input_snapshot.has_value()) {
59 return std::nullopt;
60 }
61
62 if (y_input_snapshot->texture->GetTextureDescriptor().format !=
63 PixelFormat::kR8UNormInt ||
64 uv_input_snapshot->texture->GetTextureDescriptor().format !=
65 PixelFormat::kR8G8UNormInt) {
66 return std::nullopt;
67 }
68
69 //----------------------------------------------------------------------------
70 /// Create AnonymousContents for rendering.
71 ///
72 RenderProc render_proc = [y_input_snapshot, uv_input_snapshot,
73 yuv_color_space = yuv_color_space_](
74 const ContentContext& renderer,
75 const Entity& entity, RenderPass& pass) -> bool {
76 pass.SetCommandLabel("YUV to RGB Filter");
77
78 auto options = OptionsFromPassAndEntity(pass, entity);
79 options.primitive_type = PrimitiveType::kTriangleStrip;
80 pass.SetPipeline(renderer.GetYUVToRGBFilterPipeline(options));
81
82 auto size = y_input_snapshot->texture->GetSize();
83
84 std::array<VS::PerVertexData, 4> vertices = {
85 VS::PerVertexData{Point(0, 0)},
86 VS::PerVertexData{Point(1, 0)},
87 VS::PerVertexData{Point(0, 1)},
88 VS::PerVertexData{Point(1, 1)},
89 };
90
91 auto& data_host_buffer = renderer.GetTransientsDataBuffer();
92 pass.SetVertexBuffer(CreateVertexBuffer(vertices, data_host_buffer));
93
94 VS::FrameInfo frame_info;
95 frame_info.mvp = Entity::GetShaderTransform(
96 entity.GetShaderClipDepth(), pass,
97 entity.GetTransform() * y_input_snapshot->transform *
98 Matrix::MakeScale(Vector2(size)));
99 frame_info.texture_sampler_y_coord_scale =
100 y_input_snapshot->texture->GetYCoordScale();
101
102 FS::FragInfo frag_info;
103 frag_info.yuv_color_space = static_cast<Scalar>(yuv_color_space);
104 switch (yuv_color_space) {
105 case YUVColorSpace::kBT601LimitedRange:
106 frag_info.matrix = kMatrixBT601LimitedRange;
107 break;
108 case YUVColorSpace::kBT601FullRange:
109 frag_info.matrix = kMatrixBT601FullRange;
110 break;
111 }
112
113 raw_ptr<const Sampler> sampler =
114 renderer.GetContext()->GetSamplerLibrary()->GetSampler({});
115 FS::BindYTexture(pass, y_input_snapshot->texture, sampler);
116 FS::BindUvTexture(pass, uv_input_snapshot->texture, sampler);
117
118 FS::BindFragInfo(pass, data_host_buffer.EmplaceUniform(frag_info));
119 VS::BindFrameInfo(pass, data_host_buffer.EmplaceUniform(frame_info));
120
121 return pass.Draw().ok();
122 };
123
124 CoverageProc coverage_proc =
125 [coverage](const Entity& entity) -> std::optional<Rect> {
126 return coverage.TransformBounds(entity.GetTransform());
127 };
128
129 std::shared_ptr<Contents> contents =
130 AnonymousContents::Make(render_proc, coverage_proc);
131
132 Entity sub_entity;
133 sub_entity.SetContents(std::move(contents));
134 sub_entity.SetBlendMode(entity.GetBlendMode());
135 return sub_entity;
136 }
137
138 std::optional<Rect> YUVToRGBFilterContents::GetFilterSourceCoverage(
139 const Matrix& effect_transform,
140 const Rect& output_limit) const {
141 return output_limit;
142 }
143
144 } // namespace impeller
anonymous_contents.h
impeller::AnonymousContents::Make
static std::shared_ptr< Contents > Make(RenderProc render_proc, CoverageProc coverage_proc)
Definition: anonymous_contents.cc:11
impeller::ContentContext
Definition: content_context.h:122
impeller::Contents::CoverageProc
std::function< std::optional< Rect >(const Entity &entity)> CoverageProc
Definition: contents.h:40
impeller::Contents::RenderProc
std::function< bool(const ContentContext &renderer, const Entity &entity, RenderPass &pass)> RenderProc
Definition: contents.h:39
impeller::Entity
Definition: entity.h:26
impeller::Entity::GetShaderTransform
Matrix GetShaderTransform(const RenderPass &pass) const
Definition: entity.cc:50
impeller::FilterInput::Vector
std::vector< FilterInput::Ref > Vector
Definition: filter_input.h:33
impeller::RenderPipelineHandle::VertexShader
VertexShader_ VertexShader
Definition: pipeline.h:163
impeller::RenderPipelineHandle::FragmentShader
FragmentShader_ FragmentShader
Definition: pipeline.h:164
impeller::YUVToRGBFilterContents::~YUVToRGBFilterContents
~YUVToRGBFilterContents() override
impeller::YUVToRGBFilterContents::YUVToRGBFilterContents
YUVToRGBFilterContents()
impeller::YUVToRGBFilterContents::SetYUVColorSpace
void SetYUVColorSpace(YUVColorSpace yuv_color_space)
Definition: yuv_to_rgb_filter_contents.cc:36
content_context.h
formats.h
matrix.h
impeller
Definition: allocation.cc:12
impeller::YUVColorSpace
YUVColorSpace
Definition: color.h:54
impeller::YUVColorSpace::kBT601FullRange
@ kBT601FullRange
impeller::YUVColorSpace::kBT601LimitedRange
@ kBT601LimitedRange
impeller::kMatrixBT601LimitedRange
constexpr Matrix kMatrixBT601LimitedRange
Definition: yuv_to_rgb_filter_contents.cc:17
impeller::PrimitiveType::kTriangleStrip
@ kTriangleStrip
impeller::Vector2
Point Vector2
Definition: point.h:429
impeller::Scalar
float Scalar
Definition: scalar.h:19
impeller::Rect
TRect< Scalar > Rect
Definition: rect.h:788
impeller::Point
TPoint< Scalar > Point
Definition: point.h:425
impeller::FS
LinePipeline::FragmentShader FS
Definition: line_contents.cc:14
impeller::CreateVertexBuffer
VertexBuffer CreateVertexBuffer(std::array< VertexType, size > input, HostBuffer &data_host_buffer)
Create an index-less vertex buffer from a fixed size array.
Definition: vertex_buffer_builder.h:24
impeller::PixelFormat::kR8UNormInt
@ kR8UNormInt
impeller::PixelFormat::kR8G8UNormInt
@ kR8G8UNormInt
impeller::kMatrixBT601FullRange
constexpr Matrix kMatrixBT601FullRange
Definition: yuv_to_rgb_filter_contents.cc:24
impeller::VS
LinePipeline::VertexShader VS
Definition: line_contents.cc:13
impeller::OptionsFromPassAndEntity
ContentContextOptions OptionsFromPassAndEntity(const RenderPass &pass, const Entity &entity)
Definition: contents.cc:34
render_pass.h
impeller::Matrix
A 4x4 matrix using column-major storage.
Definition: matrix.h:37
impeller::Matrix::MakeScale
static constexpr Matrix MakeScale(const Vector3 &s)
Definition: matrix.h:104
impeller::TRect< Scalar >
impeller::TRect::TransformBounds
constexpr TRect TransformBounds(const Matrix &transform) const
Creates a new bounding box that contains this transformed rectangle.
Definition: rect.h:472
vertex_buffer_builder.h
yuv_to_rgb_filter_contents.h