DisplayBuffer: overload atom_containing with a version taking an iterator to search start

1 files changed, 8 insertions, 2 deletions

diff --git a/src/display_buffer.cc b/src/display_buffer.cc
index 64b2099e..a263d8f5 100644
--- a/src/display_buffer.cc
+++ b/src/display_buffer.cc
@@ -88,10 +88,16 @@ DisplayBuffer::iterator DisplayBuffer::insert(iterator where, const DisplayAtom&
 
 DisplayBuffer::iterator DisplayBuffer::atom_containing(const BufferIterator& where)
 {
-    for (iterator it = m_atoms.begin(); it != m_atoms.end(); ++it)
+    return atom_containing(where, m_atoms.begin());
+}
+
+DisplayBuffer::iterator DisplayBuffer::atom_containing(const BufferIterator& where,
+                                                       iterator start)
+{
+    for (iterator it = start; it != m_atoms.end(); ++it)
     {
         if (it->end() > where)
-            return it;
+            return it->begin() <= where ? it : end();
     }
     return end();
 }