Remove ZLib codec support from chromoting host and client.

remoting/codec/video_encoder_verbatim.cc

 // Copyright (c) 2012 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/codec/video_encoder_row_based.h"
+#include "remoting/codec/video_encoder_verbatim.h"
 
 #include "base/logging.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;
 
-VideoEncoderRowBased* VideoEncoderRowBased::CreateZlibEncoder() {
-  return new VideoEncoderRowBased(new CompressorZlib(),
-                                  VideoPacketFormat::ENCODING_ZLIB);
+VideoEncoderVerbatim::VideoEncoderVerbatim()
+    : screen_size_(SkISize::Make(0,0)),
+      max_packet_size_(kPacketSize) {
 }
 
-VideoEncoderRowBased* VideoEncoderRowBased::CreateZlibEncoder(int packet_size) {
-  return new VideoEncoderRowBased(new CompressorZlib(),
-                                  VideoPacketFormat::ENCODING_ZLIB,
-                                  packet_size);
+void VideoEncoderVerbatim::SetMaxPacketSize(int size) {
+  max_packet_size_ = size;
 }
 
-VideoEncoderRowBased* VideoEncoderRowBased::CreateVerbatimEncoder() {
-  return new VideoEncoderRowBased(new CompressorVerbatim(),
-                                  VideoPacketFormat::ENCODING_VERBATIM);
+VideoEncoderVerbatim::~VideoEncoderVerbatim() {
 }
 
-VideoEncoderRowBased* VideoEncoderRowBased::CreateVerbatimEncoder(
-    int packet_size) {
-  return new VideoEncoderRowBased(new CompressorVerbatim(),
-                                  VideoPacketFormat::ENCODING_VERBATIM,
-                                  packet_size);
-}
-
-VideoEncoderRowBased::VideoEncoderRowBased(Compressor* compressor,
-                                           VideoPacketFormat::Encoding encoding)
-    : encoding_(encoding),
-      compressor_(compressor),
-      screen_size_(SkISize::Make(0,0)),
-      packet_size_(kPacketSize) {
-}
-
-VideoEncoderRowBased::VideoEncoderRowBased(Compressor* compressor,
-                                           VideoPacketFormat::Encoding encoding,
-                                           int packet_size)
-    : encoding_(encoding),
-      compressor_(compressor),
-      screen_size_(SkISize::Make(0,0)),
-      packet_size_(packet_size) {
-}
-
-VideoEncoderRowBased::~VideoEncoderRowBased() {}
-
-void VideoEncoderRowBased::Encode(
+void VideoEncoderVerbatim::Encode(
     scoped_refptr<CaptureData> capture_data,
     bool key_frame,
     const DataAvailableCallback& data_available_callback) {
   CHECK(capture_data->pixel_format() == media::VideoFrame::RGB32)
       << "RowBased VideoEncoder only works with RGB32. Got "
       << capture_data->pixel_format();
   capture_data_ = capture_data;
   callback_ = data_available_callback;
 
   const SkRegion& region = capture_data->dirty_region();
   SkRegion::Iterator iter(region);
   while (!iter.done()) {
     SkIRect rect = iter.rect();
     iter.next();
     EncodeRect(rect, iter.done());
   }
 
   capture_data_ = NULL;
   callback_.Reset();
 }
 
-void VideoEncoderRowBased::EncodeRect(const SkIRect& rect, bool last) {
+void VideoEncoderVerbatim::EncodeRect(const SkIRect& rect, bool last) {
   CHECK(capture_data_->data_planes().data[0]);
   CHECK_EQ(capture_data_->pixel_format(), media::VideoFrame::RGB32);
   const int strides = capture_data_->data_planes().strides[0];
   const int bytes_per_pixel = 4;
   const int row_size = bytes_per_pixel * rect.width();
 
-  compressor_->Reset();
-
   scoped_ptr<VideoPacket> packet(new VideoPacket());
   PrepareUpdateStart(rect, packet.get());
   const uint8* in = capture_data_->data_planes().data[0] +
       rect.fTop * strides + rect.fLeft * bytes_per_pixel;
   // TODO(hclam): Fill in the sequence number.
-  uint8* out = GetOutputBuffer(packet.get(), packet_size_);
+  uint8* out = GetOutputBuffer(packet.get(), max_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) {
+  while (row_y < rect.height()) {
     // Prepare a message for sending out.
     if (!packet.get()) {
       packet.reset(new VideoPacket());
-      out = GetOutputBuffer(packet.get(), packet_size_);
+      out = GetOutputBuffer(packet.get(), max_packet_size_);
       filled = 0;
     }
 
-    Compressor::CompressorFlush flush = Compressor::CompressorNoFlush;
-    if (row_y == rect.height() - 1) {
-      flush = Compressor::CompressorFinish;
+    if (row_y < rect.height()) {
+      int bytes_to_copy =
+          std::min(row_size - row_pos, max_packet_size_ - filled);
+      memcpy(out + filled, in + row_pos, bytes_to_copy);
+      row_pos += bytes_to_copy;
+      filled += bytes_to_copy;
+
+      // Jump to the next row when we've reached the end of the current row.
+      if (row_pos == row_size) {
+        row_pos = 0;
+        in += strides;
+        ++row_y;
+      }
     }
 
-    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;
+    if (row_y == rect.height()) {
+      DCHECK_EQ(row_pos, 0);
 
-    // We have reached the end of stream.
-    if (!compress_again) {
+      packet->mutable_data()->resize(filled);
       packet->set_flags(packet->flags() | VideoPacket::LAST_PACKET);
       packet->set_capture_time_ms(capture_data_->capture_time_ms());
       packet->set_client_sequence_number(
           capture_data_->client_sequence_number());
       SkIPoint dpi(capture_data_->dpi());
       if (dpi.x())
         packet->mutable_format()->set_x_dpi(dpi.x());
       if (dpi.y())
         packet->mutable_format()->set_y_dpi(dpi.y());
       if (last)
         packet->set_flags(packet->flags() | VideoPacket::LAST_PARTITION);
-      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) {
+    // If we have filled the current packet, then send it.
+    if (filled == max_packet_size_ || row_y == rect.height()) {
       packet->mutable_data()->resize(filled);
       callback_.Run(packet.Pass());
     }
-
-    // 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 VideoEncoderRowBased::PrepareUpdateStart(const SkIRect& rect,
+void VideoEncoderVerbatim::PrepareUpdateStart(const SkIRect& rect,
                                               VideoPacket* packet) {
   packet->set_flags(packet->flags() | VideoPacket::FIRST_PACKET);
 
   VideoPacketFormat* format = packet->mutable_format();
   format->set_x(rect.fLeft);
   format->set_y(rect.fTop);
   format->set_width(rect.width());
   format->set_height(rect.height());
-  format->set_encoding(encoding_);
+  format->set_encoding(VideoPacketFormat::ENCODING_VERBATIM);
   if (capture_data_->size() != screen_size_) {
     screen_size_ = capture_data_->size();
     format->set_screen_width(screen_size_.width());
     format->set_screen_height(screen_size_.height());
   }
 }
 
-uint8* VideoEncoderRowBased::GetOutputBuffer(VideoPacket* packet, size_t size) {
+uint8* VideoEncoderVerbatim::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