From 1cbfff09d7f948d83dc75c6772b8d82c4c4bce0b Mon Sep 17 00:00:00 2001
From: mwerezak <mwerezak@gmail.com>
Date: Fri, 12 Jun 2015 00:06:58 -0400
Subject: [PATCH] Adds python wrapper for dmitool

---
 tools/dmitool/dmitool.py | 91 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 91 insertions(+)
 create mode 100644 tools/dmitool/dmitool.py

diff --git a/tools/dmitool/dmitool.py b/tools/dmitool/dmitool.py
new file mode 100644
index 0000000000..f24ce2c590
--- /dev/null
+++ b/tools/dmitool/dmitool.py
@@ -0,0 +1,91 @@
+""" Python 2.7 wrapper for dmitool.
+"""
+
+import os
+from subprocess import Popen, PIPE
+
+_JAVA_PATH = ["java"]
+_DMITOOL_CMD = ["-jar", "dmitool.jar"]
+
+def _dmitool_call(*dmitool_args, **popen_args):
+    return Popen(_JAVA_PATH + _DMITOOL_CMD + list(dmitool_args), **popen_args)
+
+def _safe_parse(dict, key, deferred_value):
+    try:
+        dict[key] = deferred_value()
+    except Exception as e: 
+        print "Could not parse property '%s': %s"%(key, e)
+        return e
+    return False
+
+def version():
+    """ Prints the version to stdout. """
+    return _dmitool_call("version")
+
+def help():
+    """ Prints the help text to stdout. """
+    return _dmitool_call("help")
+
+def info(filepath):
+    """ Totally not a hack that parses the output from dmitool into a dictionary. 
+        May break at any moment.
+    """
+    subproc = _dmitool_call("info", filepath, stdout=PIPE)
+    stdout, stderr = subproc.communicate()
+    
+    result = {}
+    data = stdout.split(os.linesep)[1:]
+    #for s in data: print s
+    
+    #parse header line
+    if len(data) > 0:
+        header = data.pop(0).split(",")
+        #don't need to parse states, it's redundant
+        _safe_parse(result, "images", lambda: int(header[0].split()[0].strip()))
+        _safe_parse(result, "size", lambda: header[2].split()[1].strip())
+    
+    #parse state information
+    states = []    
+    for item in data:
+        if not len(item): continue
+    
+        stateinfo = {}
+        item = item.split(",", 3)
+        _safe_parse(stateinfo, "name", lambda: item[0].split()[1].strip(" \""))
+        _safe_parse(stateinfo, "dirs", lambda: int(item[1].split()[0].strip()))
+        _safe_parse(stateinfo, "frames", lambda: int(item[2].split()[0].strip()))
+        if len(item) > 3:
+            stateinfo["misc"] = item[3]
+        
+        states.append(stateinfo)
+    
+    result["states"] = states
+    return result
+
+def extract_state(input_path, output_path, icon_state, direction=None, frame=None):
+    """ Extracts an icon state as a png to a given path.
+        If provided direction should be a string, one of S, N, E, W, SE, SW, NE, NW.
+        If provided frame should be a frame number or a string of two frame number separated by a dash.
+    """
+    args = ["extract", input_path, icon_state, output_path]
+    if direction: args.extend(("direction" , str(direction)))
+    if frame: args.extend(("frame" , str(frame)))
+    return _dmitool_call(*args)
+
+def import_state(target_path, input_path, icon_state, replace=False, delays=None, rewind=False, loop=None, ismovement=False, direction=None):
+    """ Inserts an input png given by the input_path into the target_path.
+    """
+    args = ["import", target_path, icon_state, input_path]
+    
+    if replace: args.append("nodup")
+    if rewind: args.append("rewind")
+    if ismovement: args.append("movement")
+    if delays: args.extend(("delays", ",".join(delays)))
+    if direction: args.extend(("direction", direction))
+    
+    if loop in ("inf", "infinity"):
+        args.append("loop")
+    elif loop:
+        args.extend(("loopn", loop))
+    
+    return _dmitool_call(*args)