/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#define LOG_TAG "drmhwc"

#include "Backend.h"

#include <climits>

#include "BackendManager.h"
#include "bufferinfo/BufferInfoGetter.h"
#include "drm/DrmHwc.h"

namespace android {

namespace {

HwcLayer *GetCursorLayer(const std::vector<HwcLayer *> &layers) {
  auto it = std::find_if(layers.begin(), layers.end(),
                         [&](auto *layer) -> bool {
                           return layer->GetSfType() ==
                                  HwcLayer::CompositionType::kCursor;
                         });
  if (it == layers.end()) {
    return nullptr;
  }
  return *it;
}

}  // namespace

void Backend::ValidateDisplay(HwcDisplay *display) {
  auto layers = display->GetOrderLayersByZPos();

  auto flatcon = display->GetFlatCon();
  if (flatcon) {
    bool should_flatten = false;
    if (layers.size() <= 1)
      flatcon->Disable();
    else
      should_flatten = flatcon->NewFrame();

    if (should_flatten) {
      display->total_stats().frames_flattened++;
      MarkValidated(layers, 0, layers.size(), /*use_cursor_plane=*/false);
      return;
    }
  }

  int client_start = -1;
  size_t client_size = 0;
  auto *cursor_layer = GetCursorLayer(layers);
  auto cursor_plane = display->GetPipe().GetUsablePlanes().second;
  bool use_cursor_plane = cursor_layer != nullptr && cursor_plane != nullptr &&
                          !IsClientLayer(display, cursor_layer) &&
                          cursor_plane->Get()->IsValidForLayer(
                              &cursor_layer->GetLayerData());

  // Validates layers and creates a test composition, returning whether it
  // succeeded.
  auto validate_and_test = [&]() -> bool {
    std::tie(client_start, client_size) = GetClientLayers(display, layers,
                                                          use_cursor_plane);
    MarkValidated(layers, client_start, client_size, use_cursor_plane);

    bool testing_needed = client_start != 0 || client_size != layers.size();
    AtomicCommitArgs a_args = {.test_only = true};

    if (testing_needed) {
      return display->CreateComposition(a_args);
    }

    return true;
  };

  // Initial composition attempt.
  bool success = validate_and_test();

  // First fallback: convert cursor layer to device composition and reattempt.
  if (!success && use_cursor_plane) {
    ++display->total_stats().failed_kms_cursor_validate;
    use_cursor_plane = false;
    success = validate_and_test();
  }

  // Final fallback: convert all layers to client composition.
  if (!success) {
    ++display->total_stats().failed_kms_validate;
    client_start = 0;
    client_size = layers.size();
    MarkValidated(layers, client_start, client_size, use_cursor_plane);
  }

  display->total_stats().gpu_pixops += CalcPixOps(layers, client_start,
                                                  client_size);
  display->total_stats().total_pixops += CalcPixOps(layers, 0, layers.size());
  if (use_cursor_plane) {
    ++display->total_stats().cursor_plane_frames;
  }
}

std::tuple<int, size_t> Backend::GetClientLayers(
    HwcDisplay *display, const std::vector<HwcLayer *> &layers,
    bool use_cursor_plane) {
  int client_start = -1;
  size_t client_size = 0;

  for (size_t z_order = 0; z_order < layers.size(); ++z_order) {
    if (IsClientLayer(display, layers[z_order])) {
      if (client_start < 0)
        client_start = (int)z_order;
      client_size = (z_order - client_start) + 1;
    }
  }

  return GetExtraClientRange(display, layers, client_start, client_size,
                             use_cursor_plane);
}

bool Backend::IsClientLayer(HwcDisplay *display, HwcLayer *layer) {
  return !HardwareSupportsLayerType(layer->GetSfType()) ||
         !layer->IsLayerUsableAsDevice() || display->CtmByGpu() ||
         (layer->GetLayerData().pi.RequireScalingOrPhasing() &&
          display->GetHwc()->GetResMan().ForcedScalingWithGpu());
}

bool Backend::HardwareSupportsLayerType(HwcLayer::CompositionType comp_type) {
  return comp_type == HwcLayer::CompositionType::kDevice ||
         comp_type == HwcLayer::CompositionType::kCursor;
}

uint32_t Backend::CalcPixOps(const std::vector<HwcLayer *> &layers,
                             size_t first_z, size_t size) {
  uint32_t pixops = 0;
  for (size_t z_order = 0; z_order < layers.size(); ++z_order) {
    if (z_order >= first_z && z_order < first_z + size) {
      auto &df = layers[z_order]->GetLayerData().pi.display_frame;
      if (df.i_rect) {
        pixops += (df.i_rect->right - df.i_rect->left) *
                  (df.i_rect->bottom - df.i_rect->top);
      }
    }
  }
  return pixops;
}

void Backend::MarkValidated(std::vector<HwcLayer *> &layers,
                            size_t client_first_z, size_t client_size,
                            bool use_cursor_plane) {
  for (size_t z_order = 0; z_order < layers.size(); ++z_order) {
    if (z_order >= client_first_z && z_order < client_first_z + client_size) {
      layers[z_order]->SetValidatedType(HwcLayer::CompositionType::kClient);
    } else if (use_cursor_plane && layers[z_order]->GetSfType() ==
                                       HwcLayer::CompositionType::kCursor) {
      layers[z_order]->SetValidatedType(HwcLayer::CompositionType::kCursor);
    } else {
      layers[z_order]->SetValidatedType(HwcLayer::CompositionType::kDevice);
    }
  }
}

std::tuple<int, int> Backend::GetExtraClientRange(
    HwcDisplay *display, const std::vector<HwcLayer *> &layers,
    int client_start, size_t client_size, bool use_cursor_plane) {
  size_t avail_planes = display->GetPipe().GetUsablePlanes().first.size();
  size_t layers_size = layers.size();

  // Cursor plane is not counted among |avail_planes|, so the cursor layer
  // shouldn't be counted in |layers_size|.
  if (use_cursor_plane) {
    --layers_size;
  }

  /*
   * If more layers than planes, save one plane
   * for client composited layers
   */
  if (avail_planes < layers_size) {
    avail_planes--;
  }

  const int extra_client = int(layers_size - client_size) - int(avail_planes);

  if (extra_client > 0) {
    int start = 0;
    size_t steps = 0;
    if (client_size != 0) {
      const int prepend = std::min(client_start, extra_client);
      const int append = std::min(int(layers_size) -
                                      int(client_start + client_size),
                                  extra_client);
      start = client_start - (int)prepend;
      client_size += extra_client;
      steps = 1 + std::min(std::min(append, prepend),
                           int(layers_size) - int(start + client_size));
    } else {
      client_size = extra_client;
      steps = 1 + layers_size - extra_client;
    }

    uint32_t gpu_pixops = UINT32_MAX;
    for (size_t i = 0; i < steps; i++) {
      const uint32_t po = CalcPixOps(layers, start + i, client_size);
      if (po < gpu_pixops) {
        gpu_pixops = po;
        client_start = start + int(i);
      }
    }
  }

  return std::make_tuple(client_start, client_size);
}

// clang-format off
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables, cert-err58-cpp)
REGISTER_BACKEND("generic", Backend);
// clang-format on

}  // namespace android