diff options
Diffstat (limited to 'remoting/base/encoder_row_based.cc')
-rw-r--r-- | remoting/base/encoder_row_based.cc | 148 |
1 files changed, 148 insertions, 0 deletions
diff --git a/remoting/base/encoder_row_based.cc b/remoting/base/encoder_row_based.cc new file mode 100644 index 0000000..a7919e1 --- /dev/null +++ b/remoting/base/encoder_row_based.cc @@ -0,0 +1,148 @@ +// Copyright (c) 2010 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. + +#include "remoting/base/encoder_row_based.h" + +#include "base/logging.h" +#include "gfx/rect.h" +#include "remoting/base/capture_data.h" +#include "remoting/base/compressor_verbatim.h" +#include "remoting/base/compressor_zlib.h" +#include "remoting/base/util.h" +#include "remoting/proto/video.pb.h" + +namespace remoting { + +static const int kPacketSize = 1024 * 1024; + +EncoderRowBased* EncoderRowBased::CreateZlibEncoder() { + return new EncoderRowBased(new CompressorZlib(), + VideoPacketFormat::ENCODING_ZLIB); +} + +EncoderRowBased* EncoderRowBased::CreateVerbatimEncoder() { + return new EncoderRowBased(new CompressorVerbatim(), + VideoPacketFormat::ENCODING_VERBATIM); +} + +EncoderRowBased::EncoderRowBased(Compressor* compressor, + VideoPacketFormat::Encoding encoding) + : encoding_(encoding), + compressor_(compressor), + packet_size_(kPacketSize) { +} + +EncoderRowBased::EncoderRowBased(Compressor* compressor, + VideoPacketFormat::Encoding encoding, + int packet_size) + : encoding_(encoding), + compressor_(compressor), + packet_size_(packet_size) { +} + +EncoderRowBased::~EncoderRowBased() {} + +void EncoderRowBased::Encode(scoped_refptr<CaptureData> capture_data, + bool key_frame, + DataAvailableCallback* data_available_callback) { + CHECK(capture_data->pixel_format() == media::VideoFrame::RGB32) + << "RowBased Encoder only works with RGB32. Got " + << capture_data->pixel_format(); + capture_data_ = capture_data; + callback_.reset(data_available_callback); + + const InvalidRects& rects = capture_data->dirty_rects(); + int index = 0; + for (InvalidRects::const_iterator r = rects.begin(); + r != rects.end(); ++r, ++index) { + EncodeRect(*r, index); + } + + capture_data_ = NULL; + callback_.reset(); +} + +void EncoderRowBased::EncodeRect(const gfx::Rect& rect, size_t rect_index) { + CHECK(capture_data_->data_planes().data[0]); + const int strides = capture_data_->data_planes().strides[0]; + const int bytes_per_pixel = GetBytesPerPixel(capture_data_->pixel_format()); + const int row_size = bytes_per_pixel * rect.width(); + + compressor_->Reset(); + + VideoPacket* packet = new VideoPacket(); + PrepareUpdateStart(rect, packet); + const uint8* in = capture_data_->data_planes().data[0] + + rect.y() * strides + rect.x() * bytes_per_pixel; + // TODO(hclam): Fill in the sequence number. + uint8* out = GetOutputBuffer(packet, packet_size_); + int filled = 0; + int row_pos = 0; // Position in the current row in bytes. + int row_y = 0; // Current row. + bool compress_again = true; + while (compress_again) { + // Prepare a message for sending out. + if (!packet) { + packet = new VideoPacket(); + out = GetOutputBuffer(packet, packet_size_); + filled = 0; + } + + Compressor::CompressorFlush flush = Compressor::CompressorNoFlush; + if (row_y == rect.height() - 1) { + flush = Compressor::CompressorFinish; + } + + int consumed = 0; + int written = 0; + compress_again = compressor_->Process(in + row_pos, row_size - row_pos, + out + filled, packet_size_ - filled, + flush, &consumed, &written); + row_pos += consumed; + filled += written; + + // We have reached the end of stream. + if (!compress_again) { + packet->set_flags(packet->flags() | VideoPacket::LAST_PACKET); + DCHECK(row_pos == row_size); + DCHECK(row_y == rect.height() - 1); + } + + // If we have filled the message or we have reached the end of stream. + if (filled == packet_size_ || !compress_again) { + packet->mutable_data()->resize(filled); + callback_->Run(packet); + packet = NULL; + } + + // Reached the end of input row and we're not at the last row. + if (row_pos == row_size && row_y < rect.height() - 1) { + row_pos = 0; + in += strides; + ++row_y; + } + } +} + +void EncoderRowBased::PrepareUpdateStart(const gfx::Rect& rect, + VideoPacket* packet) { + packet->set_flags(packet->flags() | VideoPacket::FIRST_PACKET); + + VideoPacketFormat* format = packet->mutable_format(); + format->set_x(rect.x()); + format->set_y(rect.y()); + format->set_width(rect.width()); + format->set_height(rect.height()); + format->set_encoding(encoding_); +} + +uint8* EncoderRowBased::GetOutputBuffer(VideoPacket* packet, size_t size) { + packet->mutable_data()->resize(size); + // TODO(ajwong): Is there a better way to do this at all??? + return const_cast<uint8*>(reinterpret_cast<const uint8*>( + packet->mutable_data()->data())); +} + + +} // namespace remoting |