#include <QGroupBox>
#include <QRadioButton>
#include <QDialogButtonBox>
#include <QGridLayout>
#include <QPushButton>
#include <QFileDialog>
#include <QLabel>

#include "SpkDiarizationDialog.h"

SpkDiarizationDialog::SpkDiarizationDialog(const QString &title, bool local, bool view, QWidget *parent)
  : QDialog(parent),
    m_dist(UtteranceTree::Mahal),
    m_agrCrit(UtteranceTree::Ward),
    m_partMeth(UtteranceTree::Silhouette),
    m_base(false),
    m_refSpk(false),
    m_sigma(false),
    m_local(local)
{
  setWindowTitle(title);

  QGroupBox *methBox = new QGroupBox("Local clustering method:");
  QRadioButton *hier = new QRadioButton("Bottom-Up hierarchical clustering");
  QRadioButton *base = new QRadioButton("Baseline");
  QRadioButton *refSpk = new QRadioButton("Reference speakers");
  hier->setChecked(true);
  QGridLayout *methLayout = new QGridLayout;
  methLayout->addWidget(hier, 0, 0);
  methLayout->addWidget(base, 1, 0);
  methLayout->addWidget(refSpk, 2, 0);
  methBox->setLayout(methLayout);

  m_ubm = new QCheckBox(tr("New GMM/UBM"));
  m_ubm->hide();

  QGroupBox *distBox = new QGroupBox("Distance between utterance vectors:");
  m_l2 = new QRadioButton("L2");
  m_mahal = new QRadioButton("Mahalanobis");
  m_mahal->setChecked(true);
  QGridLayout *distLayout = new QGridLayout;
  distLayout->addWidget(m_l2, 0, 0);
  distLayout->addWidget(m_mahal, 0, 1);
  distBox->setLayout(distLayout);

  m_norm = new QCheckBox(tr("Normalize vectors"));
  m_norm->setChecked(true);

  QGroupBox *covBox = new QGroupBox("Covariance matrix:");
  QChar sigmaChar(0xa3, 0x03);
  QRadioButton *sigma = new QRadioButton(sigmaChar);
  QRadioButton *w = new QRadioButton("W");
  w->setChecked(true);
  QGridLayout *covLayout = new QGridLayout;
  covLayout->addWidget(sigma, 0, 0);
  covLayout->addWidget(w, 0, 1);
  covBox->setLayout(covLayout);

  QGroupBox *agrBox = new QGroupBox("Aggregation criterion:");
  QRadioButton *min = new QRadioButton("Min.");
  QRadioButton *max = new QRadioButton("Max.");
  QRadioButton *mean = new QRadioButton("Mean");
  QRadioButton *ward = new QRadioButton("Ward");
  ward->setChecked(true);
  QGridLayout *agrLayout = new QGridLayout;
  agrLayout->addWidget(min, 0, 0);
  agrLayout->addWidget(max, 0, 1);
  agrLayout->addWidget(mean, 0, 2);
  agrLayout->addWidget(ward, 0, 3);
  agrBox->setLayout(agrLayout);
  
  QGroupBox *partBox = new QGroupBox("Method used to select the best partition:");
  QRadioButton *sil = new QRadioButton("Silhouette");
  QRadioButton *bip = new QRadioButton("Bipartition");
  sil->setChecked(true);
  QGridLayout *partLayout = new QGridLayout;
  partLayout->addWidget(sil, 0, 0);
  partLayout->addWidget(bip, 0, 1);
  partBox->setLayout(partLayout);

  m_weight = new QCheckBox(tr("Weight Utterances"));

  QDialogButtonBox *buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok
						     | QDialogButtonBox::Cancel);

  QGridLayout *gridLayout = new QGridLayout;
  gridLayout->addWidget(methBox, 0, 0);
  gridLayout->addWidget(m_ubm, 1, 0);
  gridLayout->addWidget(distBox, 2, 0);
  gridLayout->addWidget(covBox, 3, 0);
  gridLayout->addWidget(m_norm, 4, 0);
  gridLayout->addWidget(agrBox, 5, 0);
  gridLayout->addWidget(partBox, 6, 0);
  gridLayout->addWidget(m_weight, 7, 0);
  gridLayout->addWidget(buttonBox, 8, 0, 1, 2, Qt::AlignHCenter);

  setLayout(gridLayout);

  // hide options in case of view request
  if (view) {
    methBox->hide();
    distBox->hide();
    covBox->hide();
    m_norm->hide();
    agrBox->hide();
    partBox->hide();
    m_weight->hide();
  }

  // connecting signals to corresponding slots
  connect(m_l2, SIGNAL(clicked()), this, SLOT(activL2()));
  connect(m_l2, SIGNAL(clicked(bool)), covBox, SLOT(setDisabled(bool)));
  connect(m_mahal, SIGNAL(clicked()), this, SLOT(activMahal()));
  connect(m_mahal, SIGNAL(clicked(bool)), covBox, SLOT(setEnabled(bool)));
  connect(sigma, SIGNAL(clicked()), this, SLOT(activSigma()));
  connect(w, SIGNAL(clicked()), this, SLOT(activW()));
  connect(min, SIGNAL(clicked()), this, SLOT(activMin()));
  connect(max, SIGNAL(clicked()), this, SLOT(activMax()));
  connect(mean, SIGNAL(clicked()), this, SLOT(activMean()));
  connect(ward, SIGNAL(clicked()), this, SLOT(activWard()));
  connect(sil, SIGNAL(clicked()), this, SLOT(activSilhouette()));
  connect(bip, SIGNAL(clicked()), this, SLOT(activBipartition()));
  connect(base, SIGNAL(clicked()), this, SLOT(activBase()));
  connect(hier, SIGNAL(clicked()), this, SLOT(activHier()));
  connect(refSpk, SIGNAL(clicked()), this, SLOT(activRefSpk()));

  connect(base, SIGNAL(clicked(bool)), m_ubm, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), distBox, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), covBox, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), m_norm, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), agrBox, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), partBox, SLOT(setDisabled(bool)));
  connect(base, SIGNAL(clicked(bool)), m_weight, SLOT(setDisabled(bool)));

  connect(refSpk, SIGNAL(clicked(bool)), m_ubm, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), distBox, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), covBox, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), m_norm, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), agrBox, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), partBox, SLOT(setDisabled(bool)));
  connect(refSpk, SIGNAL(clicked(bool)), m_weight, SLOT(setDisabled(bool)));
  
  connect(hier, SIGNAL(clicked(bool)), m_ubm, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), distBox, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), covBox, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), m_norm, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), agrBox, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), partBox, SLOT(setEnabled(bool)));
  connect(hier, SIGNAL(clicked(bool)), m_weight, SLOT(setEnabled(bool)));

  connect(buttonBox, SIGNAL(accepted()), this, SLOT(accept()));
  connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
}

///////////////
// accessors //
///////////////

QString SpkDiarizationDialog::getSpeakersFName() const
{
  return m_speakersFName;
}

UtteranceTree::DistType SpkDiarizationDialog::getDist() const
{
  return m_dist;
}

UtteranceTree::AgrCrit SpkDiarizationDialog::getAgrCrit() const
{
  return m_agrCrit;
}

UtteranceTree::PartMeth SpkDiarizationDialog::getPartMeth() const
{
  return m_partMeth;
}

bool SpkDiarizationDialog::getBase() const
{
  return m_base;
}

bool SpkDiarizationDialog::getRefSpk() const
{
  return m_refSpk;
}

bool SpkDiarizationDialog::getNorm() const
{
  return m_norm->isChecked();
}

bool SpkDiarizationDialog::getWeight() const
{
  return m_weight->isChecked();
}

bool SpkDiarizationDialog::getSigma() const
{
  return m_sigma;
}

bool SpkDiarizationDialog::getUbm() const
{
  return m_ubm->isChecked();
}

///////////
// slots //
///////////

void SpkDiarizationDialog::setSpeakersFName()
{
  m_speakersFName = QFileDialog::getOpenFileName(this, tr("Open characters labels file"), tr("/home/xavier/Dropbox/tv_series_proc_tool/tools/spkDiarization/ivecs"), tr("Data File (*.csv)"));
}

void SpkDiarizationDialog::activL2()
{
  m_dist = UtteranceTree::L2;
}

void SpkDiarizationDialog::activMahal()
{
  m_dist = UtteranceTree::Mahal;
}

void SpkDiarizationDialog::activSigma()
{
  m_sigma = true;
}

void SpkDiarizationDialog::activW()
{
  m_sigma = false;
}

void SpkDiarizationDialog::activMin()
{
  m_agrCrit = UtteranceTree::Min;
}

void SpkDiarizationDialog::activMax()
{
  m_agrCrit = UtteranceTree::Max;
}

void SpkDiarizationDialog::activMean()
{
  m_agrCrit = UtteranceTree::Mean;
}

void SpkDiarizationDialog::activWard()
{
  m_agrCrit = UtteranceTree::Ward;
  update();
}

void SpkDiarizationDialog::activSilhouette()
{
  m_partMeth = UtteranceTree::Silhouette;
}

void SpkDiarizationDialog::activBipartition()
{
  m_partMeth = UtteranceTree::Bipartition;
}

void SpkDiarizationDialog::activBase()
{
  m_base = true;
  m_refSpk = false;
}
 
void SpkDiarizationDialog::activHier()
{
  m_base = false;
  m_refSpk = false;
}

void SpkDiarizationDialog::activRefSpk()
{
  m_base = false;
  m_refSpk = true;
}