// Copyright 2014 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/host/shaped_desktop_capturer.h"

#include <utility>

#include "base/logging.h"
#include "remoting/host/desktop_shape_tracker.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_frame.h"

namespace remoting {

ShapedDesktopCapturer::ShapedDesktopCapturer(
    scoped_ptr<webrtc::DesktopCapturer> desktop_capturer,
    scoped_ptr<DesktopShapeTracker> shape_tracker)
    : desktop_capturer_(std::move(desktop_capturer)),
      shape_tracker_(std::move(shape_tracker)),
      callback_(nullptr) {}

ShapedDesktopCapturer::~ShapedDesktopCapturer() {}

void ShapedDesktopCapturer::Start(webrtc::DesktopCapturer::Callback* callback) {
  callback_ = callback;
  desktop_capturer_->Start(this);
}

void ShapedDesktopCapturer::Capture(const webrtc::DesktopRegion& region) {
  desktop_capturer_->Capture(region);
}

void ShapedDesktopCapturer::SetSharedMemoryFactory(
    rtc::scoped_ptr<webrtc::SharedMemoryFactory> shared_memory_factory) {
  desktop_capturer_->SetSharedMemoryFactory(std::move(shared_memory_factory));
}

void ShapedDesktopCapturer::OnCaptureCompleted(webrtc::DesktopFrame* frame) {
  shape_tracker_->RefreshDesktopShape();
  frame->set_shape(new webrtc::DesktopRegion(shape_tracker_->desktop_shape()));
  callback_->OnCaptureCompleted(frame);
}

}  // namespace remoting