edgetx-gx12/companion/src/simulator.cpp

/****************************************************************************
**
** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
** All rights reserved.
** Contact: Nokia Corporation (qt-info@nokia.com)
**
** This file is part of the examples of the Qt Toolkit.
**
** $QT_BEGIN_LICENSE:LGPL$
** Commercial Usage
** Licensees holding valid Qt Commercial licenses may use this file in
** accordance with the Qt Commercial License Agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and Nokia.
**
** GNU Lesser General Public License Usage
** Alternatively, this file may be used under the terms of the GNU Lesser
** General Public License version 2.1 as published by the Free Software
** Foundation and appearing in the file LICENSE.LGPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU Lesser General Public License version 2.1 requirements
** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
**
** In addition, as a special exception, Nokia gives you certain additional
** rights.  These rights are described in the Nokia Qt LGPL Exception
** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3.0 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.  Please review the following information to
** ensure the GNU General Public License version 3.0 requirements will be
** met: http://www.gnu.org/copyleft/gpl.html.
**
** If you have questions regarding the use of this file, please contact
** Nokia at qt-info@nokia.com.
** $QT_END_LICENSE$
**
****************************************************************************/

#include <QApplication>
#include <QTranslator>
#include <QLocale>
#include <QString>
#include <QDir>
#include <QFileInfo>
#include <QSplashScreen>
#include <QThread>
#include <QDebug>
#include <QTextStream>
#if defined(JOYSTICKS) || defined(SIMU_AUDIO)
  #include <SDL.h>
  #undef main
#endif
#include "simulatordialog.h"
#include "eeprominterface.h"
#include "appdata.h"
#include "qxtcommandoptions.h"

#if defined WIN32 || !defined __GNUC__
#include <windows.h>
#define sleep(x) Sleep(x*1000)
#else
#include <unistd.h>
#endif

#ifdef __APPLE__
#include <QProxyStyle> 

class MyProxyStyle : public QProxyStyle
 {
   public:
    void polish ( QWidget * w ) {
      QMenu* mn = dynamic_cast<QMenu*>(w);
      QPushButton* pb = dynamic_cast<QPushButton*>(w);
      if(!(mn || pb) && !w->testAttribute(Qt::WA_MacNormalSize))
          w->setAttribute(Qt::WA_MacSmallSize);
    }
 };
#endif

int main(int argc, char *argv[])
{
  Q_INIT_RESOURCE(companion);
  QApplication app(argc, argv);
  app.setApplicationName("OpenTX Simulator");
  app.setOrganizationName("OpenTX");
  app.setOrganizationDomain("open-tx.org");

#ifdef __APPLE__
  app.setStyle(new MyProxyStyle);
#endif

  /* QTranslator companionTranslator;
  companionTranslator.load(":/companion_" + locale);
  QTranslator qtTranslator;
  qtTranslator.load((QString)"qt_" + locale.left(2), QLibraryInfo::location(QLibraryInfo::TranslationsPath));
  app.installTranslator(&companionTranslator);
  app.installTranslator(&qtTranslator);
*/

  QTextCodec::setCodecForCStrings(QTextCodec::codecForName("UTF-8"));

#if defined(JOYSTICKS) || defined(SIMU_AUDIO)
  uint32_t sdlFlags = 0;
  #ifdef JOYSTICKS
    sdlFlags |= SDL_INIT_JOYSTICK;
  #endif
  #ifdef SIMU_AUDIO
    sdlFlags |= SDL_INIT_AUDIO;
  #endif
  if (SDL_Init(sdlFlags) < 0) {
    QTextStream stream(stderr);
    stream << "WARNING: couldn't initialize SDL: " << SDL_GetError() << endl;
  }
#endif

  SimulatorDialog *dialog;
  QString eepromFileName;
  QDir eedir;
  QFile file;

  registerSimulators();
  registerOpenTxFirmwares();

  eedir = QDir(QDesktopServices::storageLocation(QDesktopServices::DocumentsLocation));
  if (!eedir.exists("OpenTX")) {
    eedir.mkdir("OpenTX");
  }
  eedir.cd("OpenTX");

  QStringList firmwareIds;
  int currentIdx = 0;
  foreach(SimulatorFactory *factory, registered_simulators) {
    firmwareIds << factory->name();
    if (factory->name() == g.lastSimulator()) {
      currentIdx = firmwareIds.size() - 1;
    }
  }

  QxtCommandOptions options;
  options.add("radio", "radio to simulate", QxtCommandOptions::ValueRequired);
  options.alias("radio", "r");
  options.add("help", "show this help text");
  options.alias("help", "h");
  options.parse(QCoreApplication::arguments());
  if(options.count("help") || options.showUnrecognizedWarning()) {
    options.showUsage();
    // list all available radios
    QTextStream stream(stdout);
    stream << endl << "Available radios:" << endl;
    foreach(QString name, firmwareIds) {
      stream << "\t" << name << endl;
    }
    return 1;
  }


  bool ok = false;
  QString firmwareId;
  if (options.count("radio") == 1) {
    firmwareId = options.value("radio").toString();
    if (firmwareIds.contains(firmwareId)) {
      ok = true;
    }
  }
  if (!ok) {
    firmwareId = QInputDialog::getItem(0, QObject::tr("Radio type"), 
                                                QObject::tr("Which radio type do you want to simulate?"),
                                                firmwareIds, currentIdx, false, &ok);
  }
  qDebug() << "firmwareId" << firmwareId;

  if (ok && !firmwareId.isEmpty()) {
    if (firmwareId != g.lastSimulator()) {
      g.lastSimulator(firmwareId);
    }
    QString radioId;
    int pos = firmwareId.indexOf("-");
    if (pos > 0) {
      radioId = firmwareId.mid(pos+1);
      pos = radioId.lastIndexOf("-");
      if (pos > 0) {
        radioId = radioId.mid(0, pos);
      }
    }
    qDebug() << "radioId" << radioId;
    current_firmware_variant = GetFirmware(firmwareId);
    qDebug() << "current_firmware_variant" << current_firmware_variant->getName();

    if (options.positional().isEmpty()) {
      eepromFileName = QString("eeprom-%1.bin").arg(radioId);
      eepromFileName = eedir.filePath(eepromFileName.toAscii());
    }
    else {
      eepromFileName = options.positional()[0];
    }
    qDebug() << "eepromFileName" << eepromFileName;
    // TODO display used eeprom filename somewhere

    SimulatorFactory *factory = getSimulatorFactory(firmwareId);
    if (!factory) {
      QTextStream stream(stderr);
      stream << "factory not found" << endl;
      return 2;
    }
    if (factory->type() == BOARD_HORUS)
      dialog = new SimulatorDialogHorus(NULL, factory->create());
    else if (factory->type() == BOARD_FLAMENCO)
      dialog = new SimulatorDialogFlamenco(NULL, factory->create());
    else if (factory->type() == BOARD_TARANIS)
      dialog = new SimulatorDialogTaranis(NULL, factory->create(), SIMULATOR_FLAGS_S1|SIMULATOR_FLAGS_S2);
    else
      dialog = new SimulatorDialog9X(NULL, factory->create());
  }
  else {
    return 0;
  }

  dialog->show();
  dialog->start(eepromFileName.toAscii().constData());

  int result = app.exec();

  delete dialog;

  unregisterSimulators();
  unregisterOpenTxFirmwares();

#if defined(JOYSTICKS) || defined(SIMU_AUDIO)
  SDL_Quit();
#endif

  return result;
}