scatterdatamodifier.cpp Example File

  /****************************************************************************
  **
  ** Copyright (C) 2016 The Qt Company Ltd.
  ** Contact: https://www.qt.io/licensing/
  **
  ** This file is part of the Qt Data Visualization module of the Qt Toolkit.
  **
  ** $QT_BEGIN_LICENSE:GPL$
  ** Commercial License Usage
  ** Licensees holding valid commercial Qt licenses may use this file in
  ** accordance with the commercial license agreement provided with the
  ** Software or, alternatively, in accordance with the terms contained in
  ** a written agreement between you and The Qt Company. For licensing terms
  ** and conditions see https://www.qt.io/terms-conditions. For further
  ** information use the contact form at https://www.qt.io/contact-us.
  **
  ** GNU General Public License Usage
  ** Alternatively, this file may be used under the terms of the GNU
  ** General Public License version 3 or (at your option) any later version
  ** approved by the KDE Free Qt Foundation. The licenses are as published by
  ** the Free Software Foundation and appearing in the file LICENSE.GPL3
  ** included in the packaging of this file. Please review the following
  ** information to ensure the GNU General Public License requirements will
  ** be met: https://www.gnu.org/licenses/gpl-3.0.html.
  **
  ** $QT_END_LICENSE$
  **
  ****************************************************************************/

  #include "scatterdatamodifier.h"

  #include <QtDataVisualization/QScatterDataProxy>
  #include <QtDataVisualization/QValue3DAxis>
  #include <QtDataVisualization/Q3DScene>
  #include <QtDataVisualization/Q3DCamera>
  #include <QtDataVisualization/QScatter3DSeries>
  #include <QtDataVisualization/Q3DTheme>
  #include <QtCore/qmath.h>
  #include <QtCore/QTextStream>
  #include <QtCore/QDebug>

  using namespace QtDataVisualization;

  ScatterDataModifier::ScatterDataModifier(Q3DScatter *scatter)
      : m_graph(scatter),
        m_inputHandler(new CustomInputHandler())
  {
      m_graph->activeTheme()->setType(Q3DTheme::ThemeDigia);
      m_graph->setShadowQuality(QAbstract3DGraph::ShadowQualityMedium);
      m_graph->scene()->activeCamera()->setCameraPreset(Q3DCamera::CameraPresetFront);

      m_graph->setAxisX(new QValue3DAxis);
      m_graph->setAxisY(new QValue3DAxis);
      m_graph->setAxisZ(new QValue3DAxis);

      m_graph->axisX()->setRange(-10.0f, 10.0f);
      m_graph->axisY()->setRange(-5.0f, 5.0f);
      m_graph->axisZ()->setRange(-5.0f, 5.0f);

      QScatter3DSeries *series = new QScatter3DSeries;
      series->setItemLabelFormat(QStringLiteral("@xLabel, @yLabel, @zLabel"));
      series->setMesh(QAbstract3DSeries::MeshCube);
      series->setItemSize(0.15f);
      m_graph->addSeries(series);

      m_animationCameraX = new QPropertyAnimation(m_graph->scene()->activeCamera(), "xRotation");
      m_animationCameraX->setDuration(20000);
      m_animationCameraX->setStartValue(QVariant::fromValue(0.0f));
      m_animationCameraX->setEndValue(QVariant::fromValue(360.0f));
      m_animationCameraX->setLoopCount(-1);

      QPropertyAnimation *upAnimation = new QPropertyAnimation(m_graph->scene()->activeCamera(), "yRotation");
      upAnimation->setDuration(9000);
      upAnimation->setStartValue(QVariant::fromValue(5.0f));
      upAnimation->setEndValue(QVariant::fromValue(45.0f));

      QPropertyAnimation *downAnimation = new QPropertyAnimation(m_graph->scene()->activeCamera(), "yRotation");
      downAnimation->setDuration(9000);
      downAnimation->setStartValue(QVariant::fromValue(45.0f));
      downAnimation->setEndValue(QVariant::fromValue(5.0f));

      m_animationCameraY = new QSequentialAnimationGroup();
      m_animationCameraY->setLoopCount(-1);
      m_animationCameraY->addAnimation(upAnimation);
      m_animationCameraY->addAnimation(downAnimation);

      m_animationCameraX->start();
      m_animationCameraY->start();

      // Give ownership of the handler to the graph and make it the active handler
      m_graph->setActiveInputHandler(m_inputHandler);

      m_selectionTimer = new QTimer(this);
      m_selectionTimer->setInterval(10);
      m_selectionTimer->setSingleShot(false);
      QObject::connect(m_selectionTimer, &QTimer::timeout, this,
                       &ScatterDataModifier::triggerSelection);
      m_selectionTimer->start();
  }

  ScatterDataModifier::~ScatterDataModifier()
  {
      delete m_graph;
  }

  void ScatterDataModifier::start()
  {
      addData();
  }

  void ScatterDataModifier::addData()
  {
      QVector<QVector3D> itemList;

      // Read data items from the file to QVector
      QTextStream stream;
      QFile dataFile(":/data/data.txt");
      if (dataFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
          stream.setDevice(&dataFile);
          while (!stream.atEnd()) {
              QString line = stream.readLine();
              if (line.startsWith("#")) // Ignore comments
                  continue;
              QStringList strList = line.split(",", QString::SkipEmptyParts);
              // Each line has three data items: xPos, yPos and zPos value
              if (strList.size() < 3) {
                  qWarning() << "Invalid row read from data:" << line;
                  continue;
              }
              itemList.append(QVector3D(
                                   strList.at(0).trimmed().toFloat(),
                                   strList.at(1).trimmed().toFloat(),
                                   strList.at(2).trimmed().toFloat()));
          }
      } else {
          qWarning() << "Unable to open data file:" << dataFile.fileName();
      }

      // Add data from the QVector to datamodel
      QScatterDataArray *dataArray = new QScatterDataArray;
      dataArray->resize(itemList.count());
      QScatterDataItem *ptrToDataArray = &dataArray->first();
      for (int i = 0; i < itemList.count(); i++) {
          ptrToDataArray->setPosition(itemList.at(i));
          ptrToDataArray++;
      }

      m_graph->seriesList().at(0)->dataProxy()->resetArray(dataArray);
  }

  void ScatterDataModifier::toggleCameraAnimation()
  {
      if (m_animationCameraX->state() != QAbstractAnimation::Paused) {
          m_animationCameraX->pause();
          m_animationCameraY->pause();
      } else {
          m_animationCameraX->resume();
          m_animationCameraY->resume();
      }
  }

  void ScatterDataModifier::triggerSelection()
  {
      m_graph->scene()->setSelectionQueryPosition(m_inputHandler->inputPosition());
  }

  void ScatterDataModifier::shadowQualityUpdatedByVisual(QAbstract3DGraph::ShadowQuality sq)
  {
      int quality = int(sq);
      emit shadowQualityChanged(quality); // connected to a checkbox in main.cpp
  }

  void ScatterDataModifier::changeShadowQuality(int quality)
  {
      QAbstract3DGraph::ShadowQuality sq = QAbstract3DGraph::ShadowQuality(quality);
      m_graph->setShadowQuality(sq);
  }