integrate inx generation into ink/stitch proper

6 files changed, 92 insertions, 0 deletions

diff --git a/lib/inx/__init__.py b/lib/inx/__init__.py
new file mode 100644
index 00000000..32b8bfae
--- /dev/null
+++ b/lib/inx/__init__.py
@@ -0,0 +1 @@
+from generate import generate_inx_files
diff --git a/lib/inx/extensions.py b/lib/inx/extensions.py
new file mode 100755
index 00000000..ba206d8f
--- /dev/null
+++ b/lib/inx/extensions.py
@@ -0,0 +1,21 @@
+import pyembroidery
+
+from .utils import build_environment, write_inx_file
+from .outputs import pyembroidery_output_formats
+from ..extensions import extensions
+
+
+def pyembroidery_debug_formats():
+    for format in pyembroidery.supported_formats():
+        if 'writer' in format and format['category'] != 'embroidery':
+            yield format['extension'], format['description']
+
+
+def generate_extension_inx_files():
+    env = build_environment()
+
+    for extension in extensions:
+        name = extension.name()
+        template = env.get_template('%s.inx' % name)
+        write_inx_file(name, template.render(formats=pyembroidery_output_formats(),
+                                             debug_formats=pyembroidery_debug_formats()))
diff --git a/lib/inx/generate.py b/lib/inx/generate.py
new file mode 100644
index 00000000..4773a92d
--- /dev/null
+++ b/lib/inx/generate.py
@@ -0,0 +1,8 @@
+from .inputs import generate_input_inx_files
+from .outputs import generate_output_inx_files
+from .extensions import generate_extension_inx_files
+
+def generate_inx_files():
+    generate_input_inx_files()
+    generate_output_inx_files()
+    generate_extension_inx_files()
diff --git a/lib/inx/inputs.py b/lib/inx/inputs.py
new file mode 100755
index 00000000..d40ffeaf
--- /dev/null
+++ b/lib/inx/inputs.py
@@ -0,0 +1,18 @@
+import pyembroidery
+
+from .utils import build_environment, write_inx_file
+
+
+def pyembroidery_input_formats():
+    for format in pyembroidery.supported_formats():
+        if 'reader' in format and format['category'] == 'embroidery':
+            yield format['extension'], format['description']
+
+
+def generate_input_inx_files():
+    env = build_environment()
+    template = env.get_template('input.inx')
+
+    for format, description in pyembroidery_input_formats():
+        name = "input_%s" % format.upper()
+        write_inx_file(name, template.render(format=format, description=description))
diff --git a/lib/inx/outputs.py b/lib/inx/outputs.py
new file mode 100644
index 00000000..aef0c8b5
--- /dev/null
+++ b/lib/inx/outputs.py
@@ -0,0 +1,18 @@
+import pyembroidery
+
+from .utils import build_environment, write_inx_file
+
+
+def pyembroidery_output_formats():
+    for format in pyembroidery.supported_formats():
+        if 'writer' in format and format['category'] == 'embroidery':
+            yield format['extension'], format['description']
+
+
+def generate_output_inx_files():
+    env = build_environment()
+    template = env.get_template('output.inx')
+
+    for format, description in pyembroidery_output_formats():
+        name = "output_%s" % format.upper()
+        write_inx_file(name, template.render(format=format, description=description))
diff --git a/lib/inx/utils.py b/lib/inx/utils.py
new file mode 100644
index 00000000..da0a4614
--- /dev/null
+++ b/lib/inx/utils.py
@@ -0,0 +1,26 @@
+import os
+from os.path import dirname
+from jinja2 import Environment, FileSystemLoader
+
+from ..i18n import translation as inkstitch_translation
+
+
+_top_path = dirname(dirname(dirname(os.path.realpath(__file__))))
+inx_path = os.path.join(_top_path, "inx")
+template_path = os.path.join(_top_path, "templates")
+
+def build_environment():
+    env = Environment(
+        loader = FileSystemLoader(template_path),
+        autoescape = True,
+        extensions=['jinja2.ext.i18n']
+    )
+
+    env.install_gettext_translations(inkstitch_translation)
+
+    return env
+
+def write_inx_file(name, contents):
+    inx_file_name = "inkstitch_%s.inx" % name
+    with open(os.path.join(inx_path, inx_file_name), 'w') as inx_file:
+        print >> inx_file, contents