require File.dirname(__FILE__) + '/../property_helper'

class BoolEnumSampleProblem < Gecode::Model
  attr :bools
  attr :b1
  attr :b2
  
  def initialize
    @bools = bool_var_array(4)
    @b1 = bool_var
    @b2 = bool_var
    branch_on @bools
    branch_on wrap_enum([@b1, @b2])
  end
end

# Expects @stub, which contains the started constraint and @compute_result 
# which computes whether the left hand side is true or not.
describe 'bool enum relation constraint', :shared => true do
  it 'should handle being constrained to be true' do
    @stub.must_be.true
    @model.solve!
    @compute_result.call.should be_true
  end
  
  it 'should handle being constrained to be negated true' do
    @stub.must_not_be.true
    @model.solve!
    @compute_result.call.should_not be_true
  end
  
  it 'should handle being constrained to be false' do
    @stub.must_be.false
    @model.solve!
    @compute_result.call.should_not be_true
  end
  
  it 'should handle being constrained to be negated false' do
    @stub.must_not_be.false
    @model.solve!
    @compute_result.call.should be_true
  end
  
  it 'should handle being constrained to be equal to a variable' do
    @stub.must_be == @b1
    @model.solve!
    @compute_result.call.should == @b1.value
  end
  
  it 'should handle being constrained to not be equal to a variable' do
    @stub.must_not_be == @b1
    @model.solve!
    @compute_result.call.should_not == @b1.value
  end
  
  it 'should handle being constrained to be equal to be a nested expression' do
    @stub.must_be == (@b1 | @b2) & @b1
    @model.solve!
    @compute_result.call.should == (@b1.value | @b2.value) & @b1.value
  end
  
  it 'should handle being constrained to not be equal to be a nested expression' do
    @stub.must_not_be == (@b1 | @b2) & @b1
    @model.solve!
    @compute_result.call.should_not == (@b1.value | @b2.value) & @b1.value
  end
end

describe Gecode::BoolEnum::Relation, ' (conjunction)' do
  before do
    @model = BoolEnumSampleProblem.new
    @bools = @model.bools
    @b1 = @model.b1
    @b2 = @model.b2
    
    # For property spec.
    @property_types = [:bool_enum]
    @select_property = lambda do |bool_enum|
      bool_enum.conjunction
    end
    @selected_property = @bools.conjunction
    @constraint_class = Gecode::BlockConstraint
    
    # For bool enum spec.
    @stub = @selected_property
    @compute_result = lambda{ @bools.all?{ |b| b.value } }
  end
  
  it_should_behave_like 'bool enum relation constraint'
  it_should_behave_like(
    'property that produces bool operand by short circuiting equality')
end

describe Gecode::BoolEnum::Relation, ' (disjunction)' do
  before do
    @model = BoolEnumSampleProblem.new
    @bools = @model.bools
    @b1 = @model.b1
    @b2 = @model.b2
    
    # For constraint option spec.
    @property_types = [:bool_enum]
    @select_property = lambda do |bool_enum|
      bool_enum.disjunction
    end
    @selected_property = @bools.disjunction
    @constraint_class = Gecode::BlockConstraint
    
    # For bool enum spec.
    @stub = @selected_property
    @compute_result = lambda{ @bools.any?{ |b| b.value } }
  end
  
  it_should_behave_like 'bool enum relation constraint'
  it_should_behave_like(
    'property that produces bool operand by short circuiting equality')
end