summaryrefslogtreecommitdiffstats
path: root/content/browser/devtools/protocol/frame_recorder.h
blob: 56926efc09530ae5917adac589665140e19c11f3 (plain)
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
// 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 CONTENT_BROWSER_DEVTOOLS_PROTOCOL_FRAME_RECORDER_H_
#define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_FRAME_RECORDER_H_

#include <string>
#include <utility>
#include <vector>

#include "base/callback.h"
#include "base/cancelable_callback.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "content/browser/devtools/protocol/devtools_protocol_handler.h"
#include "content/public/browser/readback_types.h"

class SkBitmap;

namespace content {

class RenderViewHostImpl;

namespace devtools {
namespace page {

using EncodedFrame = std::pair<std::string, double>;

class FrameRecorder {
 public:
  using Response = DevToolsProtocolClient::Response;
  using StopRecordingFramesCallback =
      base::Callback<void(scoped_refptr<StopRecordingFramesResponse>)>;

  explicit FrameRecorder();
  virtual ~FrameRecorder();

  Response StartRecordingFrames(int max_frame_count);
  Response StopRecordingFrames(StopRecordingFramesCallback callback);
  Response CancelRecordingFrames();

  void SetRenderViewHost(RenderViewHostImpl* host);
  void OnSwapCompositorFrame();

 private:
  enum State {
    Ready,
    Recording,
    Encoding
  };

  void FrameCaptured(const SkBitmap& bitmap, ReadbackResponse response);
  void FrameEncoded(const scoped_ptr<EncodedFrame>& encoded_frame);
  void MaybeSendResponse();

  RenderViewHostImpl* host_;
  State state_;
  StopRecordingFramesCallback callback_;
  size_t inflight_requests_count_;
  size_t max_frame_count_;
  size_t captured_frames_count_;
  base::Time last_captured_frame_timestamp_;
  std::vector<scoped_refptr<devtools::page::RecordedFrame>> frames_;

  base::CancelableCallback<void(const scoped_ptr<EncodedFrame>&)>
      frame_encoded_callback_;
  base::WeakPtrFactory<FrameRecorder> weak_factory_;

  DISALLOW_COPY_AND_ASSIGN(FrameRecorder);
};

}  // namespace page
}  // namespace devtools
}  // namespace content

#endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_FRAME_RECORDER_H_