lut

Author: yxdragon

imagepy/core/manager/colormanager.py

@@ -11,22 +11,13 @@
 import sys
 from glob import glob
 
-
-
-
-# read from the lut binarycode
-# glob: return a list ot paths matching a pathname pattern.
 filenames = glob(os.path.join(root_dir,'data/luts/*.lut'))
+filenames.extend(glob(os.path.join(root_dir,'data/luts/*/*.lut')))
 keys = [os.path.split(filename)[-1][:-4] for filename in filenames]
 values = [np.fromfile(filename, dtype=np.uint8).reshape((3,256)).T.copy() for filename in filenames]
 
-filenames_others = glob(os.path.join(root_dir,'data/luts/Others/*.lut'))
-keys_others = [os.path.split(filename)[-1][:-4] for filename in filenames_others]
-values_others = [np.fromfile(filename, dtype=np.uint8).reshape((3,256)).T.copy() for filename in filenames_others]
-
 class ColorManager:
     luts = dict(zip(keys, values))
-    luts_others = dict(zip(keys_others, values_others))
     frontcolor = (255,255,0)
     backcolor = (0,0,0)
     wr, wg, wb = 1.0/3, 1.0/3, 1.0/3
@@ -64,14 +55,8 @@ def get_back(cls, one):
         return np.dot((cls.wr,cls.wg,cls.wb), cls.backcolor)
 
     @classmethod
-    def get_lut(cls, name='grays'):
-        if name=='grays':
-            lut = np.arange(256).reshape((-1,1))
-            return (lut*np.ones((1,3))).astype(np.uint8)
-        else: return cls.luts[name].copy()
-    @classmethod
-    def get_lut_others(cls, name='grays'):
+    def get_lut(cls, name='grays', set='all'):
         if name=='grays':
             lut = np.arange(256).reshape((-1,1))
             return (lut*np.ones((1,3))).astype(np.uint8)
-        else: return cls.luts_others[name].copy()
+        else: return cls.luts[name].copy()

imagepy/data/luts/Orange_Hot.lut

@@ -3,44 +3,9 @@
 Created on Mon Oct 17 21:13:42 2016
 @author: yxl
 """
-from imagepy import IPy
-import numpy as np
-from imagepy.core import ImagePlus
-from imagepy.ui.canvasframe import CanvasFrame
-from imagepy.core.engine import Free
-from imagepy.core.manager import ColorManager
+from imagepy import IPy, root_dir
+import os, glob
+from ..lookuptables_plg import Plugin
 
-class Plugin(Free):
-    def __init__(self, key):
-        self.title = key
-    
-    def load(self):
-        plus = IPy.get_ips()
-        if plus==None:
-            img = np.ones((30,1), dtype=np.uint8) * np.arange(256, dtype=np.uint8)
-            ips = ImagePlus([img], self.title)
-            ips.lut = ColorManager.get_lut_others(self.title)
-            IPy.show_ips(ips)
-            return False
-        elif plus.channels != 1:
-            IPy.alert('RGB image do not surport Lookup table!')
-            return False
-        return True
-
-    #process
-    def run(self, para = None):
-        plus = IPy.get_ips()
-        plus.lut = ColorManager.get_lut_others(self.title)
-        plus.update()
-    
-    def __call__(self):
-        return self
-          
-plgs = [Plugin(i) for i in list(ColorManager.luts_others.keys())]
-for i in range(len(plgs)):
-    if plgs[i].title == 'Grays':
-        plgs.insert(0, plgs.pop(i))
-
-if __name__ == '__main__':
-    print(list(ColorManager.luts_others.keys()))
-    
+fs = glob.glob(os.path.join(root_dir,'data/luts/Others/*.lut'))
+plgs = [Plugin(i) for i in [os.path.split(f)[-1][:-4] for f in fs]]

imagepy/data/luts/Others/Rainbow RGB.lut imagepy/data/luts/Others/Rainbow_RGB.lutimagepy/data/luts/Others/Rainbow RGB.lut renamed to imagepy/data/luts/Others/Rainbow_RGB.lut

@@ -3,12 +3,13 @@
 Created on Mon Oct 17 21:13:42 2016
 @author: yxl
 """
-from imagepy import IPy
+from imagepy import IPy, root_dir
 import numpy as np
 from imagepy.core import ImagePlus
 from imagepy.ui.canvasframe import CanvasFrame
 from imagepy.core.engine import Free
 from imagepy.core.manager import ColorManager
+import os, glob
 
 class Plugin(Free):
     def __init__(self, key):
@@ -36,7 +37,9 @@ def run(self, para = None):
     def __call__(self):
         return self
 
-plgs = [Plugin(i) for i in list(ColorManager.luts.keys())]
+fs = glob.glob(os.path.join(root_dir,'data/luts/*.lut'))
+plgs = [Plugin(i) for i in [os.path.split(f)[-1][:-4] for f in fs]]
+
 for i in range(len(plgs)):
     if plgs[i].title == 'Grays':
         plgs.insert(0, plgs.pop(i))