1// Copyright (c) 2012 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/codec/video_encoder_verbatim.h" 6 7#include "base/logging.h" 8#include "base/stl_util.h" 9#include "base/time/time.h" 10#include "remoting/base/util.h" 11#include "remoting/proto/video.pb.h" 12#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h" 13#include "third_party/webrtc/modules/desktop_capture/desktop_geometry.h" 14#include "third_party/webrtc/modules/desktop_capture/desktop_region.h" 15 16namespace remoting { 17 18static uint8_t* GetPacketOutputBuffer(VideoPacket* packet, size_t size) { 19 packet->mutable_data()->resize(size); 20 return reinterpret_cast<uint8_t*>(string_as_array(packet->mutable_data())); 21} 22 23VideoEncoderVerbatim::VideoEncoderVerbatim() {} 24VideoEncoderVerbatim::~VideoEncoderVerbatim() {} 25 26scoped_ptr<VideoPacket> VideoEncoderVerbatim::Encode( 27 const webrtc::DesktopFrame& frame) { 28 CHECK(frame.data()); 29 30 base::Time encode_start_time = base::Time::Now(); 31 32 // Create a VideoPacket with common fields (e.g. DPI, rects, shape) set. 33 scoped_ptr<VideoPacket> packet(helper_.CreateVideoPacket(frame)); 34 packet->mutable_format()->set_encoding(VideoPacketFormat::ENCODING_VERBATIM); 35 36 // Calculate output size. 37 size_t output_size = 0; 38 for (webrtc::DesktopRegion::Iterator iter(frame.updated_region()); 39 !iter.IsAtEnd(); iter.Advance()) { 40 const webrtc::DesktopRect& rect = iter.rect(); 41 output_size += rect.width() * rect.height() * 42 webrtc::DesktopFrame::kBytesPerPixel; 43 } 44 45 uint8_t* out = GetPacketOutputBuffer(packet.get(), output_size); 46 const int in_stride = frame.stride(); 47 48 // Encode pixel data for all changed rectangles into the packet. 49 for (webrtc::DesktopRegion::Iterator iter(frame.updated_region()); 50 !iter.IsAtEnd(); iter.Advance()) { 51 const webrtc::DesktopRect& rect = iter.rect(); 52 const int row_size = webrtc::DesktopFrame::kBytesPerPixel * rect.width(); 53 const uint8_t* in = frame.data() + rect.top() * in_stride + 54 rect.left() * webrtc::DesktopFrame::kBytesPerPixel; 55 for (int y = rect.top(); y < rect.top() + rect.height(); ++y) { 56 memcpy(out, in, row_size); 57 out += row_size; 58 in += in_stride; 59 } 60 } 61 62 // Note the time taken to encode the pixel data. 63 packet->set_encode_time_ms( 64 (base::Time::Now() - encode_start_time).InMillisecondsRoundedUp()); 65 66 return packet.Pass(); 67} 68 69} // namespace remoting 70