#!/usr/bin/env ruby

require File.expand_path(File.join(File.dirname(__FILE__), "spec_common"))
require 'bindata'

share_examples_for "All Integers" do

  it "should have correct num_bytes" do
    all_classes do |int_class|
      int_class.new.num_bytes.should == @nbytes
    end
  end

  it "should have a sensible value of zero" do
    all_classes do |int_class|
      int_class.new.value.should be_zero
    end
  end

  it "should avoid underflow" do
    all_classes do |int_class|
      obj = int_class.new
      obj.value = min_value - 1

      obj.value.should == min_value
    end
  end

  it "should avoid overflow" do
    all_classes do |int_class|
      obj = int_class.new
      obj.value = max_value + 1

      obj.value.should == max_value
    end
  end

  it "should assign values" do
    all_classes do |int_class|
      obj = int_class.new
      test_int = gen_test_int
      obj.assign(test_int)
      obj.value.should == test_int
    end
  end

  it "should assign values from other int objects" do
    all_classes do |int_class|
      src = int_class.new
      src.assign(gen_test_int)

      obj = int_class.new
      obj.assign(src)
      obj.value.should == src.value
    end
  end

  it "should symmetrically read and write a +ve number" do
    all_classes do |int_class|
      obj = int_class.new
      obj.value = gen_test_int

      str = obj.to_binary_s
      int_class.read(str).should == obj.value
    end
  end

  it "should symmetrically read and write a -ve number" do
    all_classes do |int_class|
      if @signed
        obj = int_class.new
        obj.value = -gen_test_int

        str = obj.to_binary_s
        int_class.read(str).should == obj.value
      end
    end
  end

  it "should convert a +ve number to string" do
    all_classes do |int_class|
      val = gen_test_int

      obj = int_class.new
      obj.value = val

      obj.to_binary_s.should == int_to_binary_str(val)
    end
  end

  it "should convert a -ve number to string" do
    all_classes do |int_class|
      if @signed
        val = -gen_test_int

        obj = int_class.new
        obj.value = val

        obj.to_binary_s.should == int_to_binary_str(val)
      end
    end
  end

  def all_classes(&block)
    @ints.each_pair do |int_class, nbytes|
      @nbytes = nbytes
      yield int_class
    end
  end

  def min_value
    if @signed
      -max_value - 1
    else
      0
    end
  end

  def max_value
    if @signed
      (1 << (@nbytes * 8 - 1)) - 1
    else
      (1 << (@nbytes * 8)) - 1
    end
  end

  def gen_test_int
    # resulting int is guaranteed to be +ve for signed or unsigned integers
    (0 ... @nbytes).inject(0) { |val, i| (val << 8) | ((val + 0x11) % 0x100) }
  end

  def int_to_binary_str(val)
    str = ""
    v = val & ((1 << (@nbytes * 8)) - 1)
    @nbytes.times do
      str.concat(v & 0xff)
      v >>= 8
    end
    (@endian == :little) ? str : str.reverse
  end

  def create_mapping_of_class_to_nbits(endian, signed)
    base = signed ? "Int" : "Uint"
    signed_sym = signed ? :signed : :unsigned
    endian_str = (endian == :little) ? "le" : "be"

    result = {}
    result[BinData.const_get("#{base}8")] = 1
    (1 .. 20).each do |nbytes|
      nbits = nbytes * 8
      class_name = "#{base}#{nbits}#{endian_str}"
      BinData::Int.define_class(nbits, endian, signed_sym)
      result[BinData.const_get(class_name)] = nbytes
    end

    result
  end
end

describe "All signed big endian integers" do
  it_should_behave_like "All Integers"

  before(:all) do
    @endian = :big
    @signed = true
    @ints = create_mapping_of_class_to_nbits(@endian, @signed)
  end
end

describe "All unsigned big endian integers" do
  it_should_behave_like "All Integers"

  before(:all) do
    @endian = :big
    @signed = false
    @ints = create_mapping_of_class_to_nbits(@endian, @signed)
  end
end

describe "All signed little endian integers" do
  it_should_behave_like "All Integers"

  before(:all) do
    @endian = :little
    @signed = true
    @ints = create_mapping_of_class_to_nbits(@endian, @signed)
  end
end

describe "All unsigned little endian integers" do
  it_should_behave_like "All Integers"

  before(:all) do
    @endian = :little
    @signed = false
    @ints = create_mapping_of_class_to_nbits(@endian, @signed)
  end
end

describe "Custom defined integers" do
  it "should fail unless bits are a multiple of 8" do
    lambda {
      BinData::Int.define_class(7, :little, :unsigned)
    }.should raise_error

    lambda {
      BinData::Int.define_class(7, :big, :unsigned)
    }.should raise_error

    lambda {
      BinData::Int.define_class(7, :little, :signed)
    }.should raise_error

    lambda {
      BinData::Int.define_class(7, :big, :signed)
    }.should raise_error
  end
end