RasterHistogramWidget.cpp

Go to the documentation of this file.

/*  Copyright (C) 2008 National Institute For Space Research (INPE) - Brazil.

    This file is part of the TerraLib - a Framework for building GIS enabled applications.

    TerraLib is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License,
    or (at your option) any later version.

    TerraLib 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with TerraLib. See COPYING. If not, write to
    TerraLib Team at <terralib-team@terralib.org>.
 */

/*!
  \file terralib/qt/widgets/rp/RasterHistogramWidget.cpp

  \brief This file has the RasterHistogramWidget class.
*/

// TerraLib
#include "../../../dataaccess/dataset/DataSet.h"
#include "../../../dataaccess/utils/Utils.h"
#include "../../../raster/Band.h"
#include "../../../raster/Raster.h"
#include "../../../raster/RasterSummaryManager.h"
#include "../../../raster/Utils.h"
#include "../charts/ChartDisplay.h"
#include "../charts/ChartStyle.h"
#include "../charts/Histogram.h"
#include "../charts/HistogramChart.h"
#include "../charts/HistogramStyle.h"
#include "RasterHistogramWidget.h"
#include "ui_RasterHistogramWidgetForm.h"

// Qt
#include <QGridLayout>

//Qwt
#include <qwt_legend.h>

//STL
#include <memory>

te::qt::widgets::RasterHistogramWidget::RasterHistogramWidget(QWidget* parent, Qt::WindowFlags f)
  : QWidget(parent, f),
    m_ui(new Ui::RasterHistogramWidgetForm)
{
  m_ui->setupUi(this);

  m_ui->m_applyToolButton->setIcon(QIcon::fromTheme("chart-bar"));

  m_inputRaster = nullptr;
  m_minValueLine = nullptr;
  m_maxValueLine = nullptr;

  //Creating and adjusting the chart Display's style.
  m_chartStyle = new te::qt::widgets::ChartStyle();
  m_chartStyle->setTitle(tr(""));
  m_chartStyle->setAxisX(tr("Gray Level"));
  m_chartStyle->setAxisY(tr("Frequency"));
  m_chartStyle->setGridChecked(true);

  //build form
  QGridLayout* layout = new QGridLayout(m_ui->m_widget);

  //Creating and adjusting the chart Display
  m_chartDisplay = new te::qt::widgets::ChartDisplay(m_ui->m_widget, QString::fromUtf8(""), m_chartStyle);
  m_chartDisplay->adjustDisplay();
  m_chartDisplay->show();
  m_chartDisplay->replot();

  layout->addWidget(m_chartDisplay, 0, 0);
  layout->setContentsMargins(0,0,0,0);

  m_histogramInput = new te::qt::widgets::Histogram();
  m_histogramChartInput = new te::qt::widgets::HistogramChart(m_histogramInput);
  m_histogramChartInput->setPen(Qt::black);
  m_histogramChartInput->setBrush(QBrush(Qt::blue));
  m_histogramChartInput->attach(m_chartDisplay);
  m_histogramChartInput->setTitle(tr("Input"));

  m_histogramOutput = new te::qt::widgets::Histogram();
  m_histogramChartOutput = new te::qt::widgets::HistogramChart(m_histogramOutput);
  m_histogramChartOutput->setPen(Qt::black, 3.);
  m_histogramChartOutput->setBrush(QBrush(QColor(255, 0, 0, 127)));
  m_histogramChartOutput->setStyle(QwtPlotHistogram::Outline);
  m_histogramChartOutput->attach(m_chartDisplay);
  m_histogramChartOutput->setTitle(tr("Output"));

  m_chartDisplay->insertLegend(new QwtLegend(), QwtPlot::RightLegend);
  m_chartDisplay->enableTools(false);

  //connects
  connect(m_ui->m_applyToolButton, SIGNAL(clicked()), this, SLOT(onApplyToolButtonClicked()));
  connect(m_chartDisplay, SIGNAL(leftPointSelected(const QPointF &)), this, SLOT(onLeftPointSelected(const QPointF &)));
  connect(m_chartDisplay, SIGNAL(rigthPointSelected(const QPointF &)), this, SLOT(onRigthPointSelected(const QPointF &)));

  //hide tool bar
  m_ui->m_frame->setVisible(false);
}

te::qt::widgets::RasterHistogramWidget::~RasterHistogramWidget() = default;

Ui::RasterHistogramWidgetForm* te::qt::widgets::RasterHistogramWidget::getForm() const
{
  return m_ui.get();
}

void te::qt::widgets::RasterHistogramWidget::setChartInputColor(int red, int green, int blue)
{
  m_histogramChartInput->setBrush(QBrush(QColor(red, green, blue)));
}

void te::qt::widgets::RasterHistogramWidget::setEnableTools(bool enable)
{
  m_chartDisplay->enableTools(enable);
}

void te::qt::widgets::RasterHistogramWidget::setEnableNavigationTools(bool enable)
{
  m_chartDisplay->enableMultiSelection(enable);
  m_chartDisplay->enableMagnifier(enable);
  m_chartDisplay->enablePan(enable);
}

void te::qt::widgets::RasterHistogramWidget::setInputRaster(te::rst::Raster* raster)
{
  if(!raster)
    return;

  m_inputRaster = raster;

  //set bands from input raster
  m_ui->m_bandComboBox->clear();

  for(unsigned int i = 0; i < m_inputRaster->getNumberOfBands(); ++i)
  {
    QString strBand;
    strBand.setNum(i);

    m_ui->m_bandComboBox->addItem(strBand);
  }
}

void te::qt::widgets::RasterHistogramWidget::setOutputRaster(te::rst::Raster* raster)
{
  m_outputRaster.reset(raster);
}

void te::qt::widgets::RasterHistogramWidget::drawHistogram(int band)
{
  QString toFind = QString::number(band);
  int idx = m_ui->m_bandComboBox->findText(toFind);
  m_ui->m_bandComboBox->setCurrentIndex(idx);

  if(m_inputRaster)
  {
    m_histogramInput->setValues(std::map<te::dt::AbstractData*, unsigned int>());

    std::map<double, unsigned int>* values = te::rst::RasterSummaryManager::getInstance().get(m_inputRaster, te::rst::SUMMARY_R_HISTOGRAM)->at(band).m_histogramR;

    for(std::map<double, unsigned int>::iterator it = (*values).begin(); it != (*values).end(); ++it)
    {
      m_histogramInput->insert(std::make_pair(new te::dt::Double(it->first), it->second));
    }
    
    m_histogramChartInput->setData();
    m_histogramChartInput->attach(m_chartDisplay);
  }
  else
    m_histogramChartInput->detach();

  if(m_outputRaster.get())
  {
    m_histogramOutput->setValues(std::map<te::dt::AbstractData*, unsigned int>());

    double max = 0.;

    std::map<double, unsigned int> values =  m_outputRaster->getBand(band)->getHistogramR();

    for(std::map<double, unsigned int>::iterator it = values.begin(); it != values.end(); ++it)
    {
      m_histogramOutput->insert(std::make_pair(new te::dt::Double(it->first), it->second));

      if(it->second > max)
        max = it->second;
    }

    m_histogramChartOutput->setData();
    m_histogramChartOutput->attach(m_chartDisplay);
  }
  else
    m_histogramChartOutput->detach();

  m_chartDisplay->updateLayout();
  m_chartDisplay->replot();
}

void te::qt::widgets::RasterHistogramWidget::setMinimumValueEnabled(bool enable)
{
  if(m_minValueLine)
  {
    if(enable)
      m_minValueLine->attach(m_chartDisplay);
    else
      m_minValueLine->detach();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMinimumValueLine(int value, bool replot)
{
  if(!m_minValueLine)
  {
    m_minValueLine = new QwtPlotMarker();
    m_minValueLine->setLabel(QString::fromUtf8("Minimum"));
    m_minValueLine->setLabelAlignment(Qt::AlignLeft | Qt::AlignVCenter);
    m_minValueLine->setLabelOrientation(Qt::Vertical);
    m_minValueLine->setLineStyle(QwtPlotMarker::VLine);
    m_minValueLine->setLinePen(Qt::darkRed, 2, Qt::DashDotLine);
    m_minValueLine->attach(m_chartDisplay);
  }

  m_minValueLine->setXValue(value);

  if(replot)
  {
    m_chartDisplay->updateLayout();

    m_chartDisplay->replot();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMinimumValueLine(double value, bool replot)
{
  if(!m_minValueLine)
  {
    m_minValueLine = new QwtPlotMarker();
    m_minValueLine->setLabel(QString::fromUtf8("Minimum"));
    m_minValueLine->setLabelAlignment(Qt::AlignLeft | Qt::AlignVCenter);
    m_minValueLine->setLabelOrientation(Qt::Vertical);
    m_minValueLine->setLineStyle(QwtPlotMarker::VLine);
    m_minValueLine->setLinePen(Qt::darkRed, 2, Qt::DashDotDotLine);
    m_minValueLine->attach(m_chartDisplay);
  }

  m_minValueLine->setXValue(value);

  if(replot)
  {
    m_chartDisplay->updateLayout();

    m_chartDisplay->replot();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMinimumValueLabel(QString value)
{
  if(m_minValueLine)
    m_minValueLine->setLabel(value);
}

void te::qt::widgets::RasterHistogramWidget::setMaximumValueEnabled(bool enable)
{
  if(m_maxValueLine)
  {
    if(enable)
      m_maxValueLine->attach(m_chartDisplay);
    else
      m_maxValueLine->detach();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMaximumValueLine(int value, bool replot)
{
  if(!m_maxValueLine)
  {
    m_maxValueLine = new QwtPlotMarker();
    m_maxValueLine->setLabel(QString::fromUtf8("Maximum"));
    m_maxValueLine->setLabelAlignment(Qt::AlignLeft | Qt::AlignVCenter);
    m_maxValueLine->setLabelOrientation(Qt::Vertical);
    m_maxValueLine->setLineStyle(QwtPlotMarker::VLine);
    m_maxValueLine->setLinePen(Qt::darkRed, 2, Qt::DashDotLine);
    m_maxValueLine->attach(m_chartDisplay);
  }

  m_maxValueLine->setXValue(value);

  if(replot)
  {
    m_chartDisplay->updateLayout();

    m_chartDisplay->replot();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMaximumValueLine(double value, bool replot)
{
  if(!m_maxValueLine)
  {
    m_maxValueLine = new QwtPlotMarker();
    m_maxValueLine->setLabel(QString::fromUtf8("Maximum"));
    m_maxValueLine->setLabelAlignment(Qt::AlignLeft | Qt::AlignVCenter);
    m_maxValueLine->setLabelOrientation(Qt::Vertical);
    m_maxValueLine->setLineStyle(QwtPlotMarker::VLine);
    m_maxValueLine->setLinePen(Qt::darkRed, 2, Qt::DashDotDotLine);
    m_maxValueLine->attach(m_chartDisplay);
  }

  m_maxValueLine->setXValue(value);

  if(replot)
  {
    m_chartDisplay->updateLayout();

    m_chartDisplay->replot();
  }
}

void te::qt::widgets::RasterHistogramWidget::updateMaximumValueLabel(QString value)
{
  if(m_maxValueLine)
    m_maxValueLine->setLabel(value);
}

void te::qt::widgets::RasterHistogramWidget::clearMinMaxLines()
{
  delete m_minValueLine;
  m_minValueLine = nullptr;

  delete m_maxValueLine;
  m_maxValueLine = nullptr;
}

void te::qt::widgets::RasterHistogramWidget::clear()
{
  m_histogramChartInput->detach();
  m_histogramChartOutput->detach();
  delete m_histogramInput;
  delete m_histogramOutput;

  clearMinMaxLines();

  m_inputRaster = nullptr;
  m_outputRaster.reset(nullptr);

  m_histogramInput = new te::qt::widgets::Histogram();
  m_histogramChartInput = new te::qt::widgets::HistogramChart(m_histogramInput);
  m_histogramChartInput->setPen(Qt::black);
  m_histogramChartInput->setBrush(QBrush(Qt::blue));
  m_histogramChartInput->attach(m_chartDisplay);
  m_histogramChartInput->setTitle(tr("Input"));

  m_histogramOutput = new te::qt::widgets::Histogram();
  m_histogramChartOutput = new te::qt::widgets::HistogramChart(m_histogramOutput);
  m_histogramChartOutput->setPen(Qt::black, 3.);
  m_histogramChartOutput->setBrush(QBrush(QColor(255, 0, 0, 127)));
  m_histogramChartOutput->setStyle(QwtPlotHistogram::Outline);
  m_histogramChartOutput->attach(m_chartDisplay);
  m_histogramChartOutput->setTitle(tr("Output"));

  m_chartDisplay->replot();
}

void te::qt::widgets::RasterHistogramWidget::onApplyToolButtonClicked()
{
  int index = m_ui->m_bandComboBox->currentIndex();

  drawHistogram(index);
}

void te::qt::widgets::RasterHistogramWidget::onLeftPointSelected(const QPointF& point)
{
  emit minValueSelected((int)point.x(), m_ui->m_bandComboBox->currentIndex());

  emit minValueSelected((double)point.x(), m_ui->m_bandComboBox->currentIndex());
}

void te::qt::widgets::RasterHistogramWidget::onRigthPointSelected(const QPointF& point)
{
  emit maxValueSelected((int)point.x(), m_ui->m_bandComboBox->currentIndex());

  emit maxValueSelected((double)point.x(), m_ui->m_bandComboBox->currentIndex());
}