// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef UI_VIEWS_CONTROLS_IMAGE_VIEW_H_
#define UI_VIEWS_CONTROLS_IMAGE_VIEW_H_

#include <memory>
#include <utility>

#include "base/macros.h"
#include "ui/base/metadata/metadata_header_macros.h"
#include "ui/base/models/image_model.h"
#include "ui/gfx/image/image_skia.h"
#include "ui/views/metadata/view_factory.h"
#include "ui/views/view.h"
#include "ui/views/views_export.h"

namespace gfx {
class Canvas;
}

namespace views {

/////////////////////////////////////////////////////////////////////////////
//
// ImageView class.
//
// An ImageView can display an image from an ImageSkia. If a size is provided,
// the ImageView will resize the provided image to fit if it is too big or will
// center the image if smaller. Otherwise, the preferred size matches the
// provided image size.
//
/////////////////////////////////////////////////////////////////////////////
class VIEWS_EXPORT ImageView : public View {
 public:
  METADATA_HEADER(ImageView);

  enum class Alignment { kLeading, kCenter, kTrailing };

  ImageView();
  explicit ImageView(const ui::ImageModel& image_model);
  ~ImageView() override;

  // Set the image that should be displayed.
  // TODO(pkasting): Change callers to pass an ImageModel and eliminate this.
  void SetImage(const gfx::ImageSkia& image) {
    SetImage(ui::ImageModel::FromImageSkia(image));
  }

  // Set the image that should be displayed from a pointer. Reset the image
  // if the pointer is NULL.
  // TODO(pkasting): Change callers to pass an ImageModel and eliminate this.
  void SetImage(const gfx::ImageSkia* image_skia) {
    SetImage(image_skia ? *image_skia : gfx::ImageSkia());
  }

  // Sets the image that should be displayed.
  void SetImage(const ui::ImageModel& image_model);

  // Sets the desired size of the image to be displayed.
  void SetImageSize(const gfx::Size& size);

  // Reset the image size to the current image dimensions.
  void ResetImageSize();

  // Returns the actual bounds of the visible image inside the view.
  gfx::Rect GetImageBounds() const;

  // Returns the image currently displayed, which can be empty if not set.
  // TODO(pkasting): Convert to an ImageModel getter.
  gfx::ImageSkia GetImage() const;

  ui::ImageModel GetImageModel() const;

  // Set / Get the horizontal alignment.
  void SetHorizontalAlignment(Alignment ha);
  Alignment GetHorizontalAlignment() const;

  // Set / Get the vertical alignment.
  void SetVerticalAlignment(Alignment va);
  Alignment GetVerticalAlignment() const;

  // Set / Get the accessible name text.
  void SetAccessibleName(const std::u16string& name);
  const std::u16string& GetAccessibleName() const;

  // Set the tooltip text.
  void SetTooltipText(const std::u16string& tooltip);
  const std::u16string& GetTooltipText() const;

  // Overridden from View:
  void OnPaint(gfx::Canvas* canvas) override;
  void GetAccessibleNodeData(ui::AXNodeData* node_data) override;
  std::u16string GetTooltipText(const gfx::Point& p) const override;
  gfx::Size CalculatePreferredSize() const override;
  views::PaintInfo::ScaleType GetPaintScaleType() const override;
  void OnBoundsChanged(const gfx::Rect& previous_bounds) override;
  void PreferredSizeChanged() override;

 protected:
  void OnThemeChanged() override;

 private:
  friend class ImageViewTest;

  void OnPaintImage(gfx::Canvas* canvas);

  // Gets an ImageSkia to paint that has proper rep for |scale|. Note that if
  // there is no existing rep of `scale`, we will utilize the image resize
  // operation to create one. The resize may be time consuming for a big image.
  gfx::ImageSkia GetPaintImage(float scale);

  // Returns true if |image_model| is the same as the last image we painted.
  // This is intended to be a quick check, not exhaustive. In other words it's
  // possible for this to return false even though the images are in fact equal.
  bool IsImageEqual(const ui::ImageModel& image_model) const;

  // Recomputes and updates the |image_origin_|.
  void UpdateImageOrigin();

  gfx::Size GetImageSize() const;

  // The origin of the image.
  gfx::Point image_origin_;

  // The current tooltip text.
  std::u16string tooltip_text_;

  // The current accessible name text.
  std::u16string accessible_name_;

  // Horizontal alignment.
  Alignment horizontal_alignment_ = Alignment::kCenter;

  // Vertical alignment.
  Alignment vertical_alignment_ = Alignment::kCenter;

  // The underlying image.
  ui::ImageModel image_model_;

  // Caches the scaled image reps.
  gfx::ImageSkia scaled_image_;

  // Scale last painted at.
  float last_paint_scale_ = 0.f;

  // Address of bytes we last painted. This is used only for comparison, so its
  // safe to cache.
  void* last_painted_bitmap_pixels_ = nullptr;

  // The requested image size.
  absl::optional<gfx::Size> image_size_;

  DISALLOW_COPY_AND_ASSIGN(ImageView);
};

BEGIN_VIEW_BUILDER(VIEWS_EXPORT, ImageView, View)
// Explicitly declare the overloaded SetImage methods in order to properly
// disambiguate between them.
BuilderT& SetImage(const gfx::ImageSkia& value) & {
  auto setter = std::make_unique<::views::internal::PropertySetter<
      ViewClass_, gfx::ImageSkia,
      decltype((static_cast<void (ViewClass_::*)(const gfx::ImageSkia&)>(
          &ViewClass_::SetImage))),
      &ViewClass_::SetImage>>(value);
  ::views::internal::ViewBuilderCore::AddPropertySetter(std::move(setter));
  return *static_cast<BuilderT*>(this);
}
BuilderT&& SetImage(const gfx::ImageSkia& value) && {
  return std::move(this->SetImage(value));
}
BuilderT& SetImage(const gfx::ImageSkia* value) & {
  auto setter = std::make_unique<::views::internal::PropertySetter<
      ViewClass_, gfx::ImageSkia*,
      decltype((static_cast<void (ViewClass_::*)(const gfx::ImageSkia*)>(
          &ViewClass_::SetImage))),
      &ViewClass_::SetImage>>(value);
  ::views::internal::ViewBuilderCore::AddPropertySetter(std::move(setter));
  return *static_cast<BuilderT*>(this);
}
BuilderT&& SetImage(const gfx::ImageSkia* value) && {
  return std::move(this->SetImage(value));
}
VIEW_BUILDER_PROPERTY(gfx::Size, ImageSize)
VIEW_BUILDER_PROPERTY(ImageView::Alignment, HorizontalAlignment)
VIEW_BUILDER_PROPERTY(ImageView::Alignment, VerticalAlignment)
VIEW_BUILDER_PROPERTY(std::u16string, AccessibleName)
VIEW_BUILDER_PROPERTY(std::u16string, TooltipText)
END_VIEW_BUILDER

}  // namespace views

DEFINE_VIEW_BUILDER(VIEWS_EXPORT, ImageView)

#endif  // UI_VIEWS_CONTROLS_IMAGE_VIEW_H_