Module qimview.imview

from qimview.utils.qt_imports import QtWidgets
import argparse
import sys
import json

import sys

from qimview.tests_utils.event_recorder import EventRecorder
from qimview.tests_utils.event_player   import EventPlayer
from qimview.tests_utils.qtdump import *
from qimview.image_viewers.qt_image_viewer import QTImageViewer
from qimview.image_viewers.gl_image_viewer import GLImageViewer
from qimview.image_viewers.image_filter_parameters import ImageFilterParameters
from qimview.image_viewers.image_filter_parameters_gui import ImageFilterParametersGui

from qimview.image_readers import gb_image_reader
from typing import Union

# define a Qt window with an OpenGL widget inside it
# class TestWindow(QtGui.QMainWindow):
class ImView(QtWidgets.QMainWindow):
    def __init__(self, events, params: dict) -> None:
        super().__init__()

        record_file = params['record']
        if record_file is not None:
            self.event_recorder = EventRecorder(filename=record_file)
            self.event_recorder.register_widget(id(self), "TestWindow")
        else:
            self.event_recorder = None

        self.main_widget = QtWidgets.QWidget()
        vertical_layout = QtWidgets.QVBoxLayout()
        self.main_widget.setLayout(vertical_layout)

        self.widget: Union[GLImageViewer, QTImageViewer]

        if params['gl']:
            self.widget = GLImageViewer(event_recorder = self.event_recorder)
        else:
            self.widget = QTImageViewer(event_recorder = self.event_recorder)
        if params['timing']:
            self.widget.display_timing = True
        if record_file is not None:
            self.event_recorder.register_widget(id(self.widget), "widget")

        # if params['input_image'] is None:
        #     # Ask for input file
        #     filename =  QtWidgets.QFileDialog.getOpenFileName(caption="imview: Select input image")
        #     params['input_image'] = filename[0]
        # im = gb_image_reader.read(params['input_image'])
        # self.widget.set_image(im)
        # put the window at the screen position (100, 100)
        self.setGeometry(0, 0, self.widget.width(), self.widget.height())
        self.setCentralWidget(self.main_widget)

        self.filter_params = ImageFilterParameters()
        self.filter_params_gui = ImageFilterParametersGui(self.filter_params, name="TestViewer")
        self.filter_params_gui.set_event_recorder(self.event_recorder)

        hor_layout = QtWidgets.QHBoxLayout()
        self.filter_params_gui.add_blackpoint(hor_layout, self.update_image_intensity_event)
        # white point adjustment
        self.filter_params_gui.add_whitepoint(hor_layout, self.update_image_intensity_event)
        # Gamma adjustment
        self.filter_params_gui.add_gamma(hor_layout, self.update_image_intensity_event)
        # G_R adjustment
        self.filter_params_gui.add_g_r(hor_layout, self.update_image_intensity_event)
        # G_B adjustment
        self.filter_params_gui.add_g_b(hor_layout, self.update_image_intensity_event)

        vertical_layout.addLayout(hor_layout)
        vertical_layout.addWidget(self.widget)

        self.show()

        if events != None:
            event_list = events['events']
            # self.event_recorder.register_widget(self.id(), "TestWindow")
            # self.event_recorder.register_widget(self.widget.id(), "widget")
            event_player = EventPlayer()
            event_player.register_widget("TestWindow", self)
            event_player.register_widget("widget", self.widget)
            self.filter_params_gui.register_event_player(event_player)
            event_player.play_events(event_list=event_list)
    
    def set_image(self, params):
        if params['input_image'] is None:
            # Ask for input file
            filename =  QtWidgets.QFileDialog.getOpenFileName(caption="imview: Select input image")
            params['input_image'] = filename[0]
        im = gb_image_reader.read(params['input_image'])
        self.widget.set_image(im)
        self.widget.image_name = params['input_image']

        if params['ref'] is not None:
            im = gb_image_reader.read(params['ref'])
            print(f"im ref {im}")
            if im:
                self.widget.set_image_ref(im)

    def update_image_intensity_event(self):
        self.widget.filter_params.copy_from(self.filter_params)
        # print(f"parameters {self.filter_params}")
        self.widget.viewer_update()

    def event(self, evt):
        if self.event_recorder is not None:
            if evt.type() in RESIZE_EVENTS:
                self.event_recorder.store_event(self, evt)
        return QtWidgets.QMainWindow.event(self, evt)

    def closeEvent(self, event):
        # Save event list
        if self.event_recorder is not None:
            self.event_recorder.save_screen(self.widget)
            self.event_recorder.save_events()
        event.accept()

def main():
    # import numpy for generating random data points
    parser = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
    parser.add_argument('input_image', nargs='?', help='input image (if not specified, will open file dialog)')
    parser.add_argument('-ref', default=None,  help='reference image')
    parser.add_argument('-p', '--play', help='events json file', default=None)
    parser.add_argument('-r', '--record', help='record events in given json file', default=None)
    parser.add_argument('--timing', action='store_true', help='display timings')
    parser.add_argument('--gl', action='store_true', help='use opengl viewer')
    args = parser.parse_args()
    _params = vars(args)
    print(f"record {_params['record']}")


    # create the Qt App and window
    QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_ShareOpenGLContexts)
    app = QtWidgets.QApplication(sys.argv)
    app.setApplicationDisplayName('imview ' + ' '.join(sys.argv[1:]))
    if _params['play'] is not None:
        with open(_params['play']) as json_file:
            events = json.load(json_file)
    else:
        events = None

    window = ImView(events, _params)
    window.set_image(_params)
    window.show()
    app.exec()

if __name__ == '__main__':
    main()