From 9cf7cb682c6e472e8906d16a5c8e0488b86ca324 Mon Sep 17 00:00:00 2001
From: Maxime Coste <frrrwww@gmail.com>
Date: Thu, 3 Nov 2011 13:44:02 +0000
Subject: [PATCH] Buffer: fix modified handling

---
 src/buffer.cc | 12 +++++++++---
 src/buffer.hh |  2 +-
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/buffer.cc b/src/buffer.cc
index d5ec4349a..f0a396b6e 100644
--- a/src/buffer.cc
+++ b/src/buffer.cc
@@ -22,7 +22,7 @@ Buffer::Buffer(const std::string& name, Type type,
                const BufferString& initial_content)
     : m_name(name), m_type(type),
       m_history(1), m_history_cursor(m_history.begin()),
-      m_content(initial_content), m_last_save_undo_group(m_history.begin())
+      m_content(initial_content), m_last_save_undo_index(0)
 {
     BufferManager::instance().register_buffer(this);
 
@@ -154,6 +154,10 @@ void Buffer::begin_undo_group()
 {
     assert(m_current_undo_group.empty());
     m_history.erase(m_history_cursor, m_history.end());
+
+    if (m_history.size() < m_last_save_undo_index)
+        m_last_save_undo_index = -1;
+
     m_history_cursor = m_history.end();
 }
 
@@ -245,13 +249,15 @@ void Buffer::delete_window(Window* window)
 
 bool Buffer::is_modified() const
 {
-    return m_last_save_undo_group != m_history_cursor
+    size_t history_cursor_index = m_history_cursor - m_history.begin();
+    return m_last_save_undo_index != history_cursor_index
            or not m_current_undo_group.empty();
 }
 
 void Buffer::notify_saved()
 {
-    m_last_save_undo_group = m_history_cursor;
+    size_t history_cursor_index = m_history_cursor - m_history.begin();
+    m_last_save_undo_index = history_cursor_index;
 }
 
 void Buffer::register_modification_listener(BufferModificationListener* listener) const
diff --git a/src/buffer.hh b/src/buffer.hh
index 31abd55e3..1d4455bcc 100644
--- a/src/buffer.hh
+++ b/src/buffer.hh
@@ -184,7 +184,7 @@ private:
 
     std::list<std::unique_ptr<Window>> m_windows;
 
-    std::vector<UndoGroup>::iterator m_last_save_undo_group;
+    size_t m_last_save_undo_index;
 
     mutable std::vector<BufferModificationListener*> m_modification_listeners;
 };