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
81
82
83
84
85
86
87
88
89
90
|
// Copyright 2020 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/offloading_video_encoder.h"
#include "base/bind_post_task.h"
#include "base/sequenced_task_runner.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "base/threading/sequenced_task_runner_handle.h"
#include "media/base/video_frame.h"
namespace media {
OffloadingVideoEncoder::OffloadingVideoEncoder(
std::unique_ptr<VideoEncoder> wrapped_encoder,
const scoped_refptr<base::SequencedTaskRunner> work_runner,
const scoped_refptr<base::SequencedTaskRunner> callback_runner)
: wrapped_encoder_(std::move(wrapped_encoder)),
work_runner_(std::move(work_runner)),
callback_runner_(std::move(callback_runner)) {
DCHECK(wrapped_encoder_);
DCHECK(work_runner_);
DCHECK(callback_runner_);
DCHECK_NE(callback_runner_, work_runner_);
}
OffloadingVideoEncoder::OffloadingVideoEncoder(
std::unique_ptr<VideoEncoder> wrapped_encoder)
: OffloadingVideoEncoder(std::move(wrapped_encoder),
base::ThreadPool::CreateSequencedTaskRunner(
{base::TaskPriority::USER_BLOCKING}),
base::SequencedTaskRunnerHandle::Get()) {}
void OffloadingVideoEncoder::Initialize(VideoCodecProfile profile,
const Options& options,
OutputCB output_cb,
StatusCB done_cb) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
work_runner_->PostTask(
FROM_HERE,
base::BindOnce(&VideoEncoder::Initialize,
base::Unretained(wrapped_encoder_.get()), profile, options,
WrapCallback(std::move(output_cb)),
WrapCallback(std::move(done_cb))));
}
void OffloadingVideoEncoder::Encode(scoped_refptr<VideoFrame> frame,
bool key_frame,
StatusCB done_cb) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
work_runner_->PostTask(
FROM_HERE,
base::BindOnce(&VideoEncoder::Encode,
base::Unretained(wrapped_encoder_.get()), std::move(frame),
key_frame, WrapCallback(std::move(done_cb))));
}
void OffloadingVideoEncoder::ChangeOptions(const Options& options,
OutputCB output_cb,
StatusCB done_cb) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
work_runner_->PostTask(
FROM_HERE, base::BindOnce(&VideoEncoder::ChangeOptions,
base::Unretained(wrapped_encoder_.get()),
options, WrapCallback(std::move(output_cb)),
WrapCallback(std::move(done_cb))));
}
void OffloadingVideoEncoder::Flush(StatusCB done_cb) {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
work_runner_->PostTask(
FROM_HERE, base::BindOnce(&VideoEncoder::Flush,
base::Unretained(wrapped_encoder_.get()),
WrapCallback(std::move(done_cb))));
}
OffloadingVideoEncoder::~OffloadingVideoEncoder() {
DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
work_runner_->DeleteSoon(FROM_HERE, std::move(wrapped_encoder_));
}
template <class T>
T OffloadingVideoEncoder::WrapCallback(T cb) {
DCHECK(callback_runner_);
return base::BindPostTask(callback_runner_, std::move(cb));
}
} // namespace media
|
