require_relative '../ffi/gdal'
require_relative 'raster_attribute_table_extensions'


module GDAL
  class RasterAttributeTable
    include RasterAttributeTableExtensions

    # @return [GDAL::RasterAttributeTable]
    def self.create
      raster_attribute_table_ptr = FFI::GDAL.GDALCreateRasterAttributeTable

      new(raster_attribute_table_ptr)
    end

    # Create an object from a ColorTable.
    #
    # @param color_table [GDAL::ColorTable, FFI::Pointer]
    # @return [GDAL::RasterAttributeTable]
    def self.from_color_table(color_table)
      color_table_ptr = GDAL._pointer(GDAL::ColorTable, color_table)
      rat_ptr = FFI::GDAL.GDALCreateRasterAttributeTable
      cpl_err = FFI::GDAL.GDALRATInitializeFromColorTable(rat_ptr, color_table_ptr)
      cpl_err.to_bool

      new(rat_ptr)
    end

    # @param raster_attribute_table [GDAL::RasterAttributeTable, FFI::Pointer]
    def initialize(raster_attribute_table=nil)
      @rat_pointer = GDAL._pointer(GDAL::RasterAttributeTable, raster_attribute_table)
    end

    def destroy!
      FFI::GDAL.GDALDestroyRasterAttributeTable(@rat_pointer)
    end

    def c_pointer
      @rat_pointer
    end

    # Clone using the C API.
    #
    # @return [GDAL::RasterAttributeTable]
    def clone
      rat_ptr = FFI::GDAL.GDALRATClone(@rat_pointer)
      return nil if rat_ptr.null?

      self.class.new(rat_ptr)
    end

    # +true+ if the changes made to this RAT have been written to the associated
    # dataset.
    #
    # @return [Boolean]
    def changes_written_to_file?
      FFI::GDAL.GDALRATChangesAreWrittenToFile(@rat_pointer)
    end

    # @return [Fixnum]
    def column_count
      FFI::GDAL.GDALRATGetColumnCount(@rat_pointer)
    end

    # @param index [Fixnum] The column number.
    # @return [String]
    def column_name(index)
      FFI::GDAL.GDALRATGetNameOfCol(@rat_pointer, index)
    end

    # @param index [Fixnum] The column number.
    # @return [GDALRATFieldUsage]
    def column_usage(index)
      FFI::GDAL.GDALRATGetUsageOfCol(@rat_pointer, index)
    end

    # @param index [Fixnum] The column number.
    # @return [GDALRATFieldType]
    def column_type(index)
      FFI::GDAL.GDALRATGetTypeOfCol(@rat_pointer, index)
    end

    # @param field_usage [GDALRATFieldUsage]
    # @return [Fixnum] The column number.
    def column_of_usage(field_usage)
      FFI::GDAL.GDALRATGetColOfUsage(@rat_pointer, field_usage)
    end

    # @param name [String]
    # @param type [FFI::GDALRATFieldType]
    # @param usage [FFI::GDALRATFieldUsage]
    # @return [Boolean]
    def create_column(name, type, usage)
      cpl_err = FFI::GDAL.GDALRATCreateColumn(@rat_pointer, name, type, usage)

      cpl_err.to_bool
    end

    # @return [Fixnum] The number of rows.
    def row_count
      FFI::GDAL.GDALRATGetRowCount(@rat_pointer)
    end

    # @return [Fixnum] The number of rows.
    def row_count=(count)
      FFI::GDAL.GDALRATSetRowCount(@rat_pointer, count)
    end

    # Get the row for a pixel value.
    #
    # @param value [Float]
    # @return [Fixnum]
    def row_of_value(value)
      FFI::GDAL.GDALRATGetRowOfValue(@rat_pointer, value)
    end

    # @param row [Fixnum]
    # @param field [Fixnum]
    # @return [String]
    def value_to_s(row, field)
      FFI::GDAL.GDALRATGetValueAsString(@rat_pointer, row, field)
    end

    # @param row [Fixnum]
    # @param field [Fixnum]
    # @return [Fixnum]
    def value_to_i(row, field)
      FFI::GDAL.GDALRATGetValueAsInt(@rat_pointer, row, field)
    end

    # @param row [Fixnum]
    # @param field [Fixnum]
    # @return [Float]
    def value_to_f(row, field)
      FFI::GDAL.GDALRATGetValueAsDouble(@rat_pointer, row, field)
    end

    # @param row [Fixnum]
    # @param field [Fixnum]
    # @param value [String, Float, Fixnum]
    def add_value(row, field, value)
      case value.class.name
      when 'String'
        FFI::GDAL.GDALRATSetValueAsString(@rat_pointer, row, field, value)
      when 'Float'
        FFI::GDAL.GDALRATSetValueAsDouble(@rat_pointer, row, field, value)
      when 'Fixnum'
        FFI::GDAL.GDALRATSetValueAsInt(@rat_pointer, row, field, value)
      else
        raise "Unknown value type for value '#{value}'"
      end
    end

    # @return [Hash{row_0_minimum => Float, bin_size => Float}]
    def linear_binning
      row_0_min_ptr = FFI::MemoryPointer.new(:double)
      bin_size_ptr = FFI::MemoryPointer.new(:double)
      FFI::GDAL.GDALRATGetLinearBinning(@rat_pointer, row_0_min_ptr, bin_size_ptr)

      {
        row_0_minimum: row_0_min_ptr.read_double,
        bin_size: bin_size_ptr.read_double
      }
    end

    # @param entry_count [Fixnum] The number of entries to produce.  The default
    #   will try to auto-determine the number.
    # @return [GDAL::ColorTable]
    def to_color_table(entry_count = -1)
      color_table_pointer = FFI::GDAL.GDALRATTranslateToColorTable(@rat_pointer, entry_count)

      GDAL::ColorTable.new(color_table_pointer)
    end

    # @param file_path [String]
    def dump_readable(file_path=nil)
      file = if file_path
        File.open(file_path, 'r')
      else
        file_path
      end
      FFI::GDAL.GDALRATDumpReadable(@rat_pointer, file)
    end
  end
end