ar/assignment1/src/a2.py

   1 #!/usr/bin/env python3
   2 pc = [(4, 2), (4, 2), (4, 2)]
   3 rc = [(9, 7), (12, 6), (10, 7), (18, 5), (20, 4), (10, 6), (8, 6), (10, 8)]
   4 mx = 29
   5 my = 22
   6 pd = 17
   7
   8 # Print preamble
   9 print("(benchmark a4.smt")
  10 print(":logic QF_UFLIA")
  11
  12 # Print variables
  13 print(":extrafuns (")
  14 for i, (w, h) in enumerate(pc+rc, 1):
  15     print("(c{}x Int)".format(i), end=' ')
  16     print("(c{}y Int)".format(i), end=' ')
  17     print("(c{}w Int)".format(i), end=' ')
  18     print("(c{}h Int)".format(i))
  19 print(")")
  20
  21 print(":formula")
  22 print("(and")
  23
  24 # Print the PC and RC subformulas
  25 for i, (w, h) in enumerate(pc+rc, 1):
  26     # Print the width and height
  27     print(('(or '
  28            '(and (= c{0}w {1}) (= c{0}h {2})) '
  29            '(and (= c{0}w {2}) (= c{0}h {1})))').format(i, w-1, h-1))
  30     # Print the bounds of the coordinates
  31     print(('(> c{0}x 0) (> c{0}y 0)'
  32            '(<= (+ c{0}x c{0}w) {1}) (<= (+ c{0}y c{0}h) {2})'
  33            ).format(i, mx, my))
  34
  35     # Print the non overlap with others
  36     for j in range(1, 1+len(pc+rc)):
  37         if i != j:
  38             print((
  39                  '(or '
  40                  '(> c{0}x (+ c{1}x c{1}w)) (< (+ c{0}x c{0}w) c{1}x) '
  41                  '(> c{0}y (+ c{1}y c{1}h)) (< (+ c{0}y c{0}h) c{1}y)'
  42                  ')').format(i, j))
  43
  44 # Print the PC distance to eachother
  45 for i, _ in enumerate(pc, 1):
  46     for j, _ in enumerate(pc, 1):
  47         if i != j:
  48             print(('(or '
  49                    '(> (- (+ c{0}x (/ c{0}w 2)) (+ c{1}x (/ c{1}w 2))) {2}) '
  50                    '(> (- (+ c{1}x (/ c{1}w 2)) (+ c{0}x (/ c{0}w 2))) {2}) '
  51                    '(> (- (+ c{0}y (/ c{0}h 2)) (+ c{1}y (/ c{1}h 2))) {2}) '
  52                    '(> (- (+ c{1}y (/ c{1}h 2)) (+ c{0}y (/ c{0}h 2))) {2})'
  53                    ')').format(i, j, pd))
  54
  55             # Print the constraint that they have to be connected to a ps
  56 for i, _ in enumerate(rc, 1+len(pc)):
  57     print('(or')
  58     for j, _ in enumerate(pc, 1):
  59         print(('(not (or '
  60                '(> c{0}x (+ c{1}x 1 c{1}w)) (< (+ c{0}x c{0}w) (- c{1}x 1)) '
  61                '(> c{0}y (+ c{1}y 1 c{1}h)) (< (+ c{0}y c{0}h) (- c{1}y 1))'
  62                '))').format(i, j))
  63     print(')')
  64
  65 # Close the and,benchmark parenthesis
  66 print("))")