components/omnibox/composebox/composebox_image_helper.cc

// Copyright 2025 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/omnibox/composebox/composebox_image_helper.h"

#include <numbers>
#include <optional>
#include <vector>

#include "base/memory/ref_counted_memory.h"
#include "base/memory/scoped_refptr.h"
#include "base/numerics/safe_math.h"
#include "components/lens/lens_bitmap_processing.h"
#include "components/lens/ref_counted_lens_overlay_client_logs.h"
#include "ui/gfx/geometry/rect.h"
#include "ui/gfx/geometry/size.h"
#include "ui/gfx/image/image_skia_operations.h"

namespace {

SkBitmap DownscaleImageIfNeeded(
    const SkBitmap& image,
    scoped_refptr<lens::RefCountedLensOverlayClientLogs> client_logs,
    const composebox::ImageEncodingOptions& image_options) {
  auto size = gfx::Size(image.width(), image.height());
  if (lens::ShouldDownscaleSize(size, image_options.max_size,
                                image_options.max_width,
                                image_options.max_height)) {
    return lens::DownscaleImage(image, image_options.max_width,
                                image_options.max_height, client_logs);
  }

  // No downscaling needed.
  return image;
}

}  // namespace

namespace composebox {

lens::ImageData DownscaleAndEncodeBitmap(
    const SkBitmap& image,
    scoped_refptr<lens::RefCountedLensOverlayClientLogs> client_logs,
    const ImageEncodingOptions& image_options) {
  lens::ImageData image_data;
  scoped_refptr<base::RefCountedBytes> data =
      base::MakeRefCounted<base::RefCountedBytes>();

  auto resized_bitmap =
      DownscaleImageIfNeeded(image, client_logs, image_options);
  if (image_options.enable_webp_encoding
          ? lens::EncodeImageMaybeWithTransparency(
                resized_bitmap, image_options.compression_quality, data,
                client_logs)
          : lens::EncodeImage(resized_bitmap, image_options.compression_quality,
                              data, client_logs)) {
    image_data.mutable_image_metadata()->set_height(resized_bitmap.height());
    image_data.mutable_image_metadata()->set_width(resized_bitmap.width());

    image_data.mutable_payload()->mutable_image_bytes()->assign(data->begin(),
                                                                data->end());
  }
  return image_data;
}

}  // namespace composebox