add comments to ongiong NFA implementation
This commit is contained in:
@@ -11,229 +11,269 @@ struct
|
||||
| ONE_OR_MORE of regex * state
|
||||
| GROUP of regex * state
|
||||
|
||||
val groupLevel = 1
|
||||
val postfixLevel = 2
|
||||
val concatLevel = 3
|
||||
val altLevel = 4
|
||||
structure NfaMatch =
|
||||
struct
|
||||
(* test to see if NFA matches.
|
||||
* Algorithm: Walk down to the leaves/subnodes of the regex tree
|
||||
* and check if the current chr turns the leaf valid.
|
||||
*
|
||||
* When we hit a concatenation node, we check if the list's hd
|
||||
* matches the same char.
|
||||
* If it is valid and we have reached the last list element,
|
||||
* then the whole concatenation is valid and we mark it as such.
|
||||
* If it is valid while we still have other nodes to test,
|
||||
* we filter the hd and the next loop with the next chr
|
||||
* checks the node's tl.
|
||||
* If it isn't valid, then we mark the whole concatenation as invalid.
|
||||
*
|
||||
* When we hit an alternation node, we check each list element at once
|
||||
* to see if any of them are valid.
|
||||
* We filter out the nodes in the alternation list that turn out to be
|
||||
* invalid.
|
||||
* At the end of the alternation loop, we check if all nodes are valid
|
||||
* and then mark the alternation as valid if so.
|
||||
* This helps us to implement "maximal munch",
|
||||
* retrieving the maximum match instead of any other.
|
||||
* *)
|
||||
local
|
||||
fun loop (tl, maxValid) =
|
||||
case tl of
|
||||
(_, VALID curValid) :: tl => loop (tl, Int.max (maxValid, curValid))
|
||||
| (_, UNTESTED) :: _ => UNTESTED
|
||||
| (_, INVALID) :: _ =>
|
||||
raise Fail
|
||||
"nfa.sml 24: \
|
||||
\should not have INVALID state in acc"
|
||||
| [] => VALID maxValid
|
||||
in
|
||||
fun getAlternationState acc =
|
||||
case acc of
|
||||
(_, VALID maxValid) :: tl => loop (tl, maxValid)
|
||||
| (_, UNTESTED) :: _ => UNTESTED
|
||||
| (_, INVALID) :: _ =>
|
||||
raise Fail
|
||||
"nfa.sml 26: \
|
||||
\should not have INVALID state in acc"
|
||||
| [] => UNTESTED
|
||||
end
|
||||
|
||||
local
|
||||
fun loop (tl, maxValid) =
|
||||
case tl of
|
||||
(_, VALID curValid) :: tl => loop (tl, Int.max (maxValid, curValid))
|
||||
| (_, UNTESTED) :: _ => UNTESTED
|
||||
| (_, INVALID) :: _ =>
|
||||
raise Fail
|
||||
"nfa.sml 24: \
|
||||
\should not have INVALID state in acc"
|
||||
| [] => VALID maxValid
|
||||
in
|
||||
fun getAlternationState acc =
|
||||
case acc of
|
||||
(_, VALID maxValid) :: tl => loop (tl, maxValid)
|
||||
| (_, UNTESTED) :: _ => UNTESTED
|
||||
| (_, INVALID) :: _ =>
|
||||
raise Fail
|
||||
"nfa.sml 26: \
|
||||
\should not have INVALID state in acc"
|
||||
| [] => UNTESTED
|
||||
end
|
||||
local
|
||||
fun rebuildConcat (lst, chr, idx) =
|
||||
case lst of
|
||||
[(hd, _)] =>
|
||||
let
|
||||
val (hd: regex, state: state) = rebuild (hd, chr, idx)
|
||||
val result = [(hd, state)]
|
||||
val concat = CONCAT (result, state)
|
||||
in
|
||||
(concat, state)
|
||||
end
|
||||
| (hd, _) :: tl =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
in
|
||||
case state of
|
||||
UNTESTED =>
|
||||
let val concat = CONCAT ((hd, state) :: tl, UNTESTED)
|
||||
in (concat, UNTESTED)
|
||||
end
|
||||
| INVALID =>
|
||||
let val concat = CONCAT ([], INVALID)
|
||||
in (concat, INVALID)
|
||||
end
|
||||
| VALID _ =>
|
||||
let val concat = CONCAT (tl, UNTESTED)
|
||||
in (concat, UNTESTED)
|
||||
end
|
||||
end
|
||||
| [] =>
|
||||
(* should never occur *)
|
||||
raise Fail
|
||||
"nfa.sml, rebuildConcat 45: \
|
||||
\should never try to rebuild empty concat list"
|
||||
|
||||
local
|
||||
fun rebuildConcat (lst, chr, idx) =
|
||||
case lst of
|
||||
[(hd, _)] =>
|
||||
and rebuildAlternation (lst, chr, idx, acc) =
|
||||
case lst of
|
||||
[(hd, _)] =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
val acc =
|
||||
case state of
|
||||
VALID _ => (hd, state) :: acc
|
||||
| UNTESTED => (hd, state) :: acc
|
||||
| INVALID => acc
|
||||
val state = getAlternationState acc
|
||||
in
|
||||
(ALTERNATION (acc, state), state)
|
||||
end
|
||||
| (hd, _) :: tl =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
val acc =
|
||||
case state of
|
||||
VALID _ => (hd, state) :: acc
|
||||
| UNTESTED => (hd, state) :: acc
|
||||
| INVALID => acc
|
||||
in
|
||||
rebuildAlternation (tl, chr, idx, acc)
|
||||
end
|
||||
| [] => (ALTERNATION ([], INVALID), INVALID)
|
||||
|
||||
and rebuild (nfa, chr, idx) =
|
||||
case nfa of
|
||||
CHAR_LITERAL (lit, UNTESTED) =>
|
||||
if chr = lit then (CHAR_LITERAL (lit, VALID idx), VALID idx)
|
||||
else (CHAR_LITERAL (lit, INVALID), INVALID)
|
||||
| CHAR_LITERAL (lit, state) => (nfa, state)
|
||||
|
||||
| CONCAT (lst, UNTESTED) => rebuildConcat (lst, chr, idx)
|
||||
| CONCAT (_, state) => (nfa, state)
|
||||
|
||||
| ALTERNATION (lst, UNTESTED) => rebuildAlternation (lst, chr, idx, [])
|
||||
| ALTERNATION (_, state) => (nfa, state)
|
||||
|
||||
| _ =>
|
||||
raise Fail "nfa.sml 69: not char literal or concat or alternation"
|
||||
|
||||
fun loop (pos, str, nfa, origNfa, startPos) =
|
||||
if pos = String.size str then
|
||||
false
|
||||
else
|
||||
let
|
||||
val (hd: regex, state: state) = rebuild (hd, chr, idx)
|
||||
val result = [(hd, state)]
|
||||
val concat = CONCAT (result, state)
|
||||
in
|
||||
(concat, state)
|
||||
end
|
||||
| (hd, _) :: tl =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
val chr = String.sub (str, pos)
|
||||
val (nfa, state) = rebuild (nfa, chr, pos)
|
||||
in
|
||||
case state of
|
||||
UNTESTED =>
|
||||
let val concat = CONCAT ((hd, state) :: tl, UNTESTED)
|
||||
in (concat, UNTESTED)
|
||||
end
|
||||
VALID _ => true
|
||||
| INVALID =>
|
||||
let val concat = CONCAT ([], INVALID)
|
||||
in (concat, INVALID)
|
||||
end
|
||||
| VALID _ =>
|
||||
let val concat = CONCAT (tl, UNTESTED)
|
||||
in (concat, UNTESTED)
|
||||
end
|
||||
(* backtrack to another position in the string
|
||||
* to see if the NFA matches that portion of the string *)
|
||||
loop (startPos + 1, str, origNfa, origNfa, startPos + 1)
|
||||
| UNTESTED => loop (pos + 1, str, nfa, origNfa, startPos)
|
||||
end
|
||||
| [] =>
|
||||
(* should never occur *)
|
||||
raise Fail
|
||||
"nfa.sml, rebuildConcat 45: \
|
||||
\should never try to rebuild empty concat list"
|
||||
|
||||
and rebuildAlternation (lst, chr, idx, acc) =
|
||||
case lst of
|
||||
[(hd, _)] =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
val acc =
|
||||
case state of
|
||||
VALID _ => (hd, state) :: acc
|
||||
| UNTESTED => (hd, state) :: acc
|
||||
| INVALID => acc
|
||||
val state = getAlternationState acc
|
||||
in
|
||||
(ALTERNATION (acc, state), state)
|
||||
end
|
||||
| (hd, _) :: tl =>
|
||||
let
|
||||
val (hd, state) = rebuild (hd, chr, idx)
|
||||
val acc =
|
||||
case state of
|
||||
VALID _ => (hd, state) :: acc
|
||||
| UNTESTED => (hd, state) :: acc
|
||||
| INVALID => acc
|
||||
in
|
||||
rebuildAlternation (tl, chr, idx, acc)
|
||||
end
|
||||
| [] => (ALTERNATION ([], INVALID), INVALID)
|
||||
|
||||
and rebuild (nfa, chr, idx) =
|
||||
case nfa of
|
||||
CHAR_LITERAL (lit, UNTESTED) =>
|
||||
if chr = lit then (CHAR_LITERAL (lit, VALID idx), VALID idx)
|
||||
else (CHAR_LITERAL (lit, INVALID), INVALID)
|
||||
| CHAR_LITERAL (lit, state) => (nfa, state)
|
||||
|
||||
| CONCAT (lst, UNTESTED) => rebuildConcat (lst, chr, idx)
|
||||
| CONCAT (_, state) => (nfa, state)
|
||||
|
||||
| ALTERNATION (lst, UNTESTED) => rebuildAlternation (lst, chr, idx, [])
|
||||
| ALTERNATION (_, state) => (nfa, state)
|
||||
|
||||
| _ => raise Fail "nfa.sml 69: not char literal or concat or alternation"
|
||||
|
||||
fun loop (pos, str, nfa, origNfa, startPos) =
|
||||
if pos = String.size str then
|
||||
false
|
||||
else
|
||||
let
|
||||
val chr = String.sub (str, pos)
|
||||
val (nfa, state) = rebuild (nfa, chr, pos)
|
||||
in
|
||||
case state of
|
||||
VALID _ => true
|
||||
| INVALID => loop (startPos + 1, str, origNfa, origNfa, startPos + 1)
|
||||
| UNTESTED => loop (pos + 1, str, nfa, origNfa, startPos)
|
||||
end
|
||||
in
|
||||
fun hasAnyMatch (str, nfa) =
|
||||
loop (0, str, nfa, nfa, 0)
|
||||
in
|
||||
fun hasAnyMatch (str, nfa) =
|
||||
loop (0, str, nfa, nfa, 0)
|
||||
end
|
||||
end
|
||||
|
||||
local
|
||||
fun loop (pos, str, openParens, closeParens) =
|
||||
structure ParseNfa =
|
||||
struct
|
||||
(* parsing through precedence climbing algorithm.
|
||||
* Todo: return a a `regex option`, and use bounds-checking
|
||||
* to ensure we don't raise an exception. *)
|
||||
|
||||
val groupLevel = 1
|
||||
val postfixLevel = 2
|
||||
val concatLevel = 3
|
||||
val altLevel = 4
|
||||
|
||||
local
|
||||
fun loop (pos, str, openParens, closeParens) =
|
||||
if pos = String.size str then
|
||||
pos
|
||||
else
|
||||
case String.sub (str, pos) of
|
||||
#"(" => loop (pos + 1, str, openParens + 1, closeParens)
|
||||
| #")" =>
|
||||
if closeParens + 1 = openParens then pos
|
||||
else loop (pos + 1, str, openParens, closeParens + 1)
|
||||
| _ => loop (pos + 1, str, openParens, closeParens)
|
||||
in
|
||||
fun getRightParenIdx (pos, str) = loop (pos, str, 1, 0)
|
||||
end
|
||||
|
||||
fun climb (pos, str, lhs, level) =
|
||||
if pos = String.size str then
|
||||
pos
|
||||
(pos, lhs)
|
||||
else
|
||||
case String.sub (str, pos) of
|
||||
#"(" => loop (pos + 1, str, openParens + 1, closeParens)
|
||||
| #")" =>
|
||||
if closeParens + 1 = openParens then pos
|
||||
else loop (pos + 1, str, openParens, closeParens + 1)
|
||||
| _ => loop (pos + 1, str, openParens, closeParens)
|
||||
in
|
||||
fun getRightParenIdx (pos, str) = loop (pos, str, 1, 0)
|
||||
#")" => (pos + 1, lhs)
|
||||
| #"(" =>
|
||||
if level < groupLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val groupEndIdx = getRightParenIdx (pos + 1, str)
|
||||
val substr = String.substring
|
||||
(str, pos + 1, groupEndIdx - pos - 1)
|
||||
val rhs = parse substr
|
||||
val rhs = GROUP (rhs, UNTESTED)
|
||||
val result = CONCAT
|
||||
([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
climb (groupEndIdx + 1, str, result, groupLevel)
|
||||
end
|
||||
| #"|" =>
|
||||
if level < altLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val chr = String.sub (str, pos + 1)
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
val (pos, rhs) = climb (pos + 2, str, chr, altLevel)
|
||||
val result =
|
||||
case rhs of
|
||||
ALTERNATION (lst, state) =>
|
||||
ALTERNATION ((lhs, UNTESTED) :: lst, UNTESTED)
|
||||
| _ =>
|
||||
ALTERNATION ([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
(pos, result)
|
||||
end
|
||||
| #"?" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ZERO_OR_ONE (lhs, UNTESTED)
|
||||
in climb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| #"*" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ZERO_OR_MORE (lhs, UNTESTED)
|
||||
in climb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| #"+" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ONE_OR_MORE (lhs, UNTESTED)
|
||||
in climb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| chr =>
|
||||
if level < concatLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
val (pos, rhs) = climb (pos + 1, str, chr, concatLevel)
|
||||
val result =
|
||||
case rhs of
|
||||
CONCAT (lst, _) => CONCAT ((lhs, UNTESTED) :: lst, UNTESTED)
|
||||
| _ => CONCAT ([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
(pos, result)
|
||||
end
|
||||
|
||||
and loop (pos, str, ast) =
|
||||
if pos = String.size str then
|
||||
ast
|
||||
else
|
||||
let val (pos, ast) = climb (pos, str, ast, altLevel)
|
||||
in loop (pos, str, ast)
|
||||
end
|
||||
|
||||
and parse str =
|
||||
let
|
||||
val chr = String.sub (str, 0)
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
in
|
||||
loop (1, str, chr)
|
||||
end
|
||||
end
|
||||
|
||||
fun helpClimb (pos, str, lhs, level) =
|
||||
if pos = String.size str then
|
||||
(pos, lhs)
|
||||
else
|
||||
case String.sub (str, pos) of
|
||||
#")" => (pos + 1, lhs)
|
||||
| #"(" =>
|
||||
if level < groupLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val groupEndIdx = getRightParenIdx (pos + 1, str)
|
||||
val substr = String.substring
|
||||
(str, pos + 1, groupEndIdx - pos - 1)
|
||||
val rhs = climb substr
|
||||
val rhs = GROUP (rhs, UNTESTED)
|
||||
val result = CONCAT ([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
helpClimb (groupEndIdx + 1, str, result, groupLevel)
|
||||
end
|
||||
| #"|" =>
|
||||
if level < altLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val chr = String.sub (str, pos + 1)
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
val (pos, rhs) = helpClimb (pos + 2, str, chr, altLevel)
|
||||
val result =
|
||||
case rhs of
|
||||
ALTERNATION (lst, state) =>
|
||||
ALTERNATION ((lhs, UNTESTED) :: lst, UNTESTED)
|
||||
| _ =>
|
||||
ALTERNATION ([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
(pos, result)
|
||||
end
|
||||
| #"?" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ZERO_OR_ONE (lhs, UNTESTED)
|
||||
in helpClimb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| #"*" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ZERO_OR_MORE (lhs, UNTESTED)
|
||||
in helpClimb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| #"+" =>
|
||||
if level < postfixLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let val lhs = ONE_OR_MORE (lhs, UNTESTED)
|
||||
in helpClimb (pos + 1, str, lhs, postfixLevel)
|
||||
end
|
||||
| chr =>
|
||||
if level < concatLevel then
|
||||
(pos, lhs)
|
||||
else
|
||||
let
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
val (pos, rhs) = helpClimb (pos + 1, str, chr, concatLevel)
|
||||
val result =
|
||||
case rhs of
|
||||
CONCAT (lst, _) => CONCAT ((lhs, UNTESTED) :: lst, UNTESTED)
|
||||
| _ => CONCAT ([(lhs, UNTESTED), (rhs, UNTESTED)], UNTESTED)
|
||||
in
|
||||
(pos, result)
|
||||
end
|
||||
|
||||
and loop (pos, str, ast) =
|
||||
if pos = String.size str then
|
||||
ast
|
||||
else
|
||||
let val (pos, ast) = helpClimb (pos, str, ast, altLevel)
|
||||
in loop (pos, str, ast)
|
||||
end
|
||||
|
||||
and climb str =
|
||||
let
|
||||
val chr = String.sub (str, 0)
|
||||
val chr = CHAR_LITERAL (chr, UNTESTED)
|
||||
in
|
||||
loop (1, str, chr)
|
||||
end
|
||||
val parse = ParseNfa.parse
|
||||
val hasAnyMatch = NfaMatch.hasAnyMatch
|
||||
end
|
||||
|
||||
Reference in New Issue
Block a user