calculator-cpp/main.cpp at master · ictxiangxin/calculator-cpp · GitHub

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
#include <iostream>
#include <fstream>
#include <sstream>
#include <list>
#include <mpfr.h>
#include "parser/boson.hpp"


void usage() {
    std::cout << "Usage: calculator -[s|f] [code|file]" << std::endl;
    exit(0);
}

int main(int argc, char** argv) {
    if (argc != 3) {
        usage();
    }
    std::string code;
    if (argv[1][0] == '-') {
        if (argv[1][1] == 's') {
            code = argv[2];
        } else if (argv[1][1] == 'f') {
            std::ifstream file;
            file.open(argv[2],  std::ifstream::in);
            if (file.good()) {
                std::stringstream buffer;
                buffer << file.rdbuf();
                code = buffer.str();
                file.close();
                buffer.clear();
            }
        } else {
            usage();
        }
    } else {
        usage();
    }
    mpfr_set_default_prec(128);
    std::unordered_map<std::string, std::string> variables;
    boson::Tokenizer tokenizer;
    boson::Parser parser;
    boson::Interpreter<mpfr_t> interpreter;
    if (tokenizer.tokenize(code) != tokenizer.no_error_index()) {
        std::cerr << "[ERROR] Lexical error, index: " << tokenizer.error_index() << std::endl;
        return -1;
    }
    boson::BosonGrammar grammar_tree = parser.parse(tokenizer.token_list());
    if (grammar_tree.get_error_index() != grammar_tree.no_error_index()) {
        int line = tokenizer.token_list()[grammar_tree.get_error_index()].line;
        std::cerr << "[ERROR] Grammar error, line: " << line << std::endl;
        for (auto &t: tokenizer.token_list()) {
            if (line == t.line) {
                std::cerr << t.text << ' ';
            }
        }
        std::cerr << std::endl;
        return -1;
    }
    interpreter.register_action("set_variable", [&variables](boson::BosonSemanticsNode<mpfr_t> &node) -> boson::BosonSemanticsNode<mpfr_t> {
        char number[1024];
        mpfr_snprintf(number, 1024, "%.32Rf", node[1].get_data());
        variables[node[0].get_text()] = number;
        return boson::BosonSemanticsNode<mpfr_t>::null_node();
    });
    interpreter.register_action("get_variable", [&variables](boson::BosonSemanticsNode<mpfr_t> &node) -> boson::BosonSemanticsNode<mpfr_t> {
        boson::BosonSemanticsNode<mpfr_t> result;
        mpfr_init_set_str(result.get_data(), variables[node[0].get_text()].c_str(), 10, GMP_RNDD);
        return result;
    });
    interpreter.register_action("function", [](boson::BosonSemanticsNode<mpfr_t> &node) -> boson::BosonSemanticsNode<mpfr_t> {
        boson::BosonSemanticsNode<mpfr_t> function_return;
        mpfr_init(function_return.get_data());
        std::string function_name = node[0].get_text();
        if (function_name == "sqrt") {
            mpfr_sqrt(function_return.get_data(), node[1].get_data(), GMP_RNDD);
        } else if (function_name == "sin") {
            mpfr_sin(function_return.get_data(), node[1].get_data(), GMP_RNDD);
        } else if (function_name == "cos") {
            mpfr_cos(function_return.get_data(), node[1].get_data(), GMP_RNDD);
        } else if (function_name == "tan") {
            mpfr_tan(function_return.get_data(), node[1].get_data(), GMP_RNDD);
        } else if (function_name == "pow") {
            mpfr_pow(function_return.get_data(), node[1].get_data(), node[2].get_data(), GMP_RNDD);
        } else if (function_name == "print") {
            mpfr_printf("%.32Rf\n", node[1].get_data());
            return boson::BosonSemanticsNode<mpfr_t>::null_node();
        }
        return function_return;
    });
    interpreter.register_action("compute", [](boson::BosonSemanticsNode<mpfr_t> &node) -> boson::BosonSemanticsNode<mpfr_t> {
        boson::BosonSemanticsNode<mpfr_t> result;
        mpfr_init(result.get_data());
        boson::BosonSemanticsNode<mpfr_t> &value_a = node[0];
        boson::BosonSemanticsNode<mpfr_t> &op = node[1];
        boson::BosonSemanticsNode<mpfr_t> &value_b = node[2];
        if (op.get_text() == "+") {
            mpfr_add(result.get_data(), value_a.get_data(), value_b.get_data(), GMP_RNDD);
        } else if (op.get_text() == "-") {
            mpfr_sub(result.get_data(), value_a.get_data(), value_b.get_data(), GMP_RNDD);
        } else if (op.get_text() == "*") {
            mpfr_mul(result.get_data(), value_a.get_data(), value_b.get_data(), GMP_RNDD);
        } else if (op.get_text() == "/") {
            mpfr_div(result.get_data(), value_a.get_data(), value_b.get_data(), GMP_RNDD);
        } else if (op.get_text() == "^") {
            mpfr_pow(result.get_data(), value_a.get_data(), value_b.get_data(), GMP_RNDD);
        }
        return result;
    });
    interpreter.register_action("number", [](boson::BosonSemanticsNode<mpfr_t> &node) -> boson::BosonSemanticsNode<mpfr_t> {
        boson::BosonSemanticsNode<mpfr_t> result;
        mpfr_init_set_str(result.get_data(), node[0].get_text().c_str(), 10, GMP_RNDD);
        return result;
    });
    interpreter.execute(grammar_tree.get_grammar_tree());
    return 0;
}