Add a helper module to the chroma.models directory that takes

inventory of all the .stl/.stl.bz2 files present and creates building
functions for each of them in the models module.

This allows usage like the following:

chroma-cam chroma.models.lionsolid
chroma-cam chroma.models.tie_interceptor6

You don't need to worry about where the chroma package was actually
installed.  Loading from STL files listed on the command line still
works, of course.

1 files changed, 20 insertions, 0 deletions

diff --git a/chroma/models/__init__.py b/chroma/models/__init__.py
new file mode 100644
index 0000000..4b19f94
--- /dev/null
+++ b/chroma/models/__init__.py
@@ -0,0 +1,20 @@
+import os.path
+import glob
+import sys
+
+from chroma.stl import mesh_from_stl
+
+class Loader(object):
+    def __init__(self, filename):
+        self.filename = filename
+    def __call__(self):
+        return mesh_from_stl(self.filename)
+
+# Create functions to load
+this_module = sys.modules[__name__]
+for filename in glob.glob(os.path.join(os.path.dirname(__file__),'*.stl*')):
+    name, ext = os.path.splitext(os.path.basename(filename))
+    while ext != '':
+        name, ext = os.path.splitext(name)
+    setattr(this_module, name, Loader(filename))
+