#!/usr/bin/env ruby
#
# Copyright (c) 2006, 2007 - R.W. van 't Veer

require File.join(File.dirname(__FILE__), 'test_helper')

class TestTIFF < Test::Unit::TestCase
  def setup    
    @t = TIFF.new(f('nikon_d1x.tif'))
  end
  
  def test_initialize
    all_test_tiffs.each do |fname|
      assert_nothing_raised do
        TIFF.new(fname)
      end
      assert_nothing_raised do
        open(fname) { |rd| TIFF.new(rd) }
      end
      assert_nothing_raised do
        TIFF.new(StringIO.new(File.read(fname)))
      end
    end
  end
  
  def test_multiple_images
    assert_equal 2, @t.size
  end
  
  def test_size
    assert_equal 269, @t.image_width
    assert_equal 269, @t.image_length
    assert_equal 269, @t.width
    assert_equal 269, @t.height
    assert_equal 120, @t[1].image_width
    assert_equal 160, @t[1].image_length
    assert_equal 120, @t[1].width
    assert_equal 160, @t[1].height
    
    @t = TIFF.new(f('plain.tif'))
    assert_equal 23, @t.image_width
    assert_equal 24, @t.image_length
    assert_equal 23, @t.width
    assert_equal 24, @t.height
  end
  
  def test_enumerable
    assert_equal @t[1], @t.find { |i| i.f_number.nil? }
  end
  
  def test_misc_fields
    assert_equal 'Canon PowerShot G3', TIFF.new(f('canon-g3.exif')).model
  end
  
  def test_dates
    (all_test_tiffs - [f('weird_date.exif'), f('plain.tif')]).each do |fname|
      assert_kind_of Time, TIFF.new(fname).date_time
    end
    assert_nil TIFF.new(f('weird_date.exif')).date_time
  end
  
  def test_orientation
    tested = 0 # count tests because not all exif samples have an orientation field
    all_test_exifs.each do |fname|
      orientation = TIFF.new(fname).orientation
      if orientation
        assert [
          TIFF::TopLeftOrientation,
          TIFF::TopRightOrientation,
          TIFF::BottomRightOrientation,
          TIFF::BottomLeftOrientation,
          TIFF::LeftTopOrientation,
          TIFF::RightTopOrientation,
          TIFF::RightBottomOrientation,
          TIFF::LeftBottomOrientation
        ].any? { |c| orientation == c }, 'not an orientation'
        assert orientation.respond_to?(:to_i)
        assert orientation.respond_to?(:transform_rmagick)
        tested += 1
      end
    end
    assert tested > 0
  end
  
  def test_gps
    t = TIFF.new(f('gps.exif'))
    assert_equal "\2\2\0\0", t.gps_version_id
    assert_equal 'N', t.gps_latitude_ref
    assert_equal 'W', t.gps_longitude_ref
    assert_equal [5355537.quo(100000), 0.quo(1), 0.quo(1)], t.gps_latitude
    assert_equal [678886.quo(100000), 0.quo(1), 0.quo(1)], t.gps_longitude
    assert_equal 'WGS84', t.gps_map_datum
    
    (all_test_exifs - [f('gps.exif')]).each do |fname|
      assert_nil TIFF.new(fname).gps_version_id
    end
  end
  
  def test_ifd_dispatch
    assert_not_nil @t.f_number
    assert_kind_of Rational, @t.f_number
    assert_not_nil @t[0].f_number
    assert_kind_of Rational, @t[0].f_number
  end
  
  def test_avoid_dispatch_to_nonexistent_ifds
    assert_nothing_raised do
      all_test_tiffs.each do |fname|
        t = TIFF.new(fname)
        TIFF::TAGS.each { |tag| t.send(tag) }
      end
    end
  end
  
  def test_to_hash
    all_test_tiffs.each do |fname|
      t = TIFF.new(fname)
      TIFF::TAGS.each do |key|
        assert_literally_equal t.send(key), t.to_hash[key], "#{key} not equal"
      end
    end
  end
  
  def test_old_style
    assert_nothing_raised do
      assert_not_nil @t[:f_number]
    end
  end
  
  def test_yaml_dump_and_load
    require 'yaml'
    
    all_test_tiffs.each do |fname|
      t = TIFF.new(fname)
      y = YAML.dump(t)
      assert_literally_equal t.to_hash, YAML.load(y).to_hash
    end
  end
  
  def test_jpeg_thumbnails
    count = 0
    all_test_tiffs.each do |fname|
      t = TIFF.new(fname)
      unless t.jpeg_thumbnails.empty?
        assert_nothing_raised do
          t.jpeg_thumbnails.each do |n|
            JPEG.new(StringIO.new(n))
          end
        end
        count += 1
      end
    end
    assert count > 0, 'no thumbnails found'
  end
end