Implement material maps for exporting objects with multiple materials

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

diff --git a/blender/io_mspgl/util.py b/blender/io_mspgl/util.py
index 000190fe05e2cb80510121c90afd945378dd3ba6..70b005f6df60e84db38ea8c35c03fc0bbcc79123 100644 (file)
--- a/blender/io_mspgl/util.py
+++ b/blender/io_mspgl/util.py
@@ -1,20 +1,72 @@
+import os
+
 class Progress:
        def __init__(self, context):
                self.task = ""
                self.start = 0.0
                self.delta = 1.0
                self.last = 0.0
-               self.window_manager = context.window_manager
-               self.window_manager.progress_begin(0.0, 1.0)
+               self.stack = []
+               if context:
+                       self.window_manager = context.window_manager
+                       self.window_manager.progress_begin(0.0, 1.0)
+               else:
+                       self.window_manager = None
+
+       def push_task(self, task, low, high):
+               self.stack.append((self.task, self.start, self.delta))
+               self.set_task(task, low, high)
+
+       def push_task_slice(self, task, index, count):
+               self.push_task(task, index/count, (index+1)/count)
+
+       def pop_task(self):
+               if not self.stack:
+                       return
+
+               self.set_progress(1.0)
+               self.task, self.start, self.delta = self.stack.pop()
 
        def set_task(self, task, low, high):
+               if self.stack:
+                       outer = self.stack[-1]
+                       if outer[0]:
+                               task = "{}: {}".format(outer[0], task)
+                       low = outer[1]+low*outer[2]
+                       high = outer[1]+high*outer[2]
+
                self.task = task
                self.start = low
                self.delta = high-low
-               self.set_progress(0)
+
+               self.set_progress(0.0)
+
+       def set_task_slice(self, task, index, count):
+               self.set_task(task, index/count, (index+1)/count)
 
        def set_progress(self, value):
                value = self.start+self.delta*value
                if value>self.last+0.01:
-                       self.window_manager.progress_update(value)
+                       if self.window_manager:
+                               self.window_manager.progress_update(value)
                        self.last = value
+
+
+def linear_to_srgb(l):
+       if l<0.0031308:
+               return 12.92*l
+       else:
+               return 1.055*(l**(1/2.4))-0.055
+
+def get_colormap(srgb):
+       if srgb:
+               return linear_to_srgb
+       else:
+               return lambda x: x
+
+def image_name(img):
+       fp = img.filepath
+       if fp:
+               return os.path.split(fp)[1]
+       else:
+               return img.name