// 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/client/frame_evictor.h"

#include <algorithm>
#include <utility>

#include "base/check.h"
#include "base/feature_list.h"
#include "build/buildflag.h"
#include "components/viz/common/features.h"

namespace viz {

FrameEvictorClient::EvictIds::EvictIds() = default;
FrameEvictorClient::EvictIds::~EvictIds() = default;

FrameEvictorClient::EvictIds::EvictIds(EvictIds&& other) = default;
FrameEvictorClient::EvictIds& FrameEvictorClient::EvictIds::operator=(
    EvictIds&& other) = default;

FrameEvictor::FrameEvictor(FrameEvictorClient* client) : client_(client) {}

FrameEvictor::~FrameEvictor() {
  OnSurfaceDiscarded();
}

void FrameEvictor::OnNewSurfaceEmbedded() {
  has_surface_ = true;
  if (!opted_out_from_frame_eviction_) {
    FrameEvictionManager::GetInstance()->AddFrame(this, visible_);
  }
}

void FrameEvictor::OnSurfaceDiscarded() {
  if (!opted_out_from_frame_eviction_) {
    FrameEvictionManager::GetInstance()->RemoveFrame(this);
  }
  has_surface_ = false;
}

void FrameEvictor::SetVisible(bool visible) {
  if (visible_ == visible) {
    return;
  }
  visible_ = visible;
  if (has_surface_ && !opted_out_from_frame_eviction_) {
    if (visible) {
      FrameEvictionManager::GetInstance()->LockFrame(this);
    } else {
      FrameEvictionManager::GetInstance()->UnlockFrame(this);
    }
  }
}

void FrameEvictor::OptOutFrameEviction() {
  if (opted_out_from_frame_eviction_) {
    return;
  }
  opted_out_from_frame_eviction_ = true;
  if (has_surface_) {
    FrameEvictionManager::GetInstance()->RemoveFrame(this);
  }
}

std::vector<SurfaceId> FrameEvictor::CollectSurfaceIdsForEviction() const {
  auto ids = client_->CollectSurfaceIdsForEviction();
  std::vector<SurfaceId> output_ids = std::move(ids.embedded_ids);
  auto current = client_->GetCurrentSurfaceId();
  DCHECK(output_ids.empty() || !current.is_valid() ||
         std::ranges::contains(output_ids, current));

  if (output_ids.empty() && current.is_valid()) {
    output_ids.push_back(current);
  }

  auto pre_nav_surface_id = client_->GetPreNavigationSurfaceId();
  if (pre_nav_surface_id.is_valid()) {
    output_ids.push_back(pre_nav_surface_id);
  }

  std::ranges::sort(output_ids.begin(), output_ids.end());

  return output_ids;
}

void FrameEvictor::EvictCurrentFrame() {
  CHECK(!opted_out_from_frame_eviction_);
  client_->EvictDelegatedFrame(CollectSurfaceIdsForEviction());
}

}  // namespace viz