rework settings, add pyqtUsrObj.py
This commit is contained in:
135
pyqtUsrObj.py
Normal file
135
pyqtUsrObj.py
Normal file
@@ -0,0 +1,135 @@
|
||||
#!/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_()
|
||||
|
||||
Reference in New Issue
Block a user