#!/usr/bin/env python

import os

import errno

import fnmatch

import shutil

import logging

logging.basicConfig(format='[%(asctime)s] %(levelname)s %(name)s: %(message)s', level=logging.INFO)

LOG = logging.getLogger()

from PIL import Image, ImageFont, ImageDraw

import vdf

import vpk

# configs

vpk_path = '/d/Steam/steamapps/common/dota 2 beta/game/dota/pak01_dir.vpk'

vpk_img_root = 'resource/flash3/images'

fill_blue = "#1f496f"

fill_green = "#26721b"

fill_gold = "#7a4b4e"

font = ImageFont.truetype('Exo-SemiBold.ttf', 13)

src_root = "./src/"

src_preimages_root = "./prerendered_items/"

out_root = "./out/"

# helper functions

def mktree(path):

return path

# clear output directory

if os.path.exists("./out"):

shutil.rmtree("./out")

LOG.info("Reading VPK from %s" % repr(vpk_path))

# load game asset index

pak1 = vpk.open(vpk_path)

LOG.info("Reading items.txt")

# load items.txt schema and save a local copy to track in repo

with pak1.get_file("scripts/npc/items.txt") as vpkfile:

vpkfile.save(os.path.join(src_root, 'items.txt'))

items = vdf.load(vpkfile)['DOTAAbilities']

pt_names = ['power_treads_str.png', 'power_treads_agi.png', 'power_treads_int.png']

def get_file(filename):

return pak1.get_file('/'.join([vpk_img_root, "items", filename]))

LOG.info("Generating images for %d items" % len(items))

# find all items with mana or health requirements

for item_name in items:

if type(items[item_name]) is not dict:

continue

item = items[item_name]

filename = "%s.png" % item_name[5:] # removes 'item_' prefix

# skip recipies, since they all sahre the same icon image

if filename.startswith("recipe_"):

continue

# it done like so to handle power treads special case

filenames = [filename]

if filename == 'power_treads.png':

filenames += pt_names

image_src_files = map(get_file, filenames)

manacost_img = None

color = fill_blue

# generate manacost banner

if 'AbilityManaCost' in item:

manacost = item['AbilityManaCost']

# some items require 0 mana, like 'blink dagger' or 'soul ring'

if manacost == '0':

# ok, we look if the items sacrifice health, e.g. soul ring

try:

for xv in item['AbilitySpecial'].itervalues():

if type(xv) is not dict:

continue

if "health_sacrifice" in xv:

manacost = xv['health_sacrifice']

raise Exception

except:

color = fill_green # sacrifice health, health is green, we use green

# handle items that are upgradable (manacosts for each level is separated by whitespaces)

else:

if ' ' in manacost:

manacosts = manacost.split(' ')

try:

idx = int(item_name[-1]) - 1 # e.g. item_dagon, item_dagon_2, item_dagon_3

except:

idx = 0

manacost = manacosts[idx]

if manacost != '0':

# center text in our little box

text_pos_padding = (3 - len(manacost)) * 4

# open the image

for filename, image_src_file in zip(filenames, image_src_files):

manacost_img = Image.open(image_src_file)

# add info in the bottom left corner

d = ImageDraw.Draw(manacost_img)

d.polygon([(0, 50), (20, 50), (25, 64), (0, 64)], fill=color)

d.text((text_pos_padding, 49), manacost, font=font, fill="#ffffff")

del d

# save the image in the ouput directory

savepath = os.path.join(out_root, 'manacost', vpk_img_root, 'items', filename)

manacost_img.save(mktree(savepath))

# generate itemcost banner

if 'ItemCost' not in item or item['ItemCost'] == '0':

continue

for filename, image_src_file in zip(filenames, image_src_files):

image_src_file.seek(0)

itemcost_img = Image.open(image_src_file)

pad = 8 * len(item['ItemCost'])

d = ImageDraw.Draw(itemcost_img)

d.polygon([(84-pad, 50), (88, 50), (88, 64), (82-pad, 64)], fill=fill_gold)

d.text((86-pad, 49), item['ItemCost'], font=font, fill='#ffffff')

del d

savepath = os.path.join(out_root, 'goldcost', vpk_img_root, 'items', filename)

itemcost_img.save(mktree(savepath))

# generate combined version if needed

if manacost_img is not None:

itemcost_img.paste(manacost_img.crop((0, 0, 30, 64)), box=(0, 0, 30, 64))

savepath = os.path.join(out_root, 'combined', vpk_img_root, 'items', filename)

itemcost_img.save(mktree(savepath))

# add color coded damage type in the top right corner

# RED physical

# BLUE magic

# PURPLE pure

dmg_type_color = {

}

LOG.info("Reading npc_abilities.txt")

# load npc_abilities.txt schema and save a local copy to track in repo

with pak1.get_file("scripts/npc/npc_abilities.txt") as vpkfile:

vpkfile.save(os.path.join(src_root, 'npc_abilities.txt'))

abilities = vdf.load(vpkfile)['DOTAAbilities']

src_files = [path for path in pak1 if path.startswith("resource/flash3/images/spellicons/")]

LOG.info("Generating images for %d abilities" % len(abilities))

for name in abilities:

if type(abilities[name]) is not dict:

continue

cur = abilities[name]

if 'AbilityUnitDamageType' not in cur:

continue

if cur['AbilityUnitDamageType'] not in dmg_type_color:

continue

color = dmg_type_color[cur['AbilityUnitDamageType']]

for filepath in fnmatch.filter(src_files, "*%s.png" % name):

img = Image.open(pak1.get_file(filepath))

d = ImageDraw.Draw(img)

d.polygon([(128-42, 0), (128, 0), (128, 42)], fill="#191C22")

d.polygon([(128-38, 0), (128, 0), (128, 38)], fill=color)

del d

# save the image in the ouput directory

filepath = mktree(os.path.join(out_root, 'damagetype', filepath))

img.save(filepath)

img.close()

LOG.info("Generating VPKs")

# create VPKs

vpk.new(os.path.join(out_root, 'manacost')).save(mktree(os.path.join(out_root, 'costmod_manacost', 'pak01_dir.vpk')))

vpk.new(os.path.join(out_root, 'goldcost')).save(mktree(os.path.join(out_root, 'costmod_goldcost', 'pak01_dir.vpk')))

vpk.new(os.path.join(out_root, 'combined')).save(mktree(os.path.join(out_root, 'costmod_combined', 'pak01_dir.vpk')))

vpk.new(os.path.join(out_root, 'damagetype')).save(mktree(os.path.join(out_root, 'costmod_damagetype', 'pak01_dir.vpk')))

LOG.info("Done")