1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
// Copyright 2018 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.
#include "media/base/video_thumbnail_decoder.h"
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "media/base/decoder_buffer.h"
#include "media/base/video_frame.h"
namespace media {
VideoThumbnailDecoder::VideoThumbnailDecoder(
std::unique_ptr<VideoDecoder> decoder,
const VideoDecoderConfig& config,
std::vector<uint8_t> encoded_data)
: decoder_(std::move(decoder)),
config_(config),
encoded_data_(std::move(encoded_data)) {
DCHECK(!encoded_data_.empty());
DCHECK(config_.IsValidConfig());
}
VideoThumbnailDecoder::~VideoThumbnailDecoder() = default;
void VideoThumbnailDecoder::Start(VideoFrameCallback video_frame_callback) {
video_frame_callback_ = std::move(video_frame_callback);
DCHECK(video_frame_callback_);
decoder_->Initialize(
config_, false, nullptr,
base::BindOnce(&VideoThumbnailDecoder::OnVideoDecoderInitialized,
weak_factory_.GetWeakPtr()),
base::BindRepeating(&VideoThumbnailDecoder::OnVideoFrameDecoded,
weak_factory_.GetWeakPtr()),
base::DoNothing());
}
void VideoThumbnailDecoder::OnVideoDecoderInitialized(Status status) {
if (!status.is_ok()) {
NotifyComplete(nullptr);
return;
}
auto buffer =
DecoderBuffer::CopyFrom(&encoded_data_[0], encoded_data_.size());
encoded_data_.clear();
decoder_->Decode(buffer,
base::BindOnce(&VideoThumbnailDecoder::OnVideoBufferDecoded,
weak_factory_.GetWeakPtr()));
}
void VideoThumbnailDecoder::OnVideoBufferDecoded(DecodeStatus status) {
if (status != DecodeStatus::OK) {
NotifyComplete(nullptr);
return;
}
// Enqueue eos since only one video frame is needed for thumbnail.
decoder_->Decode(DecoderBuffer::CreateEOSBuffer(),
base::BindOnce(&VideoThumbnailDecoder::OnEosBufferDecoded,
weak_factory_.GetWeakPtr()));
}
void VideoThumbnailDecoder::OnEosBufferDecoded(DecodeStatus status) {
if (status != DecodeStatus::OK)
NotifyComplete(nullptr);
}
void VideoThumbnailDecoder::OnVideoFrameDecoded(
scoped_refptr<VideoFrame> frame) {
NotifyComplete(std::move(frame));
}
void VideoThumbnailDecoder::NotifyComplete(scoped_refptr<VideoFrame> frame) {
DCHECK(video_frame_callback_);
std::move(video_frame_callback_).Run(std::move(frame));
}
} // namespace media
|
