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

#include "pdf/pdf_ink_cursor.h"

#include <math.h>

#include <algorithm>

#include "base/check_op.h"
#include "pdf/pdf_ink_brush.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "third_party/skia/include/core/SkCanvas.h"
#include "third_party/skia/include/core/SkColor.h"
#include "third_party/skia/include/core/SkPath.h"

namespace chrome_pdf {

namespace {

constexpr int kMinDiameter = 6;
constexpr int kMaxDiameter = 32;

void CheckCursorDiameterIsInRange(int diameter) {
  CHECK_GE(diameter, kMinDiameter);
  CHECK_LE(diameter, kMaxDiameter);
}

SkColor GetCursorOutlineColor(SkColor color) {
  // Compute the lightness value as defined by HSL.
  const std::pair<uint8_t, uint8_t> minmax =
      std::minmax({SkColorGetR(color), SkColorGetG(color), SkColorGetB(color)});
  const int lightness = 0.5 * (minmax.first + minmax.second);

  // Use lighter outline for darker `color`.
  constexpr SkColor kDarkOutlineColor = SkColorSetARGB(0xFF, 0x90, 0x90, 0x90);
  constexpr SkColor kLightOutlineColor = SkColorSetARGB(0xFF, 0xAA, 0xAA, 0xAA);
  return lightness > 127 ? kDarkOutlineColor : kLightOutlineColor;
}

}  // namespace

int CursorDiameterFromBrushSizeAndZoom(float brush_size, float zoom) {
  CHECK(PdfInkBrush::IsToolSizeInRange(brush_size));

  constexpr float kMinSize = 4;  // Cursor become very hard to see if smaller.
  float cursor_diameter = std::max(brush_size * zoom, kMinSize);

  // Fudge factors to better match sizes used in Chrome OS Gallery.
  constexpr int kSmallAdjustment = 2;
  constexpr int kLargeAdjustment = 4;
  constexpr int kLargeSize = 10;
  int adjustment =
      cursor_diameter < kLargeSize ? kSmallAdjustment : kLargeAdjustment;
  return std::clamp(static_cast<int>(round(cursor_diameter)) + adjustment,
                    kMinDiameter, kMaxDiameter);
}

SkBitmap GenerateToolCursor(SkColor color, int diameter) {
  CheckCursorDiameterIsInRange(diameter);

  SkBitmap bitmap;
  bitmap.allocPixels(
      SkImageInfo::MakeN32(diameter, diameter, kPremul_SkAlphaType));
  SkCanvas canvas(bitmap);

  SkPaint paint;
  paint.setAntiAlias(true);
  paint.setColor(GetCursorOutlineColor(color));
  paint.setStyle(SkPaint::kFill_Style);

  auto rect = SkRect::MakeWH(diameter, diameter);
  const SkPath outline_path = SkPath::Oval(rect, SkPathDirection::kCW, 1);
  canvas.drawPath(outline_path, paint);

  paint.setColor(color);
  rect.inset(1, 1);
  const SkPath fill_path = SkPath::Oval(rect, SkPathDirection::kCW, 1);
  canvas.drawPath(fill_path, paint);

  return bitmap;
}

}  // namespace chrome_pdf