persistent_range_query: first draft

libs/persistent_range_query/tests/rsq_test.rs

@@ -0,0 +1,56 @@
+use persistent_range_query::naive::*;
+use persistent_range_query::ops::rsq::*;
+use persistent_range_query::ops::SameElementsInitializer;
+use persistent_range_query::{PersistentVecStorage, VecVersion};
+use std::ops::{Mul, Range, Sub};
+
+#[derive(Copy, Clone, Debug)]
+struct K(u8);
+
+impl IndexableKey for K {
+    fn index(all_keys: &Range<Self>, key: &Self) -> usize {
+        (key.0 as usize) - (all_keys.start.0 as usize)
+    }
+
+    fn element_range(all_keys: &Range<Self>, index: usize) -> Range<Self> {
+        K(all_keys.start.0 + index as u8)..K(all_keys.start.0 + index as u8 + 1)
+    }
+}
+
+struct KDiff(i16);
+
+impl Sub<&K> for &K {
+    type Output = KDiff;
+
+    fn sub(self, rhs: &K) -> Self::Output {
+        KDiff((self.0 as i16) - (rhs.0 as i16))
+    }
+}
+
+impl<'a> Mul<KDiff> for &'a i32 {
+    type Output = i32;
+
+    fn mul(self, rhs: KDiff) -> Self::Output {
+        self * (rhs.0 as i32)
+    }
+}
+
+#[test]
+fn test_naive() {
+    let mut s: NaiveVecVersion<_, _, AddAssignModification<_>> =
+        NaiveVecStorage::new(K(0)..K(12), SameElementsInitializer::new(0i32));
+    assert_eq!(*s.get(K(0)..K(12)).sum(), 0);
+
+    s.modify(K(2)..K(5), AddAssignModification::Add(3));
+    assert_eq!(*s.get(K(0)..K(12)).sum(), 3 + 3 + 3);
+    let s_old = s.clone();
+
+    s.modify(K(3)..K(6), AddAssignModification::Assign(10));
+    assert_eq!(*s.get(K(0)..K(12)).sum(), 3 + 10 + 10 + 10);
+
+    s.modify(K(4)..K(7), AddAssignModification::Add(2));
+    assert_eq!(*s.get(K(0)..K(12)).sum(), 3 + 10 + 12 + 12 + 2);
+
+    assert_eq!(*s.get(K(4)..K(6)).sum(), 12 + 12);
+    assert_eq!(*s_old.get(K(4)..K(6)).sum(), 3);
+}