You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
package main
|
|
|
|
// takeZeroState takes the 0-state (if such a transition exists) for all states in the
|
|
// given slice. It returns the resulting states. If any of the resulting states is a 0-state,
|
|
// the second parameter is true.
|
|
func takeZeroState(states []*State) (rtv []*State, isZero bool) {
|
|
for _, state := range states {
|
|
if len(state.transitions[EPSILON]) > 0 {
|
|
rtv = append(rtv, state.transitions[EPSILON]...)
|
|
}
|
|
}
|
|
for _, state := range rtv {
|
|
if len(state.transitions[EPSILON]) > 0 {
|
|
return rtv, true
|
|
}
|
|
}
|
|
return rtv, false
|
|
}
|
|
|
|
// match tries to match the regex represented by given start-state, with
|
|
// the given string
|
|
func match(start *State, str string) (startIdx int, endIdx int, matched bool) {
|
|
currentStates := make([]*State, 0)
|
|
tempStates := make([]*State, 0) // Used to store states that should be used in next loop iteration
|
|
i := 0 // Index in string
|
|
// Increment until we hit a character matching the start state (assuming not 0-state)
|
|
if start.isEmpty == false {
|
|
for i < len(str) && int(str[i]) != start.content {
|
|
i++
|
|
}
|
|
i++ // Advance to next character (if we aren't at a 0-state, which doesn't match anything), so that we can check for transitions. If we advance at a 0-state, we will never get a chance to match the first character
|
|
}
|
|
// TODO - If start state is kleene star, try to match the next state
|
|
|
|
currentStates = append(currentStates, start)
|
|
startIdx = i
|
|
|
|
// Main loop
|
|
for i < len(str) {
|
|
zeroStates := make([]*State, 0)
|
|
// Keep taking zero-states, until there are no more left to take
|
|
zeroStates, isZero := takeZeroState(currentStates)
|
|
tempStates = append(tempStates, zeroStates...)
|
|
for isZero == true {
|
|
zeroStates, isZero = takeZeroState(tempStates)
|
|
tempStates = append(tempStates, zeroStates...)
|
|
}
|
|
|
|
currentStates = append(currentStates, tempStates...)
|
|
tempStates = nil
|
|
|
|
// Take any transitions corresponding to current character
|
|
for _, state := range currentStates {
|
|
if len(state.transitions[int(str[i])]) > 0 {
|
|
tempStates = append(tempStates, state.transitions[int(str[i])]...)
|
|
} else {
|
|
// This enables the 'greedy' behavior - last-state status is only checked if we can't match anything else
|
|
if state.isLast {
|
|
endIdx = i
|
|
return startIdx, endIdx, true
|
|
}
|
|
}
|
|
}
|
|
currentStates = make([]*State, len(tempStates))
|
|
copy(currentStates, tempStates)
|
|
tempStates = nil
|
|
|
|
i++
|
|
}
|
|
|
|
// End-of-string reached. Go to any 0-states. Then check if any of our states are in the end position.
|
|
for _, state := range currentStates {
|
|
if len(state.transitions[EPSILON]) > 0 {
|
|
tempStates = append(tempStates, state.transitions[EPSILON]...)
|
|
}
|
|
}
|
|
currentStates = append(currentStates, tempStates...)
|
|
tempStates = nil
|
|
|
|
for _, state := range currentStates {
|
|
if state.isLast {
|
|
endIdx = i
|
|
return startIdx, endIdx, true
|
|
} else {
|
|
return -1, -1, false
|
|
}
|
|
}
|
|
|
|
// Default
|
|
return -1, -1, false
|
|
}
|