blender/io_mspgl/util.py

   1 import os
   2
   3 class Progress:
   4         def __init__(self, context):
   5                 self.task = ""
   6                 self.start = 0.0
   7                 self.delta = 1.0
   8                 self.last = 0.0
   9                 self.stack = []
  10                 if context:
  11                         self.window_manager = context.window_manager
  12                         self.window_manager.progress_begin(0.0, 1.0)
  13                 else:
  14                         self.window_manager = None
  15
  16         def push_task(self, task, low, high):
  17                 self.stack.append((self.task, self.start, self.delta))
  18                 self.set_task(task, low, high)
  19
  20         def push_task_slice(self, task, index, count):
  21                 self.push_task(task, index/count, (index+1)/count)
  22
  23         def pop_task(self):
  24                 if not self.stack:
  25                         return
  26
  27                 self.set_progress(1.0)
  28                 self.task, self.start, self.delta = self.stack.pop()
  29
  30         def set_task(self, task, low, high):
  31                 if self.stack:
  32                         outer = self.stack[-1]
  33                         if outer[0]:
  34                                 task = "{}: {}".format(outer[0], task)
  35                         low = outer[1]+low*outer[2]
  36                         high = outer[1]+high*outer[2]
  37
  38                 self.task = task
  39                 self.start = low
  40                 self.delta = high-low
  41
  42                 self.set_progress(0.0)
  43
  44         def set_task_slice(self, task, index, count):
  45                 self.set_task(task, index/count, (index+1)/count)
  46
  47         def set_progress(self, value):
  48                 value = self.start+self.delta*value
  49                 if value>self.last+0.01:
  50                         if self.window_manager:
  51                                 self.window_manager.progress_update(value)
  52                         self.last = value
  53
  54
  55 def linear_to_srgb(l):
  56         if l<0.0031308:
  57                 return 12.92*l
  58         else:
  59                 return 1.055*(l**(1/2.4))-0.055
  60
  61 def get_colormap(srgb):
  62         if srgb:
  63                 return linear_to_srgb
  64         else:
  65                 return lambda x: x
  66
  67 def image_name(img):
  68         fp = img.filepath
  69         if fp:
  70                 return os.path.split(fp)[1]
  71         else:
  72                 return img.name