makemaps.py

from xml.dom.minidom import parseString
from bs4 import BeautifulSoup
import os
import re
import string
import math
import Image

# both x and y dimensions are equal
targetMapSize = 500.0
iconSize = 64.0

r = re.compile("(.*)_([0-9]{2,3}).xml")

def main():
    for filename in os.listdir("arena_defs_decoded"):
        readxml(filename)


def readxml(filename):
    file = open('arena_defs_decoded/' + filename, 'r')
    #convert to string:
    data = file.read()
    #close file because we dont need it anymore:
    file.close()

    m = r.search(filename)
    filename = m.group(2) + '_' + m.group(1)
    print filename

    soup = BeautifulSoup(data, 'xml')

    bottomLeft = soup.find("bottomLeft").contents
    upperRight = soup.find("upperRight").contents
    x2, y2 = string.split(bottomLeft[0].replace(',', '.'))
    x1, y1 = string.split(upperRight[0].replace(',', '.'))
    x2, y2 = float(x2), float(y2)
    x1, y1 = float(x1), float(y1)

    oldBoundingBox = [x1, x2, y1, y2]

    if (math.fabs(x1) + math.fabs(x2)) != (math.fabs(y1) + math.fabs(y2)):
        print "Could not process " + str(filename) + ": map dimensions do not match!"
        return

    ctf = soup.find("ctf")
    ass = soup.find("assault")
    dom = soup.find("domination")

    if ctf != None:
        #print "CTF"
        background = Image.open("mapsRaw/" + filename + "/mmap.png")
        teamBasePositions = ctf.find("teamBasePositions")
        team1 = teamBasePositions.find("team1")
        team2 = teamBasePositions.find("team2")
        team1Position = team1.find(re.compile("^position"))
        team2Position = team2.find(re.compile("^position"))
        #print team1Position, team2Position
        #scale and place on map
        foreground = Image.open("icon/" + 'base_green' + ".png")
        inx, iny = convertScaleToXY(team1Position.contents[0], oldBoundingBox)
        background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
        #put second team in
        foreground = Image.open("icon/" + 'base_red' + ".png")
        inx, iny = convertScaleToXY(team2Position.contents[0], oldBoundingBox)
        background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
        #do spawn points
        teamSpawnPoints = ctf.find("teamSpawnPoints")
        if teamSpawnPoints != None:
            team1 = teamSpawnPoints.find("team1")
            team2 = teamSpawnPoints.find("team2")
            team1SpawnPoint = team1.find_all("position")
            team2SpawnPoint = team2.find_all("position")
            #print team1SpawnPoint, team2SpawnPoint
            #team1 first
            counter = 1
            for spawn in team1SpawnPoint:
                foreground = Image.open("icon/" + 'spawn_green_' + str(counter) + ".png")
                inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
                counter += 1
            #team2 second
            counter = 1
            for spawn in team2SpawnPoint:
                foreground = Image.open("icon/" + 'spawn_red_' + str(counter) + ".png")
                inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
                counter += 1
        background.save("gridding/in/" + filename + "_" + "ctf" + ".png")

    if ass != None:
        #print "ASS"
        background = Image.open("mapsRaw/" + filename + "/mmap.png")
        teamBasePositions = ass.find("teamBasePositions")
        if teamBasePositions != None:
            team1 = teamBasePositions.find("team1")
            team2 = teamBasePositions.find("team2")
            team1Position = team1.find(re.compile("^position"))
            team2Position = team2.find(re.compile("^position"))
            #print team1Position, team2Position
            #first team: scale and place on map
            if team1Position != None:
                foreground = Image.open("icon/" + 'base_green' + ".png")
                inx, iny = convertScaleToXY(team1Position.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
            #put second team in
            if team2Position != None:
                foreground = Image.open("icon/" + 'base_red' + ".png")
                inx, iny = convertScaleToXY(team2Position.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
        #find spawn points
        teamSpawnPoints = ass.find("teamSpawnPoints")
        if teamSpawnPoints != None:
            team1 = teamSpawnPoints.find("team1")
            team2 = teamSpawnPoints.find("team2")
            team1SpawnPoint = team1.find_all("position")
            team2SpawnPoint = team2.find_all("position")
            #print team1SpawnPoint, team2SpawnPoint
            #team1 first
            counter = 1
            for spawn in team1SpawnPoint:
                foreground = Image.open("icon/" + 'spawn_green_' + str(counter) + ".png")
                inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
                counter += 1
            #team2 second
            counter = 1
            for spawn in team2SpawnPoint:
                foreground = Image.open("icon/" + 'spawn_red_' + str(counter) + ".png")
                inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
                background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
                counter += 1
        background.save("gridding/in/" + filename + "_" + "ass" + ".png")

    if dom != None:
        #print "DOM"
        background = Image.open("mapsRaw/" + filename + "/mmap.png")
        controlPoint = dom.find("controlPoint")
        #print controlPoint
        #scale and place on map
        foreground = Image.open("icon/" + 'base_neutral' + ".png")
        inx, iny = convertScaleToXY(controlPoint.contents[0], oldBoundingBox)
        background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
        #find spawn points
        teamSpawnPoints = dom.find("teamSpawnPoints")
        team1 = teamSpawnPoints.find("team1")
        team2 = teamSpawnPoints.find("team2")
        team1SpawnPoint = team1.find_all("position")
        team2SpawnPoint = team2.find_all("position")
        #print team1SpawnPoint, team2SpawnPoint
        #team1 first
        counter = 1
        for spawn in team1SpawnPoint:
            foreground = Image.open("icon/" + 'spawn_green_' + str(counter) + ".png")
            inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
            background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
            counter += 1
        #team2 second
        counter = 1
        for spawn in team2SpawnPoint:
            foreground = Image.open("icon/" + 'spawn_red_' + str(counter) + ".png")
            inx, iny = convertScaleToXY(spawn.contents[0], oldBoundingBox)
            background.paste(foreground, (int(inx), int(targetMapSize - iny)), foreground)
            counter += 1
        background.save("gridding/in/" + filename + "_" + "dom" + ".png")


def convertScaleToXY(input, oldScale):
    # oldScale is an array with values x1, x2, y1, y2
    inx, iny = string.split(input)
    inx, iny = round(float(inx.replace(',', '.'))), round(float(iny.replace(',', '.')))

    # move coordinates to start from (0,0)
    inx = inx - min(oldScale[0],oldScale[1])
    iny = iny - min(oldScale[2],oldScale[3])

    oldMaxX = math.fabs(oldScale[0]) + math.fabs(oldScale[1])
    oldMaxY = math.fabs(oldScale[2]) + math.fabs(oldScale[3])

    x = inx / oldMaxX * targetMapSize
    y = iny / oldMaxY * targetMapSize

    # set icon center at coordinates
    x -= iconSize / 2
    y += iconSize / 2

    return float(x), float(y)


if __name__ == '__main__':
    main()