From df13cf0a90a923d793c2c06ab5468882096fbb76 Mon Sep 17 00:00:00 2001
From: Antonio Scandurra <me@as-cii.com>
Date: Tue, 1 Jun 2021 17:54:55 +0200
Subject: [PATCH] WIP: Compare anchors without using FragmentId

---
 zed/src/editor/buffer.rs        | 34 +++++++++++++++++++++++++++++++++
 zed/src/editor/buffer/anchor.rs | 24 ++++++++++++++---------
 2 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/zed/src/editor/buffer.rs b/zed/src/editor/buffer.rs
index 8d4ada41c00a521ac5839abe9489f831acdf514b..c0a1df49e72c112d84f581e9b950a536f9602ff8 100644
--- a/zed/src/editor/buffer.rs
+++ b/zed/src/editor/buffer.rs
@@ -353,6 +353,7 @@ pub struct FragmentSummary {
     max_version: time::Global,
     min_insertion_version: time::Global,
     max_insertion_version: time::Global,
+    count: usize,
 }
 
 #[derive(Default, Clone, Debug, PartialEq, Eq)]
@@ -2029,6 +2030,26 @@ impl Buffer {
         }
     }
 
+    fn fragment_ix_for_anchor(&self, anchor: &Anchor) -> usize {
+        match anchor {
+            Anchor::Start => 0,
+            Anchor::End => self.fragments.extent::<FragmentCount>(&None).0,
+            Anchor::Middle {
+                offset,
+                bias,
+                version,
+            } => {
+                let mut cursor = self.fragments.cursor::<VersionedOffset, FragmentCount>();
+                cursor.seek(
+                    &VersionedOffset::Offset(*offset),
+                    bias.to_seek_bias(),
+                    &Some(version.clone()),
+                );
+                cursor.start().0
+            }
+        }
+    }
+
     pub fn point_for_offset(&self, offset: usize) -> Result<Point> {
         if offset <= self.len() {
             Ok(self.text_summary_for_range(0..offset).lines)
@@ -2503,6 +2524,7 @@ impl sum_tree::Item for Fragment {
                 max_version,
                 min_insertion_version,
                 max_insertion_version,
+                count: 1,
             }
         } else {
             FragmentSummary {
@@ -2514,6 +2536,7 @@ impl sum_tree::Item for Fragment {
                 max_version,
                 min_insertion_version,
                 max_insertion_version,
+                count: 1,
             }
         }
     }
@@ -2532,6 +2555,7 @@ impl sum_tree::Summary for FragmentSummary {
             .meet(&other.min_insertion_version);
         self.max_insertion_version
             .join(&other.max_insertion_version);
+        self.count += other.count;
     }
 }
 
@@ -2543,6 +2567,7 @@ impl Default for FragmentSummary {
             max_version: time::Global::new(),
             min_insertion_version: time::Global::new(),
             max_insertion_version: time::Global::new(),
+            count: 0,
         }
     }
 }
@@ -2641,6 +2666,15 @@ impl<'a> sum_tree::Dimension<'a, FragmentSummary> for (VersionedOffset, usize) {
     }
 }
 
+#[derive(Copy, Clone, Default, Debug, Eq, PartialEq, Ord, PartialOrd)]
+struct FragmentCount(usize);
+
+impl<'a> sum_tree::Dimension<'a, FragmentSummary> for FragmentCount {
+    fn add_summary(&mut self, summary: &'a FragmentSummary, _: &Option<time::Global>) {
+        self.0 += summary.count;
+    }
+}
+
 impl Operation {
     fn replica_id(&self) -> ReplicaId {
         self.lamport_timestamp().replica_id
diff --git a/zed/src/editor/buffer/anchor.rs b/zed/src/editor/buffer/anchor.rs
index eeacaf112464164f61f107f8067b9ac8330f6cb2..55aee9a82167e28047039e742b22bb69530396fb 100644
--- a/zed/src/editor/buffer/anchor.rs
+++ b/zed/src/editor/buffer/anchor.rs
@@ -63,21 +63,27 @@ impl Anchor {
                 Anchor::Middle {
                     offset: self_offset,
                     bias: self_bias,
-                    version: self_version,
+                    ..
                 },
                 Anchor::Middle {
                     offset: other_offset,
                     bias: other_bias,
-                    version: other_version,
+                    ..
                 },
             ) => {
-                let offset_comparison = if self_version == other_version {
-                    self_offset.cmp(other_offset)
-                } else {
-                    self.to_offset(buffer).cmp(&other.to_offset(buffer))
-                };
-
-                offset_comparison.then_with(|| self_bias.cmp(other_bias))
+                dbg!(
+                    self,
+                    other,
+                    self_offset,
+                    other_offset,
+                    buffer.fragment_ix_for_anchor(self),
+                    buffer.fragment_ix_for_anchor(other)
+                );
+                buffer
+                    .fragment_ix_for_anchor(self)
+                    .cmp(&buffer.fragment_ix_for_anchor(other))
+                    .then_with(|| self_offset.cmp(&other_offset))
+                    .then_with(|| self_bias.cmp(other_bias))
             }
         })
     }