diff options
Diffstat (limited to 'src/GF/Data/Trie.hs')
| -rw-r--r-- | src/GF/Data/Trie.hs | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/GF/Data/Trie.hs b/src/GF/Data/Trie.hs index 0a0a4472b..724df0761 100644 --- a/src/GF/Data/Trie.hs +++ b/src/GF/Data/Trie.hs @@ -10,6 +10,7 @@ module Trie ( tcompile, + collapse, Trie, trieLookup, decompose, @@ -35,6 +36,14 @@ optimize :: TrieT -> Trie optimize (TrieT (xs,res)) = Trie ([(c,optimize t) | (c,t) <- xs] |->+ empty, res) +collapse :: Trie -> [(String,[(Attr,String)])] +collapse trie = collapse' trie [] + where collapse' (Trie (map,(x:xs))) s = if (isEmpty map) then [(reverse s,(x:xs))] + else (reverse s,(x:xs)): + concat [ collapse' trie (c:s) | (c,trie) <- flatten map] + collapse' (Trie (map,[])) s + = concat [ collapse' trie (c:s) | (c,trie) <- flatten map] + tcompile :: [(String,[(Attr,String)])] -> Trie tcompile xs = optimize $ build xs emptyTrie |