LibrePilot/ground/openpilotgcs/src/plugins/setupwizard/connectiondiagram.cpp

/**
 ******************************************************************************
 *
 * @file       connectiondiagram.cpp
 * @author     The OpenPilot Team, http://www.openpilot.org Copyright (C) 2012.
 * @addtogroup
 * @{
 * @addtogroup ConnectionDiagram
 * @{
 * @brief
 *****************************************************************************/
/*
 * 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 3 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.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include <QDebug>
#include <QFile>
#include <QFileDialog>
#include "connectiondiagram.h"
#include "ui_connectiondiagram.h"

const char* ConnectionDiagram::FILE_NAME = ":/setupwizard/resources/connection-diagrams.svg";
const int ConnectionDiagram::IMAGE_PADDING = 10;

ConnectionDiagram::ConnectionDiagram(QWidget *parent, VehicleConfigurationSource *configSource) :
    QDialog(parent), ui(new Ui::ConnectionDiagram), m_configSource(configSource)
{
    ui->setupUi(this);
    setWindowTitle(tr("Connection Diagram"));
    setupGraphicsScene();
}

ConnectionDiagram::~ConnectionDiagram()
{
    delete ui;
}

void ConnectionDiagram::resizeEvent(QResizeEvent *event)
{
    QWidget::resizeEvent(event);
    fitInView();
}

void ConnectionDiagram::showEvent(QShowEvent *event)
{
    QWidget::showEvent(event);
    fitInView();
}

void ConnectionDiagram::fitInView()
{
    ui->connectionDiagram->setSceneRect(m_scene->itemsBoundingRect());
    ui->connectionDiagram->fitInView(
                m_scene->itemsBoundingRect().adjusted(-IMAGE_PADDING,-IMAGE_PADDING, IMAGE_PADDING, IMAGE_PADDING),
                Qt::KeepAspectRatio);
}

void ConnectionDiagram::setupGraphicsScene()
{
    m_renderer = new QSvgRenderer();
    if (QFile::exists(QString(FILE_NAME)) &&
        m_renderer->load(QString(FILE_NAME)) &&
        m_renderer->isValid()) {
        m_scene = new QGraphicsScene(this);
        ui->connectionDiagram->setScene(m_scene);

        QList<QString> elementsToShow;

        switch (m_configSource->getControllerType()) {
        case VehicleConfigurationSource::CONTROLLER_CC:
        case VehicleConfigurationSource::CONTROLLER_CC3D:
            elementsToShow << "controller-cc";
            break;
        case VehicleConfigurationSource::CONTROLLER_REVO:
            elementsToShow << "controller-revo";
            break;
        case VehicleConfigurationSource::CONTROLLER_NANO:
            elementsToShow << "controller-nano";
            break;
        case VehicleConfigurationSource::CONTROLLER_OPLINK:
        default:
            elementsToShow << "controller-cc";
            break;
        }

        switch (m_configSource->getVehicleType()) {
        case VehicleConfigurationSource::VEHICLE_MULTI:
            switch (m_configSource->getVehicleSubType()) {
            case VehicleConfigurationSource::MULTI_ROTOR_TRI_Y:
                elementsToShow << "tri";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_QUAD_X:
                elementsToShow << "quad-x";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_QUAD_PLUS:
                elementsToShow << "quad-p";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_QUAD_H:
                elementsToShow << "quad-h";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_HEXA:
                elementsToShow << "hexa";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_HEXA_X:
                elementsToShow << "hexa-x";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_HEXA_COAX_Y:
                elementsToShow << "hexa-y";
                break;
            case VehicleConfigurationSource::MULTI_ROTOR_HEXA_H:
                elementsToShow << "hexa-h";
                break;
            default:
                break;
            }
            break;
        case VehicleConfigurationSource::VEHICLE_FIXEDWING:
            switch (m_configSource->getVehicleSubType()) {
            case VehicleConfigurationSource::FIXED_WING_DUAL_AILERON:
                elementsToShow << "aileron";
                break;
            case VehicleConfigurationSource::FIXED_WING_AILERON:
                elementsToShow << "aileron-single";
                break;
            case VehicleConfigurationSource::FIXED_WING_ELEVON:
                elementsToShow << "elevon";
                break;
            default:
                break;
            }
        case VehicleConfigurationSource::VEHICLE_HELI:
        case VehicleConfigurationSource::VEHICLE_SURFACE:
        default:
            break;
        }

        switch (m_configSource->getInputType()) {
        case VehicleConfigurationSource::INPUT_PWM:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "pwm-nano";
            } else {
                elementsToShow << "pwm";
            }
            break;
        case VehicleConfigurationSource::INPUT_PPM:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "ppm-nano";
            } else {
                elementsToShow << "ppm";
            }
            break;
        case VehicleConfigurationSource::INPUT_SBUS:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "sbus-nano";
            } else {
                elementsToShow << "sbus";
            }
            break;
        case VehicleConfigurationSource::INPUT_DSMX10:
        case VehicleConfigurationSource::INPUT_DSMX11:
        case VehicleConfigurationSource::INPUT_DSM2:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "satellite-nano";
            } else {
                elementsToShow << "satellite";
            }
            break;
        default:
            break;
        }

        switch (m_configSource->getGpsType()) {
        case VehicleConfigurationSource::GPS_DISABLED:
            break;
        case VehicleConfigurationSource::GPS_NMEA:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "nano-generic-nmea";
            } else if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_REVO) {
                elementsToShow << "generic-nmea";
            }
            break;
        case VehicleConfigurationSource::GPS_PLATINUM:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "nano-OPGPS-v9";
            } else if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_REVO) {
                elementsToShow << "OPGPS-v9";
            }
            break;
        case VehicleConfigurationSource::GPS_UBX:
            if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                elementsToShow << "nano-OPGPS-v8-ublox";
            } else if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_REVO) {
                elementsToShow << "OPGPS-v8-ublox";
            }
            break;
        default:
            break;
        }

        if (m_configSource->getVehicleType() == VehicleConfigurationSource::VEHICLE_FIXEDWING &&
                m_configSource->getAirspeedType() != VehicleConfigurationSource::AIRSPEED_ESTIMATE) {
            switch (m_configSource->getAirspeedType()) {
            case VehicleConfigurationSource::AIRSPEED_EAGLETREE:
                if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                    elementsToShow << "nano-eagletree-speed-sensor";
                } else if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_REVO) {
                    elementsToShow << "eagletree-speed-sensor";
                }
                break;
            case VehicleConfigurationSource::AIRSPEED_MS4525:
                if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_NANO) {
                    elementsToShow << "nano-ms4525-speed-sensor";
                } else if (m_configSource->getControllerType() == VehicleConfigurationSource::CONTROLLER_REVO) {
                    elementsToShow << "ms4525-speed-sensor";
                }
                break;
            default:
                break;
            }
        }

        setupGraphicsSceneItems(elementsToShow);
        fitInView();
        qDebug() << "Scene complete";
    }
}

void ConnectionDiagram::setupGraphicsSceneItems(QList<QString> elementsToShow)
{
    qreal z = 0;

    foreach(QString elementId, elementsToShow) {
        if (m_renderer->elementExists(elementId)) {
            QGraphicsSvgItem *element = new QGraphicsSvgItem();
            element->setSharedRenderer(m_renderer);
            element->setElementId(elementId);
            element->setZValue(z++);
            element->setOpacity(1.0);

            QMatrix matrix = m_renderer->matrixForElement(elementId);
            QRectF orig    = matrix.mapRect(m_renderer->boundsOnElement(elementId));
            element->setPos(orig.x(), orig.y());
            m_scene->addItem(element);
            qDebug() << "Adding " << elementId << " to scene at " << element->pos();
        } else {
            qDebug() << "Element with id: " << elementId << " not found.";
        }
    }
}

void ConnectionDiagram::on_saveButton_clicked()
{
    QImage image(2200, 1100, QImage::Format_ARGB32);

    image.fill(Qt::white);
    QPainter painter(&image);
    m_scene->render(&painter);
    QString fileName = QFileDialog::getSaveFileName(this, tr("Save File"), "", tr("Images (*.png *.xpm *.jpg)"));
    if (!fileName.isEmpty()) {
        image.save(fileName);
    }
}