def _done(self, xxx_todo_changeme):
(x, fn, y_avg, y_vals, w_vals, i) = xxx_todo_changeme
if self.pPlotResult.value():
if self.outDisplay is None or self.outDisplay.isClosed():
self.outDisplay = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'),
title='Noise level functions')
else:
self.outDisplay.removeLayers()
if self.outDisplay2 is None or self.outDisplay2.isClosed():
self.outDisplay2 = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'), title='NLF pointdensity')
else:
self.outDisplay2.removeLayers()
for n, yi in enumerate(y_vals):
self.outDisplay.addLayer(label='plot %s' % n, data=[x, yi])
pen = pg.mkPen('r')
pen.setWidth(3)
self.outDisplay.addLayer(label='average', data=[x, y_avg], pen=pen)
pen = pg.mkPen('g')
pen.setWidth(5)
self.outDisplay.addLayer(label='fit', data=[x, fn(x)], pen=pen)
self.outDisplay2.addLayer(label='density',
data=[x, w_vals],
pen=pen)
self.pUpdate.show()
def _startruler(self, evt):
self.scene.sigMouseClicked.disconnect(self._startruler)
self.rulers.append(
pg.PlotDataItem(
symbol="+",
pen=pg.mkPen(color=self.pColor.value(), width=2),
symbolSize=16,
symbolBrush=pg.mkBrush(self.pColor.value()),
)
)
self.rulersLen.append(
pg.TextItem(
text="",
color=(0, 0, 0),
html=None,
anchor=(0, 0),
border=None,
fill=pg.mkBrush(self.pColor.value()),
angle=0,
)
)
self.view.addItem(self.rulers[-1])
self.view.addItem(self.rulersLen[-1])
self.rulersStartX, self.rulersStartY = self.mouseCoord(evt)
self.rulersLen[-1].setPos(self.rulersStartX, self.rulersStartY)
self.scene.sigMouseMoved.connect(self._moveruler)
self.scene.sigMouseClicked.connect(self._stopruler)
def _updateMenu(self):
'''
add a parameter to change the graphs symbol for each layer
'''
self.clear()
curves = self.display.widget.curves
for n, item in enumerate(curves):
name = item.label.text
if not name:
name = 'Plot %s' % str(n + 1)
pen = item.opts['pen']
if not isinstance(pen, QtGui.QPen):
pen = pg.mkPen(pen)
p = self._menu.p.addChild({
'name': name,
'type': 'int',
'autoIncrementName': True,
# set current color as first option:
'value': pen.width()})
# SET COLOR:
p.sigValueChanged.connect(
lambda param, val, item=item, pen=pen:
[pen.setWidth(val), item.setPen(pen)])
def _startruler(self, evt):
self.scene.sigMouseClicked.disconnect(self._startruler)
self.rulers.append(pg.PlotDataItem(
symbol='+',
pen=pg.mkPen(color=self.pColor.value(), width=2),
symbolSize=16,
symbolBrush=pg.mkBrush(self.pColor.value())
))
self.rulersLen.append(pg.TextItem(
text='',
color=(0, 0, 0),
html=None,
anchor=(0, 0),
border=None,
fill=pg.mkBrush(self.pColor.value()),
angle=0
))
self.view.addItem(self.rulers[-1])
self.view.addItem(self.rulersLen[-1])
self.rulersStartX, self.rulersStartY = self.mouseCoord(evt)
self.rulersLen[-1].setPos(self.rulersStartX, self.rulersStartY)
self.scene.sigMouseMoved.connect(self._moveruler)
self.scene.sigMouseClicked.connect(self._stopruler)
def _setLineRainbowColors(self):
curves = self.display.widget.curves
for n,c in enumerate(curves):
pen = c.opts['pen']
if isinstance(pen, QtGui.QPen):
pen.setColor( pg.mkColor( (n,len(curves)+1) ) )
else:
pen = pg.mkPen(n,len(curves)+1)
c.setPen(pen)
def _setLineRainbowColors(self):
curves = self.display.widget.curves
for n, c in enumerate(curves):
pen = c.opts['pen']
if isinstance(pen, QtGui.QPen):
pen.setColor(pg.mkColor((n, len(curves) + 1)))
else:
pen = pg.mkPen(n, len(curves) + 1)
c.setPen(pen)
def _done(self, xxx_todo_changeme):
(x, fn, y_avg, y_vals, w_vals, i) = xxx_todo_changeme
if self.pPlotResult.value():
if self.outDisplay is None or self.outDisplay.isClosed():
self.outDisplay = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'),
title='Noise level functions')
else:
self.outDisplay.removeLayers()
if self.outDisplay2 is None or self.outDisplay2.isClosed():
self.outDisplay2 = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'),
title='NLF pointdensity')
else:
self.outDisplay2.removeLayers()
for n, yi in enumerate(y_vals):
self.outDisplay.addLayer(label='plot %s' % n,
data=[x, yi])
pen = pg.mkPen('r')
pen.setWidth(3)
self.outDisplay.addLayer(label='average',
data=[x, y_avg],
pen=pen)
pen = pg.mkPen('g')
pen.setWidth(5)
self.outDisplay.addLayer(label='fit',
data=[x, fn(x)],
pen=pen)
self.outDisplay2.addLayer(label='density',
data=[x, w_vals],
pen=pen)
self.pUpdate.show()
def _updateMenu(self):
'''
add a parameter to change the graphs symbol for each layer
'''
self.clear()
curves = self.display.widget.curves
for n,item in enumerate(curves):
name = item.label.text
if not name:
name = 'Plot %s' %str(n+1)
pPlot = self._menu.p.addChild({
'name':name,
'type':'empty',
'autoIncrementName':True,
'expanded':False})
#LINE
pLine = pPlot.addChild({
'name':'Line',
'type':'empty'})
#LINE STYLE
pen = item.opts['pen']
if not isinstance(pen, QtGui.QPen):
pen = pg.mkPen(pen)
p = pLine.addChild({
'name':'Style',
'type':'list',
#set current color as first option:
'value': ENUM2STYLE[pen.style()],
'limits':ENUM2STYLE.values(),
'itemPen':(item,pen)})
p.sigValueChanged.connect(self._setLineStyle)
#LINE WIDTH
p = pLine.addChild({
'name':'Width',
'type':'int',
#set current color as first option:
'value':pen.width(),
'itemPen':(item,pen)})
p.sigValueChanged.connect(self._setLineWidth)
color = pen
if isinstance(color, QtGui.QPen):
color = color.color()
#LINE COLOR
p = pLine.addChild({
'name':'Color',
'type':'color',
#set current color as first option:
'value':color,
'itemPen':(item,pen)})
p.sigValueChanged.connect(self._setLineColor)
#Symbol
pSymbol = pPlot.addChild({
'name':'Symbol',
'type':'empty'})
#SYMBOL COLOR
brush = item.opts['symbolBrush']
if isinstance(brush, QtGui.QBrush):
color = brush.color()
else:
color = brush
#SYMBOL TYPE
p = pSymbol.addChild({
'name':'Type',
'type':'list',
'limits':PLOT_SYMBOLS.keys(),
'item':item,
#set current symbol as first option:
'value':item.opts['symbol']})
p.sigValueChanged.connect(self._setSymbolType)
#SYMBOL COLOR
p = pSymbol.addChild({
'name':'Color',
'type':'color',
'autoIncrementName':True,
#set current colour as first option:
'value':color,
'item':item})
p.sigValueChanged.connect(self._setSymbolColor)
def _updateMenu(self):
'''
add a parameter to change the graphs symbol for each layer
'''
self.clear()
curves = self.display.widget.curves
for n, item in enumerate(curves):
name = item.label.text
if not name:
name = 'Plot %s' % str(n + 1)
pPlot = self._menu.p.addChild({
'name': name,
'type': 'empty',
'autoIncrementName': True,
'expanded': False})
# LINE
pLine = pPlot.addChild({
'name': 'Line',
'type': 'empty'})
# LINE STYLE
pen = item.opts['pen']
if not isinstance(pen, QtGui.QPen):
pen = pg.mkPen(pen)
p = pLine.addChild({
'name': 'Style',
'type': 'list',
# set current color as first option:
'value': ENUM2STYLE[pen.style()],
'limits': list(ENUM2STYLE.values()),
'itemPen': (item, pen)})
p.sigValueChanged.connect(self._setLineStyle)
# LINE WIDTH
p = pLine.addChild({
'name': 'Width',
'type': 'int',
# set current color as first option:
'value': pen.width(),
'itemPen': (item, pen)})
p.sigValueChanged.connect(self._setLineWidth)
color = pen
if isinstance(color, QtGui.QPen):
color = color.color()
# LINE COLOR
p = pLine.addChild({
'name': 'Color',
'type': 'color',
# set current color as first option:
'value': color,
'itemPen': (item, pen)})
p.sigValueChanged.connect(self._setLineColor)
# Symbol
pSymbol = pPlot.addChild({
'name': 'Symbol',
'type': 'empty'})
# SYMBOL COLOR
brush = item.opts['symbolBrush']
if isinstance(brush, QtGui.QBrush):
color = brush.color()
else:
color = brush
# SYMBOL TYPE
p = pSymbol.addChild({
'name': 'Type',
'type': 'list',
'limits': list(PLOT_SYMBOLS.keys()),
'item': item,
# set current symbol as first option:
'value': item.opts['symbol']})
p.sigValueChanged.connect(self._setSymbolType)
# SYMBOL COLOR
p = pSymbol.addChild({
'name': 'Color',
'type': 'color',
'autoIncrementName': True,
# set current colour as first option:
'value': color,
'item': item})
p.sigValueChanged.connect(self._setSymbolColor)
class NoiseLevelFunction(Tool):
'''
Calculate the noise level function from one or more input images
...Works on unloaded images as well
'''
icon = 'noiseLevelFunction.svg'
def __init__(self, imageDisplay):
Tool.__init__(self, imageDisplay)
self.calFileTool = self.showGlobalTool(CalibrationFile)
self.camera = None
self.outDisplay = None
self.outDisplay2 = None
self._refDisplay = None
pa = self.setParameterMenu()
self.pRef = pa.addChild({
'name': 'Reference images',
'type': 'list',
'value':'None',
'limits': ['None', 'From display', 'Every 2nd'],
'tip':'Images of the same setup taken at the same exposure time'})
self.pRefFromDisplay = self.pRef.addChild({
'name':'From display',
'type':'menu',
'value':'[Choose]',
'visible':False})
self.pRefFromDisplay.aboutToShow.connect(self._buildRefFromDisplayMenu)
self.pRef.sigValueChanged.connect(lambda param, val:
self.pRefFromDisplay.show(val=='From display'))
self.pPlotResult = pa.addChild({
'name': 'Plot NLF',
'type': 'bool',
'value':True})
self.pRange = pa.addChild({
'name':'Set intensity range',
'type':'bool',
'value':False})
self.pMin = self.pRange.addChild({
'name':'Minimum',
'type':'int',
'visible':False,
'value':0})
self.pMax = self.pRange.addChild({
'name':'Maximum',
'type':'int',
'visible':False,
'value':65000})
self.pRange.sigValueChanged.connect(lambda param, val:
[ch.show(val) for ch in self.pRange.childs])
self.pUpdate = pa.addChild({
'name':'Update calibration',
'type':'action',
'visible':False})
self.pUpdate.sigActivated.connect(self.updateCalibration)
def updateCalibration(self):
self.calFileTool.updateNoise(self.fitParams)
def _buildRefFromDisplayMenu(self, menu):
'''
add an action for all layers of other ImageDisplays
'''
menu.clear()
for d in self.display.workspace.displays():
if ( d.widget.__class__ == self.display.widget.__class__
and d != self.display ):
a = menu.addAction(d.name())
a.triggered.connect(
lambda checked, d=d:
[menu.setTitle(d.name()[:20]),
self.__setattr__('_refDisplay', d)] )
def activate(self):
self.startThread(self._process, self._done)
def _process(self):
imgs1 = self.display.widget.image
if imgs1 is None:
#image not loaded
imgs1 = self.display.openFilesFunctions()
v = self.pRef.value()
if v == 'Every 2nd':
imgs1,imgs2 = imgs1[::2],imgs1[1::2]
elif v == 'None' or self._refDisplay is None:
imgs2 = None
else:#from display
imgs2 = self._refDisplay.widget.image
if imgs2 is None:
#image not loaded
imgs2 = self._refDisplay.openFilesFunctions()
mn_mx = None
if self.pRange.value():
mn_mx = (self.pMin.value(), self.pMax.value())
x, fn, y_avg, y_vals, w_vals, self.fitParams,i = NLF.estimateFromImages(
imgs1, imgs2, mn_mx=mn_mx)
print 'fitparams: %s' %self.fitParams
return x, fn, y_avg, y_vals, w_vals,i
def _done(self, (x, fn, y_avg, y_vals, w_vals,i)):
if self.pPlotResult.value():
if self.outDisplay is None or self.outDisplay.isClosed():
self.outDisplay = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'),
title='Noise level functions')
else:
self.outDisplay.removeLayers()
if self.outDisplay2 is None or self.outDisplay2.isClosed():
self.outDisplay2 = self.display.workspace.addDisplay(
axes=('intensity', 'st.dev'),
title='NLF pointdensity')
else:
self.outDisplay2.removeLayers()
for n, yi in enumerate(y_vals):
self.outDisplay.addLayer(label='plot %s' %n,
data=[x,yi])
pen = pg.mkPen('r')
pen.setWidth(3)
self.outDisplay.addLayer(label='average',
data = [x,y_avg],
pen=pen)
pen = pg.mkPen('g')
pen.setWidth(5)
self.outDisplay.addLayer(label='fit',
data = [x,fn(x)],
pen=pen)
self.outDisplay2.addLayer(label='density',
data = [x,w_vals],
pen=pen)
self.pUpdate.show()