Minor refactoring of the Blender exporter

[libs/gl.git] / blender / io_mspgl / util.py

diff --git a/blender/io_mspgl/util.py b/blender/io_mspgl/util.py
index 6e8ec7746bde7c41b5c4791dfca602d7dd65503b..e77e544751abe7b9bcdbfe8ad968ec451b86b2ca 100644 (file)
--- a/blender/io_mspgl/util.py
+++ b/blender/io_mspgl/util.py
@@ -1,3 +1,5 @@
+import os
+
 class Progress:
        def __init__(self, context):
                self.task = ""
@@ -44,7 +46,7 @@ class Progress:
 
        def set_progress(self, value):
                value = self.start+self.delta*value
-               if value>self.last+0.01:
+               if value>self.last+0.001:
                        if self.window_manager:
                                self.window_manager.progress_update(value)
                        self.last = value
@@ -61,3 +63,25 @@ def get_colormap(srgb):
                return linear_to_srgb
        else:
                return lambda x: x
+
+def basename(path):
+       if path.startswith("//"):
+               path = path[2:]
+       return os.path.basename(path)
+
+def make_unique(values):
+       seen = set()
+       result = []
+       for i in values:
+               if i not in seen:
+                       result.append(i)
+                       seen.add(i)
+       return result
+
+def get_linked_node_and_socket(node_tree, socket):
+       for l in node_tree.links:
+               if socket==l.to_socket:
+                       return (l.from_node, l.from_socket)
+               elif socket==l.from_socket:
+                       return (l.to_node, l.to_socket)
+       return (None, None)