Duet/converters.py

import fxconv

def convert(input, output, params, target):
    if params["custom-type"] == "level":
        convert_level(input, output, params, target)
        return 0
    else:
        return 1

def convert_level(input, output, params, target):
    with open(input, "r") as fp:
        raw = fp.read().split("\n")
        lines = [tempo.split(",") for tempo in raw if not tempo.startswith("#")]
        messages = [l[1:].strip() for l in raw if l.startswith("#")]
        message = "\n".join(l.replace("\\n", "\n") for l in messages)

    blocks = bytes()
    block_count = 0

    for (tempo, rects) in enumerate(lines):
        for r in [r for r in rects if r]:
            r = r.strip().split()
            shape, action, position = -1, 0, -1

            # Shape
            if "square" in r:
                shape = 0
            if "small" in r:
                shape = 2
            if "medium" in r:
                shape = 3
            if "normal" in r:
                shape = 4
            if "long" in r:
                shape = 5
            if "huge" in r:
                shape = 6
            if "long_vertical" in r:
                shape = 7

            # Position
            if "left" in r:
                position = 0
            if "right" in r:
                position = 1
            if "middle" in r:
                position = 2

            # Action
            if "rotate_left" in r:
                action = 1
            if "rotate_right" in r:
                action = 2
            if "fast_1" in r:
                action = 3
            if "fadeout" in r:
                action = 4
            if "outer_rotate_left" in r:
                action = 5
            if "outer_rotate_right" in r:
                action = 6
            if "slide" in r:
                action = 7
            if "fast_2" in r:
                action = 8
            if "fast_3" in r:
                action = 9
            if "rotate" in r and position == 0:
                action = 6
            if "rotate" in r and position == 1:
                action = 5

            # Implicit rules

            # Rotating in the middle -> long
            if shape < 0 and position == 2 and action in [1, 2]:
                shape = 5
            # Rotating on the side -> huge
            if shape < 0 and position in [0, 1] and action in [5, 6]:
                shape = 6
            # In the middle -> medium
            if shape < 0 and position == 2:
                shape = 3
            # No special properties -> normal
            if shape < 0 and action in [0, 3, 8, 9]:
                shape = 4

            if shape < 0 or action < 0 or position < 0:
                raise fxconv.FxconvError("Incomplete data: '" + " ".join(r) + "'")

            blocks += fxconv.u32(tempo)
            blocks += fxconv.u32(shape)
            blocks += fxconv.u32(position)
            blocks += fxconv.u32(action)
            block_count += 1

    o = fxconv.ObjectData()
    o += fxconv.u32(0x3f800000) # 1.0 as a float
    o += fxconv.ref(bytes(message, 'utf-8') + bytes(1), padding=4)
    o += fxconv.u32(block_count)
    o += fxconv.ref(blocks)

    fxconv.elf(o, output, "_" + params["name"], **target)