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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright 2016 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.
#ifndef EXTENSIONS_RENDERER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_VIDEO_ENCODER_H_
#define EXTENSIONS_RENDERER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_VIDEO_ENCODER_H_
#include <vector>
#include "base/memory/shared_memory.h"
#include "extensions/renderer/api/display_source/wifi_display/wifi_display_media_encoder.h"
#include "media/base/video_frame.h"
#include "media/video/video_encode_accelerator.h"
#include "third_party/wds/src/libwds/public/video_format.h"
namespace extensions {
class WiFiDisplayElementaryStreamDescriptor;
using WiFiDisplayEncodedFrame = WiFiDisplayEncodedUnit;
// This interface represents H.264 video encoder used by the
// Wi-Fi Display media pipeline.
// Threading: the client code should belong to a single thread.
class WiFiDisplayVideoEncoder : public WiFiDisplayMediaEncoder {
public:
using VideoEncoderCallback =
base::Callback<void(scoped_refptr<WiFiDisplayVideoEncoder>)>;
using ReceiveVideoEncodeAcceleratorCallback =
base::Callback<void(scoped_refptr<base::SingleThreadTaskRunner>,
std::unique_ptr<media::VideoEncodeAccelerator>)>;
using CreateVideoEncodeAcceleratorCallback =
base::Callback<void(const ReceiveVideoEncodeAcceleratorCallback&)>;
using ReceiveEncodeMemoryCallback =
base::Callback<void(std::unique_ptr<base::SharedMemory>)>;
using CreateEncodeMemoryCallback =
base::Callback<void(size_t size, const ReceiveEncodeMemoryCallback&)>;
struct InitParameters {
InitParameters();
InitParameters(const InitParameters&);
~InitParameters();
gfx::Size frame_size;
int frame_rate;
int bit_rate;
wds::H264Profile profile;
wds::H264Level level;
// Video Encode Accelerator (VEA) specific parameters.
CreateEncodeMemoryCallback create_memory_callback;
CreateVideoEncodeAcceleratorCallback vea_create_callback;
};
// Returns the list of supported video encoder profiles
// for the given frame size and frame rate.
// If hight profile is supported it is to be first in the list.
static std::vector<wds::H264Profile> FindSupportedProfiles(
const gfx::Size& frame_size,
int32_t frame_rate);
// A factory method that creates a new encoder instance from the given
// |params|, the encoder instance is returned as an argument of
// |result_callback| ('nullptr' argument means encoder creation failure).
static void Create(const InitParameters& params,
const VideoEncoderCallback& encoder_callback);
// WiFiDisplayMediaEncoder
WiFiDisplayElementaryStreamInfo CreateElementaryStreamInfo() const final;
// Encodes the given raw frame. The resulting encoded frame is passed
// as an |encoded_callback|'s argument which is set via 'SetCallbacks'
// method.
void InsertRawVideoFrame(const scoped_refptr<media::VideoFrame>& video_frame,
base::TimeTicks reference_time);
// Requests the next encoded frame to be an instantaneous decoding refresh
// (IDR) picture.
void RequestIDRPicture();
protected:
// A factory method that creates a new encoder instance which uses OpenH264
// SVC encoder for encoding.
static void CreateSVC(const InitParameters& params,
const VideoEncoderCallback& encoder_callback);
// A factory method that creates a new encoder instance which uses Video
// Encode Accelerator (VEA) for encoding.
static void CreateVEA(const InitParameters& params,
const VideoEncoderCallback& encoder_callback);
static void OnCreatedVEA(const InitParameters& params,
const VideoEncoderCallback& encoder_callback,
scoped_refptr<WiFiDisplayVideoEncoder> vea_encoder);
explicit WiFiDisplayVideoEncoder(
scoped_refptr<base::SingleThreadTaskRunner> media_task_runner);
~WiFiDisplayVideoEncoder() override;
virtual void InsertFrameOnMediaThread(
scoped_refptr<media::VideoFrame> video_frame,
base::TimeTicks reference_time,
bool send_idr) = 0;
std::vector<WiFiDisplayElementaryStreamDescriptor> descriptors_;
scoped_refptr<base::SingleThreadTaskRunner> media_task_runner_;
bool send_idr_;
};
} // namespace extensions
#endif // EXTENSIONS_RENDERER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_VIDEO_ENCODER_H_
|
