Ticket #2881: TreeCleaner.py

# <Entity uid="12">
    # <Template>skirmish/structures/default_civil_centre</Template>
    # <Player>2</Player>
    # <Position x="1008.30304" z="1879.43067"/>
    # <Orientation y="2.35621"/>
    # <Actor seed="58274"/>
# </Entity>

import sys
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import ElementTree
#Name of the most generic template your are looking for
templateName = "flora_tree_cretan_date_palm"
#Name of the map
mapName = "Two Seas (6).xml"
#Radius where there should not be any identical entity
treshold = 3;

def GetMatchingTemplates():
    treeList = [];
    for item in root[3]:
        if templateName in item[0].text:
            treeList.append([item.attrib.get("uid"),item[2].get("x"),item[2].get("z")])
    return treeList

def IsTooClose(tempXpos, tempYpos, itemXpos, itemYpos, treshold):
    return abs(tempXpos - itemXpos) < treshold and abs(tempYpos - itemYpos) < treshold

def PrintStatus(numberOfObjects,currentItemIndex):
    sys.stdout.write("[ " + '{:.1f}'.format((currentItemIndex / int(numberOfObjects)) * 100) +" % done parsing templates. ("  + str(currentItemIndex) + " / " + str(numberOfObjects) +") ]")
    sys.stdout.flush()
    sys.stdout.write("\b" * (100+2))

def GetTooCloseEntities(treeList):
    uidList = [];
    numberOfTemplates = len(treeList)
    currentItemIndex = 0;
    for item in treeList:
        itemXpos = float(item[1])
        itemYpos = float(item[2])
        itemIndex = int(treeList.index(item))
        for i in range (0,numberOfTemplates,1):
            if i != itemIndex:
                    try:
                        entityId = treeList[i][0]
                        tempXpos = float(treeList[i][1])
                        tempYpos = float(treeList[i][2])
                    except:
                        print("NoneType Error")
                        continue
                    if IsTooClose(tempXpos, tempYpos, itemXpos, itemYpos, treshold) and entityId not in uidList:
                        uidList.append(entityId)
        PrintStatus(numberOfTemplates,currentItemIndex)
        currentItemIndex = currentItemIndex + 1
    sys.stdout.write("\n")
    return uidList

def DeleteTooCloseEntities(uidList):
    numberOfTemplates = len(uidList)
    currentItemIndex = 0;
    for entityList in root.findall('Entities'):
        for entity in entityList.findall('Entity'):
            id = int(entity.attrib.get("uid"))
            if str(id) not in uidList:
                continue
            entityList.remove(entity)
            currentItemIndex = currentItemIndex + 1
            PrintStatus(numberOfTemplates,currentItemIndex)
    sys.stdout.write("\n")



tree = ElementTree()
tree.parse(mapName)
root = tree.getroot()
DeleteTooCloseEntities(GetTooCloseEntities(GetMatchingTemplates()))
file_handle = open(mapName,"wb")
tree.write(file_handle, "UTF-8", True)
file_handle.close()