TransactionScope for performance also when abandoning redo states

src/HistoryWindow.cpp

@@ -269,7 +269,7 @@ void HistoryDialog::OnDiscard(wxCommandEvent & WXUNUSED(event))
    int i = mLevels->GetValue();
 
    mSelected -= i;
-   mManager->RemoveStates(i);
+   mManager->RemoveStates(0, i);
    ProjectHistory::Get( *mProject ).SetStateTo(mSelected);
 
    while(--i >= 0)

src/ProjectManager.cpp

@@ -899,7 +899,7 @@ AudacityProject *ProjectManager::OpenProject(
       window.Zoom( window.GetZoomOfToFit() );
       // "Project was recovered" replaces "Create new project" in Undo History.
       auto &undoManager = UndoManager::Get( *pProject );
-      undoManager.RemoveStates(1);
+      undoManager.RemoveStates(0, 1);
    }
 
    return pProject;

src/UndoManager.cpp

@@ -180,18 +180,20 @@ void UndoManager::RemoveStateAt(int n)
 }
 
 
-void UndoManager::RemoveStates(int num)
+void UndoManager::RemoveStates(size_t begin, size_t end)
 {
    Optional<TransactionScope> pTrans;
    auto pConnection = ConnectionPtr::Get(mProject).mpConnection.get();
    if (pConnection)
       pTrans.emplace(*pConnection, "DiscardingUndoStates");
 
-   for (int i = 0; i < num; i++) {
+   for (size_t ii = begin; ii < end; ++ii) {
-      RemoveStateAt(0);
+      RemoveStateAt(begin);
 
-      current -= 1;
+      if (current > begin)
-      saved -= 1;
+        --current;
+      if (saved > begin)
+        --saved;
    }
 
    if (pTrans)
@@ -200,7 +202,7 @@ void UndoManager::RemoveStates(int num)
 
 void UndoManager::ClearStates()
 {
-   RemoveStates(stack.size());
+   RemoveStates(0, stack.size());
    current = -1;
    saved = -1;
 }
@@ -264,8 +266,6 @@ void UndoManager::PushState(const TrackList * l,
                             const TranslatableString &shortDescription,
                             UndoPush flags)
 {
-   unsigned int i;
-
    if ( (flags & UndoPush::CONSOLIDATE) != UndoPush::NONE &&
        // compare full translations not msgids!
        lastAction.Translation() == longDescription.Translation() &&
@@ -289,10 +289,11 @@ void UndoManager::PushState(const TrackList * l,
 
    mayConsolidate = true;
 
-   i = current + 1;
+   // Abandon redo states
-   while (i < stack.size()) {
+   if (saved >= current) {
-      RemoveStateAt(i);
+      saved = -1;
    }
+   RemoveStates( current + 1, stack.size() );
 
    // Assume tags was duplicated before any changes.
    // Just save a NEW shared_ptr to it.
@@ -304,10 +305,6 @@ void UndoManager::PushState(const TrackList * l,
 
    current++;
 
-   if (saved >= current) {
-      saved = -1;
-   }
-
    lastAction = longDescription;
 
    // wxWidgets will own the event object

src/UndoManager.h

@@ -132,7 +132,10 @@ class AUDACITY_DLL_API UndoManager final
    void ModifyState(const TrackList * l,
                     const SelectedRegion &selectedRegion, const std::shared_ptr<Tags> &tags);
    void ClearStates();
-   void RemoveStates(int num);  // removes the 'num' oldest states
+   void RemoveStates(
+      size_t begin, //!< inclusive start of range
+      size_t end    //!< exclusive end of range
+   );
    unsigned int GetNumStates();
    unsigned int GetCurrentState();
 

src/menus/FileMenus.cpp

@@ -235,7 +235,7 @@ void OnCompact(const CommandContext &context)
 
       // Now we can remove all previous states.
       auto numStates = undoManager.GetNumStates();
-      undoManager.RemoveStates(numStates - 1);
+      undoManager.RemoveStates(0, numStates - 1);
 
       // And clear the clipboard
       clipboard.Clear();