__main__.py 6.93 KB
Newer Older
Yan's avatar
Yan committed
1
2
3
4
5
6
7
#!/usr/bin/env python3

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
from matplotlib.widgets import SpanSelector
from PyQt5 import QtCore
from PyQt5 import QtWidgets
Yan's avatar
Yan committed
8
from PyQt5 import QtGui
Yan's avatar
Yan committed
9
10
11
from rawprasslib import load_raw
import prasopes.graphtools as gt
import prasopes.filetools as ft
Yan's avatar
Yan committed
12
import prasopes.zcetools as zce
Yan's avatar
Yan committed
13
import prasopes.drltools as drl
Yan's avatar
Yan committed
14
import prasopes.config as cf
Yan's avatar
Yan committed
15
import prasopes.tangoicons
Yan's avatar
Yan committed
16
17
18
import sys
import matplotlib
import logging
3Yan's avatar
3Yan committed
19
import os.path
Yan's avatar
Yan committed
20
21
22
matplotlib.use("Qt5Agg")


Yan's avatar
Yan committed
23
def load_file(chrom_plot, spc, d_set, ms_s, fn, chrom_s, settings):
Yan's avatar
Yan committed
24
    """populates dataset and plots it"""
3Yan's avatar
3Yan committed
25
26
27
28
29
30
31
32
33
34
    directory=settings.value("open_folder")
    if fn[0] is not None:
        directory=fn[0]
    filename = QtWidgets.QFileDialog.getOpenFileName(
            caption="Open spectrum",
            directory=directory,
            filter="Finnigan RAW files (*.raw, *.RAW)")[0]
    if filename is not '':
        d_set['chrom_dat'], d_set['masses'], d_set['matrix'] \
             = load_raw(filename, settings.value("tmp_location"))
Yan's avatar
Yan committed
35
        gt.populate(chrom_plot, spc, d_set, ms_s, chrom_s)
3Yan's avatar
3Yan committed
36
37
        fn[0] = filename

Yan's avatar
Yan committed
38

Yan's avatar
Yan committed
39
40
41
42
43
44
45
46
47
48
49
50
def update_spectrum(event, chromatogram, spect, ds,
                    ms, fn, chrom, config):
    if fn[0] is not None and event.key is "f5":
        slims = [spect.get_xlim(), spect.get_ylim()]
        ds['chrom_dat'], ds['masses'], ds['matrix'] \
            = load_raw(fn[0], config.value("tmp_location"))
        gt.populate(chromatogram, spect, ds, ms, chrom)
        spect.set_xlim(slims[0])
        spect.set_ylim(slims[1])
        spect.get_figure().canvas.draw()


Yan's avatar
Yan committed
51
def main():
Yan's avatar
Yan committed
52
53
54
    # ds for data_set, ms for mass_spec_dataset, chrom for chrom_dataset
    ds = dict(chrom_dat=None, masses=None, matrix=None)
    ms = dict(x=[0], y=[0], line=None, annotation=[],
Yan's avatar
Yan committed
55
              name="Spectrum", xlabel="m/z", ylabel="ion count",
Yan's avatar
Yan committed
56
              xtics=20, autoy=True)
Yan's avatar
Yan committed
57
58
    chrom = dict(x=[0], y=[0], line=None, t_start=None, t_end=None,
                 name="Chromatogram", xlabel="time(min)",
59
                 ylabel="total ion count", timesarg=None)
Yan's avatar
Yan committed
60
    filename = [None]
Yan's avatar
Yan committed
61
    drlcache = [None, None, None]
Yan's avatar
Yan committed
62

Yan's avatar
Yan committed
63
64
    config = cf.settings()

Yan's avatar
Yan committed
65
66
67
68
    p_logger = logging.getLogger('parseLogger')
    logging.basicConfig()
    p_logger.setLevel("WARN")

Yan's avatar
Yan committed
69
    graph = Figure(figsize=(5, 4), dpi=100, facecolor="None")
Yan's avatar
Yan committed
70
71
72
73
74
75
76
77
78
79
80
81

    chromatogram = graph.add_subplot(211, facecolor=(1, 1, 1, 0.8))
    spectrum = graph.add_subplot(212, facecolor=(1, 1, 1, 0.8))
    graph.tight_layout()

    mpl_canvas = FigureCanvas(graph)
    mpl_canvas.setStyleSheet("background-color:transparent;")
    mpl_canvas.setAutoFillBackground(False)
    mpl_canvas.setFocusPolicy(QtCore.Qt.ClickFocus)
    
    gt.pan_factory(chromatogram)
    gt.zoom_factory(chromatogram, 1.15)
82
    gt.arrow_factory(spectrum, chromatogram, ds, ms, chrom)
Yan's avatar
Yan committed
83
84
    gt.pan_factory(spectrum, ms)
    gt.zoom_factory(spectrum, 1.15, ms)
85
    mass_selector = SpanSelector(
Yan's avatar
Yan committed
86
        spectrum, lambda x_min, x_max: gt.pick_masses(
87
88
89
            x_min, x_max, spectrum, ms), 'horizontal', minspan=0.01,
        useblit=True, rectprops=dict(alpha=0.15, facecolor='purple'),
        button=3)
90
    time_selector = SpanSelector(
Yan's avatar
Yan committed
91
92
93
94
95
96
         chromatogram, lambda x_min, x_max: gt.pick_times(
             x_min, x_max, spectrum, ds, chromatogram, ms, chrom), 
         'horizontal', useblit=True, rectprops=dict(
             alpha=0.15, facecolor='purple'), button=3)

    app = QtWidgets.QApplication(sys.argv)
Yan's avatar
Yan committed
97
    main_window = QtWidgets.QMainWindow(windowTitle="Prasopes")
Yan's avatar
Yan committed
98

Yan's avatar
Yan committed
99
100
101
    if QtGui.QIcon.themeName() is "":
        QtGui.QIcon.setThemeName("TangoMFK")

Yan's avatar
Yan committed
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
    openact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "document-open"), "&Open...")
    openact.setShortcut(QtCore.Qt.CTRL + QtCore.Qt.Key_O)
    openact.triggered.connect(lambda: load_file(
        chromatogram, spectrum, ds, ms, filename, chrom, config))
    exportact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "document-save-as"), "&Export...")
    exportact.setShortcut(QtCore.Qt.CTRL + QtCore.Qt.Key_E)
    exportact.triggered.connect(lambda: ft.export_dial(
        ms, chrom, filename, main_window))
    settingsact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "preferences-system"), "&Settings...")
    settingsact.triggered.connect(lambda: cf.dial(main_window))
    quitact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "application-exit"), "&Quit")
    quitact.setShortcut(QtCore.Qt.CTRL + QtCore.Qt.Key_Q)
    quitact.triggered.connect(main_window.close)
    zceact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "applications-utilities"), "&TSQ zce...")
    zceact.setShortcut(QtCore.Qt.CTRL + QtCore.Qt.Key_T)
    zceact.triggered.connect(lambda: zce.dialog(
        main_window, ds, filename))
    drlact = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "applications-utilities"), "&DRL...")
    drlact.setShortcut(QtCore.Qt.CTRL + QtCore.Qt.Key_D)
    drlact.triggered.connect(lambda: drl.main_window(
        main_window, ds, filename, drlcache))
Yan's avatar
Yan committed
129
130
131
132
133
134
    autozoomy = QtWidgets.QAction(QtGui.QIcon.fromTheme(
        "zoom-original"), "Auto Zoom Y")
    autozoomy.setCheckable(True)
    autozoomy.setChecked(True)
    autozoomy.triggered.connect(lambda: gt.autozoomy(
        autozoomy.isChecked(), spectrum, ms))
Yan's avatar
Yan committed
135
136
137



Yan's avatar
Yan committed
138
139
    file_menu = QtWidgets.QMenu('&File', main_window)
    main_window.menuBar().addMenu(file_menu)
Yan's avatar
Yan committed
140
141
142
143
144
145
    file_menu.addAction(openact)
    file_menu.addAction(exportact)
    file_menu.addSeparator()
    file_menu.addAction(settingsact)
    file_menu.addSeparator()
    file_menu.addAction(quitact)
Yan's avatar
Yan committed
146
147
    tools_menu = QtWidgets.QMenu('&Tools', main_window)
    main_window.menuBar().addMenu(tools_menu)
Yan's avatar
Yan committed
148
149
    tools_menu.addAction(zceact)
    tools_menu.addAction(drlact)
Yan's avatar
Yan committed
150
151
152
153

    main_widget = QtWidgets.QWidget(main_window)
    main_window.setCentralWidget(main_widget)

Yan's avatar
Yan committed
154
155
156
157
158
    toolBar = QtWidgets.QToolBar(main_window)
    toolBar.setAllowedAreas(QtCore.Qt.TopToolBarArea)
    toolBar.setFloatable(False)
    toolBar.setMovable(False)
    toolBar.setToolButtonStyle(QtCore.Qt.ToolButtonTextBesideIcon)
Yan's avatar
Yan committed
159

Yan's avatar
Yan committed
160
161
162
163
164
    toolBar.addAction(openact)
    toolBar.addAction(exportact)
    toolBar.addSeparator()
    toolBar.addAction(zceact)
    toolBar.addAction(drlact)
Yan's avatar
Yan committed
165
166
    toolBar.addSeparator()
    toolBar.addAction(autozoomy)
Yan's avatar
Yan committed
167

Yan's avatar
Yan committed
168
    layout = QtWidgets.QVBoxLayout(main_widget)
Yan's avatar
Yan committed
169
    layout.addWidget(toolBar)
Yan's avatar
Yan committed
170
171
172
    layout.addWidget(mpl_canvas)
    mpl_canvas.setFocus()

173
    mpl_canvas.mpl_connect('key_press_event', lambda event:
Yan's avatar
Yan committed
174
175
176
177
        update_spectrum(event, chromatogram, spectrum, ds, ms,
                        filename, chrom, config))


Yan's avatar
Yan committed
178
179
180
    if len(sys.argv) == 2:
        raw_file = sys.argv[1]
        filename[0] = raw_file
Yan's avatar
Yan committed
181
182
        ds['chrom_dat'], ds['masses'], ds['matrix'] = load_raw(
                raw_file, config.value("tmp_location"))
Yan's avatar
Yan committed
183
184
185
186
187
188
189
        gt.populate(chromatogram, spectrum, ds, ms, chrom)
    else:
        gt.pop_plot(spectrum, ms)
        gt.pop_plot(chromatogram, chrom)

    main_window.show()
    sys.exit(app.exec_())
Yan's avatar
Yan committed
190

191

Yan's avatar
Yan committed
192
193
if __name__ == "__main__":
    main()