module ChunkyPNG::Canvas::Operations

def change_mask_color!(new_color)
  raise ChunkyPNG::ExpectationFailed, "This is not a mask image!" if palette.opaque_palette.size != 1
  pixels.map! { |pixel| (new_color & 0xffffff00) | ChunkyPNG::Color.a(pixel) }
  self
end

def change_theme_color!(old_theme_color, new_theme_color, bg_color = ChunkyPNG::Color::WHITE, tolerance = 5)
  base, mask = extract_mask(old_theme_color, bg_color, tolerance)
  mask.change_mask_color!(new_theme_color)
  self.replace(base.compose(mask))
end

def check_size_constraints!(other, offset_x, offset_y)
  raise ChunkyPNG::OutOfBounds, "Background image width is too small!"  if width  < other.width  + offset_x
  raise ChunkyPNG::OutOfBounds, "Background image height is too small!" if height < other.height + offset_y
end

def compose(other, offset_x = 0, offset_y = 0)
  check_size_constraints!(other, offset_x, offset_y)
  for y in 0...other.height do
    for x in 0...other.width do
      set_pixel(x + offset_x, y + offset_y, ChunkyPNG::Color.compose(other.get_pixel(x, y), get_pixel(x + offset_x, y + offset_y)))
    end
  end
  self
end

def crop(x, y, crop_width, crop_height)
  
  raise ChunkyPNG::OutOfBounds, "Image width is too small!" if crop_width + x > width
  raise ChunkyPNG::OutOfBounds, "Image width is too small!" if crop_height + y > height
  
  new_pixels = []
  for cy in 0...crop_height do
    new_pixels += pixels.slice((cy + y) * width + x, crop_width)
  end
  ChunkyPNG::Canvas.new(crop_width, crop_height, new_pixels)
end

def extract_mask(mask_color, bg_color = ChunkyPNG::Color::WHITE, tolerance = 5)
  base_pixels = []
  mask_pixels = []
  pixels.each do |pixel|
    if ChunkyPNG::Color.alpha_decomposable?(pixel, mask_color, bg_color, tolerance)
      mask_pixels << ChunkyPNG::Color.decompose_color(pixel, mask_color, bg_color, tolerance)
      base_pixels << bg_color
    else
      mask_pixels << (mask_color & 0xffffff00)
      base_pixels << pixel
    end
  end
  
  [ self.class.new(width, height, base_pixels), self.class.new(width, height, mask_pixels) ]
end

def flip_horizontally
  self.class.new(width, height).tap do |flipped|
    for y in 0...height do
      flipped.replace_row!(height - (y + 1), row(y))
    end
  end
end

def flip_vertically
  self.class.new(width, height).tap do |flipped|
    for x in 0...width do
      flipped.replace_column!(width - (x + 1), column(x))
    end
  end
end

def replace(other, offset_x = 0, offset_y = 0)
  check_size_constraints!(other, offset_x, offset_y)
  for y in 0...other.height do
    pixels[(y + offset_y) * width + offset_x, other.width] = other.pixels[y * other.width, other.width]
  end
  self
end

def rotate_180
  self.class.new(width, height).tap do |flipped|
    for y in 0...height do
      flipped.replace_row!(height - (y + 1), row(y).reverse)
    end
  end
end

def rotate_left
  self.class.new(height, width).tap do |rotated|
    for i in 0...width do
      rotated.replace_row!(width - (i + 1), column(i))
    end
  end
end

def rotate_right
  self.class.new(height, width).tap do |rotated|
    for i in 0...width do
      rotated.replace_row!(i, column(i).reverse)
    end
  end
end