// 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/host/video_frame_capturer.h"

#include <ApplicationServices/ApplicationServices.h>

#include <ostream>

#include "base/bind.h"
#include "base/callback.h"
#include "base/memory/scoped_ptr.h"
#include "remoting/base/capture_data.h"
#include "remoting/proto/control.pb.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace remoting {

// Verify that the OS is at least Snow Leopard (10.6).
// Chromoting doesn't support 10.5 or earlier.
bool CheckSnowLeopard() {
  long minorVersion, majorVersion;
  Gestalt(gestaltSystemVersionMajor, &majorVersion);
  Gestalt(gestaltSystemVersionMinor, &minorVersion);
  return majorVersion == 10 && minorVersion > 5;
}

class VideoFrameCapturerMacTest : public testing::Test {
 protected:
  virtual void SetUp() OVERRIDE {
    capturer_.reset(VideoFrameCapturer::Create());
  }

  void AddDirtyRect() {
    SkIRect rect = SkIRect::MakeXYWH(0, 0, 10, 10);
    region_.op(rect, SkRegion::kUnion_Op);
  }

  scoped_ptr<VideoFrameCapturer> capturer_;
  SkRegion region_;
};

// CapturerCallback1 verifies that the whole screen is initially dirty.
class VideoFrameCapturerCallback1 {
 public:
  VideoFrameCapturerCallback1() {}

  void CaptureDoneCallback(scoped_refptr<CaptureData> capture_data);

 private:
  DISALLOW_COPY_AND_ASSIGN(VideoFrameCapturerCallback1);
};

void VideoFrameCapturerCallback1::CaptureDoneCallback(
    scoped_refptr<CaptureData> capture_data) {
  CGDirectDisplayID mainDevice = CGMainDisplayID();
  int width = CGDisplayPixelsWide(mainDevice);
  int height = CGDisplayPixelsHigh(mainDevice);
  SkRegion initial_region(SkIRect::MakeXYWH(0, 0, width, height));
  EXPECT_EQ(initial_region, capture_data->dirty_region());
}

// VideoFrameCapturerCallback2 verifies that a rectangle explicitly marked as
// dirty is propagated correctly.
class VideoFrameCapturerCallback2 {
 public:
  explicit VideoFrameCapturerCallback2(const SkRegion& expected_dirty_region)
      : expected_dirty_region_(expected_dirty_region) {}

  void CaptureDoneCallback(scoped_refptr<CaptureData> capture_data);

 protected:
  SkRegion expected_dirty_region_;

 private:
  DISALLOW_COPY_AND_ASSIGN(VideoFrameCapturerCallback2);
};

void VideoFrameCapturerCallback2::CaptureDoneCallback(
    scoped_refptr<CaptureData> capture_data) {
  CGDirectDisplayID mainDevice = CGMainDisplayID();
  int width = CGDisplayPixelsWide(mainDevice);
  int height = CGDisplayPixelsHigh(mainDevice);

  EXPECT_EQ(expected_dirty_region_, capture_data->dirty_region());
  EXPECT_EQ(width, capture_data->size().width());
  EXPECT_EQ(height, capture_data->size().height());
  const DataPlanes &planes = capture_data->data_planes();
  EXPECT_TRUE(planes.data[0] != NULL);
  EXPECT_TRUE(planes.data[1] == NULL);
  EXPECT_TRUE(planes.data[2] == NULL);
  // Depending on the capture method, the screen may be flipped or not, so
  // the stride may be positive or negative.
  EXPECT_EQ(static_cast<int>(sizeof(uint32_t) * width),
            abs(planes.strides[0]));
  EXPECT_EQ(0, planes.strides[1]);
  EXPECT_EQ(0, planes.strides[2]);
}

class CursorCallback {
 public:
  CursorCallback() {}

  void CursorShapeChangedCallback(
      scoped_ptr<protocol::CursorShapeInfo> cursor_data);

 private:
  DISALLOW_COPY_AND_ASSIGN(CursorCallback);
};

void CursorCallback::CursorShapeChangedCallback(
    scoped_ptr<protocol::CursorShapeInfo> cursor_data) {
}

TEST_F(VideoFrameCapturerMacTest, Capture) {
  if (!CheckSnowLeopard()) {
    return;
  }

  SCOPED_TRACE("");
  CursorCallback cursor_callback;
  capturer_->Start(base::Bind(&CursorCallback::CursorShapeChangedCallback,
                              base::Unretained(&cursor_callback)));
  // Check that we get an initial full-screen updated.
  VideoFrameCapturerCallback1 callback1;
  capturer_->CaptureInvalidRegion(base::Bind(
      &VideoFrameCapturerCallback1::CaptureDoneCallback,
      base::Unretained(&callback1)));
  // Check that subsequent dirty rects are propagated correctly.
  AddDirtyRect();
  VideoFrameCapturerCallback2 callback2(region_);
  capturer_->InvalidateRegion(region_);
  capturer_->CaptureInvalidRegion(base::Bind(
      &VideoFrameCapturerCallback2::CaptureDoneCallback,
      base::Unretained(&callback2)));
  capturer_->Stop();
}

}  // namespace remoting

namespace gfx {

std::ostream& operator<<(std::ostream& out, const SkRegion& region) {
  out << "SkRegion(";
  for (SkRegion::Iterator i(region); !i.done(); i.next()) {
    const SkIRect& r = i.rect();
    out << "(" << r.fLeft << ","  << r.fTop << ","
        << r.fRight  << ","  << r.fBottom << ")";
  }
  out << ")";
  return out;
}

}  // namespace gfx