#include "toonzqt/validatedchoicedialog.h"

// Qt includes

#include <qlabel></qlabel>

#include <qpushbutton></qpushbutton>

#include <qbuttongroup></qbuttongroup>

//************************************************************************

//    ValidatedChoiceDialog  implementation

//************************************************************************

DVGui::ValidatedChoiceDialog::ValidatedChoiceDialog(QWidget *parent, Options opts)

	: Dialog(parent, true, false)

{

	setModal(true);

	m_buttonGroup = new QButtonGroup(this);

	m_buttonGroup->setExclusive(true);

	bool ret = true;

	QPushButton *okBtn = new QPushButton(QString(tr("Apply")), this);

	ret = ret && connect(okBtn, SIGNAL(clicked()), SLOT(accept()));

	addButtonBarWidget(okBtn);

	QPushButton *okToAllBtn = new QPushButton(QString(tr("Apply to All")), this);

	ret = ret && connect(okToAllBtn, SIGNAL(clicked()), SLOT(onApplyToAll()));

	addButtonBarWidget(okToAllBtn);

	QPushButton *cancelBtn = new QPushButton(QString(tr("Cancel")), this);

	ret = ret && connect(cancelBtn, SIGNAL(clicked()), SLOT(reject()));

	addButtonBarWidget(cancelBtn);

	assert(ret);

	reset();

	// Start layout

	beginVLayout();

	m_label = new QLabel(this);

	addWidget(m_label);

	// enbVLayout() must be invoked by derived classes

}

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

void DVGui::ValidatedChoiceDialog::reset()

{

	m_appliedToAllRes = NO_REQUIRED_RESOLUTION;

	m_appliedToAll = false;

}

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

QString validate(void *obj);

int DVGui::ValidatedChoiceDialog::execute(void *obj)

{

	struct Resol {

		int m_res;

		bool m_all;

		Resol(int res, bool all) : m_res(res), m_all(all) {}

	};

	Resol curResolution(NO_REQUIRED_RESOLUTION, false),

		newResolution(m_appliedToAll ? m_appliedToAllRes : NO_REQUIRED_RESOLUTION,

					  m_appliedToAll);

	bool initialize = true;

	// Loop until a resolution gets accepted

	do {

		QString err = acceptResolution(obj, curResolution.m_res, curResolution.m_all);

		if (err.isEmpty())

			break;

		if (newResolution.m_res == NO_REQUIRED_RESOLUTION) {

			// No new resolution selected - prompt for user interaction

			m_label->setText(err);

			m_applyToAll = false;

			if (initialize)

				initializeUserInteraction(obj), initialize = false;

			if (exec() == QDialog::Rejected) {

				curResolution.m_res = CANCEL;

				break;

			}

			newResolution = Resol(m_buttonGroup->checkedId(), m_applyToAll);

			assert(newResolution.m_res >= DEFAULT_RESOLUTIONS_COUNT);

		}

		// Substitute resolution and retry

		curResolution = newResolution,

		newResolution = Resol(NO_REQUIRED_RESOLUTION, false);

	} while (true);

	return curResolution.m_res;

}

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

void DVGui::ValidatedChoiceDialog::onApplyToAll()

{

	m_appliedToAllRes = m_buttonGroup->checkedId();

	m_applyToAll = m_appliedToAll = true;

	assert(m_appliedToAllRes >= DEFAULT_RESOLUTIONS_COUNT);

	accept();

}