/*    This file is a part of Calyp project
 *    Copyright (C) 2014-2019  by Joao Carreira   (jfmcarreira@gmail.com)
 *                                Luis Lucas      (luisfrlucas@gmail.com)
 *
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License along
 *   with this program; if not, write to the Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

/**
 * \file     QualityMeasurementSidebar.cpp
 * \brief    Definition of the quality measurement sidebar
 */

#include "QualityMeasurementSidebar.h"

#include "SubWindowHandle.h"
#include "VideoSubWindow.h"

#include <QtGui>

QualityMeasurementSidebar::QualityMeasurementSidebar( QWidget* parent, SubWindowHandle* windowManager )
    : QWidget( parent ), m_pcMainWindowManager( windowManager ), m_pcCurrentVideoSubWindow( NULL )
{
  // Side bar area -----------------------------------------------------

  QLabel* MetricLabel = new QLabel( tr( "Metric:" ) );
  MetricLabel->setAlignment( Qt::AlignVCenter | Qt::AlignRight );

  QLabel* RefLabel = new QLabel( tr( "Ref:" ) );
  RefLabel->setAlignment( Qt::AlignVCenter | Qt::AlignRight );

  QLabel* RecLabel = new QLabel( tr( "Rec:" ) );
  RecLabel->setAlignment( Qt::AlignVCenter | Qt::AlignRight );

  m_comboBoxMetric = new QComboBox;
  for( unsigned int i = 0; i < CalypFrame::supportedQualityMetricsList().size(); i++ )
  {
    m_comboBoxMetric->insertItem( i, CalypFrame::supportedQualityMetricsList()[i].c_str() );
  }
  m_comboBoxMetric->setSizeAdjustPolicy( QComboBox::AdjustToContents );
  m_comboBoxMetric->setSizePolicy( QSizePolicy::Expanding, QSizePolicy::Fixed );

  m_comboBoxRef = new QComboBox;
  QSpacerItem* spaceQuality = new QSpacerItem( 1, 1, QSizePolicy::Expanding, QSizePolicy::Fixed );
  m_comboBoxRef->setSizeAdjustPolicy( QComboBox::AdjustToContents );
  m_comboBoxRef->setSizePolicy( QSizePolicy::Expanding, QSizePolicy::Fixed );
  updateSubWindowList();

  QGridLayout* mainLayout = new QGridLayout;
  mainLayout->addWidget( MetricLabel, 0, 0, Qt::AlignLeft );
  mainLayout->addWidget( m_comboBoxMetric, 0, 1, Qt::AlignLeft );
  mainLayout->addItem( spaceQuality, 0, 2, Qt::AlignRight );
  mainLayout->addWidget( RefLabel, 1, 0, Qt::AlignLeft );
  mainLayout->addWidget( m_comboBoxRef, 1, 1, 1, 2, Qt::AlignLeft );

  m_ppcLabelQualityLabel[CLP_LUMA] = new QLabel( "PSNR Y:" );
  m_ppcLabelQualityLabel[CLP_LUMA]->setAlignment( Qt::AlignLeft | Qt::AlignVCenter );
  m_ppcLabelQualityValue[CLP_LUMA] = new QLabel;
  m_ppcLabelQualityValue[CLP_LUMA]->setAlignment( Qt::AlignRight | Qt::AlignVCenter );

  m_ppcLabelQualityLabel[CLP_CHROMA_U] = new QLabel( "PSNR U:" );
  m_ppcLabelQualityLabel[CLP_CHROMA_U]->setAlignment( Qt::AlignLeft | Qt::AlignVCenter );
  m_ppcLabelQualityValue[CLP_CHROMA_U] = new QLabel;
  m_ppcLabelQualityValue[CLP_CHROMA_U]->setAlignment( Qt::AlignRight | Qt::AlignVCenter );

  m_ppcLabelQualityLabel[CLP_CHROMA_V] = new QLabel( "PSNR V:" );
  m_ppcLabelQualityLabel[CLP_CHROMA_V]->setAlignment( Qt::AlignLeft | Qt::AlignVCenter );
  m_ppcLabelQualityValue[CLP_CHROMA_V] = new QLabel;
  m_ppcLabelQualityValue[CLP_CHROMA_V]->setAlignment( Qt::AlignRight | Qt::AlignVCenter );

  QGridLayout* qualityResultsLayout = new QGridLayout;
  qualityResultsLayout->addWidget( m_ppcLabelQualityLabel[CLP_LUMA], 0, 0 );
  qualityResultsLayout->addWidget( m_ppcLabelQualityValue[CLP_LUMA], 0, 1 );
  qualityResultsLayout->addWidget( m_ppcLabelQualityLabel[CLP_CHROMA_U], 1, 0 );
  qualityResultsLayout->addWidget( m_ppcLabelQualityValue[CLP_CHROMA_U], 1, 1 );
  qualityResultsLayout->addWidget( m_ppcLabelQualityLabel[CLP_CHROMA_V], 2, 0 );
  qualityResultsLayout->addWidget( m_ppcLabelQualityValue[CLP_CHROMA_V], 2, 1 );

  // QGroupBox *statisticsGroup = new QGroupBox( tr( "Quality" ) );
  QGroupBox* statisticsGroup = new QGroupBox;
  statisticsGroup->setLayout( qualityResultsLayout );
  statisticsGroup->setWhatsThis( tr( "<p>Here you can see the quality results calculated for the "
                                     "selected windows." ) );

  mainLayout->addWidget( statisticsGroup, 4, 0, 3, 3 );

  mainLayout->setRowStretch( 8, 10 );
  setLayout( mainLayout );

  connect( m_comboBoxRef, SIGNAL( activated( int ) ), this, SLOT( slotReferenceChanged( int ) ) );
  connect( m_comboBoxMetric, SIGNAL( currentIndexChanged( int ) ), this, SLOT( slotQualityMetricChanged( int ) ) );
}

QualityMeasurementSidebar::~QualityMeasurementSidebar() {}

QSize QualityMeasurementSidebar::sizeHint() const
{
  QSize currSize = size();
  QSize bestSize( 240, currSize.height() );
  if( currSize.width() < bestSize.width() )
    return bestSize;
  else
    return currSize;
}

void QualityMeasurementSidebar::updateSideBar( bool hasSubWindow )
{
  if( !hasSubWindow )
    m_comboBoxRef->clear();
  setEnabled( hasSubWindow );
}

void QualityMeasurementSidebar::updateSubWindowList()
{
  QString currSubWindowName;
  m_pcWindowListNames.clear();
  m_pcVideoWindowList.clear();

  VideoSubWindow* pcVideoSubWindow;
  QList<SubWindowAbstract*> subWindowList = m_pcMainWindowManager->findSubWindow( SubWindowAbstract::VIDEO_SUBWINDOW );
  for( int i = 0; i < subWindowList.size(); i++ )
  {
    pcVideoSubWindow = qobject_cast<VideoSubWindow*>( subWindowList.at( i ) );
    if( m_pcCurrentVideoSubWindow == pcVideoSubWindow )
    {
      continue;
    }
    if( m_pcCurrentVideoSubWindow )
    {
      if( !m_pcCurrentVideoSubWindow->getCurrFrame()->haveSameFmt( pcVideoSubWindow->getCurrFrame() ) )
      {
        continue;
      }
    }
    currSubWindowName = pcVideoSubWindow->getWindowName();
    m_pcWindowListNames.append( currSubWindowName );
    m_pcVideoWindowList.append( pcVideoSubWindow );
  }
  m_comboBoxRef->clear();
  m_comboBoxRef->insertItems( -1, m_pcWindowListNames );
  m_comboBoxRef->setCurrentIndex( -1 );
  if( m_pcCurrentVideoSubWindow )
  {
    if( VideoSubWindow* refSubWindow = m_pcCurrentVideoSubWindow->getRefSubWindow() )
    {
      int index = m_pcVideoWindowList.indexOf( refSubWindow );
      m_comboBoxRef->setCurrentIndex( index );
    }
  }
  //  m_comboBoxRef->setSizeAdjustPolicy( QComboBox::AdjustToContents );
}

void QualityMeasurementSidebar::updateCurrentWindow( VideoSubWindow* subWindow )
{
  m_pcCurrentVideoSubWindow = subWindow;
  if( m_pcCurrentVideoSubWindow )
  {
    int index = -1;
    if( VideoSubWindow* refSubWindow = m_pcCurrentVideoSubWindow->getRefSubWindow() )
    {
      index = m_pcVideoWindowList.indexOf( refSubWindow );
    }
    m_comboBoxRef->setCurrentIndex( index );
    updateSubWindowList();
    updateSidebarData();
  }
}

void QualityMeasurementSidebar::updateSidebarData()
{
  QString zeroValue( "0.0000" );
  if( m_pcCurrentVideoSubWindow )
  {
    VideoSubWindow* refSubWindow = m_pcCurrentVideoSubWindow->getRefSubWindow();
    if( refSubWindow )
    {
      QApplication::setOverrideCursor( Qt::WaitCursor );

      CalypFrame* currFrame = m_pcCurrentVideoSubWindow->getCurrFrame();
      CalypFrame* refFrame = refSubWindow->getCurrFrame();
      double quality;
      QString value;
      unsigned int component = 0;
      for( ; component < currFrame->getNumberChannels(); component++ )
      {
        quality = currFrame->getQuality( m_comboBoxMetric->currentIndex(), refFrame, component );
        m_ppcLabelQualityValue[component]->setText( value.setNum( quality, 'f', 4 ) );
      }
      for( ; component < 3; component++ )
      {
        m_ppcLabelQualityValue[component]->setText( zeroValue );
      }
      QApplication::restoreOverrideCursor();
      return;
    }
    else
    {
      m_comboBoxRef->setCurrentIndex( -1 );
    }
  }
  for( int component = 0; component < 3; component++ )
  {
    m_ppcLabelQualityValue[component]->setText( zeroValue );
  }
}

void QualityMeasurementSidebar::updateQualityMetric( int idx )
{
  m_comboBoxMetric->setCurrentIndex( idx );
  QString labelQuality( CalypFrame::supportedQualityMetricsList()[idx].c_str() );
  m_ppcLabelQualityLabel[CLP_LUMA]->setText( labelQuality + " Y" );
  m_ppcLabelQualityLabel[CLP_CHROMA_U]->setText( labelQuality + " U" );
  m_ppcLabelQualityLabel[CLP_CHROMA_V]->setText( labelQuality + " V" );
  updateSidebarData();
}

void QualityMeasurementSidebar::slotReferenceChanged( int index )
{
  if( index > -1 && m_pcCurrentVideoSubWindow )
  {
    VideoSubWindow* pcRefSubWindow = m_pcVideoWindowList.at( index );
    if( pcRefSubWindow )
    {
      if( m_pcCurrentVideoSubWindow->getCurrFrame()->haveSameFmt( pcRefSubWindow->getCurrFrame() ) )
      {
        m_pcCurrentVideoSubWindow->setRefSubWindow( pcRefSubWindow );
        updateSidebarData();
        m_comboBoxRef->setCurrentIndex( index );
        return;
      }
    }
  }
  else
  {
    m_comboBoxRef->setCurrentIndex( -1 );
  }
}

void QualityMeasurementSidebar::slotQualityMetricChanged( int idx )
{
  emit signalQualityMetricChanged( idx );
  // updateQualityMetric( idx );
}