136 lines
4.4 KiB
Python
136 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Demonstrates a variety of uses for ROI. This class provides a user-adjustable
|
|
region of interest marker. It is possible to customize the layout and
|
|
function of the scale/rotate handles in very flexible ways.
|
|
"""
|
|
|
|
#import initExample ## Add path to library (just for examples; you do not need this)
|
|
|
|
import pyqtgraph as pg
|
|
from pyqtgraph.Qt import QtCore, QtGui
|
|
import numpy as np
|
|
|
|
|
|
class BeamMark(pg.ROI):
|
|
"""A crosshair ROI whose position is at the center of the crosshairs. By default, it is scalable, rotatable and translatable."""
|
|
|
|
def __init__(self, pos=None, size=None, parent=None, **kargs):
|
|
#self._size=size
|
|
#self._pos=pos
|
|
#self._shape=None
|
|
pg.ROI.__init__(self, pos, size, **kargs)
|
|
|
|
def paint(self, p, *args):
|
|
#pg.ROI.paint(self, p, *args)
|
|
r=QtCore.QRectF(0, 0, self.state['size'][0], self.state['size'][1]).normalized()
|
|
p.setRenderHint(QtGui.QPainter.Antialiasing)
|
|
p.setPen(self.currentPen)
|
|
p.translate(r.left(), r.top())
|
|
p.scale(r.width(), r.height())
|
|
p.drawEllipse(0, 0, 1, 1)
|
|
p.drawRect(0, 0, 1, 1)
|
|
p.setPen(pg.mkPen(width=3, color=[200, 100, 100]))
|
|
p.drawLine(pg.Point(.5, 0), pg.Point(.5, 1))
|
|
p.drawLine(pg.Point( 0,.5), pg.Point( 1,.5))
|
|
#w, h = self.getState()["size"]
|
|
#v1, v2 = -(h * 0.8) / 2, (h * 0.8) / 2
|
|
#h1, h2 = -(w * 0.8) / 2, (w * 0.8) / 2
|
|
#p.setRenderHint(QtGui.QPainter.Antialiasing)
|
|
#p.setPen(pg.mkPen(width=3, color=[200, 100, 100]))
|
|
#p.drawLine(pg.Point(0, v1), pg.Point(0, v2))
|
|
#p.drawLine(pg.Point(h1, 0), pg.Point(h2, 0))
|
|
#p.setPen(self.currentPen)
|
|
##p.setPen(pg.mkPen(width=3, color=[200, 200, 100]))
|
|
#p.drawRect(-w / 2, -h / 2, w, h)
|
|
## p.drawText(-w, -h, '{:.0f}x{:.0f}'.format(*self._size))
|
|
|
|
|
|
|
|
## Start Qt event loop unless running in interactive mode or using pyside.
|
|
if __name__=='__main__':
|
|
import sys
|
|
import argparse
|
|
|
|
#(h, t)=os.path.split(sys.argv[0]);cmd='\n '+(t if len(h)>20 else sys.argv[0])+' '
|
|
#exampleCmd=('', '-m0xf -v0')
|
|
epilog=__doc__ #+'\nExamples:'+''.join(map(lambda s:cmd+s, exampleCmd))+'\n'
|
|
parser=argparse.ArgumentParser(epilog=epilog, formatter_class=argparse.RawDescriptionHelpFormatter)
|
|
parser.add_argument("--mode", "-m", type=lambda x: int(x,0), help="mode default=0x%(default)x", default=0)
|
|
args = parser.parse_args()
|
|
|
|
pg.setConfigOptions(imageAxisOrder='row-major')
|
|
|
|
|
|
def obj_info(obj):
|
|
pos=obj.pos()
|
|
scnPos=obj.scenePos()
|
|
t=obj.transform()
|
|
obj_info
|
|
print(f"obj_info:{obj}")
|
|
print(f"Pos:({pos.x():.6g},{pos.y():.6g})") #in coordinate value on the scene (no change by zooming)
|
|
print(f"scenePos:({scnPos.x():.6g},{scnPos.y():.6g})") #in pixel on the scene (changes by zooming)
|
|
print(f"QTransform:{t.m11():8.5g} {t.m12():8.5g} {t.m13():8.5g}")
|
|
print(f" {t.m21():8.5g} {t.m22():8.5g} {t.m23():8.5g}")
|
|
print(f" {t.m31():8.5g} {t.m32():8.5g} {t.m33():8.5g}")
|
|
|
|
|
|
def pt2str(p):
|
|
return (f'({p.x():.5g},{p.y():.5g})')
|
|
|
|
def childTree(obj, lvl=0):
|
|
print('+'*lvl+str(obj))
|
|
for child in obj.childItems():
|
|
childTree(child, lvl+1)
|
|
|
|
def mouse_click_event(event):
|
|
pos=event.pos()
|
|
scn=event.scenePos() # there is a small black border, that makes the difference
|
|
img=viImg.mapFromScene(scn)
|
|
roi=viUsrRoi.mapFromScene(scn)
|
|
print(f'mouse:{pt2str(pos)} {pt2str(scn)} img:{pt2str(img)} roi:{pt2str(roi)}')
|
|
#childTree(vb)
|
|
obj_info(viUsrRoi)
|
|
|
|
## Create image to display
|
|
arr=np.ones((100, 100), dtype=float)
|
|
arr[45:55, 45:55]=0
|
|
arr[25, :]=5
|
|
arr[:, 25]=5
|
|
arr[75, :]=5
|
|
arr[:, 75]=5
|
|
arr[50, :]=10
|
|
arr[:, 50]=10
|
|
arr+=np.sin(np.linspace(0, 20, 100)).reshape(1, 100)
|
|
arr+=np.random.normal(size=(100, 100))
|
|
|
|
# add an arrow for asymmetry
|
|
arr[10, :50]=10
|
|
arr[9:12, 44:48]=10
|
|
arr[8:13, 44:46]=10
|
|
|
|
## create GUI
|
|
app=QtGui.QApplication([])
|
|
w=pg.GraphicsLayoutWidget(show=True, size=(1000, 800), border=True)
|
|
w.setWindowTitle('pyqtgraph example: ROI Examples')
|
|
|
|
vb=w.addViewBox(row=1, col=0, lockAspect=True)
|
|
g=pg.GridItem()
|
|
vb.addItem(g)
|
|
|
|
viImg=pg.ImageItem(arr, border='y')
|
|
vb.addItem(viImg)
|
|
# Custom ROI for selecting an image region
|
|
viRoi=pg.ROI([20, 50], [100, 70])
|
|
vb.addItem(viRoi)
|
|
viUsrRoi=BeamMark([10, 20], [30, 20])
|
|
vb.addItem(viUsrRoi)
|
|
childTree(vb)
|
|
|
|
w.scene().sigMouseClicked.connect(mouse_click_event)
|
|
|
|
if (sys.flags.interactive!=1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
|
QtGui.QApplication.instance().exec_()
|
|
|