#include "casmconfigpanel2.h"
#include "filebrowserpopup.h"
#include "application.h"

#include "tw/mainshell.h"
#include "tw/label.h"
#include "tw/optionmenu.h"
#include "tw/button.h"
#include "tw/checkbox.h"
#include "tw/keycodes.h"

#include "tconvert.h"
#include "util.h"
#include "tsystem.h"

#include "tw/message.h"

using namespace TwConsts;

//==============================================================================

FilePathField::FilePathField(TWidget *parent, string name)
	: TTextField(parent, name)
{
	m_page = dynamic_cast<CasmConfigPanel2 *>(parent);
}

//------------------------------------------------------------------------------

void FilePathField::onFocusChange(bool on)
{
	if (!on)
		close();
	TWidget::onFocusChange(on);
}

//------------------------------------------------------------------------------

void FilePathField::close()
{
	static string oldText;
	if (oldText != toString(m_text)) {
		TFilePath fp(m_text);
		try {
			string uncString;
			try {
				uncString = convertToUncString(fp);
				m_text = toWideString(uncString);
				m_page->loadSetup(TFilePath(uncString));
			} catch (TException &) {
				TFileStatus fs(fp);
				if (fs.doesExist() && !fs.isDirectory())
					m_page->loadSetup(TFilePath(toString(fp.getWideString())));
				else {
					CasmTask2 *task = dynamic_cast<CasmTask2 *>(m_page->getTask());
					if (task)
						task->m_setupFile = toString(m_text);
				}
			}
			oldText = toString(m_text);
		} catch (...) {
			TMessage::error("boh");
		}
	}
	invalidate();
}

//------------------------------------------------------------------------------

void FilePathField::keyDown(int key, unsigned long flags, const TPoint &p)
{
	if (key == TK_Return)
		close();
	else
		TTextField::keyDown(key, flags, p);
}

//==============================================================================

enum numFieldType {
	M_FROM = 0,
	M_TO,
	M_STEP,
	M_SHRINK,
	M_RTILE,
	M_MCHUNK,
	M_COLUMN,
	M_THICK,
	M_GAMMA,
	M_CHUNKSIZE
};

enum textFieldType {
	M_TCOLUMN = 0,
	M_TSETUP,
	M_TENTRYPOINT
};

namespace
{

class NumFieldChanger : public TNumField::Action
{
	CasmConfigPanel2 *m_ccp;
	numFieldType m_type;

public:
	NumFieldChanger(CasmConfigPanel2 *ccp, numFieldType type)
		: m_ccp(ccp), m_type(type)
	{
	}

	void sendCommand(const TNumField::Event &ev);
};

void NumFieldChanger::sendCommand(const TNumField::Event &ev)
{
	m_ccp->onNumField(ev.m_value, m_type);
}

class ColumnFieldChanger : public TTextField::Action
{
	CasmConfigPanel2 *m_ccp;
	textFieldType m_type;

public:
	ColumnFieldChanger(CasmConfigPanel2 *ccp, textFieldType type)
		: m_ccp(ccp), m_type(type)
	{
	}

	void sendCommand(std::wstring value)
	{
		m_ccp->onColumnField(toString(value), m_type);
	}
};
}

//------------------------------------------------------------------------------

static bool is_num(string word)
{
	string::iterator it = word.begin();
	while (it != word.end()) {
		if ((*it < '0' || *it > '9') && *it != '.')
			return false;
		it++;
	}
	return true;
}

//==============================================================================

CasmConfigPanel2::CasmConfigPanel2(TWidget *parent)
	: TaskConfigPanel(parent), m_task(0)
{
	m_chunkSizeLabel = new TLabel(this);
	m_chunkSizeLabel->setText("Task Chunk Size:");
	m_chunkSize = new TNumField(this);
	m_chunkSize->setIsInteger(true);
	m_chunkSize->setRange(1, std::numeric_limits<int>::max());
	m_chunkSize->addAction(new NumFieldChanger(this, M_CHUNKSIZE));

	m_setupLabel = new TLabel(this);
	m_setupLabel->setText("Setup Path:");
	m_setupTextField = new FilePathField(this);
	//m_setupTextField->addAction(new ColumnFieldChanger(this, M_TSETUP));
	m_setupBrowseBtn = new TButton(this);
	m_setupBrowseBtn->setTitle("...");
	tconnect(*m_setupBrowseBtn, this, &CasmConfigPanel2::browseSetupFiles);

	m_fromLabel = new TLabel(this);
	m_fromLabel->setText("From:");
	m_from = new TNumField(this);
	m_from->setIsInteger(true);
	m_from->addAction(new NumFieldChanger(this, M_FROM));

	m_toLabel = new TLabel(this);
	m_toLabel->setText("To:");
	m_to = new TNumField(this);
	m_to->setIsInteger(true);
	m_to->addAction(new NumFieldChanger(this, M_TO));

	m_stepLabel = new TLabel(this);
	m_stepLabel->setText("Step:");
	m_step = new TNumField(this);
	m_step->setIsInteger(true);
	m_step->addAction(new NumFieldChanger(this, M_STEP));

	m_shrinkLabel = new TLabel(this);
	m_shrinkLabel->setText("Shrink:");
	m_shrink = new TNumField(this);
	m_shrink->setIsInteger(true);
	m_shrink->addAction(new NumFieldChanger(this, M_SHRINK));

	m_renderTileLabel = new TLabel(this);
	m_renderTileLabel->setText("Render Tile:");
	m_renderTile = new TNumField(this);
	m_renderTile->setIsInteger(true);
	m_renderTile->addAction(new NumFieldChanger(this, M_RTILE));

	m_memChunkLabel = new TLabel(this);
	m_memChunkLabel->setText("Mem Chunk:");
	m_memChunk = new TNumField(this);
	m_memChunk->setIsInteger(true);
	m_memChunk->addAction(new NumFieldChanger(this, M_MCHUNK));

	m_subPixelMoveLabel = new TLabel(this);
	m_subPixelMoveLabel->setText("Subpixel Move:");
	m_subPixelMove = new TOptionMenu(this);
	m_subPixelMove->addOption("None");
	m_subPixelMove->addOption("RGB");
	m_subPixelMove->addOption("All");
	m_subPixelMove->setAction(new TOptionMenuAction<CasmConfigPanel2>(this, &CasmConfigPanel2::onSubPixelMove));

	m_processingLabel = new TLabel(this);
	m_processingLabel->setText("Processing:");
	m_processing = new TOptionMenu(this);
	m_processing->addOption("32 Bit");
	m_processing->addOption("64 Bit");
	m_processing->setAction(new TOptionMenuAction<CasmConfigPanel2>(this, &CasmConfigPanel2::onProcessing));

	m_resampleQualityLabel = new TLabel(this);
	m_resampleQualityLabel->setText("Resample Quality:");
	m_resampleQuality = new TOptionMenu(this);
	m_resampleQuality->addOption("Standard");
	m_resampleQuality->addOption("Improved");
	m_resampleQuality->addOption("High");
	m_resampleQuality->setAction(new TOptionMenuAction<CasmConfigPanel2>(this, &CasmConfigPanel2::onResampleQuality));

	m_focus = new TCheckBox(this, "Constant Focus");
	tconnect<CasmConfigPanel2>(*m_focus, this, onConstantFocus);
	m_noOverwrite = new TCheckBox(this, "No Overwrite");
	tconnect<CasmConfigPanel2>(*m_noOverwrite, this, onOverwrite);
	m_multimedia = new TCheckBox(this, "Multimedia");
	tconnect<CasmConfigPanel2>(*m_multimedia, this, onMultimedia);

	m_column = new TTextField(this);
	m_column->setText("All");
	m_column->disable();
	m_column->addAction(new ColumnFieldChanger(this, M_TCOLUMN));
	m_lineart = new TCheckBox(this, "Lineart");
	m_lineart->disable();
	tconnect<CasmConfigPanel2>(*m_lineart, this, onLineart);
	m_edgeAliasing = new TCheckBox(this, "Edge Aliasing");
	m_edgeAliasing->disable();
	tconnect<CasmConfigPanel2>(*m_edgeAliasing, this, onEdgeAliasing);
	m_thickness = new TNumField(this);
	m_thickness->setIsInteger(true);
	m_thickness->setRange(0, 15);
	m_thickness->setValue(8);
	m_thickness->disable();
	m_thickness->addAction(new NumFieldChanger(this, M_THICK));

	m_gamma = new TCheckBox(this, "Gamma");
	tconnect<CasmConfigPanel2>(*m_gamma, this, onGamma);
	m_gammaValue = new TNumField(this);
	m_gammaValue->setValue(0);
	m_gammaValue->disable();
	m_gammaValue->addAction(new NumFieldChanger(this, M_GAMMA));

	m_clap = new TCheckBox(this, "Add Clap");
	//tconnect<CasmConfigPanel2>(*m_clap, this, onClap);
	/*
  m_outputScriptLabel = new TLabel(this);
  m_outputScriptLabel->setText("Output Script:");
  m_outputScript = new TOptionMenu(this);
  m_outputScript->addOption("None");
  //m_subPixelMove->setAction(new TOptionMenuAction<CasmConfigPanel2>(this, &CasmConfigPanel2::onOutputScript));
  m_entryPointLabel = new TLabel(this);
  m_entryPointLabel->setText("Entry Point:");
  m_entryPoint = new TTextField(this);
  m_entryPoint->setText("");
  m_entryPoint->addAction(new ColumnFieldChanger(this, M_TENTRYPOINT));
  */
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::configureNotify(const TDimension &d)
{
	int x0 = 20;
	int y = d.ly - 1 - 10;

	int x = x0;

	m_setupLabel->setGeometry(x, y - 15, x + 100, y);
	m_setupTextField->setGeometry(x + 104, y - 15, x + 650, y);
	m_setupBrowseBtn->setGeometry(x + 660, y - 15, x + 680, y);

	y -= 60;

	m_chunkSizeLabel->setGeometry(x, y - 15, x + 100, y);
	m_chunkSize->setGeometry(x + 110, y - 15, x + 156, y);

	y -= 40;

	m_fromLabel->setGeometry(x, y - 15, x + 50, y);
	m_from->setGeometry(x + 54, y - 15, x + 100, y);
	x += 120;

	m_toLabel->setGeometry(x, y - 15, x + 80, y);
	m_to->setGeometry(x + 84, y - 15, x + 130, y);
	x += 150;

	m_stepLabel->setGeometry(x, y - 15, x + 80, y);
	m_step->setGeometry(x + 84, y - 15, x + 130, y);

	///
	x = x0;
	y -= 40;

	m_shrinkLabel->setGeometry(x, y - 15, x + 50, y);
	m_shrink->setGeometry(x + 54, y - 15, x + 100, y);
	x += 120;

	m_renderTileLabel->setGeometry(x, y - 15, x + 80, y);
	m_renderTile->setGeometry(x + 84, y - 15, x + 130, y);
	x += 150;

	m_memChunkLabel->setGeometry(x, y - 15, x + 80, y);
	m_memChunk->setGeometry(x + 84, y - 15, x + 130, y);

	x = x0;
	y -= 40;

	m_subPixelMoveLabel->setGeometry(x, y - 15, x + 94, y);
	m_subPixelMove->setGeometry(x + 98, y - 15, x + 160, y);
	x += 180;

	m_processingLabel->setGeometry(x, y - 15, x + 74, y);
	m_processing->setGeometry(x + 78, y - 15, x + 140, y);
	x += 160;

	m_resampleQualityLabel->setGeometry(x, y - 15, x + 114, y);
	m_resampleQuality->setGeometry(x + 118, y - 15, x + 200, y);

	x = x0;
	y -= 40;

	x += 8;
	m_focus->setGeometry(x, y - 15, x + 90, y);
	x += 120;
	m_noOverwrite->setGeometry(x, y - 15, x + 80, y);
	x += 110;
	m_gamma->setGeometry(x, y - 15, x + 60, y);
	x += 70;
	m_gammaValue->setGeometry(x, y - 15, x + 30, y);
	x += 60;
	m_clap->setGeometry(x, y - 15, x + 90, y);

	x = x0;
	y -= 40;

	x += 8;
	m_multimedia->setGeometry(x, y - 15, x + 80, y);
	x += 90;
	m_column->setGeometry(x, y - 15, x + 20, y);
	x += 30;
	m_lineart->setGeometry(x, y - 15, x + 60, y);
	x += 65;
	m_edgeAliasing->setGeometry(x, y - 15, x + 80, y);
	x += 90;
	m_thickness->setGeometry(x, y - 15, x + 30, y);
	/*
  x = x0;
  y -=40;
  m_outputScriptLabel->setGeometry(x, y-15, x+94, y);
  m_outputScript->setGeometry(x+98, y-15, x+160, y);
  x +=180;

  m_entryPointLabel->setGeometry(x, y-15, x+94, y);
  m_entryPoint->setGeometry(x+98, y-15, x+160, y);
  */
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::setTask(SubmitPageTask *task)
{
	m_task = dynamic_cast<CasmTask2 *>(task);
	assert(m_task);

	m_setupTextField->setText(m_task->m_setupFile);
	m_chunkSize->setValue(m_task->m_taskChunksize);
	m_from->setValue(m_task->m_start);
	m_to->setValue(m_task->m_end);
	m_step->setValue(m_task->m_step);

	m_shrink->setValue(m_task->m_reduction);
	m_renderTile->setValue(m_task->m_renderTile);
	m_memChunk->setValue(m_task->m_memChunk);

	string move;
	if (m_task->m_moveType == CasmTask2::M_INTMOVE)
		move = "None";
	else if (m_task->m_moveType == CasmTask2::M_BG_FRACMOVE)
		move = "RGB";
	else
		move = "All";
	m_subPixelMove->setText(move);

	string bits;
	if (m_task->m_prec == 32)
		bits = "32 Bit";
	else
		bits = "64 Bit";
	m_processing->setText(bits);

	string quality = "";
	if (m_task->m_restype == CasmTask2::RES_TYPE_STANDARD)
		quality = "Standard";
	else if (m_task->m_restype == CasmTask2::RES_TYPE_IMPROVED)
		quality = "Improved";
	else if (m_task->m_restype == CasmTask2::RES_TYPE_HIGH)
		quality = "High";
	m_resampleQuality->setText(quality);

	m_focus->select(m_task->m_cf);
	m_noOverwrite->select(m_task->m_noOverwrite);
	m_multimedia->select(m_task->m_multimedia);

	if (m_task->m_numColumn < 0)
		m_column->setText("All");
	else
		m_column->setText(toString(m_task->m_numColumn));

	if (m_task->m_lineart < 0) {
		m_lineart->disable();
		m_lineart->select(false);
	} else {
		m_lineart->enable();
		m_lineart->select(true);
	}

	if (m_task->m_edgeAliasing < 0) {
		m_edgeAliasing->disable();
		m_edgeAliasing->select(false);
	} else {
		m_edgeAliasing->enable();
		m_edgeAliasing->select(true);
	}
	if (m_task->m_edgeAliasing >= 0 || m_task->m_lineart >= 0)
		m_thickness->setValue(tmax(m_task->m_edgeAliasing, m_task->m_lineart));
	else
		m_thickness->setValue(8);

	if (m_task->m_gamma < 0.) {
		m_gamma->select(false);
		m_gammaValue->setValue(0.0);
		m_gammaValue->disable();
	} else {
		m_gamma->select(true);
		m_gammaValue->setValue(m_task->m_gamma);
		m_gammaValue->enable();
	}
}

//------------------------------------------------------------------------------

SubmitPageTask *CasmConfigPanel2::getTask() const
{
	return m_task;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onNumField(double v, int type)
{
	if (!m_task)
		return;

	switch (type) {
	case M_FROM:
		m_task->m_start = (int)v;
		break;
	case M_TO:
		m_task->m_end = (int)v;
		break;
	case M_STEP:
		m_task->m_step = (int)v;
		break;
	case M_SHRINK:
		m_task->m_reduction = (int)v;
		break;
	case M_RTILE:
		m_task->m_renderTile = (int)v;
		break;
	case M_MCHUNK:
		m_task->m_memChunk = (int)v;
		break;
	case M_THICK:
		if (m_lineart->isSelected())
			m_task->m_lineart = (int)v;
		if (m_edgeAliasing->isSelected())
			m_task->m_edgeAliasing = (int)v;
		break;
	case M_GAMMA:
		m_task->m_gamma = v;
		break;
	case M_CHUNKSIZE:
		m_task->m_taskChunksize = (int)v;
		break;
	}
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onColumnField(string value, int type)
{
	if (!m_task)
		return;

	switch (type) {
	case M_TCOLUMN:
		if (value == "" || value == "ALL" || value == "All" ||
			value == "all" || value == "AL" || value == "Al" ||
			value == "a" || value == "A")
			m_task->m_numColumn = -1;
		else if (is_num(value))
			m_task->m_numColumn = atoi(value.c_str());
		else {
			m_task->m_numColumn = -1;
			m_column->setText("All");
			TMessage::error("Only \"All\" or a number is a valid argument");
		}
		break;
	case M_TSETUP:
		m_task->m_setupFile = value;
		break;
	default:
		break;
	}
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::browseSetupFiles()
{
	static FileBrowserPopup *popup = 0;

	if (!popup) {
		CasmConfigPanel2 *csp = this;
		vector<string> fileTypes;
		fileTypes.push_back("setup");
		popup = new FileBrowserPopup(csp, fileTypes);

		popup->setOkAction(
			new TFileBrowserPopupAction<CasmConfigPanel2>(csp, &CasmConfigPanel2::loadSetup));
	}

	if (!popup)
		return;

	TFilePath fp = Application::instance()->getCurrentFolder();
	/*
  E'stato necessario fare questo controllo perche' il popup non e' in grado
  di impostare opportunamente le cose
  */
	string path = toString(fp.getWideString()).c_str();
	if (fp != TFilePath() && (path[0] != '\\'))
		popup->setCurrentDir(fp);

	TDimension d = TMainshell::getMainshell()->getSize();
#ifdef WIN32
	HDC hdc = GetDC(0);
	d.lx = GetDeviceCaps(hdc, HORZRES);
	d.ly = GetDeviceCaps(hdc, VERTRES);
	ReleaseDC(0, hdc);
#endif

	d -= popup->getSize();
	//TDimension d = TMainshell::getMainshell()->getSize() - popup->getSize();
	popup->popup(TPoint(d.lx / 2, d.ly / 2));
	popup->setCaption("Load Setup");
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::loadSetup(const TFilePath &fp)
{
	if (!TFileStatus(fp).doesExist()) {
		string msg = toString(fp.getWideString()) + "\n";
		msg += "File not found\n";
		msg += "Please verify that the path and file name are correct";
		m_setupTextField->setText("");
		TMessage::error(msg);
		return;
	}

	string uncString = toString(fp.getWideString());
	if (toString(m_setupTextField->getText()) != uncString)
		try {
			uncString = convertToUncString(fp);
		} catch (TException &e) {
			TMessage::error(toString(e.getMessage()));
			return;
		}

	Application::instance()->setCurrentFolder(fp.getParentDir());

	m_setupTextField->setText(uncString);

#ifdef WIN32
	int x = uncString.find("\\", 2);
	while (x != string::npos) {
		uncString.replace(x, 1, "/", 0, 1);
		x = uncString.find("\\", x + 1);
	}
#endif

	if (m_task)
		m_task->m_setupFile = uncString;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onMultimedia(TCheckBox *box, bool on)
{
	if (!m_task)
		return;

	if (on) {
		m_task->m_multimedia = true;
		m_column->enable();
		m_lineart->enable();
		m_edgeAliasing->enable();
		if (m_lineart->isSelected() || m_edgeAliasing->isSelected())
			m_thickness->enable();
		else
			m_thickness->disable();
	} else {
		m_task->m_multimedia = false;
		m_column->disable();
		m_lineart->disable();
		m_lineart->select(false);
		m_edgeAliasing->disable();
		m_edgeAliasing->select(false);
		m_thickness->setValue(8);
		m_thickness->disable();
		m_task->m_lineart = m_task->m_edgeAliasing = -1;
	}
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onLineart(TCheckBox *box, bool on)
{
	if (!m_task)
		return;

	if (on) {
		m_task->m_lineart = (int)m_thickness->getValue();
		m_task->m_edgeAliasing = -1;
		m_lineart->enable();
		m_edgeAliasing->disable();
		m_thickness->enable();
	} else {
		m_task->m_lineart = -1;
		m_lineart->enable();
		m_edgeAliasing->enable();
		m_thickness->disable();
	}
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onEdgeAliasing(TCheckBox *box, bool on)
{
	if (!m_task)
		return;

	if (on) {
		m_task->m_edgeAliasing = (int)m_thickness->getValue();
		m_task->m_lineart = -1;
		m_lineart->disable();
		m_edgeAliasing->enable();
		m_thickness->enable();
	} else {
		m_task->m_edgeAliasing = -1;
		m_lineart->enable();
		m_edgeAliasing->enable();
		m_thickness->disable();
	}
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onConstantFocus(TCheckBox *box, bool on)
{
	if (!m_task)
		return;
	m_task->m_cf = on;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onOverwrite(TCheckBox *box, bool on)
{
	if (!m_task)
		return;

	m_task->m_noOverwrite = on;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onGamma(TCheckBox *box, bool on)
{
	if (!m_task)
		return;

	if (on)
		m_task->m_gamma = m_gammaValue->getValue();
	else
		m_task->m_gamma = -1.;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onSubPixelMove(string move)
{
	if (!m_task)
		return;

	if (move == "None")
		m_task->m_moveType = CasmTask2::M_INTMOVE;
	else if (move == "RGB")
		m_task->m_moveType = CasmTask2::M_BG_FRACMOVE;
	else
		m_task->m_moveType = CasmTask2::M_FRACMOVE;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onProcessing(string bits)
{
	if (!m_task)
		return;

	if (bits == "32 Bit")
		m_task->m_prec = 32;
	else
		m_task->m_prec = 64;
}

//------------------------------------------------------------------------------

void CasmConfigPanel2::onResampleQuality(string quality)
{
	if (!m_task)
		return;

	if (quality == "Standard")
		m_task->m_restype = CasmTask2::RES_TYPE_STANDARD;
	else if (quality == "Improved")
		m_task->m_restype = CasmTask2::RES_TYPE_IMPROVED;
	else if (quality == "High")
		m_task->m_restype = CasmTask2::RES_TYPE_HIGH;
	else
		m_task->m_restype = CasmTask2::RES_TYPE_NONE;
}