ar/assignments/src/2.py

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