dcz/main.cpp at master · CodeCs1/dcz · 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
#include <iostream>
#include <fstream>
#include <Token.h>
#include <Syntax.h>
#include <RunAST.h>

#include <config.h>
#ifdef COMPILER
#include <AST2MC.h>
#endif
#include <sstream>
#include <cstring>

#include <CLI/CLI.hpp>
#include <functional>

using namespace std;

#define VERSION "1.0"

int main(int argc, char* argv[]) {

    CLI::App app{"A dcz (duca compiler zcript) compiler + interpreter"};
    argv = app.ensure_utf8(argv);
    string filename;
    string output="out.a";
    bool VerboseEnabled=false;
    bool Compile=false;
    int memoryorigin =0;
    string mstr;
    std::string arch="8086";
    app.add_option("filename", filename, ".dcz filename");
    app.add_flag("-v,--verbose", VerboseEnabled, "Enable verbose");
    app.add_flag_function("-V,--version",[](int64_t a) { cout << VERSION <<endl; exit(0); } ,"Show version and exit");
    app.add_option("-o,--output", output, "Output file");
    auto compileopt = app.add_option_group("Compile option");
    compileopt->add_flag("-c,--compile", Compile, "Enable Compile (default is interpreter)");
    compileopt->add_flag("-a,--arch", arch, "Binary architecture (8086, x86_64)");
    compileopt->add_option("-O,--origin", mstr,"Set memory origin (e.g: 7c00)");

    CLI11_PARSE(app, argc, argv);
    if (!mstr.empty())
        memoryorigin = stoi(mstr, 0, 16);

    if (argc < 2) {
        cerr << app.help() << endl;
        return 1;
    }

    if (filename == "stdin") {
        cerr << "stdin not support yet!" << endl;
        return 1;
    }
    ifstream ifs(filename);
    if (ifs.fail()) {
        perror("ifstream");
        return -1;
    }

    string s; {
        stringstream str;
        str << ifs.rdbuf();
        s = str.str();
    }

    Token t(filename, s);
    t.ParseLexer();
    Parser pars(filename, t.ReturnLexer());
    auto x = pars.ReturnExpr();

    if (!Compile)
        RunAST(VerboseEnabled,filename,x).Run();
    else {
        #ifdef COMPILER
        auto code = AST2MC(filename, x, memoryorigin).ReturnProg();
        cout << "Opcode:" << endl;
        for (auto c : code) {
            if (c.opcode.has_value())
                cout << hex << c.opcode.value() << endl;
        }
        #else
            cerr << "Compiler option does not support. Consider enable compiler support and rebuild this program." <<endl;
            return 1;
        #endif
    }

    return 0;
}