#include <QGroupBox>
  #include <QRadioButton>
  #include <QDialogButtonBox>
  #include <QGridLayout>
  
  #include <opencv2/core/core.hpp>
  #include <opencv2/imgproc/imgproc.hpp>
  
  #include "CompareImageDialog.h"
  
  using namespace cv;
  
  CompareImageDialog::CompareImageDialog(const QString &title, int V, int H, int S, int defValue1, int defValue2, const QString &thresh1Text, const QString &thresh2Text, QWidget *parent)
    : QDialog(parent),
      m_histoType(2),
      m_metrics(CV_COMP_CORREL)
  {
    setWindowTitle(title);
  
    QGroupBox *histoSrcBox = new QGroupBox("Histogram source:");
    QRadioButton *lRB = new QRadioButton("Luminance");
    QRadioButton *hsRB = new QRadioButton("Hue/Saturation");
    QRadioButton *hsvRB = new QRadioButton("HSV");
    hsvRB->setChecked(true);
    QGridLayout *histoSrcLayout = new QGridLayout;
    histoSrcLayout->addWidget(lRB, 0, 0);
    histoSrcLayout->addWidget(hsRB, 0, 1);
    histoSrcLayout->addWidget(hsvRB, 0, 2);
    histoSrcBox->setLayout(histoSrcLayout);
  
    QLabel *blockNBinsLabel = new QLabel(tr("<b>Number of histogram bins for each channel:</b>"));
    m_nVBinsLabel = new QLabel(tr("Value"));
    m_nHBinsLabel = new QLabel(tr("Hue"));
    m_nSBinsLabel = new QLabel(tr("Saturation"));
    m_nVBinsSB = new QSpinBox;
    m_nHBinsSB = new QSpinBox;
    m_nSBinsSB = new QSpinBox;
    m_nVBinsSB->setMinimum(2);
    m_nVBinsSB->setValue(V);
    m_nVBinsSB->setMaximum(256);
    m_nHBinsSB->setMinimum(2);
    m_nHBinsSB->setValue(H);
    m_nHBinsSB->setMaximum(180);
    m_nSBinsSB->setMinimum(2);
    m_nSBinsSB->setValue(S);
    m_nSBinsSB->setMaximum(255);
    
    QLabel *blockNumberLabel = new QLabel(tr("<b>Number of subimages along each dimension:</b>"));
    QLabel *nVBlockLabel = new QLabel(tr("Vertical"));
    QLabel *nHBlockLabel = new QLabel(tr("Horizontal"));
    m_nVBlockSB = new QSpinBox;
    m_nHBlockSB = new QSpinBox;
    m_nVBlockSB->setMinimum(1);
    m_nVBlockSB->setValue(5);
    m_nVBlockSB->setMaximum(20);
    m_nHBlockSB->setMinimum(1);
    m_nHBlockSB->setValue(6);
    m_nHBlockSB->setMaximum(20);
  
    QGroupBox *metricsBox = new QGroupBox("Metrics:");
    QRadioButton *l1 = new QRadioButton("L1");
    QRadioButton *l2 = new QRadioButton("L2");
    QRadioButton *correl = new QRadioButton("Correlation");
    QRadioButton *chisqr = new QRadioButton("Chi-Square");
    QRadioButton *intersect = new QRadioButton("Intersection");
    QRadioButton *hellinger = new QRadioButton("Hellinger dist.");
    correl->setChecked(true);
    QGridLayout *metricsLayout = new QGridLayout;
    metricsLayout->addWidget(l1, 0, 0);
    metricsLayout->addWidget(l2, 0, 1);
    metricsLayout->addWidget(correl, 0, 2);
    metricsLayout->addWidget(chisqr, 1, 0);
    metricsLayout->addWidget(intersect, 1, 1);
    metricsLayout->addWidget(hellinger, 1, 2);
    metricsBox->setLayout(metricsLayout);
  
    QLabel *thresholdLabel1 = new QLabel(thresh1Text);
    m_thresholdSB1 = new QSpinBox;
    m_thresholdSB1->setMinimum(1);
    m_thresholdSB1->setValue(defValue1);
    m_thresholdSB1->setMaximum(100);
  
    QLabel *thresholdLabel2 = new QLabel(thresh2Text);
    m_thresholdSB2 = new QSpinBox;
    m_thresholdSB2->setMinimum(1);
    m_thresholdSB2->setValue(defValue2);
    m_thresholdSB2->setMaximum(100);
    
    if (thresh2Text.isEmpty()) {
      m_thresholdSB2->hide();
      thresholdLabel2->hide();
    }
  
    QGridLayout *gridLayout = new QGridLayout;
    gridLayout->addWidget(histoSrcBox, 0, 0, 1, 3);
    gridLayout->addWidget(blockNBinsLabel, 1, 0, 1, 3);
    gridLayout->addWidget(m_nVBinsLabel, 2, 0);
    gridLayout->addWidget(m_nVBinsSB, 2, 1);
    gridLayout->addWidget(m_nHBinsLabel, 3, 0);
    gridLayout->addWidget(m_nHBinsSB, 3, 1);
    gridLayout->addWidget(m_nSBinsLabel, 4, 0);
    gridLayout->addWidget(m_nSBinsSB, 4, 1);
    gridLayout->addWidget(blockNumberLabel, 5, 0, 1, 3);
    gridLayout->addWidget(nVBlockLabel, 6, 0);
    gridLayout->addWidget(m_nVBlockSB, 6, 1);
    gridLayout->addWidget(nHBlockLabel, 7, 0);
    gridLayout->addWidget(m_nHBlockSB, 7, 1);
    gridLayout->addWidget(metricsBox, 8, 0, 1, 3);
    gridLayout->addWidget(thresholdLabel1, 9, 0);
    gridLayout->addWidget(m_thresholdSB1, 10, 1);
    gridLayout->addWidget(thresholdLabel2, 11, 0);
    gridLayout->addWidget(m_thresholdSB2, 12, 1);
  
    m_iterCB = new QCheckBox(tr("Iterate under thresholds"));
    gridLayout->addWidget(m_iterCB, 13, 0);
  
    QDialogButtonBox *buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok
  						     | QDialogButtonBox::Cancel);
    gridLayout->addWidget(buttonBox, 14, 0, 1, 3, Qt::AlignHCenter);
  
    setLayout(gridLayout);
  
    connect(buttonBox, SIGNAL(accepted()), this, SLOT(accept()));
    connect(buttonBox, SIGNAL(rejected()), this, SLOT(reject()));
  
    // connecting histo type buttons to corresponding slots
    connect(lRB, SIGNAL(pressed()), this, SLOT(activLHisto()));
    connect(hsRB, SIGNAL(pressed()), this, SLOT(activHsHisto()));
    connect(hsvRB, SIGNAL(pressed()), this, SLOT(activHsvHisto()));
  
    // connecting metrics buttons to corresponding slots
    connect(l1, SIGNAL(pressed()), this, SLOT(activL1()));
    connect(l2, SIGNAL(pressed()), this, SLOT(activL2()));
    connect(correl, SIGNAL(pressed()), this, SLOT(activCorrel()));
    connect(chisqr, SIGNAL(pressed()), this, SLOT(activChiSqr()));
    connect(intersect, SIGNAL(pressed()), this, SLOT(activIntersect()));
    connect(hellinger, SIGNAL(pressed()), this, SLOT(activHellinger()));
  }
  
  ///////////
  // slots //
  ///////////
  
  void CompareImageDialog::activLHisto()
  {
    m_nVBinsSB->setEnabled(true);
    m_nHBinsSB->setEnabled(false);
    m_nSBinsSB->setEnabled(false);
    m_nVBinsLabel->setEnabled(true);
    m_nHBinsLabel->setEnabled(false);
    m_nSBinsLabel->setEnabled(false);
    m_histoType = 0;
  }
  
  void CompareImageDialog::activHsHisto()
  {
    m_nVBinsSB->setEnabled(false);
    m_nHBinsSB->setEnabled(true);
    m_nSBinsSB->setEnabled(true);
    m_nVBinsLabel->setEnabled(false);
    m_nHBinsLabel->setEnabled(true);
    m_nSBinsLabel->setEnabled(true);
    m_histoType = 1;
  }
  
  void CompareImageDialog::activHsvHisto()
  {
    m_nVBinsSB->setEnabled(true);
    m_nHBinsSB->setEnabled(true);
    m_nSBinsSB->setEnabled(true);
    m_nVBinsLabel->setEnabled(true);
    m_nHBinsLabel->setEnabled(true);
    m_nSBinsLabel->setEnabled(true);
    m_histoType = 2;
  }
  
  void CompareImageDialog::activL1()
  {
    m_metrics = 4;
  }
  
  void CompareImageDialog::activL2()
  {
    m_metrics = 5;
  }
  
  void CompareImageDialog::activCorrel()
  {
    m_metrics = CV_COMP_CORREL;
  }
   
  void CompareImageDialog::activChiSqr()
  {
    m_metrics = CV_COMP_CHISQR;
  }
   
  void CompareImageDialog::activIntersect()
  {
    m_metrics = CV_COMP_INTERSECT;
  }
   
  void CompareImageDialog::activHellinger()
  {
    m_metrics = CV_COMP_HELLINGER;
  }
  
  ///////////////
  // accessors //
  ///////////////
  
  int CompareImageDialog::getHistoType()
  {
    return m_histoType;
  }
  
  int CompareImageDialog::getMetrics()
  {
    return m_metrics;
  }
  
  int CompareImageDialog::getThreshold1()
  {
    return m_thresholdSB1->value();
  }
  
  int CompareImageDialog::getThreshold2()
  {
    return m_thresholdSB2->value();
  }
  
  bool CompareImageDialog::getIterate()
  {
    return m_iterCB->isChecked();
  }
  
  int CompareImageDialog::getNVBlock()
  {
    return m_nVBlockSB->value();
  }
  
  int CompareImageDialog::getNHBlock()
  {
    return m_nHBlockSB->value();
  }
  
  int CompareImageDialog::getNVBins()
  {
    return m_nVBinsSB->value();
  }
  
  int CompareImageDialog::getNHBins()
  {
    return m_nHBinsSB->value();
  }
  
  int CompareImageDialog::getNSBins()
  {
    return m_nSBinsSB->value();
  }