OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "ui/gfx/buffer_format_util.h" | 5 #include "ui/gfx/buffer_format_util.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "base/numerics/safe_math.h" | 8 #include "base/numerics/safe_math.h" |
9 | 9 |
10 namespace gfx { | 10 namespace gfx { |
11 namespace { | 11 namespace { |
12 | 12 |
13 const BufferFormat kBufferFormats[] = { | 13 const BufferFormat kBufferFormats[] = { |
14 BufferFormat::ATC, BufferFormat::ATCIA, | 14 BufferFormat::ATC, BufferFormat::ATCIA, |
15 BufferFormat::DXT1, BufferFormat::DXT5, | 15 BufferFormat::DXT1, BufferFormat::DXT5, |
16 BufferFormat::ETC1, BufferFormat::R_8, | 16 BufferFormat::ETC1, BufferFormat::R_8, |
17 BufferFormat::RGBA_4444, BufferFormat::RGBA_8888, | 17 BufferFormat::RGBA_4444, BufferFormat::RGBA_8888, |
18 BufferFormat::BGRX_8888, BufferFormat::BGRA_8888, | 18 BufferFormat::BGRX_8888, BufferFormat::BGRA_8888, |
19 BufferFormat::UYVY_422, BufferFormat::YUV_420_BIPLANAR, | 19 BufferFormat::UYVY_422, BufferFormat::YUV_420_BIPLANAR, |
20 BufferFormat::YUV_420}; | 20 BufferFormat::YUV_420}; |
21 | 21 |
22 static_assert(arraysize(kBufferFormats) == | 22 static_assert(arraysize(kBufferFormats) == |
23 (static_cast<int>(BufferFormat::LAST) + 1), | 23 (static_cast<int>(BufferFormat::LAST) + 1), |
24 "BufferFormat::LAST must be last value of kBufferFormats"); | 24 "BufferFormat::LAST must be last value of kBufferFormats"); |
25 | 25 |
| 26 |
| 27 bool RowSizeForBufferFormatChecked( |
| 28 size_t width, BufferFormat format, int plane, size_t* size_in_bytes) { |
| 29 base::CheckedNumeric<size_t> checked_size = width; |
| 30 switch (format) { |
| 31 case BufferFormat::ATCIA: |
| 32 case BufferFormat::DXT5: |
| 33 DCHECK_EQ(0, plane); |
| 34 *size_in_bytes = width; |
| 35 return true; |
| 36 case BufferFormat::ATC: |
| 37 case BufferFormat::DXT1: |
| 38 case BufferFormat::ETC1: |
| 39 DCHECK_EQ(0, plane); |
| 40 DCHECK_EQ(0u, width % 2); |
| 41 *size_in_bytes = width / 2; |
| 42 return true; |
| 43 case BufferFormat::R_8: |
| 44 checked_size += 3; |
| 45 if (!checked_size.IsValid()) |
| 46 return false; |
| 47 *size_in_bytes = checked_size.ValueOrDie() & ~0x3; |
| 48 return true; |
| 49 case BufferFormat::RGBA_4444: |
| 50 case BufferFormat::UYVY_422: |
| 51 checked_size *= 2; |
| 52 if (!checked_size.IsValid()) |
| 53 return false; |
| 54 *size_in_bytes = checked_size.ValueOrDie(); |
| 55 return true; |
| 56 case BufferFormat::BGRX_8888: |
| 57 case BufferFormat::RGBA_8888: |
| 58 case BufferFormat::BGRA_8888: |
| 59 checked_size *= 4; |
| 60 if (!checked_size.IsValid()) |
| 61 return false; |
| 62 *size_in_bytes = checked_size.ValueOrDie(); |
| 63 return true; |
| 64 case BufferFormat::YUV_420: |
| 65 DCHECK_EQ(0u, width % 2); |
| 66 *size_in_bytes = width / SubsamplingFactorForBufferFormat(format, plane); |
| 67 return true; |
| 68 case BufferFormat::YUV_420_BIPLANAR: |
| 69 DCHECK_EQ(width % 2, 0u); |
| 70 *size_in_bytes = width; |
| 71 return true; |
| 72 } |
| 73 NOTREACHED(); |
| 74 return false; |
| 75 } |
| 76 |
26 } // namespace | 77 } // namespace |
27 | 78 |
28 std::vector<BufferFormat> GetBufferFormats() { | 79 std::vector<BufferFormat> GetBufferFormatsForTesting() { |
29 return std::vector<BufferFormat>(kBufferFormats, | 80 return std::vector<BufferFormat>(kBufferFormats, |
30 kBufferFormats + arraysize(kBufferFormats)); | 81 kBufferFormats + arraysize(kBufferFormats)); |
31 } | 82 } |
32 | 83 |
33 size_t NumberOfPlanesForBufferFormat(BufferFormat format) { | 84 size_t NumberOfPlanesForBufferFormat(BufferFormat format) { |
34 switch (format) { | 85 switch (format) { |
35 case BufferFormat::ATC: | 86 case BufferFormat::ATC: |
36 case BufferFormat::ATCIA: | 87 case BufferFormat::ATCIA: |
37 case BufferFormat::DXT1: | 88 case BufferFormat::DXT1: |
38 case BufferFormat::DXT5: | 89 case BufferFormat::DXT5: |
(...skipping 26 matching lines...) Expand all Loading... |
65 case BufferFormat::RGBA_8888: | 116 case BufferFormat::RGBA_8888: |
66 case BufferFormat::BGRX_8888: | 117 case BufferFormat::BGRX_8888: |
67 case BufferFormat::BGRA_8888: | 118 case BufferFormat::BGRA_8888: |
68 case BufferFormat::UYVY_422: | 119 case BufferFormat::UYVY_422: |
69 return 1; | 120 return 1; |
70 case BufferFormat::YUV_420: { | 121 case BufferFormat::YUV_420: { |
71 static size_t factor[] = {1, 2, 2}; | 122 static size_t factor[] = {1, 2, 2}; |
72 DCHECK_LT(static_cast<size_t>(plane), arraysize(factor)); | 123 DCHECK_LT(static_cast<size_t>(plane), arraysize(factor)); |
73 return factor[plane]; | 124 return factor[plane]; |
74 } | 125 } |
75 case gfx::BufferFormat::YUV_420_BIPLANAR: { | 126 case BufferFormat::YUV_420_BIPLANAR: { |
76 static size_t factor[] = {1, 2}; | 127 static size_t factor[] = {1, 2}; |
77 DCHECK_LT(static_cast<size_t>(plane), arraysize(factor)); | 128 DCHECK_LT(static_cast<size_t>(plane), arraysize(factor)); |
78 return factor[plane]; | 129 return factor[plane]; |
79 } | 130 } |
80 } | 131 } |
81 NOTREACHED(); | 132 NOTREACHED(); |
82 return 0; | 133 return 0; |
83 } | 134 } |
84 | 135 |
85 size_t RowSizeForBufferFormat(size_t width, BufferFormat format, int plane) { | 136 size_t RowSizeForBufferFormat(size_t width, BufferFormat format, int plane) { |
86 size_t row_size = 0; | 137 size_t row_size = 0; |
87 bool valid = RowSizeForBufferFormatChecked(width, format, plane, &row_size); | 138 bool valid = RowSizeForBufferFormatChecked(width, format, plane, &row_size); |
88 DCHECK(valid); | 139 DCHECK(valid); |
89 return row_size; | 140 return row_size; |
90 } | 141 } |
91 | 142 |
92 bool RowSizeForBufferFormatChecked( | 143 size_t BufferSizeForBufferFormat(const Size& size, BufferFormat format) { |
93 size_t width, BufferFormat format, int plane, size_t* size_in_bytes) { | |
94 base::CheckedNumeric<size_t> checked_size = width; | |
95 switch (format) { | |
96 case BufferFormat::ATCIA: | |
97 case BufferFormat::DXT5: | |
98 DCHECK_EQ(0, plane); | |
99 *size_in_bytes = width; | |
100 return true; | |
101 case BufferFormat::ATC: | |
102 case BufferFormat::DXT1: | |
103 case BufferFormat::ETC1: | |
104 DCHECK_EQ(0, plane); | |
105 DCHECK_EQ(0u, width % 2); | |
106 *size_in_bytes = width / 2; | |
107 return true; | |
108 case BufferFormat::R_8: | |
109 checked_size += 3; | |
110 if (!checked_size.IsValid()) | |
111 return false; | |
112 *size_in_bytes = checked_size.ValueOrDie() & ~0x3; | |
113 return true; | |
114 case BufferFormat::RGBA_4444: | |
115 case BufferFormat::UYVY_422: | |
116 checked_size *= 2; | |
117 if (!checked_size.IsValid()) | |
118 return false; | |
119 *size_in_bytes = checked_size.ValueOrDie(); | |
120 return true; | |
121 case BufferFormat::BGRX_8888: | |
122 case BufferFormat::RGBA_8888: | |
123 case BufferFormat::BGRA_8888: | |
124 checked_size *= 4; | |
125 if (!checked_size.IsValid()) | |
126 return false; | |
127 *size_in_bytes = checked_size.ValueOrDie(); | |
128 return true; | |
129 case BufferFormat::YUV_420: | |
130 DCHECK_EQ(0u, width % 2); | |
131 *size_in_bytes = width / SubsamplingFactorForBufferFormat(format, plane); | |
132 return true; | |
133 case gfx::BufferFormat::YUV_420_BIPLANAR: | |
134 DCHECK_EQ(width % 2, 0u); | |
135 *size_in_bytes = width; | |
136 return true; | |
137 } | |
138 NOTREACHED(); | |
139 return false; | |
140 } | |
141 | |
142 size_t BufferSizeForBufferFormat( | |
143 const Size& size, BufferFormat format) { | |
144 size_t buffer_size = 0; | 144 size_t buffer_size = 0; |
145 bool valid = BufferSizeForBufferFormatChecked(size, format, &buffer_size); | 145 bool valid = BufferSizeForBufferFormatChecked(size, format, &buffer_size); |
146 DCHECK(valid); | 146 DCHECK(valid); |
147 return buffer_size; | 147 return buffer_size; |
148 } | 148 } |
149 | 149 |
150 bool BufferSizeForBufferFormatChecked( | 150 bool BufferSizeForBufferFormatChecked(const Size& size, |
151 const Size& size, BufferFormat format, size_t* size_in_bytes) { | 151 BufferFormat format, |
| 152 size_t* size_in_bytes) { |
152 base::CheckedNumeric<size_t> checked_size = 0; | 153 base::CheckedNumeric<size_t> checked_size = 0; |
153 size_t num_planes = NumberOfPlanesForBufferFormat(format); | 154 size_t num_planes = NumberOfPlanesForBufferFormat(format); |
154 for (size_t i = 0; i < num_planes; ++i) { | 155 for (size_t i = 0; i < num_planes; ++i) { |
155 size_t row_size = 0; | 156 size_t row_size = 0; |
156 if (!RowSizeForBufferFormatChecked(size.width(), format, i, &row_size)) | 157 if (!RowSizeForBufferFormatChecked(size.width(), format, i, &row_size)) |
157 return false; | 158 return false; |
158 base::CheckedNumeric<size_t> checked_plane_size = row_size; | 159 base::CheckedNumeric<size_t> checked_plane_size = row_size; |
159 checked_plane_size *= size.height() / | 160 checked_plane_size *= size.height() / |
160 SubsamplingFactorForBufferFormat(format, i); | 161 SubsamplingFactorForBufferFormat(format, i); |
161 if (!checked_plane_size.IsValid()) | 162 if (!checked_plane_size.IsValid()) |
162 return false; | 163 return false; |
163 checked_size += checked_plane_size.ValueOrDie(); | 164 checked_size += checked_plane_size.ValueOrDie(); |
164 if (!checked_size.IsValid()) | 165 if (!checked_size.IsValid()) |
165 return false; | 166 return false; |
166 } | 167 } |
167 *size_in_bytes = checked_size.ValueOrDie(); | 168 *size_in_bytes = checked_size.ValueOrDie(); |
168 return true; | 169 return true; |
169 } | 170 } |
170 | 171 |
| 172 int BufferOffsetForBufferFormat(const Size& size, |
| 173 BufferFormat format, |
| 174 size_t plane) { |
| 175 DCHECK_LT(plane, gfx::NumberOfPlanesForBufferFormat(format)); |
| 176 switch (format) { |
| 177 case BufferFormat::ATC: |
| 178 case BufferFormat::ATCIA: |
| 179 case BufferFormat::DXT1: |
| 180 case BufferFormat::DXT5: |
| 181 case BufferFormat::ETC1: |
| 182 case BufferFormat::R_8: |
| 183 case BufferFormat::RGBA_4444: |
| 184 case BufferFormat::RGBA_8888: |
| 185 case BufferFormat::BGRX_8888: |
| 186 case BufferFormat::BGRA_8888: |
| 187 case BufferFormat::UYVY_422: |
| 188 return 0; |
| 189 case BufferFormat::YUV_420: { |
| 190 static size_t offset_in_2x2_sub_sampling_sizes[] = {0, 4, 5}; |
| 191 DCHECK_LT(plane, arraysize(offset_in_2x2_sub_sampling_sizes)); |
| 192 return offset_in_2x2_sub_sampling_sizes[plane] * |
| 193 (size.width() / 2 + size.height() / 2); |
| 194 } |
| 195 case gfx::BufferFormat::YUV_420_BIPLANAR: { |
| 196 static size_t offset_in_2x2_sub_sampling_sizes[] = {0, 4}; |
| 197 DCHECK_LT(plane, arraysize(offset_in_2x2_sub_sampling_sizes)); |
| 198 return offset_in_2x2_sub_sampling_sizes[plane] * |
| 199 (size.width() / 2 + size.height() / 2); |
| 200 } |
| 201 } |
| 202 NOTREACHED(); |
| 203 return 0; |
| 204 } |
| 205 |
171 } // namespace gfx | 206 } // namespace gfx |
OLD | NEW |