aboutsummaryrefslogtreecommitdiff
path: root/src/utils.cc
blob: 112bc204a53bf9904af46248d262117d641b12ee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#include "utils.h"
#include "exceptions.h"

#include "tree.h"

namespace SimpleParser {

TokenType getTokenType(char tmp) {
	switch ( tmp ) {
		case '-':
			return TokenType::OPERATOR_PLUS;
		case '+':
			return TokenType::OPERATOR_MINUS;
		case '/':
			return TokenType::OPERATOR_DIVIDE;
		case '*':
			return TokenType::OPERATOR_MULTIPLY;
		case '^':
			return TokenType::OPERATOR_POWER;
		case '(':
			return TokenType::PARENTHESES_OPEN;
		case ')':
			return TokenType::PARENTHESES_CLOSE;
		case ',':
			return TokenType::VALUE_NUMBER;
		default:
			return TokenType::VALUE_NUMBER;
	}
}

std::vector<std::string> lexer(std::string term) {
	std::string tmp;
	std::string tmpNum;
	std::vector<std::string> output;
	TokenType token;
	TokenType lastToken;

	uint32_t level = 0;

	for ( auto termIter = term.begin();
	      termIter     != term.end();
	      termIter++ ) {
		token = getTokenType(*termIter);

		if ( token    == TokenType::VALUE_NUMBER   ||
		   ( token    == TokenType::OPERATOR_MINUS &&
		     termIter == term.begin() ) ) {
			if ( level > 0 ) {
				tmp += *termIter;
			} else {
				tmpNum += *termIter;
			}
		} else {
			if ( lastToken == TokenType::VALUE_NUMBER &&
			     level     == 0 ) {
				output.push_back(tmpNum);
				tmpNum.clear();
			}

			switch ( token ) {
				case TokenType::PARENTHESES_OPEN: {
					if ( level > 0 ) {
						tmp += *termIter;
					}

					level++;

					break;
				}
				case TokenType::PARENTHESES_CLOSE: {
					level--;

					if ( level == 0 ) {
						output.push_back(tmp);
						tmp.clear();
					} else {
						tmp += *termIter;
					}

					break;
				}
				default: {
					if ( level == 0 ) {
						std::string helper;
						helper = *termIter;

						output.push_back(helper);
					} else {
						tmp += *termIter;
					}

					break;
				}
			}
		}

		lastToken = token;
	}

	if ( lastToken == TokenType::VALUE_NUMBER ) {
		output.push_back(tmpNum);
	} else if ( lastToken != TokenType::PARENTHESES_CLOSE ) {
		throw operator_exception();
	}

	if ( level != 0 ) {
		throw parenthese_exception();
	}

	if ( lastToken     == TokenType::PARENTHESES_CLOSE &&
	     output.size() == 1 ) {
		output = lexer(output[0]);
	}

	return output;
}

}