#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();
}