| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/base/encoder_verbatim.h" | 5 #include "remoting/base/encoder_verbatim.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "gfx/rect.h" | 8 #include "gfx/rect.h" |
| 9 #include "media/base/data_buffer.h" | 9 #include "media/base/data_buffer.h" |
| 10 #include "remoting/base/capture_data.h" | 10 #include "remoting/base/capture_data.h" |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 45 } | 45 } |
| 46 | 46 |
| 47 // TODO(garykac): This assumes that the rect fits into a single packet. | 47 // TODO(garykac): This assumes that the rect fits into a single packet. |
| 48 // Fix this by modeling after code in encoder_zlib.cc | 48 // Fix this by modeling after code in encoder_zlib.cc |
| 49 void EncoderVerbatim::EncodeRect(const gfx::Rect& rect, size_t rect_index) { | 49 void EncoderVerbatim::EncodeRect(const gfx::Rect& rect, size_t rect_index) { |
| 50 CHECK(capture_data_->data_planes().data[0]); | 50 CHECK(capture_data_->data_planes().data[0]); |
| 51 const int stride = capture_data_->data_planes().strides[0]; | 51 const int stride = capture_data_->data_planes().strides[0]; |
| 52 const int bytes_per_pixel = GetBytesPerPixel(capture_data_->pixel_format()); | 52 const int bytes_per_pixel = GetBytesPerPixel(capture_data_->pixel_format()); |
| 53 const int row_size = bytes_per_pixel * rect.width(); | 53 const int row_size = bytes_per_pixel * rect.width(); |
| 54 | 54 |
| 55 ChromotingHostMessage* message = new ChromotingHostMessage(); | 55 VideoPacket* packet = new VideoPacket(); |
| 56 RectangleUpdatePacket* update = message->mutable_rectangle_update(); | 56 PrepareUpdateStart(rect, packet); |
| 57 PrepareUpdateStart(rect, update); | |
| 58 | 57 |
| 59 const uint8* in = capture_data_->data_planes().data[0] + | 58 const uint8* in = capture_data_->data_planes().data[0] + |
| 60 rect.y() * stride + | 59 rect.y() * stride + |
| 61 rect.x() * bytes_per_pixel; | 60 rect.x() * bytes_per_pixel; |
| 62 // TODO(hclam): Fill in the sequence number. | 61 // TODO(hclam): Fill in the sequence number. |
| 63 uint8* out = GetOutputBuffer(update, packet_size_); | 62 uint8* out = GetOutputBuffer(packet, packet_size_); |
| 64 int total_bytes = 0; | 63 int total_bytes = 0; |
| 65 for (int y = 0; y < rect.height(); y++) { | 64 for (int y = 0; y < rect.height(); y++) { |
| 66 memcpy(out, in, row_size); | 65 memcpy(out, in, row_size); |
| 67 out += row_size; | 66 out += row_size; |
| 68 in += stride; | 67 in += stride; |
| 69 total_bytes += row_size; | 68 total_bytes += row_size; |
| 70 } | 69 } |
| 71 | 70 |
| 72 // We have reached the end of stream. | 71 // We have reached the end of stream. |
| 73 update->set_flags(update->flags() | RectangleUpdatePacket::LAST_PACKET); | 72 packet->set_flags(packet->flags() | VideoPacket::LAST_PACKET); |
| 74 | 73 |
| 75 // If we have filled the message or we have reached the end of stream. | 74 // If we have filled the message or we have reached the end of stream. |
| 76 message->mutable_rectangle_update()->mutable_encoded_rect()-> | 75 packet->mutable_data()->resize(total_bytes); |
| 77 resize(total_bytes); | 76 SubmitMessage(packet, rect_index); |
| 78 SubmitMessage(message, rect_index); | |
| 79 } | 77 } |
| 80 | 78 |
| 81 void EncoderVerbatim::PrepareUpdateStart(const gfx::Rect& rect, | 79 void EncoderVerbatim::PrepareUpdateStart(const gfx::Rect& rect, |
| 82 RectangleUpdatePacket* update) { | 80 VideoPacket* packet) { |
| 83 | 81 |
| 84 update->set_flags(update->flags() | RectangleUpdatePacket::FIRST_PACKET); | 82 packet->set_flags(packet->flags() | VideoPacket::FIRST_PACKET); |
| 85 RectangleFormat* format = update->mutable_format(); | 83 VideoPacketFormat* format = packet->mutable_format(); |
| 86 | 84 |
| 87 format->set_x(rect.x()); | 85 format->set_x(rect.x()); |
| 88 format->set_y(rect.y()); | 86 format->set_y(rect.y()); |
| 89 format->set_width(rect.width()); | 87 format->set_width(rect.width()); |
| 90 format->set_height(rect.height()); | 88 format->set_height(rect.height()); |
| 91 format->set_encoding(EncodingNone); | 89 format->set_encoding(VideoPacketFormat::ENCODING_VERBATIM); |
| 92 format->set_pixel_format(capture_data_->pixel_format()); | 90 format->set_pixel_format(capture_data_->pixel_format()); |
| 93 } | 91 } |
| 94 | 92 |
| 95 uint8* EncoderVerbatim::GetOutputBuffer(RectangleUpdatePacket* update, | 93 uint8* EncoderVerbatim::GetOutputBuffer(VideoPacket* packet, size_t size) { |
| 96 size_t size) { | 94 packet->mutable_data()->resize(size); |
| 97 update->mutable_encoded_rect()->resize(size); | |
| 98 // TODO(ajwong): Is there a better way to do this at all??? | 95 // TODO(ajwong): Is there a better way to do this at all??? |
| 99 return const_cast<uint8*>(reinterpret_cast<const uint8*>( | 96 return const_cast<uint8*>(reinterpret_cast<const uint8*>( |
| 100 update->mutable_encoded_rect()->data())); | 97 packet->mutable_data()->data())); |
| 101 } | 98 } |
| 102 | 99 |
| 103 void EncoderVerbatim::SubmitMessage(ChromotingHostMessage* message, | 100 void EncoderVerbatim::SubmitMessage(VideoPacket* packet, size_t rect_index) { |
| 104 size_t rect_index) { | 101 callback_->Run(packet); |
| 105 EncodingState state = EncodingInProgress; | |
| 106 const RectangleUpdatePacket& update = message->rectangle_update(); | |
| 107 if (rect_index == 0 && | |
| 108 (update.flags() | RectangleUpdatePacket::FIRST_PACKET)) { | |
| 109 state |= EncodingStarting; | |
| 110 } | |
| 111 if (rect_index == capture_data_->dirty_rects().size() - 1 && | |
| 112 (update.flags() | RectangleUpdatePacket::LAST_PACKET)) { | |
| 113 state |= EncodingEnded; | |
| 114 } | |
| 115 callback_->Run(message, state); | |
| 116 } | 102 } |
| 117 | 103 |
| 118 } // namespace remoting | 104 } // namespace remoting |
| OLD | NEW |