def setUp(self):
application = QtGui.QApplication.instance()
if application is None:
self.application = QtGui.QApplication([])
else:
self.application = application
DrawingImageTester.setUp(self)
def get_app_qt4(*args, **kwargs):
"""Create a new qt4 app or return an existing one."""
from pyface.qt import QtGui
app = QtGui.QApplication.instance()
if app is None:
if not args:
args = ([''],)
app = QtGui.QApplication(*args, **kwargs)
return app
def setUp(self):
from pyface.qt import QtGui
application = QtGui.QApplication.instance()
if application is None:
self.application = QtGui.QApplication([])
else:
self.application = application
unittest.TestCase.setUp(self)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('-o', '--output', default=None)
args = parser.parse_args()
# Create a QApplication instance so that the QPainter backend can be tested
app = QtGui.QApplication(sys.argv) # noqa: F841
outdir = args.output
if outdir is not None and not os.path.isdir(outdir):
os.mkdir(outdir)
results = benchmark(outdir=outdir)
if outdir is not None:
publish(results, outdir)
def demo_main(demo_class, size=(400, 400), title="Chaco plot"):
"Takes the class of the demo to run as an argument."
app = wx.PySimpleApp()
frame = demo_class(None, size=size, title=title)
app.SetTopWindow(frame)
app.MainLoop()
elif ETSConfig.toolkit == 'qt4':
from pyface.qt import QtGui
_app = QtGui.QApplication.instance()
if _app is None:
import sys
_app = QtGui.QApplication(sys.argv)
class DemoFrame(QtGui.QWidget):
def __init__(self, parent, **kw):
QtGui.QWidget.__init__(self)
# Create the subclass's window
self.plot_window = self._create_window()
layout = QtGui.QVBoxLayout()
layout.setMargin(0)
layout.addWidget(self.plot_window.control)
self.setLayout(layout)
if 'size' in kw:
# Get appropriate hemisphere
pial = scipy.io.loadmat(
os.path.join(self.subj_dir, 'Meshes',
self.hem + '_pial_trivert.mat'))
ctmr_gauss_plot(pial['tri'], pial['vert'], opacity=0.8)
#rh = scipy.io.loadmat(os.path.join(self.subj_dir, 'Meshes', 'rh_pial_trivert.mat'))
#ctmr_gauss_plot(rh['tri'], rh['vert'], opacity=0.8, new_fig=False)
# Plot the electrodes we have so far
vmax = 17.
for i, dev in enumerate(self.devices):
elecfile = os.path.join(self.subj_dir, 'elecs', 'individual_elecs',
dev + '.mat')
e = scipy.io.loadmat(elecfile)['elecmatrix']
num = self.devices.index(dev)
c = self.elec_colors(num / vmax)
el_add(e,
color=tuple(c[:3]),
msize=4,
numbers=1 + np.arange(e.shape[0]))
if __name__ == '__main__':
app = QtGui.QApplication([])
path_to_this_func = os.path.dirname(os.path.realpath(__file__))
app.setWindowIcon(
QtGui.QIcon(os.path.join(path_to_this_func, 'icons', 'leftbrain.png')))
subj_dir = sys.argv[1]
hem = sys.argv[2]
e = acpc_picker(subj_dir=subj_dir, hem=hem)
def setUpClass(cls):
cls.qapp = QtGui.QApplication.instance() or QtGui.QApplication([])
