Add notion of UI modifications to a capture, saved in .rdc sections

qrenderdoc/Windows/MainWindow.cpp

@@ -220,7 +220,8 @@ MainWindow::MainWindow(ICaptureContext &ctx) : QMainWindow(NULL), ui(new Ui::Mai
 
   m_Ctx.AddCaptureViewer(this);
 
-  ui->action_Save_Capture->setEnabled(false);
+  ui->action_Save_Capture_Inplace->setEnabled(false);
+  ui->action_Save_Capture_As->setEnabled(false);
   ui->action_Close_Capture->setEnabled(false);
 
   QList<QAction *> actions = ui->menuBar->actions();
@@ -282,6 +283,14 @@ void MainWindow::on_action_Open_Capture_triggered()
     LoadFromFilename(filename, false);
 }
 
+void MainWindow::captureModified()
+{
+  // once the capture is modified, enable the save-in-place option. It might already have been
+  // enabled if this capture was a temporary one
+  if(m_Ctx.IsCaptureLoaded())
+    ui->action_Save_Capture_Inplace->setEnabled(true);
+}
+
 void MainWindow::LoadFromFilename(const QString &filename, bool temporary)
 {
   QFileInfo path(filename);
@@ -652,6 +661,53 @@ bool MainWindow::PromptCloseCapture()
       deletepath = temppath;
     m_OwnTempCapture = false;
   }
+  else if(m_Ctx.GetCaptureModifications() != CaptureModifications::NoModifications)
+  {
+    QMessageBox::StandardButton res = QMessageBox::No;
+
+    QString text = tr("This capture has the following modifications:\n\n");
+
+    CaptureModifications mods = m_Ctx.GetCaptureModifications();
+
+    if(mods & CaptureModifications::Renames)
+      text += tr("Resources have been renamed.\n");
+    if(mods & CaptureModifications::Bookmarks)
+      text += tr("Bookmarks have been changed.\n");
+    if(mods & CaptureModifications::Notes)
+      text += tr("Capture notes have been changed.\n");
+
+    bool saveas = false;
+
+    if(m_Ctx.IsCaptureLocal())
+    {
+      text += tr("\nWould you like to save those changes to '%1'?").arg(m_Ctx.GetCaptureFilename());
+    }
+    else
+    {
+      saveas = true;
+      text +=
+          tr("\nThe capture is on a remote host, would you like to save these changes locally?");
+    }
+
+    res = RDDialog::question(NULL, tr("Save changes to capture?"), text,
+                             QMessageBox::Yes | QMessageBox::No | QMessageBox::Cancel);
+
+    if(res == QMessageBox::Cancel)
+      return false;
+
+    if(res == QMessageBox::Yes)
+    {
+      bool success = false;
+
+      if(saveas)
+        success = PromptSaveCaptureAs();
+      else
+        success = m_Ctx.SaveCaptureTo(m_Ctx.GetCaptureFilename());
+
+      if(!success)
+        return false;
+    }
+  }
 
   CloseCapture();
 
@@ -665,7 +721,8 @@ void MainWindow::CloseCapture()
 {
   m_Ctx.CloseCapture();
 
-  ui->action_Save_Capture->setEnabled(false);
+  ui->action_Save_Capture_Inplace->setEnabled(false);
+  ui->action_Save_Capture_As->setEnabled(false);
 }
 
 void MainWindow::SetTitle(const QString &filename)
@@ -1226,7 +1283,11 @@ void MainWindow::statusDoubleClicked(QMouseEvent *event)
 
 void MainWindow::OnCaptureLoaded()
 {
-  ui->action_Save_Capture->setEnabled(true);
+  // at first only allow the default save for temporary captures. It should be disabled if we have
+  // loaded a 'permanent' capture from disk and haven't made any changes. It will be enabled as soon
+  // as any changes are made.
+  ui->action_Save_Capture_Inplace->setEnabled(m_Ctx.IsCaptureTemporary());
+  ui->action_Save_Capture_As->setEnabled(true);
   ui->action_Close_Capture->setEnabled(true);
 
   // don't allow changing context while capture is open
@@ -1247,8 +1308,6 @@ void MainWindow::OnCaptureLoaded()
       ui->action_Resolve_Symbols->setEnabled(hasResolver);
       ui->action_Resolve_Symbols->setText(hasResolver ? tr("Resolve Symbols")
                                                       : tr("Resolve Symbols - None in capture"));
-
-      ui->action_Save_Capture->setEnabled(true);
     });
   });
 
@@ -1261,7 +1320,8 @@ void MainWindow::OnCaptureLoaded()
 
 void MainWindow::OnCaptureClosed()
 {
-  ui->action_Save_Capture->setEnabled(false);
+  ui->action_Save_Capture_Inplace->setEnabled(false);
+  ui->action_Save_Capture_As->setEnabled(false);
   ui->action_Close_Capture->setEnabled(false);
 
   ui->action_Start_Replay_Loop->setEnabled(false);
@@ -1361,7 +1421,28 @@ void MainWindow::on_action_Close_Capture_triggered()
   PromptCloseCapture();
 }
 
-void MainWindow::on_action_Save_Capture_triggered()
+void MainWindow::on_action_Save_Capture_Inplace_triggered()
+{
+  bool saved = false;
+
+  if(m_Ctx.IsCaptureTemporary() || !m_Ctx.IsCaptureLocal())
+  {
+    saved = PromptSaveCaptureAs();
+  }
+  else
+  {
+    if(m_Ctx.GetCaptureModifications() != CaptureModifications::NoModifications &&
+       m_Ctx.IsCaptureLocal())
+    {
+      saved = m_Ctx.SaveCaptureTo(m_Ctx.GetCaptureFilename());
+    }
+  }
+
+  if(saved)
+    ui->action_Save_Capture_Inplace->setEnabled(false);
+}
+
+void MainWindow::on_action_Save_Capture_As_triggered()
 {
   PromptSaveCaptureAs();
 }

qrenderdoc/Windows/MainWindow.h

@@ -68,6 +68,7 @@ public:
 
   void setProgress(float val);
   void takeCaptureOwnership() { m_OwnTempCapture = true; }
+  void captureModified();
   void LoadFromFilename(const QString &filename, bool temporary);
   void LoadCapture(const QString &filename, bool temporary, bool local);
   void CloseCapture();
@@ -101,7 +102,8 @@ private slots:
   void on_action_Exit_triggered();
   void on_action_About_triggered();
   void on_action_Open_Capture_triggered();
-  void on_action_Save_Capture_triggered();
+  void on_action_Save_Capture_Inplace_triggered();
+  void on_action_Save_Capture_As_triggered();
   void on_action_Close_Capture_triggered();
   void on_action_Mesh_Output_triggered();
   void on_action_API_Inspector_triggered();

qrenderdoc/Windows/MainWindow.ui

@@ -79,7 +79,8 @@
     <addaction name="action_Inject_into_Process"/>
     <addaction name="separator"/>
     <addaction name="action_Open_Capture"/>
-    <addaction name="action_Save_Capture"/>
+    <addaction name="action_Save_Capture_Inplace"/>
+    <addaction name="action_Save_Capture_As"/>
     <addaction name="action_Close_Capture"/>
     <addaction name="separator"/>
     <addaction name="menu_Recent_Capture_Files"/>
@@ -181,7 +182,7 @@
     <string>Ctrl+O</string>
    </property>
   </action>
-  <action name="action_Save_Capture">
+  <action name="action_Save_Capture_Inplace">
    <property name="text">
     <string>&amp;Save Capture</string>
    </property>
@@ -409,6 +410,11 @@
     <string>&amp;Resource Inspector</string>
    </property>
   </action>
+  <action name="action_Save_Capture_As">
+   <property name="text">
+    <string>Sa&amp;ve Capture As</string>
+   </property>
+  </action>
  </widget>
  <layoutdefault spacing="6" margin="11"/>
  <customwidgets>

qrenderdoc/Windows/PythonShell.cpp

@@ -60,6 +60,10 @@ struct CaptureContextInvoker : ICaptureContext
   virtual bool IsCaptureTemporary() override { return m_Ctx.IsCaptureTemporary(); }
   virtual bool IsCaptureLoading() override { return m_Ctx.IsCaptureLoading(); }
   virtual QString GetCaptureFilename() override { return m_Ctx.GetCaptureFilename(); }
+  virtual CaptureModifications GetCaptureModifications() override
+  {
+    return m_Ctx.GetCaptureModifications();
+  }
   virtual const FrameDescription &FrameInfo() override { return m_Ctx.FrameInfo(); }
   virtual const APIProperties &APIProps() override { return m_Ctx.APIProps(); }
   virtual uint32_t CurSelectedEvent() override { return m_Ctx.CurSelectedEvent(); }