blueprintList.py

import sys
import os

from library import *

def main():

    # Read dlc dict
    dlcs, dlcDict = itemsDLC()
    print(dlcs)
    newDlcsDict = dlcDict[-2]+dlcDict[-1]

    itemDir = "./ExpandedCDB-BP/item/"
    # 1. Melee items
    melee = []
    for item in os.listdir(itemDir+"/Melee"):
        if "StartSword" not in item and "AdminWeapon" not in item:
            melee.append(item[6:item.index(".")])
    
    # 2. Ranged items
    ranged = []
    for item in os.listdir(itemDir+"/Ranged"):
        if "StartBow" not in item:
            ranged.append(item[6:item.index(".")])
    
    # 3. Shield items
    shield = []
    for item in os.listdir(itemDir+"/Shield"):
        if "StartShield" not in item:
            shield.append(item[6:item.index(".")])
    
    # 4. Trap items
    trap = []
    for item in os.listdir(itemDir+"/DeployedTrap"):
        trap.append(item[6:item.index(".")])

    # 5. Grenade items
    grenade = []
    for item in os.listdir(itemDir+"/Grenade"):
        grenade.append(item[6:item.index(".")])
    
    # 6. Power items
    power = []
    for item in os.listdir(itemDir+"/Power"):
        power.append(item[6:item.index(".")])
    
    # 7. Skins 
    skins = []
    for item in os.listdir(itemDir+"/Skin"):
        skins.append(item[6:item.index(".")])
    
    # 8. Meta stuff
    metas = []
    lastItem = ""
    for item in os.listdir(itemDir+"/Meta"):
        item = item[6:item.index(".")]
        if not isInBlacklist(item):
            metas.append(item)
        # if item[0:-1] == lastItem[0:-1]:
            # metas[-1] = item
        # else:
            # metas.append(item)
            # lastItem=item

    # 9. Mutations/Perks
    mutations = []
    for item in os.listdir(itemDir+"/Perk"):
        item = item[6:item.index(".")]
        mutations.append(item)
    
    
    # Read base room file
    f = open('src/PrisonFlaskRoom.tmx','r')
    lines = f.readlines()
    f.close()

    # Create new room file
    f = open("PrisonFlaskRoom-BP-Mod.tmx",'w')
    f.writelines(lines[:-2])

    # Get start id
    for line in lines:
        if "object id" in line:
            id = int(line.split('"')[1])+1
    
    print("Start id:", id)

    # Generate Armory list
    # Remove duplicates
    weapons = []
    for item in melee+ranged+shield:
        if not isInBlacklist(item):
            if item not in weapons and item not in newDlcsDict:
                # Item in the latest DLC is not included here
                weapons.append(item)
    sizeOfWeapons = len(weapons)

    skills = []
    for item in trap+grenade+power:
        if not isInBlacklist(item):
            if item not in newDlcsDict: 
                # Item in the latest DLC is not included here
                skills.append(item)
    sizeOfSkills  = len(skills)

    temp = skins
    skins = []
    for item in temp:
        if item not in newDlcsDict:
            # Item in the latest DLC is not included here
            skins.append(item)
    sizeOfSkins  = len(skins)
    
    sizeOfMetas  = len(metas)

    print("Found Weapons: {0}".format(sizeOfWeapons))
    print("      Skills : {0}".format(sizeOfSkills))
    print("      Skins  : {0}".format(sizeOfSkins))
    print("      Metas  : {0}".format(sizeOfMetas))

    print (len(dlcDict[0]),len(dlcDict[1]),len(dlcDict[2]),len(dlcDict[3]))

    # Change this when the map is resized in the height direction (Add rows to the top)
    nAdd = 3

    # Armory showcase at y direction
    yList = [41,44,47,50,53,56,59]
    yList = [y+nAdd for y in yList]

    # Weapons showcase
    xRange = [2,34]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            weaponIndex = j*nInLine+i
            if weaponIndex < sizeOfWeapons:
                x = xRange[0]+i*2+1
                y = yList[j]
                #f.writelines(jsonWrapper(x,y,weapons[weaponIndex]))
                f.writelines(xmlWrapperBP(id,x,y,weapons[weaponIndex]))
                id += 1

    # Skills showcase 1
    yList = [41,44,47,50,53,56,59]
    yList = [y+nAdd for y in yList]

    xRange = [36,64]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            skillIndex = j*nInLine+i
            if skillIndex < sizeOfSkills:
                x = xRange[0]+i*2+1
                y = yList[j]
                #f.writelines(jsonWrapper(x,y,skills[skillIndex]))
                f.writelines(xmlWrapperBP(id,x,y,skills[skillIndex]))
                id += 1

    # Skins showcase
    yList = [2,5,8,11]
    xRange = [2,64]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            skinIndex = j*nInLine+i
            if skinIndex < len(skins):
                x = xRange[0]+i*2+1
                y = yList[j]
                #f.writelines(jsonWrapper(x,y,skins[skinIndex]))
                f.writelines(xmlWrapperBP(id,x,y,skins[skinIndex]))
                id += 1

    # Metas and Mutation showcase
    comb = metas+mutations
    yList = [62,65,68,71]
    yList = [y+nAdd for y in yList]

    xRange = [2,64]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            combIndex = j*nInLine+i
            if combIndex < len(comb):
                x = xRange[0]+i*2+1
                y = yList[j]
                if comb[combIndex] in metas:
                    # Runes
                    #f.writelines(jsonWrapperLoot(x,y,comb[combIndex]))
                    f.writelines(xmlWrapperLoot(id,x,y,comb[combIndex]))
                    id += 1
                else:
                    # Mutations
                    #f.writelines(jsonWrapper(x,y,comb[combIndex]))
                    f.writelines(xmlWrapperBP(id,x,y,comb[combIndex]))
                    id += 1

    # New dlc items showcase
    ## QoS
    yList = [75,78]
    yList = [y+nAdd for y in yList]
    
    xRange = [3,46]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            itemIndex = j*nInLine+i
            # Insert exception handler
            if itemIndex < len(dlcDict[-2]):
                item = dlcDict[-2][itemIndex]
                if not isInBlacklist(item):
                    x = xRange[0]+i*2+1
                    y = yList[j]
                    f.writelines(xmlWrapperBP(id, x, y, item))
                    id += 1

    ## Purple
    yList = [81,84]
    yList = [y+nAdd for y in yList]
    
    xRange = [3,46]
    nInLine = int((xRange[-1]-xRange[0])/2)
    for j in range(len(yList)):
        for i in range(nInLine):
            itemIndex = j*nInLine+i
            if itemIndex < len(dlcDict[-1]):
                item = dlcDict[-1][itemIndex]
                if not isInBlacklist(item):
                    x = xRange[0]+i*2+1
                    y = yList[j]
                    f.writelines(xmlWrapperBP(id, x, y, item))
                    id += 1
   
    tail = lines[-2:]
    f.writelines(tail)

    f.close()

    # Copy file to ExtendedCDB
    os.system("cp PrisonFlaskRoom-BP-Mod.tmx ./bp-tmx/tmx/Prison/PrisonFlaskRoom.tmx")

    print("Done!")

if __name__  == "__main__":
    main()