コード例 #1
0
ファイル: magic_tol.py プロジェクト: xsyp/imagepy
    def mouse_down(self, ips, x, y, btn, **key):
        lim = 5.0 / key['canvas'].get_scale()
        if btn == 1 or btn == 3:
            if ips.roi != None:
                self.curobj = ips.roi.pick(x, y, lim)
            if not self.curobj in (None, True): return
            if ips.roi == None:
                msk = floodfill(ips.img, x, y, self.para['tor'],
                                self.para['con'] == '8-connect')
                conts = find_contours(msk, 0, 'high')
                ips.roi = shape2roi(polygonize(conts, btn == 3))
            elif hasattr(ips.roi, 'topolygon'):
                shp = roi2shape(ips.roi.topolygon())
                oper = ''
                if key['shift']: oper = '+'
                elif key['ctrl']: oper = '-'
                elif self.curobj: return
                else: ips.roi = None

                msk = floodfill(ips.img, x, y, self.para['tor'],
                                self.para['con'] == '8-connect')
                conts = find_contours(msk, 0, 'high')
                cur = polygonize(conts, btn == 3)
                if oper == '+':
                    ips.roi = shape2roi(shp.union(cur))
                elif oper == '-':
                    ips.roi = shape2roi(shp.difference(cur))
                else:
                    ips.roi = shape2roi(cur)

            else:
                ips.roi = None

        ips.update()
コード例 #2
0
 def run(self, ips, imgs, para = None):
     shp = convert.roi2shape(ips.roi)
     m = ips.img.mat
     m = [m[0,1], 0, 0, m[1,2], m[0,0], m[1,0]]
     shp = affine_transform(shp, m)
     shp = affine_transform(shp, [0,1,1,0,0,0])
     gdf = gpd.GeoDataFrame([[shp]], columns=['geometry'], crs=ips.img.crs)
     gdf.to_file(para['path'])        
コード例 #3
0
    def mouse_down(self, ips, x, y, btn, **key): 
        lim = 5.0/key['canvas'].get_scale() 
        if btn==1 or btn==3:
            if ips.roi!= None:
                self.curobj = ips.roi.pick(x, y, ips.cur, lim)
                ips.roi.info(ips, self.curobj)
            if not self.curobj in (None,True):return
            if ips.roi == None:
                connectivity=(self.para['con']=='8-connect')+1
                img = ips.img.reshape((ips.img.shape+(1,))[:3])
                msk = np.ones(img.shape[:2], dtype=np.bool)
                for i in range(img.shape[2]):
                    msk &= flood(img[:,:,i], (int(y),int(x)), 
                        connectivity=connectivity, tolerance=self.para['tor'])
                conts = find_contours(msk, 0, 'high')
                ips.roi = shape2roi(polygonize(conts, btn==3))
            elif hasattr(ips.roi, 'topolygon'):
                shp = roi2shape(ips.roi.topolygon())
                oper = ''
                if key['shift']: oper = '+'
                elif key['ctrl']: oper = '-'
                elif self.curobj: return
                else: ips.roi=None

                connectivity=(self.para['con']=='8-connect')+1
                img = ips.img.reshape((ips.img.shape+(1,))[:3])
                msk = np.ones(img.shape[:2], dtype=np.bool)
                for i in range(img.shape[2]):
                    msk &= flood(img[:,:,i], (int(y),int(x)), 
                        connectivity=connectivity, tolerance=self.para['tor'])
                conts = find_contours(msk, 0, 'high')
                cur = polygonize(conts, btn==3)
                if oper == '+':
                    ips.roi = shape2roi(shp.union(cur))
                elif oper == '-':
                    ips.roi = shape2roi(shp.difference(cur))
                else: ips.roi = shape2roi(cur)

            else: ips.roi = None

        ips.update()