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
|
// Copyright 2014 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 REMOTING_HOST_VIDEO_FRAME_RECORDER_H_
#define REMOTING_HOST_VIDEO_FRAME_RECORDER_H_
#include <stdint.h>
#include <list>
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/single_thread_task_runner.h"
namespace webrtc {
class DesktopFrame;
}
namespace remoting {
class VideoEncoder;
// Allows sequences of DesktopFrames passed to a VideoEncoder to be recorded.
//
// VideoFrameRecorder is designed to support applications which use a dedicated
// thread for video encoding, but need to manage that process from a "main"
// or "control" thread.
//
// On the control thread:
// 1. Create the VideoFrameRecorder.
// 2. Specify the amount of memory that may be used for recording.
// 3. Call WrapVideoEncoder(), passing the actual VideoEncoder that will be
// used to encode frames.
// 4. Hand off the returned wrapper VideoEncoder to the video encoding thread,
// to call in place of the actual VideoEncoder.
// 5. Start/stop frame recording as necessary.
// 6. Use NextFrame() to read each recorded frame in sequence.
//
// The wrapper VideoEncoder is designed to be handed off to the video encoding
// thread, and used and torn down there.
//
// The VideoFrameRecorder and VideoEncoder may be torn down in any order; frame
// recording will stop as soon as either is destroyed.
class VideoFrameRecorder {
public:
VideoFrameRecorder();
virtual ~VideoFrameRecorder();
// Wraps the supplied VideoEncoder, returning a replacement VideoEncoder that
// will route frames to the recorder, as well as passing them for encoding.
// Each VideoFrameRecorder supports at most one wrapper at a time; if a new
// wrapper is required then any existing one must be deleted, or detached via
// DetachVideoEncoderWrapper(), before WrapVideoEncoder() is called again.
scoped_ptr<VideoEncoder> WrapVideoEncoder(scoped_ptr<VideoEncoder> encoder);
// Detaches the existing VideoEncoder wrapper, stopping it from recording.
// The detached wrapper remains owned by the caller and will continue to
// pass-through frames to the wrapped encoder, without recording them.
void DetachVideoEncoderWrapper();
// Enables/disables frame recording. Frame recording is initially disabled.
void SetEnableRecording(bool enable_recording);
// Sets the maximum number of bytes of pixel data that may be recorded.
// When this maximum is reached older frames will be discarded to make space
// for new ones.
void SetMaxContentBytes(int64_t max_content_bytes);
// Pops the next recorded frame in the sequence, and returns it.
scoped_ptr<webrtc::DesktopFrame> NextFrame();
private:
class RecordingVideoEncoder;
friend class RecordingVideoEncoder;
void SetEncoderTaskRunner(scoped_refptr<base::TaskRunner> task_runner);
void RecordFrame(scoped_ptr<webrtc::DesktopFrame> frame);
// The recorded frames, in sequence.
std::list<webrtc::DesktopFrame*> recorded_frames_;
// Size of the recorded frames' content, in bytes.
int64_t content_bytes_;
// Size that recorded frames' content must not exceed.
int64_t max_content_bytes_;
// True if recording is started, false otherwise.
bool enable_recording_;
// Task runner on which the wrapper VideoEncoder is being run.
scoped_refptr<base::TaskRunner> encoder_task_runner_;
// Weak reference to the wrapper VideoEncoder, to use to control it.
base::WeakPtr<RecordingVideoEncoder> recording_encoder_;
scoped_refptr<base::SingleThreadTaskRunner> caller_task_runner_;
base::WeakPtrFactory<VideoFrameRecorder> weak_factory_;
DISALLOW_COPY_AND_ASSIGN(VideoFrameRecorder);
};
} // namespace remoting
#endif // REMOTING_HOST_VIDEO_FRAME_RECORDER_H_
|
