__main__.py

#!/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
from rawprasslib import load_raw
import prasopes.graphtools as gt
import prasopes.filetools as ft
import prasopes.zcetools as zce
import prasopes.drltools as drl
import sys
import matplotlib
import logging
matplotlib.use("Qt5Agg")


def load_file(chrom_plot, spc, d_set, ms_s, fn, chrom_s):
    """populates dataset and plots it"""
    lf = ft.open_dial(d_set)
    if lf is not None:
        gt.populate(chrom_plot, spc, d_set, ms_s, chrom_s)
        fn[0] = lf


def main():
    # 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=[],
              name="Spectrum", xlabel="m/z", ylabel="ion count",
              xtics=20)
    chrom = dict(x=[0], y=[0], line=None, t_start=None, t_end=None,
                 name="Chromatogram", xlabel="time(min)",
                 ylabel="total ion count")
    filename = [None]
    drlcache = [None, None, None]

    p_logger = logging.getLogger('parseLogger')
    logging.basicConfig()
    p_logger.setLevel("WARN")

    graph = Figure(figsize=(5, 4), dpi=100, facecolor="None")

    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)
    gt.pan_factory(spectrum, ms)
    gt.zoom_factory(spectrum, 1.15, ms)
    mass_selector = SpanSelector(
        spectrum, lambda x_min, x_max: gt.pick_masses(
            x_min, x_max, spectrum, ms), 'horizontal', minspan=0.01,
        useblit=True, rectprops=dict(alpha=0.15, facecolor='purple'),
        button=3)
    time_selector = SpanSelector(
         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)
    main_window = QtWidgets.QMainWindow(windowTitle="Prasopes")

    file_menu = QtWidgets.QMenu('&File', main_window)
    main_window.menuBar().addMenu(file_menu)
    file_menu.addAction('&Open..', lambda: load_file(
        chromatogram, spectrum, ds, ms, filename, chrom),
        QtCore.Qt.CTRL + QtCore.Qt.Key_O)
    file_menu.addAction('&Export..', lambda: ft.export_dial(
        ms, chrom, filename, main_window),
        QtCore.Qt.CTRL + QtCore.Qt.Key_E)
    file_menu.addAction('&Quit', main_window.close,
                        QtCore.Qt.CTRL + QtCore.Qt.Key_Q)
    tools_menu = QtWidgets.QMenu('&Tools', main_window)
    main_window.menuBar().addMenu(tools_menu)
    tools_menu.addAction('&TSQ zce', lambda: zce.dialog(
        main_window, ds, filename), QtCore.Qt.CTRL + QtCore.Qt.Key_T)
    tools_menu.addAction('&DRL..', lambda: drl.main_window(
        main_window, ds, filename, drlcache),
        QtCore.Qt.CTRL + QtCore.Qt.Key_D)

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

    layout = QtWidgets.QVBoxLayout(main_widget)
    layout.addWidget(mpl_canvas)
    mpl_canvas.setFocus()

    if len(sys.argv) == 2:
        raw_file = sys.argv[1]
        filename[0] = raw_file
        ds['chrom_dat'], ds['masses'], ds['matrix'] = load_raw(raw_file)
        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_())


if __name__ == "__main__":
    main()