| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/protocol/protobuf_video_writer.h" | 5 #include "remoting/protocol/protobuf_video_writer.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/task.h" | 8 #include "base/task.h" |
| 9 #include "net/socket/stream_socket.h" | 9 #include "net/socket/stream_socket.h" |
| 10 #include "remoting/base/constants.h" | 10 #include "remoting/base/constants.h" |
| 11 #include "remoting/proto/video.pb.h" | 11 #include "remoting/proto/video.pb.h" |
| 12 #include "remoting/protocol/rtp_writer.h" | 12 #include "remoting/protocol/rtp_writer.h" |
| 13 #include "remoting/protocol/session.h" | 13 #include "remoting/protocol/session.h" |
| 14 #include "remoting/protocol/util.h" | 14 #include "remoting/protocol/util.h" |
| 15 | 15 |
| 16 namespace remoting { | 16 namespace remoting { |
| 17 namespace protocol { | 17 namespace protocol { |
| 18 | 18 |
| 19 ProtobufVideoWriter::ProtobufVideoWriter() { } | 19 ProtobufVideoWriter::ProtobufVideoWriter(base::MessageLoopProxy* message_loop) |
| 20 : buffered_writer_(new BufferedSocketWriter(message_loop)) { |
| 21 } |
| 20 | 22 |
| 21 ProtobufVideoWriter::~ProtobufVideoWriter() { } | 23 ProtobufVideoWriter::~ProtobufVideoWriter() { } |
| 22 | 24 |
| 23 void ProtobufVideoWriter::Init(protocol::Session* session, | 25 void ProtobufVideoWriter::Init(protocol::Session* session, |
| 24 const InitializedCallback& callback) { | 26 const InitializedCallback& callback) { |
| 25 initialized_callback_ = callback; | 27 initialized_callback_ = callback; |
| 26 | 28 |
| 27 session->CreateStreamChannel( | 29 session->CreateStreamChannel( |
| 28 kVideoChannelName, | 30 kVideoChannelName, |
| 29 base::Bind(&ProtobufVideoWriter::OnChannelReady, base::Unretained(this))); | 31 base::Bind(&ProtobufVideoWriter::OnChannelReady, base::Unretained(this))); |
| 30 } | 32 } |
| 31 | 33 |
| 32 void ProtobufVideoWriter::OnChannelReady(net::StreamSocket* socket) { | 34 void ProtobufVideoWriter::OnChannelReady(net::StreamSocket* socket) { |
| 33 if (!socket) { | 35 if (!socket) { |
| 34 initialized_callback_.Run(false); | 36 initialized_callback_.Run(false); |
| 35 return; | 37 return; |
| 36 } | 38 } |
| 37 | 39 |
| 38 DCHECK(!channel_.get()); | 40 DCHECK(!channel_.get()); |
| 39 channel_.reset(socket); | 41 channel_.reset(socket); |
| 40 buffered_writer_ = new BufferedSocketWriter(); | |
| 41 // TODO(sergeyu): Provide WriteFailedCallback for the buffered writer. | 42 // TODO(sergeyu): Provide WriteFailedCallback for the buffered writer. |
| 42 buffered_writer_->Init(socket, NULL); | 43 buffered_writer_->Init(socket, NULL); |
| 43 | 44 |
| 44 initialized_callback_.Run(true); | 45 initialized_callback_.Run(true); |
| 45 } | 46 } |
| 46 | 47 |
| 47 void ProtobufVideoWriter::Close() { | 48 void ProtobufVideoWriter::Close() { |
| 48 buffered_writer_->Close(); | 49 buffered_writer_->Close(); |
| 49 channel_.reset(); | 50 channel_.reset(); |
| 50 } | 51 } |
| 51 | 52 |
| 52 void ProtobufVideoWriter::ProcessVideoPacket(const VideoPacket* packet, | 53 void ProtobufVideoWriter::ProcessVideoPacket(const VideoPacket* packet, |
| 53 Task* done) { | 54 Task* done) { |
| 54 buffered_writer_->Write(SerializeAndFrameMessage(*packet), done); | 55 buffered_writer_->Write(SerializeAndFrameMessage(*packet), done); |
| 55 } | 56 } |
| 56 | 57 |
| 57 int ProtobufVideoWriter::GetPendingPackets() { | 58 int ProtobufVideoWriter::GetPendingPackets() { |
| 58 return buffered_writer_->GetBufferChunks(); | 59 return buffered_writer_->GetBufferChunks(); |
| 59 } | 60 } |
| 60 | 61 |
| 61 } // namespace protocol | 62 } // namespace protocol |
| 62 } // namespace remoting | 63 } // namespace remoting |
| OLD | NEW |