import os
import tempfile
import wave

import numpy as np
from PyQt5.QtCore import QTimer
from PyQt5.QtWidgets import QApplication

from tests.QtTestCase import QtTestCase
from urh.controller.CSVImportDialogController import CSVImportDialogController
from urh.controller.OptionsController import OptionsController


class TestMaincontrollerGUI(QtTestCase):
    def test_open_recent_file(self):
        # Ensure we have at least one recent action
        self.add_signal_to_form("esaver.complex")
        self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 1)

        self.form.recentFileActionList[0].trigger()
        self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 2)

    def test_open_rect_directory(self):
        test_dir = os.path.join(tempfile.gettempdir(), "project_test")
        try:
            os.mkdir(test_dir)
        except OSError:
            pass

        self.form.project_manager.set_project_folder(test_dir, ask_for_new_project=False)

        self.assertIn("project_test", self.form.recentFileActionList[0].text())

    def test_update_decodings(self):
        self.form.update_decodings()
        self.assertTrue(True)

    def test_options_changed(self):
        self.add_signal_to_form("esaver.complex")
        self.form.on_options_changed({"show_pause_as_time": True, "default_view": 2})
        QApplication.instance().processEvents()
        self.assertEqual(self.form.signal_tab_controller.signal_frames[0].ui.cbProtoView.currentIndex(), 2)

    def test_open_plain_bits(self):
        bits = ["1010111000110001010101010101",
                "1010111110010010101",
                "1111010100101010101010101010"]

        filename = os.path.join(tempfile.gettempdir(), "test_plain_bits.txt")
        with open(filename, "w") as f:
            f.write(os.linesep.join(bits))

        self.form.compare_frame_controller.ui.cbProtoView.setCurrentIndex(0)
        self.wait_before_new_file()
        self.form.add_files([filename])

        for i, bit_seq in enumerate(bits):
            table_data = "".join(map(str, self.form.compare_frame_controller.protocol_model.display_data[i]))
            self.assertEqual(bit_seq, table_data)

    def test_open_options_dialog(self):
        self.form.show_options_dialog_specific_tab(1)
        w = next((w for w in QApplication.topLevelWidgets() if isinstance(w, OptionsController)), None) # type: OptionsController
        self.assertIsNotNone(w)
        self.assertEqual(w.ui.tabWidget.currentIndex(), 1)
        w.close()

    def __accept_csv_dialog(self):
        w = next((w for w in QApplication.topLevelWidgets() if isinstance(w, CSVImportDialogController)), None)
        w.accept()

    def test_import_csv(self):
        timer = QTimer()
        timer.setInterval(10)
        timer.setSingleShot(True)
        timer.timeout.connect(self.__accept_csv_dialog)

        self.assertEqual(self.form.signal_tab_controller.num_frames, 0)
        timer.start()
        self.form.add_files([self.get_path_for_filename("csvtest.csv")])

        self.assertEqual(self.form.signal_tab_controller.signal_frames[0].signal.num_samples, 100)
        self.assertTrue(os.path.isfile(self.get_path_for_filename("csvtest.complex")))
        timer.start()
        self.form.add_files([self.get_path_for_filename("csvtest.csv")])

        self.assertEqual(self.form.signal_tab_controller.num_frames, 2)
        self.assertTrue(os.path.isfile(self.get_path_for_filename("csvtest_1.complex")))

        os.remove(self.get_path_for_filename("csvtest.complex"))
        os.remove(self.get_path_for_filename("csvtest_1.complex"))


    def test_load_single_channel_wav(self):
        filename = os.path.join(tempfile.gettempdir(), "test_single_channel.wav")
        f = wave.open(filename, "w")
        f.setnchannels(1)
        f.setsampwidth(1)
        f.setframerate(2e6)
        f.writeframes(np.array([1, 2, 3, 4], dtype=np.uint8))
        f.close()

        self.add_signal_to_form(filename)
        sig_frame = self.form.signal_tab_controller.signal_frames[0]
        self.assertEqual(sig_frame.signal.sample_rate, 2e6)
        self.assertEqual(sig_frame.signal.num_samples, 4)
        self.assertNotEqual(sig_frame.signal.data.real.sum(), 0)
        self.assertEqual(sig_frame.signal.data.imag.sum(), 0)

    def test_load_stereo_wav(self):
        filename = os.path.join(tempfile.gettempdir(), "test_stereo.wav")
        f = wave.open(filename, "w")
        f.setnchannels(2)
        f.setsampwidth(2)
        f.setframerate(10e6)
        f.writeframes(np.array([23556, 2132, 111, 11111, 3455, 2324], dtype=np.int16))
        f.close()

        self.add_signal_to_form(filename)
        sig_frame = self.form.signal_tab_controller.signal_frames[0]
        self.assertEqual(sig_frame.signal.sample_rate, 10e6)
        self.assertEqual(sig_frame.signal.num_samples, 3)
        self.assertNotEqual(sig_frame.signal.data.real.sum(), 0)
        self.assertNotEqual(sig_frame.signal.data.imag.sum(), 0)