#pragma once

#include <QtWidgets/QMainWindow>
#include "Driver.h"
#include "ui_QtGuiApplication.h"
#include "SceneGraphEdit.h"

class QtGuiApplication : public QMainWindow
{
	Q_OBJECT

public:
	QtGuiApplication(QWidget *parent = Q_NULLPTR);

protected:
	void closeEvent(QCloseEvent *event) { Driver::setStatus(DriverStatus::Finished); }

public slots:
void singleStep() { Driver::changeSingleStep(); }
void stop() { Driver::setStatus(DriverStatus::Stopped); }
void run() { Driver::setStatus(DriverStatus::Runnable); }
void reset() { Driver::setStatus(DriverStatus::Reset); }
void noneLearn() { learningNode->setLearnMethod(LearnMethod::Disable); }
void staticLearn() { learningNode->setLearnMethod(LearnMethod::Static); }
void dynamicLearn() { learningNode->setLearnMethod(LearnMethod::Dynamic); }
void hideFemForce() { femForce->hide(); }
void showFemForce() { femForce->show(); }
void hidePdmForce() { pdmForce->hide(); }
void showPdmForce() { pdmForce->show(); }
void hideDeltaForce() { deltaForce->hide(); }
void showDeltaForce() { deltaForce->show(); }
void FEMMesh() { if (surfaceFEM->isHidden())surfaceFEM->show(); else surfaceFEM->hide(); }
void PDMMesh() { if (pointPDM->isHidden())pointPDM->show(); else pointPDM->hide(); }
void setGravity();
void setExtForce();
void setSlipDirection();
void setDamping();
void saveFEM() { dynamic_cast<SurfaceRenderCuda*>(surfaceFEM)->setwritable(std::stoi(ui.saveFrames->text().toStdString())); ui.saveFEMSurf->setStyleSheet("background-color: rgb(10,10,10);"); }
void savePDM() { dynamic_cast<SurfaceRenderCuda*>(surfacePDM)->setwritable(std::stoi(ui.saveFrames->text().toStdString())); ui.savePDMSurf->setStyleSheet("background-color: rgb(10,10,10);"); }
void saveMaterial();
void loadMaterial();
void fixPlane(int index);
void externalForcePlane(int index);
void slipPlane(int index);
void exportLearnData();


private:
	Ui::QtGuiApplicationClass ui;
};