1 files changed, 76 insertions, 0 deletions
diff --git a/utils/update-map-db.py b/utils/update-map-db.py
new file mode 100644
index 0000000..a17f5f1
--- /dev/null
+++ b/utils/update-map-db.py
@@ -0,0 +1,76 @@
+#!/usr/bin/python2
+
+"""
+Update pickle file containing maps information.
+It is needed, because for some reason pytmx doesn't work
+properly on android.
+
+Author: Joseph Botosh <rumly111@gmail.com>
+Licence: GPLv2.
+"""
+
+import os
+import sys
+import zipfile
+import pytmx
+
+try:
+    import cPickle as pickle
+except:
+    import pickle
+
+try:
+    import zlib
+    compression = zipfile.ZIP_DEFLATED
+    del zlib
+except:
+    compression = zipfile.ZIP_STORED
+
+
+modes = { zipfile.ZIP_DEFLATED: 'deflated',
+          zipfile.ZIP_STORED:   'stored' }
+
+
+def LoadTmxMap(filename, maptag):
+    m = pytmx.TiledMap(filename=filename)
+
+    tmx = {
+        'tag': maptag,
+        'name': m.properties['name'],
+        'width': m.width,
+        'height': m.height,
+        'collisions': m.get_layer_by_name('Collision').data
+    }
+
+    return tmx
+
+
+def PrintHelp():
+    print('Usage: {} <maps-dir> <outfile.pickle>'.format(sys.argv[0]))
+
+if __name__ == '__main__':
+    if len(sys.argv) < 3:
+        PrintHelp()
+        sys.exit(0)
+    dirname = sys.argv[1]
+    outfile = sys.argv[2]
+
+    n = 0
+    zf = zipfile.ZipFile(outfile, 'w')
+
+    for tmx in (filter(lambda f: f.endswith('.tmx'),
+                       os.listdir(dirname))):
+        path = os.path.join(dirname, tmx)
+        maptag = tmx[:-4]
+        print("Loading map {} ...".format(path))
+        try:
+            c = LoadTmxMap(path, maptag)
+            print("\tname={} tag={} size=({},{})".format(
+                c['name'], c['tag'], c['width'], c['height']))
+            zf.writestr(maptag + '.pickle', pickle.dumps(c), compression)
+            n += 1
+        except KeyError as e:
+            print("Error loading {}: {}".format(tmx, e))
+
+    zf.close()
+    print("Done processing {} maps".format(n))