summaryrefslogtreecommitdiff
path: root/src/Erebos
diff options
context:
space:
mode:
authorRoman Smrž <roman.smrz@seznam.cz>2024-08-10 19:46:01 +0200
committerRoman Smrž <roman.smrz@seznam.cz>2024-08-10 20:40:12 +0200
commit5c67b5800a29b10b4a27fdf467cd5a2ecd4fe40a (patch)
tree635dfb859ccde7377bbb9582953b75163c72ff71 /src/Erebos
parent8743ed1b3713a764ccb191e83cb006a7c4ac60d4 (diff)
Include passed objects in generations and ancestors functions
Diffstat (limited to 'src/Erebos')
-rw-r--r--src/Erebos/Storage/Merge.hs7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/Erebos/Storage/Merge.hs b/src/Erebos/Storage/Merge.hs
index 9d9db13..a3b0fd7 100644
--- a/src/Erebos/Storage/Merge.hs
+++ b/src/Erebos/Storage/Merge.hs
@@ -97,13 +97,16 @@ storedGeneration x =
doLookup x
+-- |Returns list of sets starting with the set of given objects and
+-- intcrementally adding parents.
generations :: Storable a => [Stored a] -> [Set (Stored a)]
generations = unfoldr gen . (,S.empty)
- where gen (hs, cur) = case filter (`S.notMember` cur) $ previous =<< hs of
+ where gen (hs, cur) = case filter (`S.notMember` cur) hs of
[] -> Nothing
added -> let next = foldr S.insert cur added
- in Just (next, (added, next))
+ in Just (next, (previous =<< added, next))
+-- |Returns set containing all given objects and their ancestors
ancestors :: Storable a => [Stored a] -> Set (Stored a)
ancestors = last . (S.empty:) . generations