#include "QCreateInSARImagePlaneXYZRDialog.h"
#include "ui_QCreateInSARImagePlaneXYZRDialog.h"
#include <QFileDialog>
#include <QMessageBox>

#include "SatelliteOribtModel.h"
#include "SARSimulationTaskSetting.h"
#include "ImageOperatorBase.h"
#include "FileOperator.h"
#include "BaseConstVariable.h"
#include "GPUTool.cuh"
#include "LookTableSimulationComputer.cuh"
#include <QDebug>

#include "ImageShowDialogClass.h"
#include "QToolProcessBarDialog.h"

QCreateInSARImagePlaneXYZRDialog::QCreateInSARImagePlaneXYZRDialog(QWidget* parent)
	: QDialog(parent), ui(new Ui::QCreateInSARImagePlaneXYZRDialogClass)
{
	ui->setupUi(this);
	connect(ui->buttonBox, SIGNAL(accepted()), this, SLOT(onaccepted()));
	connect(ui->buttonBox, SIGNAL(rejected()), this, SLOT(onrejected()));
	connect(ui->pushButtonDEM, SIGNAL(clicked(bool)), this, SLOT(onpushButtonDEMClicked(bool)));
	connect(ui->pushButtonSloper, SIGNAL(clicked(bool)), this, SLOT(onpushButtonSloperClicked(bool)));
	connect(ui->pushButtonOrbitModel, SIGNAL(clicked(bool)), this, SLOT(onpushButtonOrbitModelClicked(bool)));
	connect(ui->pushButtonOutDir, SIGNAL(clicked(bool)), this, SLOT(onpushButtonOutDirClicked(bool)));
	connect(ui->pushButtonSataSetting, SIGNAL(clicked(bool)), this, SLOT(onpushButtonSataSettingClicked(bool)));



}

QCreateInSARImagePlaneXYZRDialog::~QCreateInSARImagePlaneXYZRDialog()
{

}

void QCreateInSARImagePlaneXYZRDialog::onrejected()
{
	this->close();
}

void QCreateInSARImagePlaneXYZRDialog::onpushButtonOrbitModelClicked(bool)
{
	// 调用文件选择对话框并选择一个 .tif 文件
	QString fileName = QFileDialog::getOpenFileName(this,
		u8"GPS Orbit Model xml",          // 对话框标题
		"",                        // 初始目录，可以设置为路径
		u8"xml Files (*.xml)"); // 文件类型过滤器

	if (!fileName.isEmpty()) {
		this->ui->OrbitModelPathLineEdit->setText(fileName);
	}
	else {
		QMessageBox::information(this, u8"没有选择文件", u8"没有选择任何文件");
	}
}

void QCreateInSARImagePlaneXYZRDialog::onpushButtonSataSettingClicked(bool)
{
	// 调用文件选择对话框并选择一个 .tif 文件
	QString fileName = QFileDialog::getOpenFileName(this,
		u8"Satellite Params setting xml",          // 对话框标题
		"",                        // 初始目录，可以设置为路径
		u8"xml Files (*.xml)"); // 文件类型过滤器

	if (!fileName.isEmpty()) {
		this->ui->SateSettingLineEdit->setText(fileName);
	}
	else {
		QMessageBox::information(this, u8"没有选择文件", u8"没有选择任何文件");
	}
}

void QCreateInSARImagePlaneXYZRDialog::onpushButtonDEMClicked(bool)
{
	// 调用文件选择对话框并选择一个 .tif 文件
	QString fileName = QFileDialog::getOpenFileName(this,
		u8"DEM XYZ Raster Select",          // 对话框标题
		"",                        // 初始目录，可以设置为路径
		u8"tiff Files (*.tiff);;tif Files (*.tif);;dat Files (*.dat);;All Files (*.*)"); // 文件类型过滤器

	if (!fileName.isEmpty()) {
		this->ui->DEMLineEdit->setText(fileName);
	}
	else {
		QMessageBox::information(this, u8"没有选择文件", u8"没有选择任何文件");
	}
}

void QCreateInSARImagePlaneXYZRDialog::onpushButtonSloperClicked(bool)
{
	// 调用文件选择对话框并选择一个 .tif 文件
	QString fileName = QFileDialog::getOpenFileName(this,
		u8"DEM Sloper Raster Select",          // 对话框标题
		"",                        // 初始目录，可以设置为路径
		u8"tiff Files (*.tiff);;tif Files (*.tif);;dat Files (*.dat);;All Files (*.*)"); // 文件类型过滤器

	if (!fileName.isEmpty()) {
		this->ui->SloperLineEdit->setText(fileName);
	}
	else {
		QMessageBox::information(this, u8"没有选择文件", u8"没有选择任何文件");
	}
}

void QCreateInSARImagePlaneXYZRDialog::onpushButtonOutDirClicked(bool)
{
	// 调用文件选择对话框并选择一个 .tif 文件
	QString fileName = QFileDialog::getExistingDirectory(this,
		u8"DEM Raster Select",          // 对话框标题
		""                       // 初始目录，可以设置为路径
	);
	if (!fileName.isEmpty()) {
		this->ui->outDirLineEdit->setText(fileName);
	}
	else {
		QMessageBox::information(this, u8"没有选择文件", u8"没有选择任何文件");
	}
}

void QCreateInSARImagePlaneXYZRDialog::LookTableSimualtionMainProcess(QString sateName, QString orbitpath, QString SatePath, QString DEMPath, QString outDirPath)
{
 
	if (!isExists(orbitpath)) {
		qDebug() << "Orbit model file is not exist !!!";
		return;
	}
	if (!isExists(SatePath)) {
		qDebug() << "Satellite Model file is not exist !!!";
		return;
	}
	if (!isExists(DEMPath)) {
		qDebug() << "DEM file is not exist !!!";
		return;
	}

	// 读取轨道模型
	qDebug() << "load orbit model params from xml :" << orbitpath;
	PolyfitSatelliteOribtModel orbitmodel;
	orbitmodel.loadFromXml(orbitpath);

	// 轨道参数
	long double			OribtStartTime = orbitmodel.getOribtStartTime();
	std::vector<double> PolyfitPx = orbitmodel.getPolyfitPx();
	std::vector<double> PolyfitPy = orbitmodel.getPolyfitPy();
	std::vector<double> PolyfitPz = orbitmodel.getPolyfitPz();
	std::vector<double> PolyfitVx = orbitmodel.getPolyfitVx();
	std::vector<double> PolyfitVy = orbitmodel.getPolyfitVy();
	std::vector<double> PolyfitVz = orbitmodel.getPolyfitVz();


	// 参数模型
	qDebug() << "load simulation setting params from xml :" << orbitpath;
	std::shared_ptr<AbstractSARSatelliteModel> SARSetting = ReadSimulationSettingsXML(SatePath);

	// 多普勒参数
	double dopplerRefrenceTime = SARSetting->getDopplerParametersReferenceTime();
	std::vector<double> DopplerCentroidCoefficients = SARSetting->getDopplerCentroidCoefficients();
	std::vector<double> DopplerRateValuesCoefficient = SARSetting->getDopplerRateValuesCoefficients();

	// 仿真成像参数计算
	double startTime = SARSetting->getSARImageStartTime();
	double endTime = SARSetting->getSARImageStartTime();

	double PRF = SARSetting->getPRF();
	double Fs = SARSetting->getFs();
	double nearRange = SARSetting->getNearRange();
	double farRange = SARSetting->getFarRange();
	double lamda = SARSetting->getCenterLamda();
	// 输出结果处理
	QString outLookTablePath = "";
	QString outIncPath = "";
	gdalImage demimg(DEMPath);


	outLookTablePath = JoinPath(outDirPath, sateName + "_looktable.bin");

	this->ui->label_tip->setText(u8"look table create...");
 
	this->LookTableSimulationDopplerProcess(
		DEMPath,
		outLookTablePath,
		OribtStartTime,
		PolyfitPx, PolyfitPy, PolyfitPz,
		PolyfitVx, PolyfitVy, PolyfitVz,
		dopplerRefrenceTime,
		DopplerCentroidCoefficients,
		startTime,
		endTime,
		nearRange,
		farRange,
		PRF,
		Fs,
		lamda
	);
}

void QCreateInSARImagePlaneXYZRDialog::LookTableSimulationDopplerProcess(QString DEMPath, QString outLookTablePath, long double OribtStartTime, std::vector<double> PolyfitPx, std::vector<double> PolyfitPy, std::vector<double> PolyfitPz, std::vector<double> PolyfitVx, std::vector<double> PolyfitVy, std::vector<double> PolyfitVz, double dopplerRefrenceTime, std::vector<double> DopplerCentroidCoefficients, double starttime, double endtime, double nearRange, double farRange, double PRF, double Fs, double lamda)
{
	qDebug() << "generate look table ";
	qDebug() << "DEMPath\t" << DEMPath;
	qDebug() << "outLookTablePath\t" << outLookTablePath;

	gdalImage demimg(DEMPath);
	gdalImage outLookTable = CreategdalImageDouble( // 创建查找表
		outLookTablePath,
		demimg.height, demimg.width, 4,
		demimg.gt,
		demimg.projection,
		true,
		true,
		true
	);


	starttime = starttime - OribtStartTime; // 处理坐标时间
	endtime = endtime - OribtStartTime;

	// 轨道模型
	double Xp0 = 0, Yp0 = 0, Zp0 = 0, Xv0 = 0, Yv0 = 0, Zv0 = 0;
	double Xp1 = 0, Yp1 = 0, Zp1 = 0, Xv1 = 0, Yv1 = 0, Zv1 = 0;
	double Xp2 = 0, Yp2 = 0, Zp2 = 0, Xv2 = 0, Yv2 = 0, Zv2 = 0;
	double Xp3 = 0, Yp3 = 0, Zp3 = 0, Xv3 = 0, Yv3 = 0, Zv3 = 0;
	double Xp4 = 0, Yp4 = 0, Zp4 = 0, Xv4 = 0, Yv4 = 0, Zv4 = 0;
	double Xp5 = 0, Yp5 = 0, Zp5 = 0, Xv5 = 0, Yv5 = 0, Zv5 = 0;
	int degree = PolyfitPx.size();
	switch (degree) {
	case(6):
		Xp5 = PolyfitPx[5];
		Yp5 = PolyfitPy[5];
		Zp5 = PolyfitPz[5];
		Xv5 = PolyfitVx[5];
		Yv5 = PolyfitVy[5];
		Zv5 = PolyfitVz[5];
	case(5):
		Xp4 = PolyfitPx[4];
		Yp4 = PolyfitPy[4];
		Zp4 = PolyfitPz[4];
		Xv4 = PolyfitVx[4];
		Yv4 = PolyfitVy[4];
		Zv4 = PolyfitVz[4];
	case(4):
		Xp3 = PolyfitPx[3];
		Yp3 = PolyfitPy[3];
		Zp3 = PolyfitPz[3];
		Xv3 = PolyfitVx[3];
		Yv3 = PolyfitVy[3];
		Zv3 = PolyfitVz[3];
	case(3):
		Xp2 = PolyfitPx[2];
		Yp2 = PolyfitPy[2];
		Zp2 = PolyfitPz[2];
		Xv2 = PolyfitVx[2];
		Yv2 = PolyfitVy[2];
		Zv2 = PolyfitVz[2];
	case(2):
		Xp1 = PolyfitPx[1];
		Yp1 = PolyfitPy[1];
		Zp1 = PolyfitPz[1];
		Xv1 = PolyfitVx[1];
		Yv1 = PolyfitVy[1];
		Zv1 = PolyfitVz[1];
	case(1):
		Xp0 = PolyfitPx[0];
		Yp0 = PolyfitPy[0];
		Zp0 = PolyfitPz[0];
		Xv0 = PolyfitVx[0];
		Yv0 = PolyfitVy[0];
		Zv0 = PolyfitVz[0];
	default:
		break;
	}

	// 多普勒参数
	double r0 = 0, r1 = 0, r2 = 0, r3 = 0, r4 = 0;
	degree = DopplerCentroidCoefficients.size();
	switch (degree)
	{
	case(5):
		r4 = DopplerCentroidCoefficients[4];
	case(4):
		r3 = DopplerCentroidCoefficients[3];
	case(3):
		r2 = DopplerCentroidCoefficients[2];
	case(2):
		r1 = DopplerCentroidCoefficients[1];
	case(1):
		r0 = DopplerCentroidCoefficients[0];
	default:
		break;
	}



	// 处理分块
	long GPUMemoryline = floor((Memory1MB * 2.0 / 8.0 / 3.0 / demimg.width * 2000));//2GB
	GPUMemoryline = GPUMemoryline < 1 ? 1 : GPUMemoryline;

	// 内存预分配

	double fact_lamda = 1 / lamda;
	for (long rid = 0; rid < demimg.height; rid = rid + GPUMemoryline) {
		long rowcount = GPUMemoryline;
		long colcount = demimg.width;
		qDebug() << "computer read file : " << rid << "~" << rowcount + rid << "\t:" << demimg.height;
		//double* tmep = new double[rowcount * colcount];
		std::shared_ptr<double> demX = readDataArr<double>(demimg, rid, 0, rowcount, colcount, 1, GDALREADARRCOPYMETHOD::VARIABLEMETHOD);// 行列数修改
		std::shared_ptr<double> demY = readDataArr<double>(demimg, rid, 0, rowcount, colcount, 2, GDALREADARRCOPYMETHOD::VARIABLEMETHOD);
		std::shared_ptr<double> demZ = readDataArr<double>(demimg, rid, 0, rowcount, colcount, 3, GDALREADARRCOPYMETHOD::VARIABLEMETHOD);

		// 
		std::shared_ptr<double> host_R((double*)mallocCUDAHost(sizeof(double) * rowcount * demimg.width), FreeCUDAHost);

		std::shared_ptr<double> device_R((double*)mallocCUDADevice(sizeof(double) * rowcount * demimg.width), FreeCUDADevice);

		// 
		std::shared_ptr<double> host_demX((double*)mallocCUDAHost(sizeof(double) * rowcount * demimg.width), FreeCUDAHost);
		std::shared_ptr<double> host_demY((double*)mallocCUDAHost(sizeof(double) * rowcount * demimg.width), FreeCUDAHost);
		std::shared_ptr<double> host_demZ((double*)mallocCUDAHost(sizeof(double) * rowcount * demimg.width), FreeCUDAHost);

		std::shared_ptr<double> device_demX((double*)mallocCUDADevice(sizeof(double) * rowcount * demimg.width), FreeCUDADevice);
		std::shared_ptr<double> device_demY((double*)mallocCUDADevice(sizeof(double) * rowcount * demimg.width), FreeCUDADevice);
		std::shared_ptr<double> device_demZ((double*)mallocCUDADevice(sizeof(double) * rowcount * demimg.width), FreeCUDADevice);

		// 数据复制
		memcpy(host_demX.get(), demX.get(), sizeof(double) * rowcount * colcount);
		memcpy(host_demY.get(), demY.get(), sizeof(double) * rowcount * colcount);
		memcpy(host_demZ.get(), demZ.get(), sizeof(double) * rowcount * colcount);

		//内存->GPU
		HostToDevice(host_demX.get(), device_demX.get(), sizeof(double) * rowcount * demimg.width);
		HostToDevice(host_demY.get(), device_demY.get(), sizeof(double) * rowcount * demimg.width);
		HostToDevice(host_demZ.get(), device_demZ.get(), sizeof(double) * rowcount * demimg.width);

		qDebug() << "GPU computer start: " << rid << "~" << rowcount + rid << "\t:" << demimg.height;
		RDProcess_dopplerGPU_InSARImagePlaneXYZR(
			device_demX.get(), device_demY.get(), device_demZ.get(),
			device_R.get(), 
			rowcount, colcount,
			starttime, nearRange, farRange,
			PRF, Fs,
			fact_lamda,
			Xp0, Yp0, Zp0, Xv0, Yv0, Zv0,
			Xp1, Yp1, Zp1, Xv1, Yv1, Zv1,
			Xp2, Yp2, Zp2, Xv2, Yv2, Zv2,
			Xp3, Yp3, Zp3, Xv3, Yv3, Zv3,
			Xp4, Yp4, Zp4, Xv4, Yv4, Zv4,
			Xp5, Yp5, Zp5, Xv5, Yv5, Zv5,
			dopplerRefrenceTime, r0, r1, r2, r3, r4);
		// GPU -> 内存
		DeviceToHost(host_R.get(), device_R.get(), sizeof(double) * rowcount * demimg.width);
		qDebug() << "GPU computer finished!!: " << rid << "~" << rowcount + rid << "\t:" << demimg.height;
		//exit(-1);
		// 数据存储
		outLookTable.saveImage(demX, rid, 0, rowcount, colcount, 1);
		outLookTable.saveImage(demY, rid, 0, rowcount, colcount, 2);
		outLookTable.saveImage(demZ, rid, 0, rowcount, colcount, 3);
		outLookTable.saveImage(host_R, rid, 0, rowcount, colcount, 4);

		qDebug() << "GPU computer result write finished: " << rid << " ~ " << rowcount + rid << "\t:" << demimg.height;

 	}

	qDebug() << "look table computed finished!!!";
 }

void QCreateInSARImagePlaneXYZRDialog::onaccepted()
{
	QString orbitpath = this->ui->OrbitModelPathLineEdit->text();
	QString SatePath = this->ui->SateSettingLineEdit->text();
	QString DEMPath = this->ui->DEMLineEdit->text();
	QString outDirPath = this->ui->outDirLineEdit->text();

	QString simulationName = this->ui->lineEditLookName->text();
	this->LookTableSimualtionMainProcess(
		simulationName,
		orbitpath, SatePath, DEMPath, outDirPath
	);
	QMessageBox::information(this, u8"info", u8"completed!!!");

}