Chris@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@0: /*
Chris@0:     Tony
Chris@0:     An intonation analysis and annotation tool
Chris@0:     Centre for Digital Music, Queen Mary, University of London.
Chris@0:     This file copyright 2006-2012 Chris Cannam and QMUL.
Chris@0:     
Chris@0:     This program is free software; you can redistribute it and/or
Chris@0:     modify it under the terms of the GNU General Public License as
Chris@0:     published by the Free Software Foundation; either version 2 of the
Chris@0:     License, or (at your option) any later version.  See the file
Chris@0:     COPYING included with this distribution for more information.
Chris@0: */
Chris@0: 
Chris@0: #include "MainWindow.h"
Chris@0: 
Chris@0: #include "system/System.h"
Chris@0: #include "system/Init.h"
Chris@0: #include "base/TempDirectory.h"
Chris@0: #include "base/PropertyContainer.h"
Chris@0: #include "base/Preferences.h"
Chris@0: #include "widgets/TipDialog.h"
Chris@259: #include "widgets/InteractiveFileFinder.h"
Chris@70: #include "transform/TransformFactory.h"
Chris@520: #include "svcore/plugin/PluginScan.h"
Chris@0: 
Chris@0: #include <QMetaType>
Chris@0: #include <QApplication>
Chris@572: #include <QScreen>
Chris@0: #include <QMessageBox>
Chris@0: #include <QTranslator>
Chris@0: #include <QLocale>
Chris@0: #include <QSettings>
Chris@0: #include <QIcon>
Chris@0: #include <QSessionManager>
Chris@0: #include <QSplashScreen>
Chris@0: #include <QFileOpenEvent>
Chris@0: #include <QDir>
Chris@0: 
Chris@0: #include <iostream>
Chris@0: #include <signal.h>
Chris@458: #include <cstdlib>
Chris@0: 
Chris@587: #include "../version.h"
Chris@587: 
Chris@443: #include <vamp-hostsdk/PluginHostAdapter.h>
Chris@443: 
Chris@0: static QMutex cleanupMutex;
Chris@70: static bool cleanedUp = false;
Chris@0: 
Chris@0: static void
Chris@0: signalHandler(int /* signal */)
Chris@0: {
Chris@0:     // Avoid this happening more than once across threads
Chris@0: 
Chris@70:     cerr << "signalHandler: cleaning up and exiting" << endl;
Chris@609: 
Chris@609:     if (cleanupMutex.tryLock(5000)) {
Chris@609:         if (!cleanedUp) {
Chris@609:             TempDirectory::getInstance()->cleanup();
Chris@609:             cleanedUp = true;
Chris@609:         }
Chris@609:         cleanupMutex.unlock();
Chris@70:     }
Chris@609:     
Chris@70:     exit(0);
Chris@0: }
Chris@0: 
Chris@0: class TonyApplication : public QApplication
Chris@0: {
Chris@0: public:
Chris@361:     TonyApplication(int &argc, char **argv) :
Chris@0:         QApplication(argc, argv),
Chris@0:         m_mainWindow(0),
Chris@0:         m_readyForFiles(false)
Chris@0:     {
Chris@500:         // tidier without, I reckon
Chris@500:         setAttribute(Qt::AA_DontShowIconsInMenus);
Chris@0:     }
Chris@0:     virtual ~TonyApplication() {
Chris@0:     }
Chris@0: 
Chris@0:     void setMainWindow(MainWindow *mw) { m_mainWindow = mw; }
Chris@0:     void releaseMainWindow() { m_mainWindow = 0; }
Chris@0: 
Chris@0:     virtual void commitData(QSessionManager &manager) {
Chris@0:         if (!m_mainWindow) return;
Chris@0:         bool mayAskUser = manager.allowsInteraction();
Chris@0:         bool success = m_mainWindow->commitData(mayAskUser);
Chris@0:         manager.release();
Chris@0:         if (!success) manager.cancel();
Chris@0:     }
Chris@0: 
Chris@0:     void readyForFiles() {
Chris@0:         m_readyForFiles = true;
Chris@0:     }
Chris@0: 
Chris@0:     void handleFilepathArgument(QString path, QSplashScreen *splash);
Chris@0: 
Chris@0:     void handleQueuedPaths(QSplashScreen *splash) {
Chris@0:         foreach (QString f, m_filepathQueue) {
Chris@0:             handleFilepathArgument(f, splash);
Chris@0:         }
Chris@0:     }
Chris@0:     
Chris@0: protected:
Chris@0:     MainWindow *m_mainWindow;
Chris@0:     
Chris@0:     bool m_readyForFiles;
Chris@0:     QStringList m_filepathQueue;
Chris@0: 
Chris@0:     virtual bool event(QEvent *event) {
Chris@355: 
Chris@399:         if (event->type() == QEvent::FileOpen) {
Chris@0:             QString path = static_cast<QFileOpenEvent *>(event)->file();
Chris@0:             if (m_readyForFiles) {
Chris@0:                 handleFilepathArgument(path, NULL);
Chris@0:             } else {
Chris@0:                 m_filepathQueue.append(path);
Chris@0:             }
Chris@0:             return true;
Chris@399:         } else {
Chris@0:             return QApplication::event(event);
Chris@0:         }
Chris@0:     }
Chris@0: };
Chris@443: 
Chris@541: static QString
Chris@541: getEnvQStr(QString variable)
Chris@541: {
Chris@541: #ifdef Q_OS_WIN32
Chris@541:     std::wstring wvar = variable.toStdWString();
Chris@541:     wchar_t *value = _wgetenv(wvar.c_str());
Chris@541:     if (!value) return QString();
Chris@548:     else return QString::fromStdWString(std::wstring(value));
Chris@541: #else
Chris@541:     std::string var = variable.toStdString();
Chris@541:     return QString::fromUtf8(qgetenv(var.c_str()));
Chris@541: #endif
Chris@541: }
Chris@541: 
Chris@541: static void
Chris@541: putEnvQStr(QString assignment)
Chris@541: {
Chris@541: #ifdef Q_OS_WIN32
Chris@541:     std::wstring wassignment = assignment.toStdWString();
Chris@548:     _wputenv(_wcsdup(wassignment.c_str()));
Chris@541: #else
Chris@541:     putenv(strdup(assignment.toUtf8().data()));
Chris@541: #endif
Chris@541: }
Chris@541: 
Chris@443: static void
Chris@443: setupTonyVampPath()
Chris@443: {
Chris@594:     QString myVampPath = getEnvQStr("TONY_VAMP_PATH");
Chris@541: 
Chris@443: #ifdef Q_OS_WIN32
Chris@443:     QChar sep(';');
Chris@443: #else
Chris@443:     QChar sep(':');
Chris@443: #endif
Chris@541:     
Chris@594:     if (myVampPath == "") {
Chris@592:         
Chris@592:         QString appName = QApplication::applicationName();
Chris@592:         QString myDir = QApplication::applicationDirPath();
Chris@592:         QString binaryName = QFileInfo(QCoreApplication::arguments().at(0))
Chris@592:             .fileName();
Chris@541: 
Chris@541: #ifdef Q_OS_WIN32
Chris@541:         QString programFiles = getEnvQStr("ProgramFiles");
Chris@541:         if (programFiles == "") programFiles = "C:\\Program Files";
Chris@592:         QString pfPath(programFiles + "\\" + appName);
Chris@592:         myVampPath = myDir + sep + pfPath;
Chris@548: #else
cannam@567: #ifdef Q_OS_MAC
Chris@592:         myVampPath = myDir + "/../Resources";
Chris@621:         (void)sep; // unused
cannam@567: #else
Chris@592:         if (binaryName != "") {
Chris@592:             myVampPath =
Chris@592:                 myDir + "/../lib/" + binaryName + sep;
Chris@592:         }
Chris@592:         myVampPath = myVampPath +
Chris@592:             myDir + "/../lib/" + appName + sep +
Chris@592:             myDir;
Chris@541: #endif
Chris@548: #endif
Chris@443:     }
Chris@443: 
Chris@594:     SVCERR << "Setting VAMP_PATH to " << myVampPath
Chris@541:            << " for Tony plugins" << endl;
Chris@458: 
Chris@594:     QString env = "VAMP_PATH=" + myVampPath;
Chris@458: 
Chris@458:     // Windows lacks setenv, must use putenv (different arg convention)
Chris@541:     putEnvQStr(env);
Chris@443: }
Chris@0:         
Chris@0: int
Chris@0: main(int argc, char **argv)
Chris@0: {
Chris@587:     if (argc == 2 && (QString(argv[1]) == "--version" ||
Chris@587:                       QString(argv[1]) == "-v")) {
Chris@587:         cerr << TONY_VERSION << endl;
Chris@587:         exit(0);
Chris@587:     }
Chris@587: 
Chris@0:     svSystemSpecificInitialisation();
Chris@98: 
Chris@541:     TonyApplication application(argc, argv);
Chris@541: 
Chris@579:     QApplication::setOrganizationName("sonic-visualiser");
Chris@579:     QApplication::setOrganizationDomain("sonicvisualiser.org");
Chris@579:     QApplication::setApplicationName("Tony");
Chris@579: 
Chris@443:     setupTonyVampPath();
Chris@443: 
Chris@196:     QStringList args = application.arguments();
Chris@0: 
Chris@0:     signal(SIGINT,  signalHandler);
Chris@0:     signal(SIGTERM, signalHandler);
Chris@0: 
chris@52: #ifndef Q_OS_WIN32
Chris@0:     signal(SIGHUP,  signalHandler);
Chris@0:     signal(SIGQUIT, signalHandler);
Chris@0: #endif
Chris@0: 
Chris@0:     bool audioOutput = true;
matthiasm@364:     bool sonification = true;
matthiasm@364:     bool spectrogram = true;
Chris@0: 
Chris@0:     if (args.contains("--help") || args.contains("-h") || args.contains("-?")) {
Chris@0:         std::cerr << QApplication::tr(
matthiasm@366:             "\nTony is a program for interactive note and pitch analysis and annotation.\n\nUsage:\n\n  %1 [--no-audio] [--no-sonification] [--no-spectrogram] [<file> ...]\n\n  --no-audio: Do not attempt to open an audio output device\n  --no-sonification: Disable sonification of pitch tracks and notes and hide their toggles.\n  --no-spectrogram: Disable spectrogram.\n  <file>: One or more Tony (.ton) and audio files may be provided.").arg(argv[0]).toStdString() << std::endl;
Chris@0:         exit(2);
Chris@0:     }
Chris@0: 
Chris@0:     if (args.contains("--no-audio")) audioOutput = false;
Chris@0: 
matthiasm@364:     if (args.contains("--no-sonification")) sonification = false;
matthiasm@364: 
matthiasm@364:     if (args.contains("--no-spectrogram")) spectrogram = false;
matthiasm@364: 
Chris@672:     if (args.contains("--first-run")) {
Chris@672:         QSettings settings;
Chris@672:         settings.clear();
Chris@672:     }
Chris@672: 
Chris@259:     InteractiveFileFinder::getInstance()->setApplicationSessionExtension("ton");
Chris@259: 
Chris@484:     QApplication::setAttribute(Qt::AA_UseHighDpiPixmaps);
Chris@484: 
Chris@0:     QSplashScreen *splash = 0;
Chris@0:     // If we had a splash screen, we would show it here
Chris@0: 
Chris@0:     QIcon icon;
Chris@0:     int sizes[] = { 16, 22, 24, 32, 48, 64, 128 };
Chris@0:     for (size_t i = 0; i < sizeof(sizes)/sizeof(sizes[0]); ++i) {
Chris@0:         icon.addFile(QString(":icons/tony-%1x%2.png").arg(sizes[i]).arg(sizes[i]));
Chris@0:     }
Chris@0:     QApplication::setWindowIcon(icon);
Chris@0: 
Chris@0:     QString language = QLocale::system().name();
Chris@0: 
Chris@0:     QTranslator qtTranslator;
Chris@0:     QString qtTrName = QString("qt_%1").arg(language);
Chris@0:     std::cerr << "Loading " << qtTrName.toStdString() << "..." << std::endl;
Chris@0:     bool success = false;
Chris@0:     if (!(success = qtTranslator.load(qtTrName))) {
Chris@0:         QString qtDir = getenv("QTDIR");
Chris@0:         if (qtDir != "") {
Chris@0:             success = qtTranslator.load
Chris@0:                 (qtTrName, QDir(qtDir).filePath("translations"));
Chris@0:         }
Chris@0:     }
Chris@0:     if (!success) {
Chris@0:         std::cerr << "Failed to load Qt translation for locale" << std::endl;
Chris@0:     }
Chris@0:     application.installTranslator(&qtTranslator);
Chris@0: 
Chris@0:     StoreStartupLocale();
Chris@520:     
Chris@0:     // Permit size_t and PropertyName to be used as args in queued signal calls
Chris@0:     qRegisterMetaType<size_t>("size_t");
Chris@0:     qRegisterMetaType<PropertyContainer::PropertyName>("PropertyContainer::PropertyName");
Chris@0: 
Chris@635:     MainWindow::AudioMode audioMode = 
Chris@635:         MainWindow::AUDIO_PLAYBACK_NOW_RECORD_LATER;
Chris@635: 
Chris@635:     if (!audioOutput) {
Chris@635:         audioMode = MainWindow::AUDIO_NONE;
Chris@635:     } 
Chris@474:     
Chris@635:     MainWindow *gui = new MainWindow(audioMode, sonification, spectrogram);
Chris@0:     application.setMainWindow(gui);
Chris@0:     if (splash) {
Chris@0:         QObject::connect(gui, SIGNAL(hideSplash()), splash, SLOT(hide()));
Chris@0:     }
Chris@0: 
Chris@572:     QScreen *screen = QApplication::primaryScreen();
Chris@572:     QRect available = screen->availableGeometry();
Chris@0: 
Chris@0:     int width = (available.width() * 2) / 3;
Chris@0:     int height = available.height() / 2;
Chris@0:     if (height < 450) height = (available.height() * 2) / 3;
Chris@0:     if (width > height * 2) width = height * 2;
Chris@0: 
Chris@0:     QSettings settings;
Chris@0:     settings.beginGroup("MainWindow");
Chris@0:     QSize size = settings.value("size", QSize(width, height)).toSize();
Chris@0:     gui->resizeConstrained(size);
Chris@0:     if (settings.contains("position")) {
Chris@0:         QRect prevrect(settings.value("position").toPoint(), size);
Chris@0:         if (!(available & prevrect).isEmpty()) {
Chris@0:             gui->move(prevrect.topLeft());
Chris@0:         }
Chris@0:     }
Chris@0:     settings.endGroup();
Chris@0:     
Chris@0:     gui->show();
Chris@0: 
Chris@0:     application.readyForFiles();
Chris@0:     
Chris@0:     for (QStringList::iterator i = args.begin(); i != args.end(); ++i) {
Chris@0: 
Chris@0:         if (i == args.begin()) continue;
Chris@0:         if (i->startsWith('-')) continue;
Chris@0: 
Chris@0:         QString path = *i;
Chris@0: 
Chris@0:         application.handleFilepathArgument(path, splash);
Chris@0:     }
Chris@0: 
Chris@0:     application.handleQueuedPaths(splash);
Chris@0:         
Chris@0:     if (splash) splash->finish(gui);
Chris@0:     delete splash;
Chris@0: 
Chris@0:     int rv = application.exec();
Chris@0: 
Chris@0:     gui->hide();
Chris@0: 
Chris@0:     cleanupMutex.lock();
Chris@70: 
Chris@70:     if (!cleanedUp) {
Chris@70:         TransformFactory::deleteInstance();
Chris@70:         TempDirectory::getInstance()->cleanup();
Chris@70:         cleanedUp = true;
Chris@70:     }
Chris@70: 
Chris@0:     application.releaseMainWindow();
Chris@0: 
Chris@0:     delete gui;
Chris@0: 
Chris@70:     cleanupMutex.unlock();
Chris@70: 
Chris@0:     return rv;
Chris@0: }
Chris@0: 
Chris@0: /** Application-global handler for filepaths passed in, e.g. as
Chris@0:  * command-line arguments or apple events */
Chris@0: 
Chris@0: void TonyApplication::handleFilepathArgument(QString path,
Chris@0:                                              QSplashScreen *splash)
Chris@0: {
Chris@0:     static bool haveSession = false;
Chris@0:     static bool haveMainModel = false;
Chris@0:     static bool havePriorCommandLineModel = false;
Chris@0: 
Chris@103:     if (!m_mainWindow) return;
Chris@103: 
Chris@0:     MainWindow::FileOpenStatus status = MainWindow::FileOpenFailed;
Chris@0: 
Chris@196: #ifdef Q_OS_WIN32
Chris@196:     path.replace("\\", "/");
Chris@196: #endif
Chris@196: 
Chris@0:     if (path.endsWith("ton")) {
Chris@0:         if (!haveSession) {
Chris@362:             status = m_mainWindow->openSessionPath(path);
Chris@0:             if (status == MainWindow::FileOpenSucceeded) {
Chris@0:                 haveSession = true;
Chris@0:                 haveMainModel = true;
Chris@0:             }
Chris@0:         } else {
Chris@0:             std::cerr << "WARNING: Ignoring additional session file argument \"" << path << "\"" << std::endl;
Chris@0:             status = MainWindow::FileOpenSucceeded;
Chris@0:         }
Chris@0:     }
Chris@0:     if (status != MainWindow::FileOpenSucceeded) {
Chris@0:         if (!haveMainModel) {
Chris@387:             status = m_mainWindow->openPath(path, MainWindow::ReplaceSession);
Chris@0:             if (status == MainWindow::FileOpenSucceeded) {
Chris@0:                 haveMainModel = true;
Chris@0:             }
Chris@0:         } else {
Chris@0:             if (haveSession && !havePriorCommandLineModel) {
Chris@387:                 status = m_mainWindow->openPath(path, MainWindow::AskUser);
Chris@0:                 if (status == MainWindow::FileOpenSucceeded) {
Chris@0:                     havePriorCommandLineModel = true;
Chris@0:                 }
Chris@0:             } else {
Chris@387:                 status = m_mainWindow->openPath(path, MainWindow::CreateAdditionalModel);
Chris@0:             }
Chris@0:         }
Chris@0:     }
Chris@0:     if (status == MainWindow::FileOpenFailed) {
Chris@0:         if (splash) splash->hide();
Chris@0:         QMessageBox::critical
Chris@0:             (m_mainWindow, QMessageBox::tr("Failed to open file"),
Chris@0:              QMessageBox::tr("File or URL \"%1\" could not be opened").arg(path));
Chris@0:     } else if (status == MainWindow::FileOpenWrongMode) {
Chris@0:         if (splash) splash->hide();
Chris@0:         QMessageBox::critical
Chris@0:             (m_mainWindow, QMessageBox::tr("Failed to open file"),
Chris@0:              QMessageBox::tr("<b>Audio required</b><p>Please load at least one audio file before importing annotation data"));
Chris@0:     }
Chris@0: }
Chris@0: