#include "DEM_ASSESS.h"

#include <QSettings>

void WorkThreadObject::runAssessWork(QString model, QString PRE, QString label, QString index, QString result)
{
	qDebug() << "start work:" << "\npredict: " << PRE << "\nLabel:" << label;
	QFileInfo file(index);
	file.absolutePath();
	assess_txt_path = file.absolutePath() + "/assessment.txt";
	//QDir pluginsDir = QDir(qApp->applicationDirPath());
	QDir pluginsDir = QDir("D:\\code\\RsSurvey\\RsSurvey_Build\\x64\\Release\\");
	qDebug() << "----" << pluginsDir.currentPath();
	if (!pluginsDir.cd("models\\envs"))
	{
		qDebug() << "no folder models\\envs";
		return;
	}
	QString exeDirName = pluginsDir.absoluteFilePath("miou_vegetation.exe");
	QString model_dir = " --model_dir " + model;
	QString img_data_dir = " --img_data_dir " + PRE + "/";
	QString label_dir = " --label_dir " + label + "/";
	QString val_list = " --val_list " + index;
	QString out_result = " --result_file " + result + "/";
	QString ss = exeDirName + model_dir + img_data_dir + label_dir + val_list + out_result;
	qDebug() << ss;
	QProcess* pProces = new QProcess(this);
	connect(pProces, SIGNAL(readyReadStandardOutput()), this, SLOT(writeAssessTxt()));
	pProces->start(ss);

	emit process(2);
}

void WorkThreadObject::writeAssessTxt()
{
	mProces = (QProcess*)sender();
	QString output = QString::fromLocal8Bit(mProces->readAllStandardOutput());
	output.chop(2);
	if (output.toFloat() != 0)
	{
		qDebug() << "process:" << output.toFloat();
		emit process(output.toFloat());
	}
	else
		qDebug() << "unsolved exe out:" << output;
}

void WorkThreadObject::on_cancel()
{
	if (mProces == nullptr)
	{
		qDebug() << "--mProces  null";
	}
	else
	{
		QString KillStr = "taskkill /f /im miou_vegetation.exe";
		mProces->startDetached(KillStr);
		qDebug() << "--kill Proces";
	}
}
///////////////////////////////////////////////////////

ModelAssessment::ModelAssessment()
{
	//ui.setupUi(this);
}

ModelAssessment::~ModelAssessment()
{

}

QString ModelAssessment::PannelName()
{
	return QString::fromLocal8Bit("");
}

QString ModelAssessment::CategoryName()
{
	return QString::fromLocal8Bit("DEM模块");
}

QString ModelAssessment::EnglishName()
{
	return QString::fromLocal8Bit("DEM_Module");
}

QString ModelAssessment::ChineseName()
{
	return QString::fromLocal8Bit("结果评估");
}

QString ModelAssessment::Information()
{
	return "结果评估";
}

QString ModelAssessment::IconPath()
{
	return ":/DEM_ASSESS/resources/assessment.svg";
}

QWidget* ModelAssessment::CenterWidget()
{
	QString gdal_path = qApp->applicationDirPath().toLocal8Bit() + "/share/gdal";
	QString pro_lib_path = qApp->applicationDirPath().toLocal8Bit() + "/share/proj";
	qputenv("GDAL_DATA", gdal_path.toLocal8Bit());
	qputenv("PROJ_LIB", pro_lib_path.toLocal8Bit());

	bool showWin = false;
	if (myWidget == nullptr)
	{
		myWidget = new QDialog();
		showWin = true;
		qDebug() << "new QDialog()";
	}
	else
	{
		qDebug() << "already have myWidget";
		//激活窗口并提升至顶层
		myWidget->activateWindow();
		myWidget->raise();
		return myWidget;
	}
	ui.setupUi(myWidget);

	myWidget->setWindowTitle(QString::fromLocal8Bit("结果评估"));
	myWidget->setWindowFlags(Qt::CustomizeWindowHint | Qt::WindowCloseButtonHint);
	myWidget->setWindowIcon(QIcon(":/DEM_ASSESS/resources/assessment.svg"));
	myWidget->setAttribute(Qt::WA_QuitOnClose, false);

	myWidget->setAttribute(Qt::WA_DeleteOnClose);
	connect(myWidget, &QDialog::destroyed, this, [=] {
		qDebug() << "----DEM assess window close----";

		QDir pluginsDir = QDir(qApp->applicationDirPath());
		if (pluginsDir.cd("srsplugins\\DemModel"))
		{
			QString strConfigPath = pluginsDir.absoluteFilePath("dem_config.ini");
			QFile f(strConfigPath);
			if (f.exists())
			{
				WriteConfigPaths(strConfigPath);
				f.close();
			}
		}

		if (mWorkThread != nullptr)
		{
			//mWorkThread->quit();
			//mWorkThread->wait();//调用wait后先调用finished信号对应的槽函数，执行完成后再往下走
			//mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater，不需要delete
			//mWorker = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
			mWorkThread->requestInterruption();//请求线程中断
			mWorkThread->quit();
			mWorkThread->wait();//调用wait后先调用finished信号对应的槽函数，执行完成后再往下走
			mWorker->on_cancel();
			mWorkThread = nullptr;//mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater，不需要delete
			mWorker = nullptr;//mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater
		}
		//qDebug() << "--destroyed";
		myWidget->close();
		myWidget = nullptr;
		});
	connect(ui.pbtInPredict, &QPushButton::pressed, this, &ModelAssessment::choosePredictPath);
	connect(ui.pbtInLabel, &QPushButton::pressed, this, &ModelAssessment::chooseLabelPath);
	connect(ui.pbtIndex, &QPushButton::pressed, this, &ModelAssessment::chooseAssessFile);
	connect(ui.pbtModel, &QPushButton::pressed, this, &ModelAssessment::chooseModelFile);
	connect(ui.pbtOutResult, &QPushButton::pressed, this, &ModelAssessment::chooseResultPath);

	connect(ui.pushButton_ok, &QPushButton::pressed, this, &ModelAssessment::readAndStart);
	connect(ui.pushButton_cancel, &QPushButton::pressed, this, &ModelAssessment::pbCancel);

	ui.progressBar->setTextVisible(true);
	ui.progressBar->setRange(0, 100);

	QFile qssFile(":/DEM_ASSESS/DEM_ASSESS.qss");
	qssFile.open(QFile::ReadOnly); //以只读方式打开
	if (qssFile.isOpen())
	{
		QString qss = QLatin1String(qssFile.readAll());
		myWidget->setStyleSheet(qss);
		qssFile.close();
	}
	else
		qDebug() << "无法打开文件";

	QDir pluginsDir = QDir(qApp->applicationDirPath());
	if (pluginsDir.cd("srsplugins\\DemModel"))
	{
		QString strConfigPath = pluginsDir.absoluteFilePath("dem_config.ini");
		QFile f(strConfigPath);
		if (f.exists())
		{
			ReadConfigHistoryPaths(strConfigPath);
			f.close();
		}
	}

	if (showWin)
		myWidget->show();

	return myWidget;
}

void ModelAssessment::startWorkThread()
{
	if (mWorker != nullptr)
	{
		return;
	}
	mWorkThread = new QThread();
	mWorker = new WorkThreadObject();
	mWorker->moveToThread(mWorkThread);
	connect(mWorkThread, &QThread::finished, mWorkThread, &QObject::deleteLater);
	connect(mWorkThread, &QThread::finished, mWorker, &QObject::deleteLater);
	connect(mWorker, &WorkThreadObject::process, myWidget, [=](int value) {
		ui.progressBar->setValue(value);
		if (value == 100)
		{
			QString txtDir = ui.lineOutResult->text() + "/assessment.txt";
			QFile file(txtDir);
			QString outLines = "";
			if (file.open(QIODevice::ReadOnly))
			{
				QTextStream stream(&file);
				// recall 召回率
				// precision 精确率
				// mIoU 平均交并比
				// mAP 平均精度均值
				// f1_score
				while (!stream.atEnd())
				{
					QString line = stream.readLine();
					//strs.push_back(line);
					if (line.contains("===>mIoU"))
					{
						//===>mIoU:68.45; mAP: 74.92
						line.remove("===>");
						line.replace(QStringLiteral("mAP"), QStringLiteral("平均精度均值mAP"));
						line.replace(QStringLiteral("mIoU"), QStringLiteral("平均交并比mIoU"));
						outLines = outLines + line + "\n";
					}
					else if (line.contains("===>recall"))
					{
						//===>recall:66.62; precision: 51.2
						line.remove("===>");
						line.replace(QStringLiteral("recall"), QStringLiteral("召回率recall"));
						line.replace(QStringLiteral("precision"), QStringLiteral("精确率precision"));
						outLines = outLines + line + "\n";
					}
					else if (line.contains("===>f1_score"))
					{
						//===>f1_score:57.9
						line.remove("===>");
						line.replace(QStringLiteral("f1_score"), QStringLiteral("F1分数f1_score"));
						outLines = outLines + line;
					}
				}
				file.close();
			}

			QMessageBox mess(QMessageBox::Information,
				QString::fromLocal8Bit("评估结束"),
				QString::fromLocal8Bit("评价结果:\n") + txtDir + "\n"
				+ outLines);
			mess.setWindowFlags(Qt::Drawer);
			mess.setTextInteractionFlags(Qt::TextSelectableByMouse);
			mess.setStandardButtons(QMessageBox::Yes);
			mess.button(QMessageBox::StandardButton::Yes)->setText(QString::fromLocal8Bit("确认"));
			//mess.setTextInteractionFlags(Qt::TextSelectableByMouse);
			int result = mess.exec();
			pbCancel();
		}
		});
	connect(this, &ModelAssessment::startPreAssessment, mWorker, &WorkThreadObject::runAssessWork);

	mWorkThread->start();
}

void ModelAssessment::chooseModelFile()
{
	QString fileModel = QFileDialog::getOpenFileName(ui.pbtModel, QString::fromLocal8Bit("选择输入模型文件"), "", "*.pth");
	if (fileModel != "")
		ui.lineModel->setText(fileModel);
}
void ModelAssessment::choosePredictPath()
{
	QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInPredict, QString::fromLocal8Bit("选择输入评价文件路径"), "");
	if (dirDom != "")
		ui.lineInPredict->setText(dirDom);
}
void ModelAssessment::chooseLabelPath()
{
	QString dirDom = QFileDialog::getExistingDirectory(ui.pbtInLabel, QString::fromLocal8Bit("选择输入Label文件路径"), "");
	if (dirDom != "")
		ui.lineInLabel->setText(dirDom);
}
void ModelAssessment::chooseAssessFile()
{
	QString dirDom = QFileDialog::getOpenFileName(ui.pbtIndex, QString::fromLocal8Bit("选择输入评估文件索引"), "", "*.txt");
	if (dirDom != "")
		ui.lineIndex->setText(dirDom);
}
void ModelAssessment::chooseResultPath()
{
	QString dirResult = QFileDialog::getExistingDirectory(ui.pbtOutResult, QString::fromLocal8Bit("选择评估结果输出路径"), "");
	if (dirResult != "")
		ui.lineOutResult->setText(dirResult);
}

void ModelAssessment::ReadConfigHistoryPaths(QString strPath)
{
	QSettings configIni(strPath, QSettings::IniFormat);

	//打开标题为：[DemTrain] 的组，读取DemTrain输出结果TrainResult字段
	configIni.beginGroup("DemTrain");

	QString strDemTrainResult = configIni.value("TrainResult").toString();
	QDir srcDir = QDir(strDemTrainResult);
	//更新评价文件列表
	if (srcDir.absoluteFilePath("val_list.txt") != "")
		ui.lineIndex->setText(srcDir.absoluteFilePath("val_list.txt"));
	//更新输入Images
	srcDir = QDir(strDemTrainResult);
	if (srcDir.cd("Images"))
		ui.lineInPredict->setText(srcDir.absolutePath());
	//更新输入Labels
	srcDir = QDir(strDemTrainResult);
	if (srcDir.cd("Labels"))
		ui.lineInLabel->setText(srcDir.absolutePath());

	configIni.endGroup();//关闭组

	//打开标题为：[DemAssess] 的组，读取AssessResult字段
	configIni.beginGroup("DemAssess");
	//更新评价结果输出
	ui.lineOutResult->setText(configIni.value("AssessResult").toString());

	configIni.endGroup();//关闭组
}

void ModelAssessment::WriteConfigPaths(QString strPath)
{
	QSettings configIni(strPath, QSettings::IniFormat);
	configIni.setIniCodec("utf-8");
	//打开标题为：[DemAssess] 的组
	configIni.beginGroup("DemAssess");

	//更新输入模型路径
	QString temp = ui.lineOutResult->text();
	if (temp != "")
		configIni.setValue("AssessResult", temp);

	configIni.endGroup();//关闭组
}

void ModelAssessment::readAndStart()
{
	QString dirModel = ui.lineModel->text();
	QString dirIndex = ui.lineIndex->text();
	QString dirLabel = ui.lineInLabel->text();
	QString dirPredict = ui.lineInPredict->text();
	QString dirResult = ui.lineOutResult->text();

	ui.progressBar->setValue(0);

	if (dirLabel == "" || dirPredict == "" || dirIndex == "" || dirModel == "" || dirResult == "")
	{
		QMessageBox mess(QMessageBox::NoIcon, QString::fromLocal8Bit("错误"), QString::fromLocal8Bit("请检查输入路径"));
		mess.setWindowFlags(Qt::Drawer);
		int result = mess.exec();
		return;
	}
	if (mWorkThread == nullptr)
	{
		qDebug() << "--startThread";
		startWorkThread();
	}
	emit startPreAssessment(dirModel, dirPredict, dirLabel, dirIndex, dirResult);
}

void ModelAssessment::pbCancel()
{
	delete myWidget;//调起&QDialog::destroyed
}