aboutsummaryrefslogtreecommitdiff
path: root/lib/elixir_math_parser.ex
blob: 91f67f9166e736c826fc6671e04803d5c51e88ca (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
defmodule ElixirMathParser do
  @moduledoc """
  Documentation for `ElixirMathParser`.
  """

  defp reduce_to_value({:int, _line, value}, _state) do
    {:ok, value}
  end

  defp reduce_to_value({:atom, _line, atom}, state) do
    if !Map.has_key?(state, atom) do
      {:error, "value not found for " <> to_string(atom)}
    else
      {:ok, state[atom]}
    end
  end

  defp reduce_to_value({:add_op, lhs, rhs}, state) do
    {:ok, op1} = reduce_to_value(lhs, state)
    {:ok, op2} = reduce_to_value(rhs, state)
    {:ok, op1 + op2}
  end

  defp reduce_to_value({:sub_op, lhs, rhs}, state) do
    {:ok, op1} = reduce_to_value(lhs, state)
    {:ok, op2} = reduce_to_value(rhs, state)
    {:ok, op1 - op2}
  end

  defp reduce_to_value({:mul_op, lhs, rhs}, state) do
    {:ok, op1} = reduce_to_value(lhs, state)
    {:ok, op2} = reduce_to_value(rhs, state)
    {:ok, op1 / op2}
  end

  defp reduce_to_value({:div_op, lhs, rhs}, state) do
    {:ok, op1} = reduce_to_value(lhs, state)
    {:ok, op2} = reduce_to_value(rhs, state)
    {:ok, op1 / op2}
  end

  defp evaluate_tree([{:assign, {:atom, _line, lhs}, rhs} | tail], state) do
    {:ok, rhs_value} = reduce_to_value(rhs, state)
    evaluate_tree(tail, Map.merge(state, %{lhs => rhs_value}))
  end

  defp evaluate_tree([], state) do
    state
  end

  def process_tree(tree) do
    evaluate_tree(tree, %{})
  end

  def parse_file(filename) do
    text = File.read!(filename)
    {:ok, tokens, _line} = :elixir_math_parser_lexer.string(String.to_charlist(text))
    {:ok, tree} = :elixir_math_parser.parse(tokens)
    process_tree(tree)
  end
end