def test_adding_to_plot(self):
s = ScanningParam()
a = AxisParam()
a.length = 5
a.increment = 0.5
a.start = -1.5
a.unit = 'nm'
s.axis1 = a
s.axis2 = a
s.axis3 = a
data = np.random.rand(5, 5, 5)
m = Measurement(data, s)
map_ = ThothMapItem(m)
from guiqwt.image import ImagePlot
plot = ImagePlot(title="Test")
plot.add_item(map_)
plot.show()
def __init__(self,
parent=None,
title="",
xlabel=("", ""),
ylabel=("", ""),
zlabel=None,
xunit=("", ""),
yunit=("", ""),
zunit=None,
yreverse=True,
colormap="jet",
aspect_ratio=1.0,
lock_aspect_ratio=True,
show_contrast=False,
show_itemlist=False,
show_xsection=False,
show_ysection=False,
xsection_pos="top",
ysection_pos="right",
gridparam=None,
curve_antialiasing=None,
**kwargs):
if PYQT5:
super(BaseImageWidget, self).__init__(parent, **kwargs)
self.setOrientation(Qt.Vertical)
else:
QSplitter.__init__(self, Qt.Vertical, parent)
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
self.sub_splitter = QSplitter(Qt.Horizontal, self)
self.plot = ImagePlot(parent=self,
title=title,
xlabel=xlabel,
ylabel=ylabel,
zlabel=zlabel,
xunit=xunit,
yunit=yunit,
zunit=zunit,
yreverse=yreverse,
aspect_ratio=aspect_ratio,
lock_aspect_ratio=lock_aspect_ratio,
gridparam=gridparam)
if curve_antialiasing is not None:
self.plot.set_antialiasing(curve_antialiasing)
from guiqwt.cross_section import YCrossSection
self.ycsw = YCrossSection(self,
position=ysection_pos,
xsection_pos=xsection_pos)
self.ycsw.setVisible(show_ysection)
from guiqwt.cross_section import XCrossSection
self.xcsw = XCrossSection(self)
self.xcsw.setVisible(show_xsection)
self.xcsw.SIG_VISIBILITY_CHANGED.connect(self.xcsw_is_visible)
self.xcsw_splitter = QSplitter(Qt.Vertical, self)
if xsection_pos == "top":
self.xcsw_splitter.addWidget(self.xcsw)
self.xcsw_splitter.addWidget(self.plot)
else:
self.xcsw_splitter.addWidget(self.plot)
self.xcsw_splitter.addWidget(self.xcsw)
self.xcsw_splitter.splitterMoved.connect(
lambda pos, index: self.adjust_ycsw_height())
self.ycsw_splitter = QSplitter(Qt.Horizontal, self)
if ysection_pos == "left":
self.ycsw_splitter.addWidget(self.ycsw)
self.ycsw_splitter.addWidget(self.xcsw_splitter)
else:
self.ycsw_splitter.addWidget(self.xcsw_splitter)
self.ycsw_splitter.addWidget(self.ycsw)
configure_plot_splitter(self.xcsw_splitter,
decreasing_size=xsection_pos == "bottom")
configure_plot_splitter(self.ycsw_splitter,
decreasing_size=ysection_pos == "right")
self.sub_splitter.addWidget(self.ycsw_splitter)
self.itemlist = PlotItemList(self)
self.itemlist.setVisible(show_itemlist)
self.sub_splitter.addWidget(self.itemlist)
# Contrast adjustment (Levels histogram)
from guiqwt.histogram import ContrastAdjustment
self.contrast = ContrastAdjustment(self)
self.contrast.setVisible(show_contrast)
self.addWidget(self.contrast)
configure_plot_splitter(self)
configure_plot_splitter(self.sub_splitter)
