#pragma once

#include <QtWidgets/QMainWindow>
#include "ui_colorization_test.h"

class colorization_test : public QMainWindow
{
	Q_OBJECT

public:
	colorization_test( QWidget* parent = nullptr );
	~colorization_test();

	bool eventFilter( QObject* obj, QEvent* evt ) override;

signals:
	void proc_over_sig(float ms);
	void proc_error_sig( const QString& error_msg );
    void check_env_sig(const QString& msg);

private:
	void proc_over_slot(float ms);
	void proc_error_slot( const QString& error_msg );
    void check_env_slot(const QString& msg);

    // 环境检测
    void checkENV();

    // AI 处理
    void processing();
	void colorization();
	void denoise();

    bool save();
private:
	QString m_opened_file_name;
	QImage m_befor_img;
	QImage m_after_img;

	bool m_b_pfm;
	char m_file_path[ 1024 ] = { 0 };
    int m_device_type;
    int m_quality_type;
    int m_run_times;
private:
	Ui::colorization_testClass ui;
};