// Copyright 2015 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 "cc/output/compositor_frame.h"
#include "cc/output/copy_output_result.h"
#include "cc/output/delegated_frame_data.h"
#include "cc/quads/render_pass.h"
#include "cc/resources/shared_bitmap_manager.h"
#include "cc/surfaces/display.h"
#include "cc/surfaces/display_client.h"
#include "cc/surfaces/surface.h"
#include "cc/surfaces/surface_factory.h"
#include "cc/surfaces/surface_factory_client.h"
#include "cc/surfaces/surface_id_allocator.h"
#include "cc/surfaces/surface_manager.h"
#include "cc/test/fake_output_surface.h"
#include "cc/test/test_shared_bitmap_manager.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace cc {
namespace {

class EmptySurfaceFactoryClient : public SurfaceFactoryClient {
 public:
  void ReturnResources(const ReturnedResourceArray& resources) override {}
};

class DisplayTest : public testing::Test {
 public:
  DisplayTest() : factory_(&manager_, &empty_client_) {}

  void SetUp() override {
    output_surface_ = FakeOutputSurface::CreateSoftware(
        make_scoped_ptr(new SoftwareOutputDevice));
    shared_bitmap_manager_.reset(new TestSharedBitmapManager);
    output_surface_ptr_ = output_surface_.get();
  }

 protected:
  void SubmitFrame(RenderPassList* pass_list, SurfaceId surface_id) {
    scoped_ptr<DelegatedFrameData> frame_data(new DelegatedFrameData);
    pass_list->swap(frame_data->render_pass_list);

    scoped_ptr<CompositorFrame> frame(new CompositorFrame);
    frame->delegated_frame_data = frame_data.Pass();

    factory_.SubmitFrame(surface_id, frame.Pass(),
                         SurfaceFactory::DrawCallback());
  }

  SurfaceManager manager_;
  EmptySurfaceFactoryClient empty_client_;
  SurfaceFactory factory_;
  scoped_ptr<FakeOutputSurface> output_surface_;
  FakeOutputSurface* output_surface_ptr_;
  scoped_ptr<SharedBitmapManager> shared_bitmap_manager_;
};

class TestDisplayClient : public DisplayClient {
 public:
  TestDisplayClient() : damaged(false), swapped(false) {}
  ~TestDisplayClient() override {}

  void DisplayDamaged() override { damaged = true; }
  void DidSwapBuffers() override { swapped = true; }
  void DidSwapBuffersComplete() override {}
  void CommitVSyncParameters(base::TimeTicks timebase,
                             base::TimeDelta interval) override {}
  void OutputSurfaceLost() override {}
  void SetMemoryPolicy(const ManagedMemoryPolicy& policy) override {}

  bool damaged;
  bool swapped;
};

void CopyCallback(bool* called, scoped_ptr<CopyOutputResult> result) {
  *called = true;
}

// Check that frame is damaged and swapped only under correct conditions.
TEST_F(DisplayTest, DisplayDamaged) {
  TestDisplayClient client;
  RendererSettings settings;
  settings.partial_swap_enabled = true;
  Display display(&client, &manager_, shared_bitmap_manager_.get(), nullptr,
                  settings);

  display.Initialize(output_surface_.Pass());

  SurfaceId surface_id(7u);
  EXPECT_FALSE(client.damaged);
  display.SetSurfaceId(surface_id, 1.f);
  EXPECT_TRUE(client.damaged);

  client.damaged = false;
  display.Resize(gfx::Size(100, 100));
  EXPECT_TRUE(client.damaged);

  factory_.Create(surface_id);

  // First draw from surface should have full damage.
  RenderPassList pass_list;
  scoped_ptr<RenderPass> pass = RenderPass::Create();
  pass->output_rect = gfx::Rect(0, 0, 100, 100);
  pass->damage_rect = gfx::Rect(10, 10, 1, 1);
  pass->id = RenderPassId(1, 1);
  pass_list.push_back(pass.Pass());

  client.damaged = false;
  SubmitFrame(&pass_list, surface_id);
  EXPECT_TRUE(client.damaged);

  EXPECT_FALSE(client.swapped);
  EXPECT_EQ(0u, output_surface_ptr_->num_sent_frames());
  display.Draw();
  EXPECT_TRUE(client.swapped);
  EXPECT_EQ(1u, output_surface_ptr_->num_sent_frames());
  SoftwareFrameData* software_data =
      output_surface_ptr_->last_sent_frame().software_frame_data.get();
  ASSERT_NE(nullptr, software_data);
  EXPECT_EQ(gfx::Size(100, 100).ToString(), software_data->size.ToString());
  EXPECT_EQ(gfx::Rect(0, 0, 100, 100).ToString(),
            software_data->damage_rect.ToString());

  {
    // Only damaged portion should be swapped.
    pass = RenderPass::Create();
    pass->output_rect = gfx::Rect(0, 0, 100, 100);
    pass->damage_rect = gfx::Rect(10, 10, 1, 1);
    pass->id = RenderPassId(1, 1);

    pass_list.push_back(pass.Pass());
    client.damaged = false;
    SubmitFrame(&pass_list, surface_id);
    EXPECT_TRUE(client.damaged);

    client.swapped = false;
    display.Draw();
    EXPECT_TRUE(client.swapped);
    EXPECT_EQ(2u, output_surface_ptr_->num_sent_frames());
    software_data =
        output_surface_ptr_->last_sent_frame().software_frame_data.get();
    ASSERT_NE(nullptr, software_data);
    EXPECT_EQ(gfx::Size(100, 100).ToString(), software_data->size.ToString());
    EXPECT_EQ(gfx::Rect(10, 10, 1, 1).ToString(),
              software_data->damage_rect.ToString());
  }

  {
    // Pass has no damage so shouldn't be swapped.
    pass = RenderPass::Create();
    pass->output_rect = gfx::Rect(0, 0, 100, 100);
    pass->damage_rect = gfx::Rect(10, 10, 0, 0);
    pass->id = RenderPassId(1, 1);

    pass_list.push_back(pass.Pass());
    client.damaged = false;
    SubmitFrame(&pass_list, surface_id);
    EXPECT_TRUE(client.damaged);

    client.swapped = false;
    display.Draw();
    EXPECT_TRUE(client.swapped);
    EXPECT_EQ(2u, output_surface_ptr_->num_sent_frames());
  }

  {
    // Pass is wrong size so shouldn't be swapped.
    pass = RenderPass::Create();
    pass->output_rect = gfx::Rect(0, 0, 99, 99);
    pass->damage_rect = gfx::Rect(10, 10, 10, 10);
    pass->id = RenderPassId(1, 1);

    pass_list.push_back(pass.Pass());
    client.damaged = false;
    SubmitFrame(&pass_list, surface_id);
    EXPECT_TRUE(client.damaged);

    client.swapped = false;
    display.Draw();
    EXPECT_TRUE(client.swapped);
    EXPECT_EQ(2u, output_surface_ptr_->num_sent_frames());
  }

  {
    // Pass has copy output request so should be swapped.
    pass = RenderPass::Create();
    pass->output_rect = gfx::Rect(0, 0, 100, 100);
    pass->damage_rect = gfx::Rect(10, 10, 0, 0);
    bool copy_called = false;
    pass->copy_requests.push_back(CopyOutputRequest::CreateRequest(
        base::Bind(&CopyCallback, &copy_called)));
    pass->id = RenderPassId(1, 1);

    pass_list.push_back(pass.Pass());
    client.damaged = false;
    SubmitFrame(&pass_list, surface_id);
    EXPECT_TRUE(client.damaged);

    client.swapped = false;
    display.Draw();
    EXPECT_TRUE(client.swapped);
    EXPECT_EQ(3u, output_surface_ptr_->num_sent_frames());
    EXPECT_TRUE(copy_called);
  }

  // Pass has latency info so should be swapped.
  {
    pass = RenderPass::Create();
    pass->output_rect = gfx::Rect(0, 0, 100, 100);
    pass->damage_rect = gfx::Rect(10, 10, 0, 0);
    pass->id = RenderPassId(1, 1);

    pass_list.push_back(pass.Pass());
    client.damaged = false;
    scoped_ptr<DelegatedFrameData> frame_data(new DelegatedFrameData);
    pass_list.swap(frame_data->render_pass_list);

    scoped_ptr<CompositorFrame> frame(new CompositorFrame);
    frame->delegated_frame_data = frame_data.Pass();
    frame->metadata.latency_info.push_back(ui::LatencyInfo());

    factory_.SubmitFrame(surface_id, frame.Pass(),
                         SurfaceFactory::DrawCallback());
    EXPECT_TRUE(client.damaged);

    client.swapped = false;
    display.Draw();
    EXPECT_TRUE(client.swapped);
    EXPECT_EQ(4u, output_surface_ptr_->num_sent_frames());
  }

  factory_.Destroy(surface_id);
}

}  // namespace
}  // namespace cc