Garmelon/profold
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix folding

Browse source
This commit is contained in:
Joscha 2020-03-01 00:22:25 +00:00
parent 64aa07e04c
commit dc9353a374
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/Profold/LineNode.hs
View file

@ -27,7 +27,10 @@ flatten ln
| lineFolded ln = V.singleton ([], ln)
| otherwise =
V.cons ([], ln) $
V.imap (\i (is, n) -> (i : is, n)) $ V.concatMap flatten $ lineChildren ln
V.concat $ V.toList $ V.imap flattenChild $ lineChildren ln
where
flattenChild :: Int -> LineNode -> V.Vector (Path, LineNode)
flattenChild i c = V.map (\(is, n) -> (i : is, n)) $ flatten c
modify :: (LineNode -> LineNode) -> Path -> LineNode -> LineNode
modify f [] ln = f ln