#include "toonzqt/validatedchoicedialog.h"

// Qt includes
#include <QLabel>
#include <QPushButton>
#include <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();
}