// Copyright 2010 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/paint_manager.h"

#include <stddef.h>
#include <stdint.h>

#include <algorithm>
#include <cmath>
#include <utility>

#include "base/auto_reset.h"
#include "base/check.h"
#include "base/check_op.h"
#include "base/feature_list.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "base/location.h"
#include "base/metrics/histogram_functions.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/single_thread_task_runner.h"
#include "base/time/time.h"
#include "pdf/paint_ready_rect.h"
#include "pdf/pdf_features.h"
#include "third_party/skia/include/core/SkCanvas.h"
#include "third_party/skia/include/core/SkImage.h"
#include "third_party/skia/include/core/SkRect.h"
#include "third_party/skia/include/core/SkRefCnt.h"
#include "third_party/skia/include/core/SkSamplingOptions.h"
#include "third_party/skia/include/core/SkSurface.h"
#include "ui/gfx/blit.h"
#include "ui/gfx/geometry/point.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/geometry/size.h"
#include "ui/gfx/geometry/skia_conversions.h"
#include "ui/gfx/geometry/vector2d.h"
#include "ui/gfx/geometry/vector2d_f.h"
#if BUILDFLAG(ARKWEB_PDF)
#include "arkweb/chromium_ext/cc/input/input_handler_utils.h"
#include "arkweb/chromium_ext/pdf/paint_manager_for_include.cc"
#include "base/trace_event/trace_event.h"
#endif

namespace chrome_pdf {

PaintManager::PaintManager(Client* client) : client_(client) {
  DCHECK(client_);
}

PaintManager::~PaintManager() = default;

// static
gfx::Size PaintManager::GetNewContextSize(const gfx::Size& current_context_size,
                                          const gfx::Size& plugin_size) {
  // The amount of additional space in pixels to allocate to the right/bottom of
  // the context.
  constexpr int kBufferSize = 50;

  // Default to returning the same size.
  gfx::Size result = current_context_size;

  // The minimum size of the plugin before resizing the context to ensure we
  // aren't wasting too much memory. We deduct twice the kBufferSize from the
  // current context size which gives a threshhold that is kBufferSize below
  // the plugin size when the context size was last computed.
  gfx::Size min_size(
      std::max(current_context_size.width() - 2 * kBufferSize, 0),
      std::max(current_context_size.height() - 2 * kBufferSize, 0));

  // If the plugin size is bigger than the current context size, we need to
  // resize the context. If the plugin size is smaller than the current
  // context size by a given threshhold then resize the context so that we
  // aren't wasting too much memory.
  if (plugin_size.width() > current_context_size.width() ||
      plugin_size.height() > current_context_size.height() ||
      plugin_size.width() < min_size.width() ||
      plugin_size.height() < min_size.height()) {
    // Create a larger context than needed so that if we only resize by a
    // small margin, we don't need a new context.
    result = gfx::Size(plugin_size.width() + kBufferSize,
                       plugin_size.height() + kBufferSize);
  }

  return result;
}

void PaintManager::SetSize(const gfx::Size& new_size, float device_scale) {
  if (GetEffectiveSize() == new_size &&
      GetEffectiveDeviceScale() == device_scale) {
    return;
  }

  has_pending_resize_ = true;
  pending_size_ = new_size;
  pending_device_scale_ = device_scale;

  view_size_changed_waiting_for_paint_ = true;

  Invalidate();
}

void PaintManager::SetTransform(float scale,
                                const gfx::Point& origin,
                                const gfx::Vector2d& translate,
                                bool schedule_flush) {
  if (!surface_) {
    return;
  }

  CHECK_GT(scale, 0.0f);
  // translate_with_origin = origin - scale * origin - translate
  gfx::Vector2dF translate_with_origin = origin.OffsetFromOrigin();
  translate_with_origin.Scale(1.0f - scale);
  translate_with_origin.Subtract(translate);

  // TODO(crbug.com/40203030): Should update be deferred until `Flush()`?
  client_->UpdateLayerTransform(scale, translate_with_origin);

  if (!schedule_flush) {
    return;
  }

  if (flush_pending_) {
    flush_requested_ = true;
    return;
  }
  Flush();
}

void PaintManager::ClearTransform() {
  SetTransform(1.f, gfx::Point(), gfx::Vector2d(), false);
}

void PaintManager::Invalidate() {
  if (!surface_ && !has_pending_resize_) {
    return;
  }

  EnsureCallbackPending();
  aggregator_.InvalidateRect(gfx::Rect(GetEffectiveSize()));
}

void PaintManager::InvalidateRect(const gfx::Rect& rect) {
  DCHECK(!in_paint_);

  if (!surface_ && !has_pending_resize_) {
    return;
  }

  // Clip the rect to the device area.
  gfx::Rect clipped_rect =
      gfx::IntersectRects(rect, gfx::Rect(GetEffectiveSize()));
  if (clipped_rect.IsEmpty()) {
    return;  // Nothing to do.
  }

  EnsureCallbackPending();
  aggregator_.InvalidateRect(clipped_rect);
}

void PaintManager::ScrollRect(const gfx::Rect& clip_rect,
                              const gfx::Vector2d& amount) {
  DCHECK(!in_paint_);

  if (!surface_ && !has_pending_resize_) {
    return;
  }

  EnsureCallbackPending();

  aggregator_.ScrollRect(clip_rect, amount);
}

gfx::Size PaintManager::GetEffectiveSize() const {
  return has_pending_resize_ ? pending_size_ : plugin_size_;
}

float PaintManager::GetEffectiveDeviceScale() const {
  return has_pending_resize_ ? pending_device_scale_ : device_scale_;
}

void PaintManager::EnsureCallbackPending() {
  // The best way for us to do the next update is to get a notification that
  // a previous one has completed. So if we're already waiting for one, we
  // don't have to do anything differently now.
  if (flush_pending_) {
    return;
  }

  // If no flush is pending, we need to do a manual call to get back to the
  // main thread. We may have one already pending, or we may need to schedule.
  if (manual_callback_pending_) {
    return;
  }

  base::SingleThreadTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE, base::BindOnce(&PaintManager::OnManualCallbackComplete,
                                weak_factory_.GetWeakPtr()));
  manual_callback_pending_ = true;
}

#if BUILDFLAG(ARKWEB_PDF)
void PaintManager::DoPaint(bool is_repainting) {
  TRACE_EVENT0("media", "PaintManager::DoPaint");
#else
void PaintManager::DoPaint() {
#endif
  base::AutoReset<bool> auto_reset_in_paint(&in_paint_, true);

  DCHECK(aggregator_.HasPendingUpdate());

  // Apply any pending resize. Setting the graphics to this class must happen
  // before asking the plugin to paint in case it requests invalides or resizes.
  // However, the bind must not happen until afterward since we don't want to
  // have an unpainted device bound. The needs_binding flag tells us whether to
  // do this later.
  //
  // Note that `has_pending_resize_` will always be set on the first DoPaint().
  DCHECK(surface_ || has_pending_resize_);
  if (has_pending_resize_) {
    plugin_size_ = pending_size_;
    // Only create a new graphics context if the current context isn't big
    // enough or if it is far too big. This avoids creating a new context if
    // we only resize by a small amount.
    gfx::Size old_size = surface_
                             ? gfx::Size(surface_->width(), surface_->height())
                             : gfx::Size();
    gfx::Size new_size = GetNewContextSize(old_size, pending_size_);
    if (old_size != new_size || !surface_) {
      surface_ = SkSurfaces::Raster(
          SkImageInfo::MakeN32Premul(new_size.width(), new_size.height()));
      DCHECK(surface_);

      // TODO(crbug.com/40222665): Can we guarantee repainting some other way?
      client_->InvalidatePluginContainer();

      device_scale_ = 1.0f;

      // Since we're binding a new one, all of the callbacks have been canceled.
      manual_callback_pending_ = false;
      flush_pending_ = false;
      weak_factory_.InvalidateWeakPtrs();
    }

    if (pending_device_scale_ != device_scale_) {
      client_->UpdateScale(1.0f / pending_device_scale_);
    }
    device_scale_ = pending_device_scale_;

    // This must be cleared before calling into the plugin since it may do
    // additional invalidation or sizing operations.
    has_pending_resize_ = false;
    pending_size_ = gfx::Size();
  }

  PaintAggregator::PaintUpdate update = aggregator_.GetPendingUpdate();

  // Set a timer here to check how long it takes to do all the paint operations
  // below, except for the flush.
  const base::TimeTicks begin_time = base::TimeTicks::Now();

  std::vector<PaintReadyRect> ready_rects;
  std::vector<gfx::Rect> pending_rects;
#if BUILDFLAG(ARKWEB_PDF)
  if (update.has_scroll) {
    last_update_rect.has_scroll = update.has_scroll;
    last_update_rect.scroll_delta = update.scroll_delta;
    last_update_rect.scroll_rect = update.scroll_rect;
    last_update_rect.paint_rects = update.paint_rects;
  }
  if (is_repainting && last_update_rect.has_scroll &&
      (std::abs(update.scroll_delta.x()) != 0 ||
      std::abs(update.scroll_delta.y()) == 0)) {
    if (last_update_rect.paint_rects.empty()) {
      last_update_rect.paint_rects = update.paint_rects;
    }
    for (auto& rect : last_update_rect.paint_rects) {
      rect = client_->GetAvailableArea();
    }
    client_->OnPaint(last_update_rect.paint_rects, ready_rects, pending_rects);
    last_update_rect.paint_rects.clear();
    last_update_rect.has_scroll = false;
  } else {
#endif
  client_->OnPaint(update.paint_rects, ready_rects, pending_rects);
#if BUILDFLAG(ARKWEB_PDF)
  }
#endif

  if (ready_rects.empty() && pending_rects.empty()) {
    return;  // Nothing was painted, don't schedule a flush.
  }

  std::vector<PaintReadyRect> ready_now;
  if (pending_rects.empty()) {
    aggregator_.SetIntermediateResults(std::move(ready_rects),
                                       std::move(pending_rects));
    ready_now = aggregator_.TakeReadyRects();
    aggregator_.ClearPendingUpdate();

    // First, apply any scroll amount less than the surface's size.
    if (update.has_scroll &&
        std::abs(update.scroll_delta.x()) < surface_->width() &&
        std::abs(update.scroll_delta.y()) < surface_->height()) {
      // TODO(crbug.com/40203030): Use `SkSurface::notifyContentWillChange()`.
      gfx::ScrollCanvas(surface_->getCanvas(), update.scroll_rect,
                        update.scroll_delta);
    }

    view_size_changed_waiting_for_paint_ = false;
  } else {
    std::vector<PaintReadyRect> ready_later;
    for (auto& ready_rect : ready_rects) {
      // Don't flush any part (i.e. scrollbars) if we're resizing the browser,
      // as that'll lead to flashes.  Until we flush, the browser will use the
      // previous image, but if we flush, it'll revert to using the blank image.
      // We make an exception for the first paint since we want to show the
      // default background color instead of the pepper default of black.
      if (ready_rect.flush_now() &&
          (!view_size_changed_waiting_for_paint_ || first_paint_)) {
        ready_now.push_back(std::move(ready_rect));
      } else {
        ready_later.push_back(std::move(ready_rect));
      }
    }
    // Take the rectangles, except the ones that need to be flushed right away,
    // and save them so that everything is flushed at once.
    aggregator_.SetIntermediateResults(std::move(ready_later),
                                       std::move(pending_rects));

    if (ready_now.empty()) {
      EnsureCallbackPending();
      return;
    }
  }

  for (const auto& ready_rect : ready_now) {
    const SkRect skia_rect = gfx::RectToSkRect(ready_rect.rect());

    // Paint the page's white background, and then paint the page's contents.
    // If `ready_rect.image()` has transparencies, this is necessary to paint
    // over the stale data in `skia_rect` in `surface_`.
    SkPaint paint;
    paint.setColor(SK_ColorWHITE);
    surface_->getCanvas()->drawRect(skia_rect, paint);

#if BUILDFLAG(ARKWEB_PDF) && !defined(COMPONENT_BUILD)  // FIXME
    if (!base::ohos::IsPcDevice()) {
      if (!cc::InputHandlerUtils::PdfOverSpeed() || is_repainting) {
        surface_->getCanvas()->drawImageRect(
            ready_rect.image(), skia_rect, skia_rect, SkSamplingOptions(), nullptr,
            SkCanvas::kStrict_SrcRectConstraint);
      }
    } else {
#endif
    surface_->getCanvas()->drawImageRect(
        ready_rect.image(), skia_rect, skia_rect, SkSamplingOptions(), nullptr,
        SkCanvas::kStrict_SrcRectConstraint);
#if BUILDFLAG(ARKWEB_PDF) && !defined(COMPONENT_BUILD)  // FIXME
    }
#endif
  }

  base::UmaHistogramMediumTimes("PDF.RenderAndPaintTime",
                                base::TimeTicks::Now() - begin_time);

  Flush();

  base::UmaHistogramMediumTimes("PDF.RenderPaintAndFlushTime",
                                base::TimeTicks::Now() - begin_time);

  first_paint_ = false;
}

void PaintManager::Flush() {
  flush_requested_ = false;

  sk_sp<SkImage> snapshot = surface_->makeImageSnapshot();
  surface_->getCanvas()->drawImage(snapshot.get(), /*x=*/0, /*y=*/0,
                                   SkSamplingOptions(), /*paint=*/nullptr);
  client_->UpdateSnapshot(std::move(snapshot));

  // TODO(crbug.com/40251507): Complete flush synchronously.
  base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
      FROM_HERE, base::BindOnce(&PaintManager::OnFlushComplete,
                                weak_factory_.GetWeakPtr()));
  flush_pending_ = true;
}

void PaintManager::OnFlushComplete() {
  DCHECK(flush_pending_);
  flush_pending_ = false;

  // If more paints were enqueued while we were waiting for the flush to
  // complete, execute them now.
  if (aggregator_.HasPendingUpdate())
#if BUILDFLAG(ARKWEB_PDF)
    DoPaint(false);
#else
    DoPaint();
#endif

  // If there was another flush request while flushing we flush again.
  if (flush_requested_) {
    Flush();
  }
}

void PaintManager::OnManualCallbackComplete() {
  DCHECK(manual_callback_pending_);
  manual_callback_pending_ = false;

  // Just because we have a manual callback doesn't mean there are actually any
  // invalid regions. Even though we only schedule this callback when something
  // is pending, a Flush callback could have come in before this callback was
  // executed and that could have cleared the queue.
  if (aggregator_.HasPendingUpdate()) {
#if BUILDFLAG(ARKWEB_PDF)
    DoPaint(false);
#else
    DoPaint();
#endif
  }
}

}  // namespace chrome_pdf