pyqtgraph/examples/test_ImageView.py

47 lines
1.2 KiB
Python
Raw Normal View History

2010-03-22 05:48:52 +00:00
# -*- coding: utf-8 -*-
## Add path to library (just for examples; you do not need this)
import sys, os
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..'))
2010-03-22 05:48:52 +00:00
import numpy as np
import scipy
2010-03-22 05:48:52 +00:00
from PyQt4 import QtCore, QtGui
import pyqtgraph as pg
2010-03-22 05:48:52 +00:00
app = QtGui.QApplication([])
## Create window with ImageView widget
2010-03-22 05:48:52 +00:00
win = QtGui.QMainWindow()
imv = pg.ImageView()
2010-03-22 05:48:52 +00:00
win.setCentralWidget(imv)
win.show()
## Create random 3D data set with noisy signals
img = scipy.ndimage.gaussian_filter(np.random.normal(size=(200, 200)), (5, 5)) * 20 + 100
img = img[np.newaxis,:,:]
decay = np.exp(-np.linspace(0,0.3,100))[:,np.newaxis,np.newaxis]
data = np.random.normal(size=(100, 200, 200))
data += img * decay
#for i in range(data.shape[0]):
#data[i] += 10*exp(-(2.*i)/data.shape[0])
data += 2
## Add time-varying signal
sig = np.zeros(data.shape[0])
sig[30:] += np.exp(-np.linspace(1,10, 70))
sig[40:] += np.exp(-np.linspace(1,10, 60))
sig[70:] += np.exp(-np.linspace(1,10, 30))
sig = sig[:,np.newaxis,np.newaxis] * 3
data[:,50:60,50:60] += sig
## Display the data
imv.setImage(data, xvals=np.linspace(1., 3., data.shape[0]))
## Start Qt event loop unless running in interactive mode.
if sys.flags.interactive != 1:
app.exec_()