comparison src/MainWindow.cpp @ 522:8f016d8c021a saveall

Add Save All function, as proposed by Yi Ting Tan

     action->setStatusTip(tr("Export note data to a CSV, RDF, layer XML, or MIDI file"));
     connect(action, SIGNAL(triggered()), this, SLOT(exportNoteLayer()));
     connect(this, SIGNAL(canExportNotes(bool)), action, SLOT(setEnabled(bool)));
     menu->addAction(action);
 
+    action = new QAction(tr("Save Session and All Layers"), this);
+    action->setStatusTip(tr("Save the current session, pitch and note layers with the same filename as the audio but different extensions."));
+    connect(action, SIGNAL(triggered()), this, SLOT(saveAll()));
+    connect(this, SIGNAL(canSaveAll(bool)), action, SLOT(setEnabled(bool)));
+    menu->addAction(action);
+
     menu->addSeparator();
     
     action = new QAction(tr("Browse Recorded Audio"), this);
     action->setStatusTip(tr("Open the Recorded Audio folder in the system file browser"));
     connect(action, SIGNAL(triggered()), this, SLOT(browseRecordedAudio()));

         m_analyser->getLayer(Analyser::PitchTrack);
 
     bool haveNotes = 
         m_analyser->isVisible(Analyser::Notes) &&
         m_analyser->getLayer(Analyser::Notes);
+
+    emit canSaveAll(haveMainModel && havePitchTrack && haveNotes);
 
     emit canExportPitchTrack(havePitchTrack);
     emit canExportNotes(haveNotes);
     emit canSnapNotes(haveSelection && haveNotes);
 

 }
 
 void
 MainWindow::saveSessionInAudioPath()
 {
-    if (m_audioFile == "") return;
-
-    if (!waitForInitialAnalysis()) return;
+    (void)trySaveSessionInAudioPath();
+}
+
+bool
+MainWindow::trySaveSessionInAudioPath()
+{
+    if (m_audioFile == "") return false;
+
+    if (!waitForInitialAnalysis()) return false;
 
     // We do not want to save mid-analysis regions -- that would cause
     // confusion on reloading
     m_analyser->clearReAnalysis();
     clearSelection();

     if (shouldVerify && QFileInfo(path).exists()) {
         if (QMessageBox::question(0, tr("File exists"),
                                   tr("<b>File exists</b><p>The file \"%1\" already exists.\nDo you want to overwrite it?").arg(path),
                                   QMessageBox::Ok,
                                   QMessageBox::Cancel) != QMessageBox::Ok) {
-            return;
+            return false;
         }
+    }
+
+    if (!saveSessionFile(path)) {
+        QMessageBox::critical(this, tr("Failed to save file"),
+                              tr("Session file \"%1\" could not be saved.").arg(path));
+        return false;
+    } else {
+        setWindowTitle(tr("%1: %2")
+                       .arg(QApplication::applicationName())
+                       .arg(QFileInfo(path).fileName()));
+        m_sessionFile = path;
+        CommandHistory::getInstance()->documentSaved();
+        documentRestored();
+        m_recentFiles.addFile(path);
+        return true;
+    }
+}
+
+void
+MainWindow::saveSessionAs()
+{
+    // We do not want to save mid-analysis regions -- that would cause
+    // confusion on reloading
+    m_analyser->clearReAnalysis();
+    clearSelection();
+
+    QString path = getSaveFileName(FileFinder::SessionFile);
+
+    if (path == "") {
+        return;
     }
 
     if (!waitForInitialAnalysis()) {
         QMessageBox::warning(this, tr("File not saved"),
                              tr("Wait cancelled: the session has not been saved."));
+        return;
     }
 
     if (!saveSessionFile(path)) {
         QMessageBox::critical(this, tr("Failed to save file"),
                               tr("Session file \"%1\" could not be saved.").arg(path));

         m_recentFiles.addFile(path);
     }
 }
 
 void
-MainWindow::saveSessionAs()
-{
-    // We do not want to save mid-analysis regions -- that would cause
-    // confusion on reloading
-    m_analyser->clearReAnalysis();
-    clearSelection();
-
-    QString path = getSaveFileName(FileFinder::SessionFile);
-
-    if (path == "") {
-        return;
-    }
-
-    if (!waitForInitialAnalysis()) {
-        QMessageBox::warning(this, tr("File not saved"),
-                             tr("Wait cancelled: the session has not been saved."));
-        return;
-    }
-
-    if (!saveSessionFile(path)) {
-        QMessageBox::critical(this, tr("Failed to save file"),
-                              tr("Session file \"%1\" could not be saved.").arg(path));
-    } else {
-        setWindowTitle(tr("%1: %2")
-                       .arg(QApplication::applicationName())
-                       .arg(QFileInfo(path).fileName()));
-        m_sessionFile = path;
-        CommandHistory::getInstance()->documentSaved();
-        documentRestored();
-        m_recentFiles.addFile(path);
-    }
+MainWindow::saveAll()
+{
+    if (!trySaveSessionInAudioPath()) return;
+    
+    QString filepath = QFileInfo(m_audioFile).absoluteDir().canonicalPath();
+    QString basename = QFileInfo(m_audioFile).completeBaseName();
+
+    QString pitchPath = QDir(filepath).filePath(basename + "_track.txt");
+    QString notesPath = QDir(filepath).filePath(basename + "_notes.txt");
+
+    exportPitchLayerTo(pitchPath);
+    exportNoteLayerTo(notesPath);
 }
 
 QString
 MainWindow::exportToSVL(QString path, Layer *layer)
 {

 
     QString path = getSaveFileName(type);
 
     if (path == "") return;
 
+    exportPitchLayerTo(path);
+}
+
+void
+MainWindow::exportPitchLayerTo(QString path)
+{
     if (!waitForInitialAnalysis()) return;
     
     if (QFileInfo(path).suffix() == "") path += ".svl";
 
     QString suffix = QFileInfo(path).suffix().toLower();
 
     QString error;
+
+    Layer *layer = m_analyser->getLayer(Analyser::PitchTrack);
+    if (!layer) return;
+
+    SparseTimeValueModel *model =
+        qobject_cast<SparseTimeValueModel *>(layer->getModel());
+    if (!model) return;
 
     if (suffix == "xml" || suffix == "svl") {
 
         error = exportToSVL(path, layer);
 

 
     QString path = getSaveFileName(type);
 
     if (path == "") return;
 
+    exportNoteLayerTo(path);
+}
+
+void
+MainWindow::exportNoteLayerTo(QString path)
+{
     if (QFileInfo(path).suffix() == "") path += ".svl";
 
     QString suffix = QFileInfo(path).suffix().toLower();
 
     QString error;
+
+    Layer *layer = m_analyser->getLayer(Analyser::Notes);
+    if (!layer) return;
+
+    FlexiNoteModel *model = qobject_cast<FlexiNoteModel *>(layer->getModel());
+    if (!model) return;
 
     if (suffix == "xml" || suffix == "svl") {
 
         error = exportToSVL(path, layer);