humza/sml-projects
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

extract more duplicated code into reusable functions (in src/string-set.sml)

Browse Source
This commit is contained in:
Humza Shahid
2024-09-09 18:17:07 +01:00
parent 272ccf1b03
commit 6228a259bc
1 changed files with 61 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

155
src/string-set.sml
View File

@@ -583,6 +583,21 @@ struct
| NO_INSERT_MATCH => trie) | NO_INSERT_MATCH => trie)
end end
and helpInsertChildren
(insKey, keyPos, keys, children, trie, parentConstructor) =
let
val findChr = String.sub (insKey, keyPos)
in
(case insertBinSearch (findChr, keyPos, keys) of
INSERT_NEW_CHILD insIdx =>
insertNewChild (keys, insIdx, insKey, children, parentConstructor)
| FOUND_INSERT_POS insIdx =>
foundInsertPos
(keys, children, keyPos, insKey, insIdx, trie, parentConstructor)
| APPEND_NEW_CHILD =>
appendNewChild (keys, insKey, children, parentConstructor))
end
and helpInsert (insKey, keyPos, trie) : t = and helpInsert (insKey, keyPos, trie) : t =
case trie of case trie of
FOUND => FOUND =>
@@ -594,39 +609,10 @@ struct
, children = Vector.fromList [FOUND] , children = Vector.fromList [FOUND]
} }
| CHILDREN {keys, children} => | CHILDREN {keys, children} =>
let helpInsertChildren (insKey, keyPos, keys, children, trie, CHILDREN)
val findChr = String.sub (insKey, keyPos)
in
(case insertBinSearch (findChr, keyPos, keys) of
INSERT_NEW_CHILD insIdx =>
insertNewChild (keys, insIdx, insKey, children, CHILDREN)
| FOUND_INSERT_POS insIdx =>
foundInsertPos
(keys, children, keyPos, insKey, insIdx, trie, CHILDREN)
| APPEND_NEW_CHILD =>
appendNewChild (keys, insKey, children, CHILDREN))
end
| FOUND_WITH_CHILDREN {keys, children} => | FOUND_WITH_CHILDREN {keys, children} =>
let helpInsertChildren
val findChr = String.sub (insKey, keyPos) (insKey, keyPos, keys, children, trie, FOUND_WITH_CHILDREN)
in
(case insertBinSearch (findChr, keyPos, keys) of
INSERT_NEW_CHILD insIdx =>
insertNewChild
(keys, insIdx, insKey, children, FOUND_WITH_CHILDREN)
| FOUND_INSERT_POS insIdx =>
foundInsertPos
( keys
, children
, keyPos
, insKey
, insIdx
, trie
, FOUND_WITH_CHILDREN
)
| APPEND_NEW_CHILD =>
appendNewChild (keys, insKey, children, FOUND_WITH_CHILDREN))
end
fun insert (insKey, trie) = fun insert (insKey, trie) =
if String.size insKey > 0 then if String.size insKey > 0 then
@@ -750,71 +736,52 @@ struct
removeWhenChildIsMadeEmpty removeWhenChildIsMadeEmpty
(idx, keys, children, isFoundWithChildren, parentConstructor) (idx, keys, children, isFoundWithChildren, parentConstructor)
fun helpRemove (removeKey, keyPos, trie) = fun helpRemoveChildren
(removeKey, keyPos, keys, children, isFoundWithChildren, parentConstructor) =
let
val findChr = String.sub (removeKey, keyPos)
in
(case findBinSearch (findChr, keyPos, keys) of
SOME idx =>
let
val trieKey = Vector.sub (keys, idx)
in
(case searchKeyMatch (removeKey, trieKey, keyPos + 1) of
(* no search match means nothing to delete *)
NO_SEARCH_MATCH => UNCHANGED
| FULL_SEARCH_MATCH =>
removeWhenFullMatch
( idx
, keys
, children
, isFoundWithChildren
, parentConstructor
)
| SEARCH_KEY_CONTAINS_TRIE_KEY =>
removeWhenSearchKeyContainsTrieKey
( helpRemove
( removeKey
, String.size trieKey
, Vector.sub (children, idx)
)
, idx
, keys
, children
, isFoundWithChildren
, parentConstructor
)
| TRIE_KEY_CONTAINS_SEARCH_KEY => UNCHANGED)
end
| NONE => UNCHANGED)
end
and helpRemove (removeKey, keyPos, trie) =
case trie of case trie of
CHILDREN {keys, children} => CHILDREN {keys, children} =>
let helpRemoveChildren (removeKey, keyPos, keys, children, false, CHILDREN)
val findChr = String.sub (removeKey, keyPos)
in
(case findBinSearch (findChr, keyPos, keys) of
SOME idx =>
let
val trieKey = Vector.sub (keys, idx)
in
(case searchKeyMatch (removeKey, trieKey, keyPos + 1) of
(* no search match means nothing to delete *)
NO_SEARCH_MATCH => UNCHANGED
| FULL_SEARCH_MATCH =>
removeWhenFullMatch (idx, keys, children, false, CHILDREN)
| SEARCH_KEY_CONTAINS_TRIE_KEY =>
removeWhenSearchKeyContainsTrieKey
( helpRemove
( removeKey
, String.size trieKey
, Vector.sub (children, idx)
)
, idx
, keys
, children
, false
, CHILDREN
)
| TRIE_KEY_CONTAINS_SEARCH_KEY => UNCHANGED)
end
| NONE => UNCHANGED)
end
| FOUND_WITH_CHILDREN {keys, children} => | FOUND_WITH_CHILDREN {keys, children} =>
let helpRemoveChildren
val findChr = String.sub (removeKey, keyPos) (removeKey, keyPos, keys, children, true, FOUND_WITH_CHILDREN)
in
(case findBinSearch (findChr, keyPos, keys) of
SOME idx =>
let
val trieKey = Vector.sub (keys, idx)
in
(case searchKeyMatch (removeKey, trieKey, keyPos + 1) of
(* no search match means nothing to delete *)
NO_SEARCH_MATCH => UNCHANGED
| FULL_SEARCH_MATCH =>
removeWhenFullMatch
(idx, keys, children, true, FOUND_WITH_CHILDREN)
| SEARCH_KEY_CONTAINS_TRIE_KEY =>
removeWhenSearchKeyContainsTrieKey
( helpRemove
( removeKey
, String.size trieKey
, Vector.sub (children, idx)
)
, idx
, keys
, children
, true
, FOUND_WITH_CHILDREN
)
| TRIE_KEY_CONTAINS_SEARCH_KEY => UNCHANGED)
end
| NONE => UNCHANGED)
end
| FOUND => | FOUND =>
(* (*
* This case should only occur if we recurse in a node * This case should only occur if we recurse in a node