ai brush

Author: yxdragon

imagepy/core/manager/colormanager.py

@@ -50,7 +50,7 @@ def get_front(cls, one=False):
         return np.dot((cls.wr,cls.wg,cls.wb), cls.frontcolor)
 
     @classmethod
-    def get_back(cls, one):
+    def get_back(cls, one=False):
         if not one:return cls.backcolor
         return np.dot((cls.wr,cls.wg,cls.wb), cls.backcolor)
 

imagepy/core/mark/mark.py

@@ -251,7 +251,6 @@ def draw_text(pts, dc, f, **key):
 	brush.SetColour(pen.GetColour())
 	brush.SetStyle(100)
 	if 'fcolor' in pts:
-		print('hahaha')
 		dc.SetTextBackground(pts['fcolor'])
 	if 'size' in pts:
 		font.SetPointSize(pts['size'])

imagepy/ipyalg/graph/connect.py

@@ -1,5 +1,6 @@
 import numpy as np
 from numba import jit
+import random
 from scipy.ndimage import generate_binary_structure
 
 def neighbors(shape, conn=1):
@@ -13,26 +14,45 @@ def neighbors(shape, conn=1):
     return np.dot(idx, acc[::-1])
 
 @jit(nopython=True)
-def search(img, nbs):
+def unique(idx):
+    msk = idx[:,0]<idx[:,1]
+    key = idx[:,0]<<16
+    key += idx[:,1]
+    sort = np.argsort(key)
+    idx[:] = idx[sort]
+    s = i = 1
+    while s<len(idx):
+        if key[sort[s]]!=key[sort[s-1]]:
+            idx[i,0] = idx[s,0]
+            idx[i,1] = idx[s,1]
+            i += 1
+        s += 1
+    return i
+    
+@jit(nopython=True)
+def search(img, nbs, back):
     s, line = 0, img.ravel()
-    rst = np.zeros((len(line),2), img.dtype)
+    rst = np.zeros((len(line)//2, 2), img.dtype)
     for i in range(len(line)):
         if line[i]==0:continue
         for d in nbs:
-            if line[i+d]==0: continue
+            if not back and line[i+d]==0: continue
             if line[i]==line[i+d]: continue
             rst[s,0] = line[i]
             rst[s,1] = line[i+d]
             s += 1
+            
+            if s==len(rst):
+                s = unique(rst)
     return rst[:s]
-
-def connect(img, conn=1):
+                            
+def connect_graph(img, conn=1, back=False):
     buf = np.pad(img, 1, 'constant')
     nbs = neighbors(buf.shape, conn)
-    rst = search(buf, nbs)
-    if len(rst)==0: return rst
-    rst.sort()
-    return np.unique(rst, axis=0)
+    rst = search(buf, nbs, back)
+    if len(rst)<2: return rst
+    rst.sort(axis=1)
+    return rst[:unique(rst)].copy()
 
 def mapidx(idx):
     dic = {}
@@ -47,5 +67,5 @@ def mapidx(idx):
                     [1,1,2,2,1],
                     [1,3,0,0,1],
                     [1,3,1,1,4]])
-    rst = connect(img, 2)
+    rst = connect_graph(img, 2, True)
     print(rst)

imagepy/ipyalg/graph/render.py

@@ -0,0 +1,27 @@
+import random
+
+def node_render(conmap, n=4, rand=10, shuffle=True):
+    nodes = list(conmap.keys())
+    colors = dict(zip(nodes, [0]*len(nodes)))
+    counter = dict(zip(nodes, [0]*len(nodes)))
+    if shuffle: random.shuffle(nodes)
+    while len(nodes)>0:
+        k = nodes.pop(0)
+        counter[k] += 1
+        hist = [1e4] + [0] * n
+        for p in conmap[k]:
+            hist[colors[p]] += 1
+        if min(hist)==0:
+            colors[k] = hist.index(min(hist))
+            counter[k] = 0
+            continue
+        hist[colors[k]] = 1e4
+        minc = hist.index(min(hist))
+        if counter[k]==rand:
+            counter[k] = 0
+            minc = random.randint(1,4)
+        colors[k] = minc
+        for p in conmap[k]:
+            if colors[p] == minc:
+                nodes.append(p)
+    return colors

imagepy/menus/Analysis/Region Analysis/connect_plg.py

@@ -4,16 +4,17 @@
 from skimage.measure import regionprops
 from imagepy.core.mark import GeometryMark
 from scipy.ndimage import label, generate_binary_structure
-from imagepy.ipyalg.graph.connect import connect, mapidx
+from imagepy.ipyalg.graph.connect import connect_graph, mapidx
 import pandas as pd
 
 # center, area, l, extent, cov
 class Plugin(Simple):
     title = 'Connective Analysis'
     note = ['8-bit', '16-bit', 'int']
-    para = {'con':'8-connect', 'labled':False, 'slice':False}
+    para = {'con':'8-connect', 'labled':False, 'nozero':True, 'slice':False}
     view = [(list, 'con', ['4-connect', '8-connect'], str, 'conection', 'pix'),
             (bool, 'labled', 'it is a label image'),
+            (bool, 'nozero', 'nonzero'),
             (bool, 'slice', 'slice')]
 
     #process
@@ -27,7 +28,7 @@ def run(self, ips, imgs, para = None):
         for i in range(len(imgs)):
             if para['labled']: buf = imgs[i]
             else: label(imgs[i], generate_binary_structure(2, 1), output=buf)
-            conarr = connect(buf, 1 if para['con']=='4-connect' else 2)
+            conarr = connect(buf, 1 if para['con']=='4-connect' else 2, not self.para['nozero'])
             conmap = mapidx(conarr)
 
             ls = regionprops(buf)

imagepy/menus/Analysis/label_plg.py

@@ -5,38 +5,56 @@
 """
 import numpy as np
 from scipy.ndimage import label, generate_binary_structure
-
 from imagepy import IPy
-from imagepy.core.engine import Filter
-from imagepy.ui.canvasframe import CanvasFrame
+from imagepy.core import ImagePlus
+from imagepy.core.engine import Filter, Simple
+from imagepy.ipyalg.graph import connect, render
 
-class Plugin(Filter):
+class Label(Simple):
     title = 'Label Image'
-    note = ['8-bit', 'not_slice', 'preview']
-    
-    para = {'thr':128, 'con':'4-Connect'}
-    view = [('slide', 'thr', (0,255), 0, 'Threshold'),
-            (list, 'con', ['4-Connect','8-Connect'], str, 'Structure', 'connect')]
-        
-    def load(self, ips):
-        self.lut = ips.lut
-        ips.lut = self.lut.copy()
-        return True
-        
-    def preview(self, ips, para):
-        ips.lut[:] = self.lut
-        ips.lut[para['thr']:] = [255,0,0]
-        ips.update()
-        
-    def run(self, ips, snap, img, para = None):
-        if para == None: para = self.para
-        ips.lut = self.lut
-        msk = img>para['thr']
-        con = 1 if para['con']=='4-Connect' else 2
-        strc = generate_binary_structure(2, con)
-        msk = label(msk, strc, output = np.int16)
+    note = ['8-bit', '16-bit']    
+    para = {'slice':False, 'con':'4-Connect'}
+    view = [(list, 'con', ['4-Connect','8-Connect'], str, 'Structure', 'connect'),
+            (bool, 'slice', 'slice')]
 
-        IPy.show_img([msk[0]], ips.title+'-label') 
+    def run(self, ips, imgs, para = None):
+        if not para['slice']:  imgs = [ips.img]
+        labels = []
+        for i in range(len(imgs)):
+            self.progress(i, len(imgs))
+            con = 1 if para['con']=='4-Connect' else 2
+            strc = generate_binary_structure(2, con)
+            lab, n = label(imgs[i], strc, output = np.int32)
+            labels.append(lab)
+        IPy.show_img(labels, ips.title+'-label') 
 
+class Render(Simple):
+    title = 'Label Render'
+    note = ['8-bit', '16-bit', 'int']    
+    para = {'slice':False, 'con':'8-Connect', 'colors':6, 'back':False}
+    view = [(list, 'con', ['4-Connect','8-Connect'], str, 'Structure', 'connect'),
+            (int, 'colors', (4, 255), 0, 'colors', 'n'),
+            (bool, 'back', 'background'),
+            (bool, 'slice', 'slice')]
 
-        
+    def run(self, ips, imgs, para = None):
+        if not para['slice']:  imgs = [ips.img]
+        print(para)
+        labels = []
+        for i in range(len(imgs)):
+            self.progress(i, len(imgs))
+            con = 1 if para['con']=='4-Connect' else 2
+            idx = connect.connect_graph(imgs[i], con, para['back'])
+            idx = connect.mapidx(idx)
+            cmap = render.node_render(idx, para['colors'], 10)
+
+            lut = np.ones(imgs[i].max()+1, dtype=np.uint8)
+            lut[0] = 0
+            for j in cmap: lut[j] = cmap[j]
+            labels.append(lut[imgs[i]])
+
+        ips = ImagePlus(labels, ips.title+'-render')
+        ips.range = (0, para['colors'])
+        IPy.show_ips(ips) 
+
+plgs = [Label, Render]

imagepy/menus/Process/Segment/shift_plgs.py

@@ -37,7 +37,7 @@ class Felzenszwalb(Filter):
     note = ['all', 'not_slice', 'auto_snap', 'auto_msk', 'not_channel', 'preview']
 
     para = {'scale':1.0, 'sigma':0.8, 'min_size':20}
-    view = [(float, 'scale', (0.01, 10), 2, 'scale', ''),
+    view = [(float, 'scale', (0.01, 1024), 2, 'scale', ''),
             (float, 'sigma', (0, 30), 2, 'sigma', ''),
             (int, 'min_size', (1, 1024), 0, 'min_size', '')]
 
@@ -51,7 +51,7 @@ class SLICLab(Simple):
 
     para = {'n_segments':100, 'compactness':10.0, 'max_iter':10, 'sigma':0, 'stack':False}
     view = [(int, 'n_segments', (1, 1e8), 0, 'segments', 'n'),
-            (float, 'compactness', (0.01, 100), 2, 'campactness', 'color-space'),
+            (float, 'compactness', (0.01, 1024), 2, 'campactness', 'color-space'),
             (int, 'max_iter', (3, 50), 0, 'max_iter', 'n'),
             (float, 'sigma', (0, 30), 1, 'sigma', 'smooth'),
             (bool, 'stack', 'stack')]

imagepy/tools/Draw/__init__.py

@@ -1 +1 @@
-catlog = ['magic_tol', 'floodfill_tol', 'flood3d_tol']
+catlog = ['magic_tol', 'floodfill_tol', 'flood3d_tol', 'aibrush_tol']