z3 0.0.20160323 → 0.0.20160330

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3ec7feb43474477a1c3a7220fbbd5c6e11246cd2
-  data.tar.gz: 687aedb2c8aaeca944108d4361459bba5ba08eb5
+  metadata.gz: 9b616eadc04d3acebed002b5e30969543ec87288
+  data.tar.gz: 51b12ceac3ca413e11d445c21524aa709e0d0c86
 SHA512:
-  metadata.gz: 9e641cf575e0810de111a96f394747b1316dc2f9cebd63c7a52390c2439c7836abf1660d67b52998f5e4a048fea74a569121e7f9fd9e4a2e32bce82367082362
-  data.tar.gz: 9d1c79300d21645a5ed1e121a8f069fd0a4504c5731ff07c5e5f6a6727647c94ae64fb885c4f4e5bfb2b4641183b2e697f8128d3f66f784cdc9ab6dcce3361fe
+  metadata.gz: 755c4a975d120202b93efe203d3eeea53dc7898a0a15b8008c287bc7821c773f5e88a73680dd32eb4acfcd55208659c55591ee9151a06056884e2760101492a3
+  data.tar.gz: 5bc45fa1e31598830ce699b55cccbfac82ed9f83b72d9700fd454ecc2d05f68800bad0d2f8c2dc9f6485163b1ada994cfdd233323a52e49af1f142fa8107673e

data/examples/knights_puzzle_solver

@@ -0,0 +1,122 @@
+#!/usr/bin/env ruby
+
+require_relative "../lib/z3"
+
+class KnightsPuzzle
+  def initialize
+    @x_size = 4
+    @y_size = 4
+    @num_moves = 12
+    @solver = Z3::Solver.new
+    @boards = (@num_moves+1).times.map{|t| board(t) }
+    @moves = @num_moves.times.map{|t| move(t)}
+    setup_board @boards[0], """
+    bbb.
+    xb.w
+    ..ww
+    x.xw
+    """
+    setup_board @boards[-1], """
+    www.
+    xw.b
+    ..bb
+    x.xb
+    """
+    @num_moves.times do |t|
+      setup_board_move @boards[t], @boards[t+1], @moves[t]
+    end
+  end
+
+  def print_board(t)
+    puts "State #{t}:"
+    puts @boards[t].transpose.map{|row| row.map{|c| ".wbx"[@model[c].to_i]}.join }.join("\n")
+  end
+
+  def print_move(t)
+    move = Hash[@moves[t].map{|k,v| [k,@model[v].to_i]}]
+    figure = " wbx"[move[:figure]]
+    puts "#{figure}: #{move[:start_x]},#{move[:start_y]} -> #{move[:end_x]},#{move[:end_y]}"
+    puts ""
+  end
+
+  def solve!
+    if @solver.check == :sat
+      @model = @solver.model
+      puts "Solved"
+      @num_moves.times do |t|
+        print_board(t)
+        print_move(t)
+      end
+      print_board(@num_moves)
+    else
+      puts "Puzzle can't be solved"
+    end
+  end
+
+  private
+
+  def setup_board_move(board_start, board_end, move)
+    @x_size.times.map do |x|
+      @y_size.times.map do |y|
+        # If not part of the move, stays the same
+        starts_here = (move[:start_x] == x) & (move[:start_y] == y)
+        ends_here = (move[:end_x] == x) & (move[:end_y] == y)
+        @solver.assert Z3::Implies(~starts_here & ~ends_here, board_start[x][y] == board_end[x][y])
+        # Every move starts with figure, ends on empty space
+        @solver.assert Z3::Implies(starts_here, board_start[x][y] == move[:figure])
+        @solver.assert Z3::Implies(starts_here, board_end[x][y] == 0)
+        @solver.assert Z3::Implies(ends_here, board_start[x][y] == 0)
+        @solver.assert Z3::Implies(ends_here, board_end[x][y] == move[:figure])
+      end
+    end
+  end
+
+  def board(t)
+    @x_size.times.map do |x|
+      @y_size.times.map do |y|
+        v = Z3::Int("b[#{t},#{x},#{y}]")
+        @solver.assert v >= 0
+        @solver.assert v <= 3
+        v
+      end
+    end
+  end
+
+  def setup_board(vars, str)
+    str = str.split.map(&:chomp)
+    4.times do |x|
+      4.times do |y|
+        c = str[y][x]
+        i = ".wbx".index(c) or raise "Unknown character #{c}"
+        @solver.assert vars[x][y] == i
+      end
+    end
+  end
+
+  def abs(t)
+    Z3::IfThenElse(t >= 0, t, -t)
+  end
+
+  def move(t)
+    move_start_x = Z3::Int("move_start_x_#{t}")
+    move_start_y = Z3::Int("move_start_y_#{t}")
+    move_end_x   = Z3::Int("move_end_x_#{t}")
+    move_end_y   = Z3::Int("move_end_y_#{t}")
+    figure       = Z3::Int("move_figure_#{t}")
+    @solver.assert (figure >= 1) & (figure <= 2)
+    @solver.assert (move_start_x >= 0) & (move_start_x < @x_size)
+    @solver.assert (move_start_y >= 0) & (move_start_y < @y_size)
+    @solver.assert (move_end_x >= 0) & (move_end_x < @x_size)
+    @solver.assert (move_end_y >= 0) & (move_end_y < @y_size)
+
+    # 8 possible moves
+    @solver.assert (
+      ((abs(move_start_x - move_end_x) == 2) & (abs(move_start_y - move_end_y) == 1)) |
+      ((abs(move_start_x - move_end_x) == 1) & (abs(move_start_y - move_end_y) == 2))
+    )
+
+    {start_x: move_start_x, start_y: move_start_y, end_x: move_end_x, end_y: move_end_y, figure: figure}
+  end
+end
+
+KnightsPuzzle.new.solve!

data/lib/z3/interface.rb

@@ -75,6 +75,12 @@ module Z3
     BoolSort.new.new(Z3::LowLevel.mk_iff(a, b))
   end
 
+  def IfThenElse(a, b, c)
+    a, = coerce_to_same_bool_sort(a)
+    b, c = coerce_to_same_sort(b, c)
+    b.sort.new(Z3::LowLevel.mk_ite(a, b, c))
+  end
+
   def Add(*args)
     raise ArgumentError if args.empty?
     args = coerce_to_same_sort(*args)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: z3
 version: !ruby/object:Gem::Version
-  version: 0.0.20160323
+  version: 0.0.20160330
 platform: ruby
 authors:
 - Tomasz Wegrzanowski
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-23 00:00:00.000000000 Z
+date: 2016-03-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -41,6 +41,7 @@ files:
 - examples/geometry_problem
 - examples/kakuro_solver
 - examples/kinematics_problems
+- examples/knights_puzzle_solver
 - examples/letter_connections_solver
 - examples/light_up_solver
 - examples/minisudoku_solver