How to Build a Math Evaluation Engine

Development

Reading Time: 5 minutes

One of the first things you learn to do with Ruby is to evaluate a mathematical expression in irb. But what’s the magic behind this seemingly trivial operation? Let’s create our own evaluation engine and find out!

Note: You could cheat and just use eval, but that would defeat the point of this exercise.

Tokenizing Our Input

We are given a string as input, but what we need are numbers and operators. We can use the StringScanner class and ‘tag’ the different parts of the input so we can recognize them.

This is the main method of the Tokenizer class:

def parse(input)
  @buffer = StringScanner.new(input)

  until @buffer.eos?
    skip_spaces
    read_tokens
  end

  @tokens
end

In this method, I made a StringScanner object with the input string as an argument, then I have a loop which keeps going until the end of the string is reached. Inside this loop, I call two methods: skip_spaces and read_tokens.

The skip_spaces method is very simple:

def skip_spaces
  @buffer.skip(/\s+/)
end

The /\s+/ part is a regular expression that matches spaces (including tabs and newlines). Then we have read_tokens, which is where most of the work happens:

RULES.each do |regex, type|
  token = @buffer.scan(regex)

  if token
    @tokens << Token.new(type, token)
  end
end

In this method, we try to match part of the input string against one of our parsing rules, which are just regular expressions. When a match is found, we create a new token object with the type and value. For example, for the number ‘5’ we would have Token.new(:int, 5).

RULES = {
  /\d+/      => :int,
  /[\/+\-*]/ => :op
}

And that’s all we need to convert our input string into tokens. Now let’s see how we can turn those tokens into something more useful.

Sign up for a free Codeship Account

The Shunting Yard Algorithm

Now that we have our tokens, we need to convert them into a format we can work with. The normal format for a math expression is called ‘infix’. The problem with this format is that it’s hard to evaluate because of operation precedence. There is another way to represent a mathematical expression: the ‘postfix’ notation.

Examples:

Infix ___________Postfix
3 + 43 4 +
9 * 49 4 *
2 * 5 + 12 5 * 1 +

We are going to use the Shunting-yard algorithm to convert from the infix notation to postfix. The way this algorithm works is by using two stacks, one for numbers and another for operators.

Follow these steps:

  1. Read a token.
  2. If it’s a number, put it in the numbers stack.
  3. If it’s an operator, put it in the operators stack, but first, check the precedence of the operator on the top of the stack. Move it into the numbers stack if the precedence is higher or equal.
  4. When you’ve read all the tokens, pop all the operators and put them into the numbers stack.
  5. Done!

Let’s take a look at the code, starting with the initialize method:

def initialize(tokens)
  @tokens    = tokens
  @output    = []
  @operators = []
end

In this method, we do our initial setup by using the tokens from the previous phase (tokenization) and creating two arrays that will serve as our stacks. Let’s continue our exploration of the code:

@tokens.each do |token|
  push_number(token)      if token.type == :int
  process_operator(token) if token.type == :op
end

Inside the run method, we iterate over our tokens array and put them in the correct stacks.

def operator_priority_is_less_or_equal_than(token)
  @operators.last && (token.priority <= @operators.last.priority)
end

This method helps us handle operator precedence. The priority is a method defined in the Token class.

def priority
  return 1 if value == '+' || value == '-'
  return 2 if value == '*' || value == '/'
end

When we’re done, we just dump the remaining operators into our output stack.

@output += @operators.reverse

We need to call reverse here. Putting things into a stack and taking them out reverses them, but in this case, we want the original order.

And that’s all it takes to convert to postfix notation (also know as Reverse Polish Notation or RPN).

Evaluation

Now that we have our expression in postfix notation, it’s very easy to evaluate using a single stack.

Algorithm:

  1. We put all the numbers we find on the numbers stack.
  2. Whenever we find an operator, we pop two numbers from the stack.
  3. Then we calculate the result and put the result back on the top of the stack.
  4. Repeat until done. Last number will be the final result.
tokens.each do |token|
  @numbers << token if token.type == :int

  if token.type == :op
    right_num = @numbers.pop
    left_num  = @numbers.pop

    raise 'Invalid postfix expression' unless right_num && left_num

    result = evaluate(right_num.value, left_num.value, token)
    @numbers << Token.new(:int, result)
  end
end

This is the main loop for the evaluation method. Once we find an operation, we call the evaluate method and create a new token object with the results.

def evaluate(right_num, left_num, operation)
  case operation.value
  when '+' then left_num + right_num
  when '-' then left_num - right_num
  when '*' then left_num * right_num
  when '/' then left_num / right_num
  end
end

There is nothing special about this method; we just apply the operation and return the result.

@numbers.last.value

Once all the operators have been processed, we’ll have one token left on our numbers stack. This token is the final result.

Conclusion

In this post, you learned about tokenization, which lets you break a string into its component values. This is what most programming languages (like Ruby or JavaScript) do behind the scenes to your source code so they can turn it into computer instructions.

Then you learned about two different ways to arrange a mathematical expression, ‘infix’ and ‘postfix’, and how to convert between them by using the ‘Shunting-Yard algorithm’. You’ve also learned about the ‘stack’ data structure and what it can be useful for.

I hope you enjoyed this thought exercise! You can find the final code here: https://github.com/matugm/math-eval

Subscribe via Email

Over 60,000 people from companies like Netflix, Apple, Spotify and O'Reilly are reading our articles.
Subscribe to receive a weekly newsletter with articles around Continuous Integration, Docker, and software development best practices.



We promise that we won't spam you. You can unsubscribe any time.

Join the Discussion

Leave us some comments on what you think about this topic or if you like to add something.