[go: nahoru, domu]
Google Git
Sign in
chromium / chromium / src / a8feeae1ac6aeed523003080be0561cc0357525b / . / media / gpu / vp9_decoder.cc
blob: 6a7227ffb180308f3933ff673a1822e4112b5dfc [file] [log] [blame]
Avi Drissmand387f0922022-09-14 20:51:31[diff] [blame]1// Copyright 2015 The Chromium Authors
posciakd94b2b082015-09-18 04:03:40[diff] [blame]2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
markdittmer6e70beb82016-05-02 05:40:47[diff] [blame]5#include "media/gpu/vp9_decoder.h"
mostynb6682b1c42016-04-19 10:17:30[diff] [blame]6
7#include <memory>
8
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]9#include "base/feature_list.h"
Avi Drissmand70f89a2023-01-11 23:52:55[diff] [blame]10#include "base/functional/bind.h"
mostynb6682b1c42016-04-19 10:17:30[diff] [blame]11#include "base/logging.h"
Qiu Jianlin2153ac02022-08-18 04:13:15[diff] [blame]12#include "base/metrics/histogram_functions.h"
Hirokazu Honda967b9192019-08-27 00:59:42[diff] [blame]13#include "build/build_config.h"
Yuta Hijikata29f67812020-10-24 01:15:44[diff] [blame]14#include "build/chromeos_buildflags.h"
posciakd94b2b082015-09-18 04:03:40[diff] [blame]15#include "media/base/limits.h"
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]16#include "media/base/media_switches.h"
markdittmer6e70beb82016-05-02 05:40:47[diff] [blame]17#include "media/gpu/vp9_decoder.h"
posciakd94b2b082015-09-18 04:03:40[diff] [blame]18
markdittmer6e70beb82016-05-02 05:40:47[diff] [blame]19namespace media {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]20
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]21namespace {
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]22bool GetSpatialLayerFrameSize(const DecoderBuffer& decoder_buffer,
23 std::vector<uint32_t>& frame_sizes) {
24 frame_sizes.clear();
25
Jeffrey Kardatzkea8feeae2023-08-15 20:10:16[diff] [blame^]26 if (!decoder_buffer.has_side_data() ||
27 decoder_buffer.side_data()->spatial_layers.empty()) {
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]28 return true;
Jeffrey Kardatzkea8feeae2023-08-15 20:10:16[diff] [blame^]29 }
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]30
Hirokazu Hondabbf7f402022-04-04 08:13:49[diff] [blame]31 bool enable_vp9_ksvc =
Shuhai Pengfe10fd02021-11-11 03:20:59[diff] [blame]32// On windows, currently only d3d11 supports decoding VP9 kSVC stream, we
33// shouldn't combine the switch kD3D11Vp9kSVCHWDecoding to kVp9kSVCHWDecoding
34// due to we want keep returning false to MediaCapability.
Xiaohan Wangd4983e82022-01-15 06:19:37[diff] [blame]35#if BUILDFLAG(IS_WIN)
Hirokazu Hondabbf7f402022-04-04 08:13:49[diff] [blame]36 base::FeatureList::IsEnabled(media::kD3D11Vp9kSVCHWDecoding);
Xiaohan Wangc33487932022-04-22 23:37:21[diff] [blame]37#elif BUILDFLAG(IS_CHROMEOS) && defined(ARCH_CPU_ARM_FAMILY)
Hirokazu Hondabbf7f402022-04-04 08:13:49[diff] [blame]38 // V4L2 stateless API decoder is not capable of decoding VP9 k-SVC stream.
39 false;
Shuhai Pengfe10fd02021-11-11 03:20:59[diff] [blame]40#else
Hirokazu Hondabbf7f402022-04-04 08:13:49[diff] [blame]41 base::FeatureList::IsEnabled(media::kVp9kSVCHWDecoding);
42#endif
43
44 if (!enable_vp9_ksvc) {
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]45 DLOG(ERROR) << "Vp9 k-SVC hardware decoding is disabled";
46 return false;
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]47 }
48
Jeffrey Kardatzkea8feeae2023-08-15 20:10:16[diff] [blame^]49 size_t num_of_layers = decoder_buffer.side_data()->spatial_layers.size();
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]50 if (num_of_layers > 3u) {
51 DLOG(WARNING) << "The maximum number of spatial layers in VP9 is three";
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]52 return false;
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]53 }
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]54
Jeffrey Kardatzkea8feeae2023-08-15 20:10:16[diff] [blame^]55 frame_sizes = decoder_buffer.side_data()->spatial_layers;
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]56 return true;
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]57}
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]58
59VideoCodecProfile VP9ProfileToVideoCodecProfile(uint8_t profile) {
60 switch (profile) {
61 case 0:
62 return VP9PROFILE_PROFILE0;
63 case 1:
64 return VP9PROFILE_PROFILE1;
65 case 2:
66 return VP9PROFILE_PROFILE2;
67 case 3:
68 return VP9PROFILE_PROFILE3;
69 default:
70 return VIDEO_CODEC_PROFILE_UNKNOWN;
71 }
72}
73
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]74bool IsValidBitDepth(uint8_t bit_depth, VideoCodecProfile profile) {
75 // Spec 7.2.
76 switch (profile) {
77 case VP9PROFILE_PROFILE0:
78 case VP9PROFILE_PROFILE1:
79 return bit_depth == 8u;
80 case VP9PROFILE_PROFILE2:
81 case VP9PROFILE_PROFILE3:
82 return bit_depth == 10u || bit_depth == 12u;
83 default:
Peter Boström0ce6af602023-05-15 21:27:24[diff] [blame]84 NOTREACHED_NORETURN();
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]85 }
86}
Hirokazu Honda790d98e2020-12-04 03:41:48[diff] [blame]87
Qiu Jianlin2153ac02022-08-18 04:13:15[diff] [blame]88VideoChromaSampling GetVP9ChromaSampling(const Vp9FrameHeader& frame_header) {
89 // Spec section 7.2.2
90 uint8_t subsampling_x = frame_header.subsampling_x;
91 uint8_t subsampling_y = frame_header.subsampling_y;
92 if (subsampling_x == 0 && subsampling_y == 0) {
93 return VideoChromaSampling::k444;
94 } else if (subsampling_x == 1u && subsampling_y == 0u) {
95 return VideoChromaSampling::k422;
96 } else if (subsampling_x == 1u && subsampling_y == 1u) {
97 return VideoChromaSampling::k420;
98 } else {
99 DLOG(WARNING) << "Unknown chroma sampling format.";
100 return VideoChromaSampling::kUnknown;
101 }
Hirokazu Honda790d98e2020-12-04 03:41:48[diff] [blame]102}
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]103} // namespace
104
posciakd94b2b082015-09-18 04:03:40[diff] [blame]105VP9Decoder::VP9Accelerator::VP9Accelerator() {}
106
107VP9Decoder::VP9Accelerator::~VP9Accelerator() {}
108
Miguel Casas71b9dc42022-03-05 02:45:41[diff] [blame]109bool VP9Decoder::VP9Accelerator::SupportsContextProbabilityReadback() const {
110 return false;
111}
112
Fredrik Hubinette3cebc622018-10-27 01:01:12[diff] [blame]113VP9Decoder::VP9Decoder(std::unique_ptr<VP9Accelerator> accelerator,
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]114 VideoCodecProfile profile,
Drew Davenporteb9cb952023-01-31 00:29:39[diff] [blame]115 const VideoColorSpace& container_color_space)
posciak77118c92016-08-28 13:18:39[diff] [blame]116 : state_(kNeedStreamMetadata),
Fredrik Hubinette3cebc622018-10-27 01:01:12[diff] [blame]117 container_color_space_(container_color_space),
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]118 // TODO(hiroh): Set profile to UNKNOWN.
119 profile_(profile),
Miguel Casas2d5ecad82018-02-22 19:03:05[diff] [blame]120 accelerator_(std::move(accelerator)),
Miguel Casas4eab0212022-03-13 17:32:52[diff] [blame]121 parser_(accelerator_->NeedsCompressedHeaderParsed(),
Miguel Casas71b9dc42022-03-05 02:45:41[diff] [blame]122 accelerator_->SupportsContextProbabilityReadback()) {}
posciakd94b2b082015-09-18 04:03:40[diff] [blame]123
Miguel Casas2d5ecad82018-02-22 19:03:05[diff] [blame]124VP9Decoder::~VP9Decoder() = default;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]125
Hirokazu Honda23132c32019-07-09 14:31:20[diff] [blame]126void VP9Decoder::SetStream(int32_t id, const DecoderBuffer& decoder_buffer) {
127 const uint8_t* ptr = decoder_buffer.data();
128 const size_t size = decoder_buffer.data_size();
129 const DecryptConfig* decrypt_config = decoder_buffer.decrypt_config();
130
posciakd94b2b082015-09-18 04:03:40[diff] [blame]131 DCHECK(ptr);
132 DCHECK(size);
Pawel Osciakec6e21b2018-03-19 09:13:06[diff] [blame]133 DVLOG(4) << "New input stream id: " << id << " at: " << (void*)ptr
134 << " size: " << size;
135 stream_id_ = id;
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]136 std::vector<uint32_t> frame_sizes;
137 if (!GetSpatialLayerFrameSize(decoder_buffer, frame_sizes)) {
138 SetError();
139 return;
Ted Meyer0b35c5fd2018-11-27 22:29:29[diff] [blame]140 }
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]141
142 parser_.SetStream(ptr, size, frame_sizes,
143 decrypt_config ? decrypt_config->Clone() : nullptr);
posciakd94b2b082015-09-18 04:03:40[diff] [blame]144}
145
146bool VP9Decoder::Flush() {
147 DVLOG(2) << "Decoder flush";
148 Reset();
149 return true;
150}
151
152void VP9Decoder::Reset() {
153 curr_frame_hdr_ = nullptr;
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]154 decrypt_config_.reset();
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]155 pending_pic_.reset();
Sreerenj Balachandran1beb5482019-04-03 03:40:05[diff] [blame]156
157 ref_frames_.Clear();
posciakd94b2b082015-09-18 04:03:40[diff] [blame]158
159 parser_.Reset();
160
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]161 if (state_ == kDecoding) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]162 state_ = kAfterReset;
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]163 }
posciakd94b2b082015-09-18 04:03:40[diff] [blame]164}
165
166VP9Decoder::DecodeResult VP9Decoder::Decode() {
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]167 while (true) {
Hirokazu Hondafb2681c2021-10-29 03:53:59[diff] [blame]168 if (state_ == kError)
169 return kDecodeError;
170
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]171 // If we have a pending picture to decode, try that first.
172 if (pending_pic_) {
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]173 VP9Accelerator::Status status =
174 DecodeAndOutputPicture(std::move(pending_pic_));
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]175 if (status == VP9Accelerator::Status::kFail) {
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]176 SetError();
177 return kDecodeError;
178 }
179 if (status == VP9Accelerator::Status::kTryAgain)
180 return kTryAgain;
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]181 }
182
posciakd94b2b082015-09-18 04:03:40[diff] [blame]183 // Read a new frame header if one is not awaiting decoding already.
184 if (!curr_frame_hdr_) {
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]185 gfx::Size allocate_size;
xhwang60f96672016-06-14 21:47:53[diff] [blame]186 std::unique_ptr<Vp9FrameHeader> hdr(new Vp9FrameHeader());
Ted Meyer0b35c5fd2018-11-27 22:29:29[diff] [blame]187 Vp9Parser::Result res =
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]188 parser_.ParseNextFrame(hdr.get(), &allocate_size, &decrypt_config_);
posciakd94b2b082015-09-18 04:03:40[diff] [blame]189 switch (res) {
xhwang60f96672016-06-14 21:47:53[diff] [blame]190 case Vp9Parser::kOk:
brusi_roy8e7155962016-09-14 18:22:58[diff] [blame]191 curr_frame_hdr_ = std::move(hdr);
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]192 curr_frame_size_ = allocate_size;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]193 break;
194
xhwang60f96672016-06-14 21:47:53[diff] [blame]195 case Vp9Parser::kEOStream:
posciakd94b2b082015-09-18 04:03:40[diff] [blame]196 return kRanOutOfStreamData;
197
xhwang60f96672016-06-14 21:47:53[diff] [blame]198 case Vp9Parser::kInvalidStream:
posciakd94b2b082015-09-18 04:03:40[diff] [blame]199 DVLOG(1) << "Error parsing stream";
200 SetError();
201 return kDecodeError;
kcwue5a9a6292016-08-17 03:40:02[diff] [blame]202
203 case Vp9Parser::kAwaitingRefresh:
posciak77118c92016-08-28 13:18:39[diff] [blame]204 DVLOG(4) << "Awaiting context update";
205 return kNeedContextUpdate;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]206 }
207 }
208
209 if (state_ != kDecoding) {
210 // Not kDecoding, so we need a resume point (a keyframe), as we are after
211 // reset or at the beginning of the stream. Drop anything that is not
212 // a keyframe in such case, and continue looking for a keyframe.
Sreerenj Balachandran57163302018-11-28 02:45:57[diff] [blame]213 // Only exception is when the stream/sequence starts with an Intra only
214 // frame.
215 if (curr_frame_hdr_->IsKeyframe() ||
216 (curr_frame_hdr_->IsIntra() && pic_size_.IsEmpty())) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]217 state_ = kDecoding;
218 } else {
219 curr_frame_hdr_.reset();
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]220 decrypt_config_.reset();
posciakd94b2b082015-09-18 04:03:40[diff] [blame]221 continue;
222 }
223 }
224
225 if (curr_frame_hdr_->show_existing_frame) {
226 // This frame header only instructs us to display one of the
227 // previously-decoded frames, but has no frame data otherwise. Display
228 // and continue decoding subsequent frames.
kcwue5a9a6292016-08-17 03:40:02[diff] [blame]229 size_t frame_to_show = curr_frame_hdr_->frame_to_show_map_idx;
Sreerenj Balachandran1beb5482019-04-03 03:40:05[diff] [blame]230 if (frame_to_show >= kVp9NumRefFrames ||
231 !ref_frames_.GetFrame(frame_to_show)) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]232 DVLOG(1) << "Request to show an invalid frame";
233 SetError();
234 return kDecodeError;
235 }
236
Chih-Yu Huangcd5181e2018-10-23 06:33:06[diff] [blame]237 // Duplicate the VP9Picture and set the current bitstream id to keep the
238 // correct timestamp.
Sreerenj Balachandran1beb5482019-04-03 03:40:05[diff] [blame]239 scoped_refptr<VP9Picture> pic =
240 ref_frames_.GetFrame(frame_to_show)->Duplicate();
Chih-Yu Huangcd5181e2018-10-23 06:33:06[diff] [blame]241 pic->set_bitstream_id(stream_id_);
Dan Sanders9d1177622023-08-15 19:39:11[diff] [blame]242 pic->frame_hdr = std::move(curr_frame_hdr_);
Chih-Yu Huangcd5181e2018-10-23 06:33:06[diff] [blame]243 if (!accelerator_->OutputPicture(std::move(pic))) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]244 SetError();
245 return kDecodeError;
246 }
247
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]248 decrypt_config_.reset();
posciakd94b2b082015-09-18 04:03:40[diff] [blame]249 continue;
250 }
251
Hirokazu Honda10df95d2019-07-19 19:31:43[diff] [blame]252 gfx::Size new_pic_size = curr_frame_size_;
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]253 gfx::Rect new_render_rect(curr_frame_hdr_->render_width,
254 curr_frame_hdr_->render_height);
Hirokazu Hondacffd0ef2020-02-28 03:32:31[diff] [blame]255 // For safety, check the validity of render size or leave it as pic size.
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]256 if (!gfx::Rect(new_pic_size).Contains(new_render_rect)) {
257 DVLOG(1) << "Render size exceeds picture size. render size: "
258 << new_render_rect.ToString()
259 << ", picture size: " << new_pic_size.ToString();
Hirokazu Hondacffd0ef2020-02-28 03:32:31[diff] [blame]260 new_render_rect = gfx::Rect(new_pic_size);
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]261 }
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]262 VideoCodecProfile new_profile =
263 VP9ProfileToVideoCodecProfile(curr_frame_hdr_->profile);
264 if (new_profile == VIDEO_CODEC_PROFILE_UNKNOWN) {
265 VLOG(1) << "Invalid profile: " << curr_frame_hdr_->profile;
266 return kDecodeError;
267 }
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]268 if (!IsValidBitDepth(curr_frame_hdr_->bit_depth, new_profile)) {
269 DVLOG(1) << "Invalid bit depth="
270 << base::strict_cast<int>(curr_frame_hdr_->bit_depth)
271 << ", profile=" << GetProfileName(new_profile);
272 return kDecodeError;
273 }
Qiu Jianlin2153ac02022-08-18 04:13:15[diff] [blame]274 VideoChromaSampling new_chroma_sampling =
275 GetVP9ChromaSampling(*curr_frame_hdr_);
276 if (new_chroma_sampling != chroma_sampling_) {
277 chroma_sampling_ = new_chroma_sampling;
278 base::UmaHistogramEnumeration(
279 "Media.PlatformVideoDecoding.ChromaSampling", chroma_sampling_);
280 }
281
282 if (chroma_sampling_ != VideoChromaSampling::k420) {
Hirokazu Honda790d98e2020-12-04 03:41:48[diff] [blame]283 DVLOG(1) << "Only YUV 4:2:0 is supported";
284 return kDecodeError;
285 }
posciakd94b2b082015-09-18 04:03:40[diff] [blame]286
Saifuddin Hitawalae2e6d482023-05-31 19:33:28[diff] [blame]287 VideoColorSpace new_color_space;
288 // For VP9, container color spaces override video stream color spaces.
289 if (container_color_space_.IsSpecified()) {
290 new_color_space = container_color_space_;
291 } else if (curr_frame_hdr_->GetColorSpace().IsSpecified()) {
292 new_color_space = curr_frame_hdr_->GetColorSpace();
293 }
294
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]295 DCHECK(!new_pic_size.IsEmpty());
Miguel Casasc4d75842023-04-25 04:37:00[diff] [blame]296
297 const bool is_pic_size_different = new_pic_size != pic_size_;
298 const bool is_pic_size_larger = new_pic_size.width() > pic_size_.width() ||
299 new_pic_size.height() > pic_size_.height();
300 const bool is_new_configuration_different_enough =
301 (ignore_resolution_changes_to_smaller_for_testing_
302 ? is_pic_size_larger
303 : is_pic_size_different) ||
304 new_profile != profile_ || curr_frame_hdr_->bit_depth != bit_depth_;
305
306 if (is_new_configuration_different_enough) {
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]307 DVLOG(1) << "New profile: " << GetProfileName(new_profile)
Miguel Casasc4d75842023-04-25 04:37:00[diff] [blame]308 << ", new resolution: " << new_pic_size.ToString()
309 << ", new bit depth: "
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]310 << base::strict_cast<int>(curr_frame_hdr_->bit_depth);
posciakd94b2b082015-09-18 04:03:40[diff] [blame]311
Sreerenj Balachandran57163302018-11-28 02:45:57[diff] [blame]312 if (!curr_frame_hdr_->IsKeyframe() &&
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]313 !(curr_frame_hdr_->IsIntra() && pic_size_.IsEmpty())) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]314 // TODO(posciak): This is doable, but requires a few modifications to
315 // VDA implementations to allow multiple picture buffer sets in flight.
Dongseong Hwang4f20ebb2018-06-07 00:28:20[diff] [blame]316 // http://crbug.com/832264
Sreerenj Balachandran57163302018-11-28 02:45:57[diff] [blame]317 DVLOG(1) << "Resolution change currently supported for keyframes and "
318 "sequence begins with Intra only when there is no prior "
319 "frames in the context";
Dongseong Hwang4f20ebb2018-06-07 00:28:20[diff] [blame]320 if (++size_change_failure_counter_ > kVPxMaxNumOfSizeChangeFailures) {
321 SetError();
322 return kDecodeError;
323 }
324
325 curr_frame_hdr_.reset();
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]326 decrypt_config_.reset();
Dongseong Hwang4f20ebb2018-06-07 00:28:20[diff] [blame]327 return kRanOutOfStreamData;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]328 }
329
330 // TODO(posciak): This requires us to be on a keyframe (see above) and is
331 // required, because VDA clients expect all surfaces to be returned before
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]332 // they can cycle surface sets after receiving kConfigChange.
posciakd94b2b082015-09-18 04:03:40[diff] [blame]333 // This is only an implementation detail of VDAs and can be improved.
Sreerenj Balachandran1beb5482019-04-03 03:40:05[diff] [blame]334 ref_frames_.Clear();
posciakd94b2b082015-09-18 04:03:40[diff] [blame]335
336 pic_size_ = new_pic_size;
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]337 visible_rect_ = new_render_rect;
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]338 profile_ = new_profile;
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]339 bit_depth_ = curr_frame_hdr_->bit_depth;
Saifuddin Hitawalae2e6d482023-05-31 19:33:28[diff] [blame]340 picture_color_space_ = new_color_space;
Dongseong Hwang4f20ebb2018-06-07 00:28:20[diff] [blame]341 size_change_failure_counter_ = 0;
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]342 return kConfigChange;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]343 }
344
Saifuddin Hitawalae2e6d482023-05-31 19:33:28[diff] [blame]345 if (new_color_space.IsSpecified() &&
346 new_color_space != picture_color_space_ &&
347 curr_frame_hdr_->IsKeyframe()) {
348 // TODO(posciak): This requires us to be on a keyframe (see above) and is
349 // required, because VDA clients expect all surfaces to be returned before
350 // they can cycle surface sets after receiving kConfigChange.
351 // This is only an implementation detail of VDAs and can be improved.
352 ref_frames_.Clear();
353
354 picture_color_space_ = new_color_space;
355 return kColorSpaceChange;
356 }
357
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]358 scoped_refptr<VP9Picture> pic = accelerator_->CreateVP9Picture();
359 if (!pic) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]360 return kRanOutOfSurfaces;
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]361 }
johnylin74410872017-06-19 13:05:30[diff] [blame]362 DVLOG(2) << "Render resolution: " << new_render_rect.ToString();
363
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]364 pic->set_visible_rect(new_render_rect);
365 pic->set_bitstream_id(stream_id_);
Fredrik Hubinette3cebc622018-10-27 01:01:12[diff] [blame]366
Jeffrey Kardatzke48ebdb692021-01-11 22:05:02[diff] [blame]367 pic->set_decrypt_config(std::move(decrypt_config_));
Ted Meyer0b35c5fd2018-11-27 22:29:29[diff] [blame]368
Saifuddin Hitawalae2e6d482023-05-31 19:33:28[diff] [blame]369 // Set the color space for the picture.
370 pic->set_colorspace(picture_color_space_);
posciakd94b2b082015-09-18 04:03:40[diff] [blame]371
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]372 pic->frame_hdr = std::move(curr_frame_hdr_);
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]373
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]374 VP9Accelerator::Status status = DecodeAndOutputPicture(std::move(pic));
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]375 if (status == VP9Accelerator::Status::kFail) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]376 SetError();
377 return kDecodeError;
378 }
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]379 if (status == VP9Accelerator::Status::kTryAgain)
380 return kTryAgain;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]381 }
382}
383
posciak77118c92016-08-28 13:18:39[diff] [blame]384void VP9Decoder::UpdateFrameContext(
Ted Meyer4fac4f62019-05-08 22:57:15[diff] [blame]385 scoped_refptr<VP9Picture> pic,
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]386 Vp9Parser::ContextRefreshCallback context_refresh_cb) {
Dale Curtise25163812018-09-21 22:13:39[diff] [blame]387 DCHECK(context_refresh_cb);
posciak77118c92016-08-28 13:18:39[diff] [blame]388 Vp9FrameContext frame_ctx;
389 memset(&frame_ctx, 0, sizeof(frame_ctx));
390
Ted Meyer4fac4f62019-05-08 22:57:15[diff] [blame]391 if (!accelerator_->GetFrameContext(std::move(pic), &frame_ctx)) {
posciak77118c92016-08-28 13:18:39[diff] [blame]392 SetError();
393 return;
394 }
395
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]396 std::move(context_refresh_cb).Run(frame_ctx);
posciak77118c92016-08-28 13:18:39[diff] [blame]397}
398
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]399VP9Decoder::VP9Accelerator::Status VP9Decoder::DecodeAndOutputPicture(
400 scoped_refptr<VP9Picture> pic) {
posciakd94b2b082015-09-18 04:03:40[diff] [blame]401 DCHECK(!pic_size_.IsEmpty());
402 DCHECK(pic->frame_hdr);
403
Chih-Yu Huangcc6b1b92019-12-19 08:57:16[diff] [blame]404 base::OnceClosure done_cb;
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]405 Vp9Parser::ContextRefreshCallback context_refresh_cb =
posciak77118c92016-08-28 13:18:39[diff] [blame]406 parser_.GetContextRefreshCb(pic->frame_hdr->frame_context_idx);
Jose Lopes24d1cda82020-02-21 13:22:36[diff] [blame]407 if (context_refresh_cb) {
408 done_cb =
409 base::BindOnce(&VP9Decoder::UpdateFrameContext, base::Unretained(this),
410 pic, std::move(context_refresh_cb));
411 }
posciak77118c92016-08-28 13:18:39[diff] [blame]412
413 const Vp9Parser::Context& context = parser_.context();
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]414 VP9Accelerator::Status status = accelerator_->SubmitDecode(
415 pic, context.segmentation(), context.loop_filter(), ref_frames_,
416 std::move(done_cb));
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]417 if (status != VP9Accelerator::Status::kOk) {
418 if (status == VP9Accelerator::Status::kTryAgain)
419 pending_pic_ = std::move(pic);
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]420 return status;
Jeffrey Kardatzke4ed4e332020-12-05 00:32:15[diff] [blame]421 }
posciakd94b2b082015-09-18 04:03:40[diff] [blame]422
423 if (pic->frame_hdr->show_frame) {
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]424 if (!accelerator_->OutputPicture(pic))
425 return VP9Accelerator::Status::kFail;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]426 }
427
Ted Meyer4fac4f62019-05-08 22:57:15[diff] [blame]428 ref_frames_.Refresh(std::move(pic));
Jeffrey Kardatzke5e36baf2020-11-20 22:17:13[diff] [blame]429 return status;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]430}
431
432void VP9Decoder::SetError() {
433 Reset();
434 state_ = kError;
435}
436
437gfx::Size VP9Decoder::GetPicSize() const {
438 return pic_size_;
439}
440
Hirokazu Honda436cf27b2019-05-21 10:44:39[diff] [blame]441gfx::Rect VP9Decoder::GetVisibleRect() const {
442 return visible_rect_;
443}
444
Hirokazu Honda7ec32652019-11-22 09:40:21[diff] [blame]445VideoCodecProfile VP9Decoder::GetProfile() const {
446 return profile_;
447}
448
Hirokazu Honda963303782020-12-03 05:57:04[diff] [blame]449uint8_t VP9Decoder::GetBitDepth() const {
450 return bit_depth_;
451}
452
Qiu Jianlina790b3d2022-07-29 03:09:44[diff] [blame]453VideoChromaSampling VP9Decoder::GetChromaSampling() const {
Qiu Jianlin2153ac02022-08-18 04:13:15[diff] [blame]454 return chroma_sampling_;
Qiu Jianlina790b3d2022-07-29 03:09:44[diff] [blame]455}
456
Saifuddin Hitawalae2e6d482023-05-31 19:33:28[diff] [blame]457VideoColorSpace VP9Decoder::GetVideoColorSpace() const {
458 return picture_color_space_;
459}
460
Sida Zhuc4276442022-12-03 03:54:24[diff] [blame]461absl::optional<gfx::HDRMetadata> VP9Decoder::GetHDRMetadata() const {
462 // VP9 only allow HDR metadata exists in the container.
463 return absl::nullopt;
464}
465
posciakd94b2b082015-09-18 04:03:40[diff] [blame]466size_t VP9Decoder::GetRequiredNumOfPictures() const {
Miguel Casas477a7062019-01-04 19:13:45[diff] [blame]467 constexpr size_t kPicsInPipeline = limits::kMaxVideoFrames + 1;
468 return kPicsInPipeline + GetNumReferenceFrames();
469}
470
471size_t VP9Decoder::GetNumReferenceFrames() const {
Miguel Casas3dd7e562019-02-14 17:38:46[diff] [blame]472 // Maximum number of reference frames
473 return kVp9NumRefFrames;
posciakd94b2b082015-09-18 04:03:40[diff] [blame]474}
475
markdittmer6e70beb82016-05-02 05:40:47[diff] [blame]476} // namespace media