// Aseprite Document Library
// Copyright (c) 2018-2024 Igara Studio S.A.
// Copyright (c) 2001-2016 David Capello
//
// This file is released under the terms of the MIT license.
// Read LICENSE.txt for more information.

#ifndef DOC_IMAGE_H_INCLUDED
#define DOC_IMAGE_H_INCLUDED
#pragma once

#include "doc/color.h"
#include "doc/color_mode.h"
#include "doc/image_buffer.h"
#include "doc/image_iterators2.h"
#include "doc/image_spec.h"
#include "doc/object.h"
#include "doc/pixel_format.h"
#include "gfx/clip.h"
#include "gfx/rect.h"
#include "gfx/size.h"

namespace doc {

template<typename ImageTraits>
class ImageBits;
class Palette;
class Pen;
class RgbMap;

class Image : public Object {
public:
  enum LockType {
    ReadLock,     // Read-only lock
    WriteLock,    // Write-only lock
    ReadWriteLock // Read and write
  };

  static Image* create(PixelFormat format,
                       int width,
                       int height,
                       const ImageBufferPtr& buffer = ImageBufferPtr());
  static Image* create(const ImageSpec& spec, const ImageBufferPtr& buffer = ImageBufferPtr());
  static Image* createCopy(const Image* image, const ImageBufferPtr& buffer = ImageBufferPtr());

  virtual ~Image();

  const ImageSpec& spec() const { return m_spec; }
  ColorMode colorMode() const { return m_spec.colorMode(); }
  PixelFormat pixelFormat() const { return (PixelFormat)colorMode(); }
  bool isTilemap() const { return m_spec.colorMode() == ColorMode::TILEMAP; }
  int width() const { return m_spec.width(); }
  int height() const { return m_spec.height(); }
  gfx::Size size() const { return m_spec.size(); }
  gfx::Rect bounds() const { return m_spec.bounds(); }
  color_t maskColor() const { return m_spec.maskColor(); }
  void setMaskColor(color_t c) { m_spec.setMaskColor(c); }
  void setColorSpace(const gfx::ColorSpaceRef& cs) { m_spec.setColorSpace(cs); }

  // Number of bytes to store one pixel of this image.
  int bytesPerPixel() const { return m_spec.bytesPerPixel(); }

  // Number of bytes to store all visible pixels on each row.
  int widthBytes() const { return m_spec.widthBytes(); }

  // Number of bytes for each row of this image on memory (some
  // bytes for each row might be hidden/just for alignment to
  // "base_alignment").
  int rowBytes() const { return m_rowBytes; }

  // Number of pixels for each row (some of these pixels are hidden
  // when width() < rowPixels()).
  int rowPixels() const { return m_rowBytes / bytesPerPixel(); }

  virtual int getMemSize() const override;

  template<typename ImageTraits>
  ImageBits<ImageTraits> lockBits(LockType lockType, const gfx::Rect& bounds)
  {
    return ImageBits<ImageTraits>(this, bounds);
  }

  template<typename ImageTraits>
  ImageBits<ImageTraits> lockBits(LockType lockType, const gfx::Rect& bounds) const
  {
    return ImageBits<ImageTraits>(const_cast<Image*>(this), bounds);
  }

  template<typename ImageTraits>
  void unlockBits(ImageBits<ImageTraits>& imageBits)
  {
    // Do nothing
  }

  // Warning: These functions doesn't have (and shouldn't have)
  // bounds checks. Use the primitives defined in doc/primitives.h
  // in case that you need bounds check.
  virtual uint8_t* getPixelAddress(int x, int y) const = 0;
  virtual color_t getPixel(int x, int y) const = 0;
  virtual void putPixel(int x, int y, color_t color) = 0;
  virtual void clear(color_t color) = 0;
  virtual void copy(const Image* src, gfx::Clip area) = 0;
  virtual void drawHLine(int x1, int y, int x2, color_t color) = 0;
  virtual void fillRect(int x1, int y1, int x2, int y2, color_t color) = 0;
  virtual void blendRect(int x1, int y1, int x2, int y2, color_t color, int opacity) = 0;

  ReadIterator readArea() const { return ReadIterator(this, this->bounds()); }
  WriteIterator writeArea() { return WriteIterator(this, this->bounds()); }

  ReadIterator readArea(const gfx::Rect& bounds,
                        const IteratorStart start = IteratorStart::TopLeft) const
  {
    return ReadIterator(this, bounds, start);
  }

  WriteIterator writeArea(const gfx::Rect& bounds,
                          const IteratorStart start = IteratorStart::TopLeft)
  {
    return WriteIterator(this, bounds, start);
  }

protected:
  Image(const ImageSpec& spec);

  // Number of bytes for each row.
  size_t m_rowBytes;

private:
  ImageSpec m_spec;
};

} // namespace doc

#include "doc/image_bits.h"
#include "doc/image_iterator.h"

#endif
