MPDiagram.cpp

/*
 *  This file is part of the AiBO+ project
 *
 *  Copyright (C) 2005-2016 Csaba Kertész (csaba.kertesz@gmail.com)
 *
 *  AiBO+ 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.
 *
 *  AiBO+ 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., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 *
 */

#include "MPDiagram.hpp"

#include <qwt_legend.h>
#include <qwt_plot_marker.h>
#include <qwt_scale_widget.h>
#include <qwt_text_label.h>

#include <qlayout.h>

#include <boost/scoped_ptr.hpp>

QwtText MPTimeScaleDraw::label(double value) const
{
  QString NumberStr;

  NumberStr.sprintf("%1.2f", value);
  if (NumberStr.endsWith(QLatin1String(".00")))
  {
    NumberStr.remove(NumberStr.size()-3, 3);
  }
  QwtText Text(NumberStr);

  Text.setColor(Qt::black);
  return Text;
}


MPPlotCurve::MPPlotCurve() : QwtPlotCurve()
{
  setRenderHint(QwtPlotItem::RenderAntialiased);
  setLegendAttribute(QwtPlotCurve::LegendShowLine, true);
  // Make the line legend item longer than the default
  setLegendIconSize(QSize(20, 5));
  setStyle(QwtPlotCurve::Lines);
}


void MPLegendLabel::drawText(QPainter* painter, const QRectF& rect)
{
  QRectF newRect(rect);

  // Does this defect only occur on Ubuntu where Qt4 and Qt5 live together?
#if defined(__unix__) && !defined(__ANDROID__)
  // Fix the vertical alignment problem of the legend labels,
  if (text().text().contains("<sub>"))
  {
    newRect.setY(newRect.y()-14);
  } else {
    newRect.setY(newRect.y()-6);
  }
#endif
  QwtTextLabel::drawText(painter, newRect);
}


void MPLegendLabel::paintEvent(QPaintEvent* event)
{
  // Don't paint the clicked state
  if (isDown())
  {
    setDown(false);
  }
  QwtLegendLabel::paintEvent(event);
}


QWidget* MPLegend::createWidget(const QwtLegendData& legend_data) const
{
  QwtLegendLabel* Label = new MPLegendLabel();

  // Remove all margins and indents to spare space on the UI
  Label->setMargin(0);
  Label->setIndent(0);
  Label->setItemMode(defaultItemMode());
  Label->setData(legend_data);
  connect(Label, SIGNAL(clicked()), SLOT(itemClicked()));
  connect(Label, SIGNAL(checked(bool)), SLOT(itemChecked(bool)));
  return Label;
}


MPDiagram::MPDiagram(bool enable_legend) : QwtPlot(), MinorTicksGrid(new QwtPlotGrid),
  MajorTicksGrid(new QwtPlotGrid)
{
  if (enable_legend)
  {
    insertLegend(new MPLegend(), QwtPlot::TopLegend);
    QWidget* ContentsWidget = static_cast<QwtLegend*>(legend())->contentsWidget();
#if defined(__ANDROID__)
    boost::scoped_ptr<QFont> Font(new QFont("Courier New", 7));

    setAxisFont(QwtPlot::xBottom, *Font);
    setAxisFont(QwtPlot::yLeft, *Font);
    boost::scoped_ptr<QFont> TitleFont(new QFont("Courier New", 8, QFont::Bold));
    QwtText Title;

    Title.setFont(*TitleFont);
    setAxisTitle(QwtPlot::xBottom, Title);
    ContentsWidget->setStyleSheet(ContentsWidget->styleSheet()+" QWidget { font: 8pt \"Courier New\"; }");
#else
    ContentsWidget->setStyleSheet(ContentsWidget->styleSheet()+" QWidget { font: 10.5pt \"Courier New\"; }");
#endif
    // Remove the spacing between the legend elements
    ContentsWidget->layout()->setMargin(0);
    ContentsWidget->layout()->setSpacing(0);
  }
  setContentsMargins(10, 10, 10, 10);
  // Set helper grids
  MinorTicksGrid->setMajorPen(QPen(Qt::gray, 0, Qt::DotLine));
  MinorTicksGrid->setMinorPen(QPen(Qt::gray, 0, Qt::DotLine));
  MinorTicksGrid->attach(this);
  SetMinorGridEnabled(true);
  MajorTicksGrid->setMajorPen(QPen(Qt::black, 0, Qt::DotLine));
  MajorTicksGrid->attach(this);
  SetMajorGridEnabled(true);
  // Set axis
  QwtText AxisTitle = axisTitle(QwtPlot::xBottom);

  AxisTitle.setColor(Qt::black);
  setAxisTitle(QwtPlot::xBottom, AxisTitle);
  setAxisScaleDraw(QwtPlot::xBottom, new MPTimeScaleDraw);
  setAxisScaleDraw(QwtPlot::yLeft, new MPTimeScaleDraw);
  axisWidget(QwtPlot::xBottom)->setPalette(Qt::white);
  axisWidget(QwtPlot::yLeft)->setPalette(Qt::white);
}


MPDiagram::~MPDiagram()
{
  for (auto iter = MissingDataMarkers.begin(); iter != MissingDataMarkers.end(); ++iter)
  {
    iter.value()->detach();
    delete iter.value();
  }
  MissingDataMarkers.clear();
  for (auto iter = StickMarkers.begin(); iter != StickMarkers.end(); ++iter)
  {
    iter.value()->detach();
    delete iter.value();
  }
  StickMarkers.clear();
}


void MPDiagram::SetHorizontalTicks(double first, double last)
{
  QList<double> Ticks;

  for (int i = 0; i <= 10; ++i)
  {
    Ticks.push_back(first+i*(last-first) / 10);
  }
  QwtScaleDiv ScaleDiv = axisScaleDiv(QwtPlot::xBottom);

  ScaleDiv.setTicks(QwtScaleDiv::MajorTick, Ticks);
  Ticks.clear();
  for (int i = 0; i <= 40; ++i)
  {
    Ticks.push_back(first+i*(last-first) / 40);
  }
  ScaleDiv.setTicks(QwtScaleDiv::MinorTick, Ticks);
  setAxisScaleDiv(QwtPlot::xBottom, ScaleDiv);
  MinorTicksGrid->setXDiv(ScaleDiv);
  MajorTicksGrid->setXDiv(ScaleDiv);
}


void MPDiagram::SetVerticalTicks(double first, double last)
{
  QList<double> Ticks;

  for (int i = 0; i <= 10; ++i)
  {
    Ticks.push_back(first+i*(last-first) / 10);
  }
  QwtScaleDiv ScaleDiv = axisScaleDiv(QwtPlot::yLeft);

  ScaleDiv.setTicks(QwtScaleDiv::MajorTick, Ticks);
  Ticks.clear();
  ScaleDiv.setTicks(QwtScaleDiv::MediumTick, Ticks);
  Ticks.clear();
  ScaleDiv.setTicks(QwtScaleDiv::MinorTick, Ticks);
  setAxisScaleDiv(QwtPlot::yLeft, ScaleDiv);
  MinorTicksGrid->setYDiv(ScaleDiv);
  MajorTicksGrid->setYDiv(ScaleDiv);
}


void MPDiagram::SetMinorGridEnabled(bool new_state)
{
  if (new_state)
  {
    MinorTicksGrid->enableX(true);
    MinorTicksGrid->enableY(true);
    MinorTicksGrid->enableXMin(false);
    MinorTicksGrid->enableYMin(true);
  } else {
    MinorTicksGrid->enableX(false);
    MinorTicksGrid->enableY(false);
    MinorTicksGrid->enableXMin(false);
    MinorTicksGrid->enableYMin(false);
  }
}


void MPDiagram::SetMajorGridEnabled(bool new_state)
{
  if (new_state)
  {
    MajorTicksGrid->enableX(false);
    MajorTicksGrid->enableY(true);
    MajorTicksGrid->enableXMin(false);
    MajorTicksGrid->enableYMin(false);
  } else {
    MajorTicksGrid->enableX(false);
    MajorTicksGrid->enableY(false);
    MajorTicksGrid->enableXMin(false);
    MajorTicksGrid->enableYMin(false);
  }
}


QwtPlotMarker* MPDiagram::AddMissingDataMarker(float value)
{
  if (MissingDataMarkers.contains(value))
    return MissingDataMarkers[value];

  QwtPlotMarker* NewMarker = new QwtPlotMarker();

  NewMarker->setLineStyle(QwtPlotMarker::VLine);
  NewMarker->setLabelAlignment(Qt::AlignRight | Qt::AlignBottom);
  NewMarker->setLinePen(QPen(QColor(0x90, 0x90, 0x90), 0, Qt::DashLine));
  NewMarker->setValue(value, 0.0);
  NewMarker->attach(this);
  MissingDataMarkers.insert(value, NewMarker);
  return NewMarker;
}


QwtPlotMarker* MPDiagram::AddStickMarker(float value, const QColor& color, bool big_stick)
{
  if (StickMarkers.contains(value))
    return StickMarkers[value];

  QwtPlotMarker* NewMarker = new QwtPlotMarker();

  if (big_stick)
    NewMarker->setLineStyle(QwtPlotMarker::VDoubleStick);
  else
    NewMarker->setLineStyle(QwtPlotMarker::VStick);
  NewMarker->setLabelAlignment(Qt::AlignRight | Qt::AlignBottom);
  NewMarker->setLinePen(QPen(color, 0, Qt::SolidLine));
  NewMarker->setValue(value, 0.0);
  NewMarker->attach(this);
  StickMarkers.insert(value, NewMarker);
  return NewMarker;
}


void MPDiagram::RemoveMissingDataMarker(float value)
{
  auto Iter = MissingDataMarkers.find(value);

  if (Iter == MissingDataMarkers.end())
    return;

  Iter.value()->detach();
  delete Iter.value();
  MissingDataMarkers.erase(Iter);
}


void MPDiagram::RemoveStickMarker(float value)
{
  auto Iter = StickMarkers.find(value);

  if (Iter == StickMarkers.end())
    return;

  Iter.value()->detach();
  delete Iter.value();
  StickMarkers.erase(Iter);
}