OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "remoting/ios/bridge/frame_consumer_bridge.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/logging.h" | |
9 #include "base/synchronization/waitable_event.h" | |
10 #include "remoting/base/util.h" | |
11 #include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" | |
12 | |
13 namespace remoting { | |
14 | |
15 FrameConsumerBridge::FrameConsumerBridge(OnFrameCallback callback) | |
16 : callback_(callback), frame_producer_(NULL) {} | |
17 | |
18 FrameConsumerBridge::~FrameConsumerBridge() { | |
19 // The producer should now return any pending buffers. At this point, however, | |
20 // the buffers are returned via tasks which may not be scheduled before the | |
21 // producer, so we free all the buffers once the producer's queue is empty. | |
22 // And the scheduled tasks will die quietly. | |
23 if (frame_producer_) { | |
24 base::WaitableEvent done_event(true, false); | |
25 frame_producer_->RequestReturnBuffers(base::Bind( | |
26 &base::WaitableEvent::Signal, base::Unretained(&done_event))); | |
27 done_event.Wait(); | |
28 } | |
29 } | |
30 | |
31 void FrameConsumerBridge::Initialize(FrameProducer* producer) { | |
32 DCHECK(!frame_producer_); | |
33 frame_producer_ = producer; | |
34 DCHECK(frame_producer_); | |
35 } | |
36 | |
37 void FrameConsumerBridge::ApplyBuffer(const webrtc::DesktopSize& view_size, | |
38 const webrtc::DesktopRect& clip_area, | |
39 webrtc::DesktopFrame* buffer, | |
40 const webrtc::DesktopRegion& region, | |
41 const webrtc::DesktopRegion& shape) { | |
42 DCHECK(frame_producer_); | |
43 if (!view_size_.equals(view_size)) { | |
44 // Drop the frame, since the data belongs to the previous generation, | |
45 // before SetSourceSize() called SetOutputSizeAndClip(). | |
46 ReturnBuffer(buffer); | |
47 return; | |
48 } | |
49 | |
50 // This call completes synchronously. | |
51 callback_.Run(view_size, buffer, region); | |
52 | |
53 // Recycle |buffer| by returning it to |frame_producer_| as the next buffer | |
54 frame_producer_->DrawBuffer(buffer); | |
55 } | |
56 | |
57 void FrameConsumerBridge::ReturnBuffer(webrtc::DesktopFrame* buffer) { | |
58 DCHECK(frame_producer_); | |
59 ScopedVector<webrtc::DesktopFrame>::iterator it = | |
60 std::find(buffers_.begin(), buffers_.end(), buffer); | |
61 | |
62 DCHECK(it != buffers_.end()); | |
63 buffers_.erase(it); | |
64 } | |
65 | |
66 void FrameConsumerBridge::SetSourceSize(const webrtc::DesktopSize& source_size, | |
67 const webrtc::DesktopVector& dpi) { | |
68 DCHECK(frame_producer_); | |
69 view_size_ = source_size; | |
70 webrtc::DesktopRect clip_area = webrtc::DesktopRect::MakeSize(view_size_); | |
71 frame_producer_->SetOutputSizeAndClip(view_size_, clip_area); | |
72 | |
73 // Now that the size is well known, ask the producer to start drawing | |
74 DrawWithNewBuffer(); | |
75 } | |
76 | |
77 FrameConsumerBridge::PixelFormat FrameConsumerBridge::GetPixelFormat() { | |
78 return FORMAT_RGBA; | |
79 } | |
80 | |
81 void FrameConsumerBridge::DrawWithNewBuffer() { | |
82 DCHECK(frame_producer_); | |
83 webrtc::DesktopFrame* buffer = new webrtc::BasicDesktopFrame(view_size_); | |
84 buffers_.push_back(buffer); | |
85 frame_producer_->DrawBuffer(buffer); | |
86 } | |
87 | |
88 } // namespace remoting | |
OLD | NEW |