// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/viz/test/fake_surface_observer.h"

#include "components/viz/service/surfaces/surface_manager.h"
#include "ui/latency/latency_info.h"

namespace viz {

FakeSurfaceObserver::FakeSurfaceObserver(SurfaceManager* manager,
                                         bool damage_display)
    : damage_display_(damage_display) {
  observer_registration_.Observe(manager);
}

FakeSurfaceObserver::~FakeSurfaceObserver() = default;

void FakeSurfaceObserver::Reset() {
  last_ack_ = BeginFrameAck();
  damaged_surfaces_.clear();
  surface_subtree_damaged_.clear();
  last_surface_info_ = SurfaceInfo();
  last_created_surface_id_ = SurfaceId();
}

bool FakeSurfaceObserver::IsSurfaceDamaged(const SurfaceId& surface_id) const {
  return damaged_surfaces_.count(surface_id) > 0;
}

bool FakeSurfaceObserver::OnSurfaceDamaged(
    const SurfaceId& surface_id,
    const BeginFrameAck& ack,
    HandleInteraction handle_interaction,
    const std::vector<ui::LatencyInfo>& latency_info) {
  if (ack.has_damage)
    damaged_surfaces_.insert(surface_id);
  last_ack_ = ack;
  return ack.has_damage && damage_display_;
}

void FakeSurfaceObserver::OnFirstSurfaceActivation(
    const SurfaceInfo& surface_info) {
  last_created_surface_id_ = surface_info.id();
  last_surface_info_ = surface_info;
}

void FakeSurfaceObserver::OnSurfaceActivated(const SurfaceId& surface_id) {}

}  // namespace viz