Doom 3 map generation utils for python

genmap.py

#!/usr/bin/python3.7

TEXTURED = True

def main():
    playerstart = (32, 32, 16)

    brushes = [
        # generateRect3d((-64, -64, -8), (128, 128, 8)), #bottom
        # generateRect3d((-64 - 8, -64, 0), (8, 128, 128)), #left (player back)
        # generateCube(-64, -192, 0, 128), #front (player right)
        # generateCube(64, -64, 0, 128), #right (player front)
        # generateCube(-64, 64, 0, 128), #back (player left)
        # generateRect3d((-64, -64, 128), (128, 128, 8)), #top
        # generateRect3d((256, 256, 256), (128, 128, 1)),

        # generateSafeLine((8, 8), (8, 16)),
        # generateSafeLine((0, 0), (0, 64)),
        # generateSafeLine((0, 0), (64, 0)),

        generateBox(0, 0, 0, 256),
    ]

    with open('generated.map', 'w') as _out:
        _out.write(generateMapFromBrushes(brushes, playerstart))

def getPlaneString(x: int, y: int, z: int, distance: int) -> str:
    return f'( {x} {y} {z} {distance} ) ( ( 0.125 0 -5 ) ( 0 0.125 57 ) ) "{"textures/alphalabs/a_enwall13c" if TEXTURED else "_none"}" 0 0 0'

def generateBrushDef3(brushes: tuple, comment='// primitive', indent=4) -> str:
    ''' generatesBrushDef3 from 6 *brushes* (normalx, normaly, normalz, distancefromorigin) '''

    if len(brushes) != 6: raise Exception("There must be 6 brushes")

    lines = [
        comment,
        '{',
        '    brushDef3 {',
    ]

    for brush in brushes:
        lines.append('        ' + getPlaneString(*brush))

    lines.append('    }')
    lines.append('}')

    return ' ' * indent + ('\n' + ' ' * indent).join(lines)

def generateSafeLine(v1:tuple, v2:tuple, height:tuple=(0, 8), indent=4):
    if v1[0] == v2[0]: # vertical
        if v2[1] < v1[1]: v1, v2 = v2, v1
        sizex = 8
        sizey = v2[1] - v1[1] + 8
        return generateRect3d((v1[0], v1[1], height[0]), (sizex, sizey, height[1] - height[0]))
    elif v1[1] == v2[1]: # horizontal
        if v2[0] < v1[0]: v1, v2 = v2, v1
        sizex = v2[0] - v1[0] + 8
        sizey = 8
        return generateRect3d((v1[0], v1[1], height[0]), (sizex, sizey, height[1] - height[0]))
    else: raise Exception('Not a safe line')

def generateLine(v1:tuple, v2:tuple, height:tuple=(0, 8), indent=4): # FIXME:
    x1, y1 = v1
    x2, y2 = v2

    bottom = (0, 0, -1, height[0])
    top = (0, 0, 1, -height[1])
    front = (0, -1, 0, y1)
    right = (1, 0, 0, -x2)
    back = (0, 1, 0, -y2)
    left = (-1, 0, 0, x1)

    return generateBrushDef3((bottom, top, front, right, back, left), f'// Line(({x1}, {y1}), ({x2}, {y2}), ({height[0]}, {height[1]}))', indent=indent)

def generateRect3d(position: tuple, size: tuple, indent=4) -> str:
    x, y, z = position
    width, depth, height = size

    bottom = (0, 0, -1, z)
    top = (0, 0, 1, -(z + height))
    front = (0, -1, 0, y)
    right = (1, 0, 0, -(x + width))
    back = (0, 1, 0, -(y + depth))
    left = (-1, 0, 0, x)

    return generateBrushDef3((bottom, top, front, right, back, left), f'// Rect3d(({x}, {y}, {z}), ({width}, {depth}, {height})', indent=indent)

def generateCube(x, y, z, size, indent=4) -> str:
    bottom = (0, 0, -1, z)
    top = (0, 0, 1, -(z + size))
    front = (0, -1, 0, y)
    right = (1, 0, 0, -(x + size)   )
    back = (0, 1, 0, -(y + size))
    left = (-1, 0, 0, x)

    return generateBrushDef3((bottom, top, front, right, back, left), f'// cube({x}, {y}, {z}, {size})', indent=indent)

def generateBox(x, y, z, size, width=8, indent=4) -> str:
    brushes = [
        generateRect3d((x, y, z - width), (size, size, width)), #bottom
        generateRect3d((x - width, y, z), (width, size, size)), #left (player back)
        generateRect3d((x, y - width, z), (size, width, size)), #front (player right)
        generateRect3d((x + size, y, z), (width, size, size)), #right (player front)
        generateRect3d((x, y + size, z), (size, width, size)), #back (player left)
        generateRect3d((x, y, z + size), (size, size, width)), #top
    ]

    return '\n'.join(brushes)



def generateMapFromBrushes(brushes: list, playerstart: tuple):
    px, py, pz = playerstart
    lines = []
    lines.append('Version 2')
    lines.append('// Map')
    lines.append('{')
    lines.append('    "classname" "worldspawn"')

    for brush in brushes: lines.append(brush)

    lines.append('}')

    lines.append('// PlayerStart')
    lines.append('{')
    lines.append('    "classname" "info_player_start"')
    lines.append('    "name" "info_player_start_1"')
    lines.append(f'    "origin" "{px} {py} {pz}"')
    lines.append('}')

    lines.append('// Player light')
    lines.append('{')
    lines.append('    "classname" "light"')
    lines.append('    "name" "light1"')
    lines.append(f'    "origin" "{px} {py} {pz}"')
    lines.append('}')

    return '\n'.join(lines)

if __name__ == "__main__":
    main()