/
data_exploration_wizards.py
74 lines (54 loc) · 2.81 KB
/
data_exploration_wizards.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
'''
Created on Jan 6, 2014
@author: george
'''
from PyQt4 import QtCore, QtGui
from PyQt4.Qt import *
from common_wizard_pages.histogram_page import HistogramPage
from common_wizard_pages.scatterplot_page import ScatterPlotPage
from common_wizard_pages.histogram_dataselect_page import HistogramDataSelectPage
from common_wizard_pages.scatterplot_dataselect_page import ScatterplotDataSelectPage
# Histogram wizard ids
[Page_HistogramDataSelectPage, Page_HistogramParams,] = range(2)
# scatterplot wizard ids
[Page_ScatterplotDataSelectPage, Page_ScatterplotParams,] = range(2)
class HistogramWizard(QtGui.QWizard):
def __init__(self, model, old_histogram_params={}, prev_hist_var=None, parent=None):
super(HistogramWizard, self).__init__(parent)
self.model=model
self.setWindowTitle("Histogram Wizard")
self.setWizardStyle(QWizard.ClassicStyle)
self.data_select_page = HistogramDataSelectPage(model=model, prev_hist_var=prev_hist_var)
self.histogram_page = HistogramPage(old_histogram_params=old_histogram_params)
self.setPage(Page_HistogramDataSelectPage, self.data_select_page)
self.setPage(Page_HistogramParams, self.histogram_page)
def nextId(self):
if self.currentId()==Page_HistogramDataSelectPage:
return Page_HistogramParams
elif self.currentId() == Page_HistogramParams:
return -1
def get_selected_var(self):
return self.data_select_page.get_selected_var()
def get_histogram_params(self):
return self.histogram_page.get_parameters()
class ScatterPlotWizard(QtGui.QWizard):
def __init__(self, model, old_scatterplot_params={}, prev_scatterplot_data=None, parent=None):
super(ScatterPlotWizard, self).__init__(parent)
self.model=model
self.setWindowTitle("Scatterplot Wizard")
self.setWizardStyle(QWizard.ClassicStyle)
self.data_select_page = ScatterplotDataSelectPage(model=model, prev_scatterplot_data=prev_scatterplot_data)
self.scatterplot_page = ScatterPlotPage(old_scatterplot_params=old_scatterplot_params)
self.setPage(Page_ScatterplotDataSelectPage, self.data_select_page)
self.setPage(Page_ScatterplotParams, self.scatterplot_page)
def nextId(self):
if self.currentId()==Page_ScatterplotDataSelectPage:
return Page_ScatterplotParams
elif self.currentId() == Page_ScatterplotParams:
return -1
def get_selected_vars(self):
return self.data_select_page.get_selected_vars()
def get_scatterplot_params(self):
return self.scatterplot_page.get_parameters()
def get_annotate_plot_with_study_labels(self):
return self.scatterplot_page.get_annotate_plot_with_study_labels()