Clone lookaroundNFA when cloning a state; use compiled regex for
lookarounds instead of compiling a new one
This commit is contained in:
12
nfa.go
12
nfa.go
@@ -32,6 +32,7 @@ type State struct {
|
|||||||
allChars bool // Whether or not the state represents all characters (eg. a 'dot' metacharacter). A 'dot' node doesn't store any contents directly, as it would take up too much space
|
allChars bool // Whether or not the state represents all characters (eg. a 'dot' metacharacter). A 'dot' node doesn't store any contents directly, as it would take up too much space
|
||||||
except []rune // Only valid if allChars is true - match all characters _except_ the ones in this block. Useful for inverting character classes.
|
except []rune // Only valid if allChars is true - match all characters _except_ the ones in this block. Useful for inverting character classes.
|
||||||
lookaroundRegex string // Only for lookaround states - Contents of the regex that the lookaround state holds
|
lookaroundRegex string // Only for lookaround states - Contents of the regex that the lookaround state holds
|
||||||
|
lookaroundNFA *State // Holds the NFA of the lookaroundRegex - if it exists
|
||||||
}
|
}
|
||||||
|
|
||||||
// Clones the NFA starting from the given state.
|
// Clones the NFA starting from the given state.
|
||||||
@@ -47,6 +48,9 @@ func cloneStateHelper(state *State, cloneMap map[*State]*State) *State {
|
|||||||
if clone, exists := cloneMap[state]; exists {
|
if clone, exists := cloneMap[state]; exists {
|
||||||
return clone
|
return clone
|
||||||
}
|
}
|
||||||
|
if state == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
// Recursive case - if the clone doesn't exist, create it, add it to the map,
|
// Recursive case - if the clone doesn't exist, create it, add it to the map,
|
||||||
// and recursively call for each of the transition states.
|
// and recursively call for each of the transition states.
|
||||||
clone := &State{
|
clone := &State{
|
||||||
@@ -80,6 +84,10 @@ func cloneStateHelper(state *State, cloneMap map[*State]*State) *State {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if state.lookaroundNFA == state {
|
||||||
|
clone.lookaroundNFA = clone
|
||||||
|
}
|
||||||
|
clone.lookaroundNFA = cloneStateHelper(state.lookaroundNFA, cloneMap)
|
||||||
return clone
|
return clone
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -104,9 +112,7 @@ func (s State) checkAssertion(str []rune, idx int) bool {
|
|||||||
// 1. Compile the regex stored in the state's contents.
|
// 1. Compile the regex stored in the state's contents.
|
||||||
// 2. Run it on the test string.
|
// 2. Run it on the test string.
|
||||||
// 3. Based on the kind of lookaround (and the indices we get), determine what action to take.
|
// 3. Based on the kind of lookaround (and the indices we get), determine what action to take.
|
||||||
regex := s.lookaroundRegex
|
startState := s.lookaroundNFA
|
||||||
re_postfix := shuntingYard(regex)
|
|
||||||
startState := thompson(re_postfix)
|
|
||||||
matchIndices := findAllMatches(startState, str)
|
matchIndices := findAllMatches(startState, str)
|
||||||
|
|
||||||
numMatchesFound := 0
|
numMatchesFound := 0
|
||||||
|
Reference in New Issue
Block a user