view LightTools/build.py @ 51:efd2b1ca5b77

Clean up gui
author koryspansel <koryspansel@bendbroadband.com>
date Tue, 27 Sep 2011 09:42:01 -0700
parents a379bce1aeb1
children 2444937929ae
line wrap: on
line source

#
# build
#

import sys
import string
import struct
import os
import functools
import re

_pattern = re.compile('\(\s*(\d+)\s*,\s*(\d+)\s*\)')

class MapDefinition:
    Parse = functools.partial(string.split, sep=',')

    def __init__(self, size, position, direction, grid):
        self.Size = self._ParseVector(size)
        self.Position = self._ParseVector(position)
        self.Direction = self._ParseVector(direction)
        self.Grid = self._ParseList(grid)

    def Write(self, filename):
        with open(filename, 'wb') as handle:
            handle.write(struct.pack('i' * len(self.Size),      *self.Size))            # 8
            handle.write(struct.pack('i' * len(self.Position),  *self.Position))        # 8
            handle.write(struct.pack('i' * len(self.Direction), *self.Direction))       # 4

            for tower in self.Grid:
                handle.write(struct.pack('i' * len(tower), *tower))                     # 8 * Size.X * Size.Y

    def _ParseVector(self, line):
        return map(int, self.Parse(line))

    def _ParseList(self, lines):
        #return [self._ParseVector(line) for line in lines]
        tuples = []

        #print ''.join(lines)
        #print _pattern.split(''.join(lines))

        for item in _pattern.finditer(''.join(lines)):
            #print item.groups()
            #tuples.append((int(item.group(1)), int(item.group(2))))
            tuples.append(map(int, item.groups()))

        #print tuples
        return tuples
        #print tuples

        #return [map(int, item.groups()) for item in _pattern.finditer(''.join(lines))]

def ReadDefinition(filename):
    def FilterLines(lines):
        for line in map(string.strip, lines):
            if line and not line.startswith('#'):
                yield line.split('#')[0]

    with open(filename, 'rt') as handle:
        lines = handle.readlines()

    definition = tuple(FilterLines(lines))

    size = definition[0]
    position = definition[1]
    direction = definition[2]
    grid = definition[3:]

    return MapDefinition(size, position, direction, grid)

if __name__ == '__main__':
    path_root   = os.path.dirname(sys.argv[0])
    path_assets = os.path.abspath(os.path.join(path_root, '..', 'Assets', 'Maps'))
    path_build  = os.path.abspath(os.path.join(path_root, '..', 'Data', 'Maps'))

    if os.path.isdir(path_assets):
        for filename in os.listdir(path_assets):
            if not filename.endswith('.def'):
                continue

            definition = ReadDefinition(os.path.join(path_assets, filename))
            if not definition:
                continue

            output = os.path.join(path_build, os.path.splitext(filename)[0] + '.map')
            folder = os.path.dirname(output)

            if not os.path.isdir(folder):
                os.makedirs(folder)

            print 'Building %s from %s' % (os.path.basename(output), os.path.basename(filename))
            definition.Write(output)