107 lines
3.1 KiB
C++
107 lines
3.1 KiB
C++
#include "Lexer.h"
|
|
#include <cassert>
|
|
|
|
Lexer::Lexer() {
|
|
//Do nothing
|
|
currentPosition = 0;
|
|
}
|
|
|
|
Lexer::Lexer(std::string inputString) {
|
|
input = inputString;
|
|
currentPosition = 0;
|
|
}
|
|
|
|
Lexer::~Lexer() {
|
|
//No cleanup necessary
|
|
}
|
|
|
|
void Lexer::setInput(std::string inputString) {
|
|
input = inputString;
|
|
}
|
|
|
|
void Lexer::addRegEx(std::string regExString) {
|
|
regExs.push_back(new RegEx(regExString));
|
|
}
|
|
|
|
Symbol Lexer::next() {
|
|
//std::cout << "Current at is \"" << input.substr(currentPosition) << "\" currentPos is " << currentPosition << " out of " << input.length() <<std::endl;
|
|
//If we're at the end, return an eof
|
|
if (currentPosition >= input.length())
|
|
return Symbol("$EOF$", true);
|
|
int longestMatch = -1;
|
|
RegEx* longestRegEx = NULL;
|
|
std::string remainingString = input.substr(currentPosition);
|
|
for (std::vector<RegEx*>::size_type i = 0; i < regExs.size(); i++) {
|
|
//std::cout << "Trying regex " << regExs[i]->getPattern() << std::endl;
|
|
int currentMatch = regExs[i]->longMatch(remainingString);
|
|
if (currentMatch > longestMatch) {
|
|
longestMatch = currentMatch;
|
|
longestRegEx = regExs[i];
|
|
}
|
|
}
|
|
if (longestRegEx != NULL) {
|
|
std::string eatenString = input.substr(currentPosition, longestMatch);
|
|
currentPosition += longestMatch;
|
|
//std::cout << "Current at is \"" << input.substr(currentPosition) << "\" currentPos is " << currentPosition <<std::endl;
|
|
return Symbol(longestRegEx->getPattern(), true, eatenString);
|
|
} else {
|
|
//std::cout << "Found no applicable regex" << std::endl;
|
|
//std::cout << "Remaining is ||" << input.substr(currentPosition) << "||" << std::endl;
|
|
return Symbol();
|
|
}
|
|
}
|
|
|
|
void Lexer::test() {
|
|
Symbol s;
|
|
{
|
|
Lexer lex;
|
|
lex.addRegEx("b");
|
|
lex.setInput("bb");
|
|
s = lex.next();
|
|
assert(s.getName() == "b" && s.getValue() == "b");
|
|
s = lex.next();
|
|
assert(s.getName() == "b" && s.getValue() == "b");
|
|
assert(lex.next() == Symbol("$EOF$", true));
|
|
}
|
|
|
|
{
|
|
Lexer lex;
|
|
lex.addRegEx("a*");
|
|
lex.addRegEx("b");
|
|
lex.setInput("aaabaabb");
|
|
s = lex.next();
|
|
assert(s.getName() == "a*" && s.getValue() == "aaa");
|
|
s = lex.next();
|
|
assert(s.getName() == "b" && s.getValue() == "b");
|
|
s = lex.next();
|
|
assert(s.getName() == "a*" && s.getValue() == "aa");
|
|
s = lex.next();
|
|
assert(s.getName() == "b" && s.getValue() == "b");
|
|
s = lex.next();
|
|
assert(s.getName() == "b" && s.getValue() == "b");
|
|
assert(lex.next() == Symbol("$EOF$", true));
|
|
}
|
|
|
|
// Test a lexer error condition.
|
|
{
|
|
Lexer lex;
|
|
lex.addRegEx("a|b");
|
|
lex.setInput("blah");
|
|
s = lex.next();
|
|
assert(s.getName() == "a|b" && s.getValue() == "b");
|
|
assert(lex.next() == Symbol());
|
|
}
|
|
|
|
// Lexer can consume all the input at once.
|
|
{
|
|
Lexer lex;
|
|
lex.addRegEx("xyzzy");
|
|
lex.setInput("xyzzy");
|
|
s = lex.next();
|
|
assert(s.getName() == "xyzzy" && s.getValue() == "xyzzy");
|
|
assert(lex.next() == Symbol("$EOF$", true));
|
|
}
|
|
|
|
std::cout << "Lexer tests passed\n";
|
|
}
|