futhark/generators/exercises/robot_simulator.py at 45e64eae17f1508784bb83577b670456d01f38b0 · keiravillekode/futhark · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
MAIN = """\
local def encode_direction (d: direction): i32 =
    match (d)
      case #north -> 0
      case #east -> 1
      case #south -> 2
      case #west -> 3

local def decode_direction (i: i32): direction =
    match (i)
      case 0 -> #north
      case 1 -> #east
      case 2 -> #south
      case 3 -> #west
      case _ -> assert false #north

def main (start: []i32) (instructions: []u8): []i32 =
  let start_position = (start[0], start[1])
  let start_direction = decode_direction start[2]
  let result = move (start_position, start_direction) instructions
  in
    [result.0.0, result.0.1, encode_direction result.1]
"""


DIRECTIONS = ["north", "east", "south", "west"]


def gen_test_case(prop, description, inp, expected, f):
    in_x = inp["position"]["x"]
    in_y = inp["position"]["y"]
    in_d = DIRECTIONS.index(inp["direction"])
    if "instructions" in inp.keys():
        instructions = inp["instructions"]
    else:
        instructions = ""

    out_x = expected["position"]["x"]
    out_y = expected["position"]["y"]
    out_d = DIRECTIONS.index(expected["direction"])

    f.write(f"-- {description}\n")
    f.write("-- ==\n")
    f.write(f'-- input  {{ [{in_x}, {in_y}, {in_d}] "{instructions}" }}\n')
    f.write(f"-- output {{ [{out_x}, {out_y}, {out_d}] }}\n\n")


def gen_main(f):
    f.write(MAIN)