Skip to content
GitLab
Select Git revision
  • main default
Find file BlameHistoryPermalink
Driver.cpp 3.97 KiB
1 
#include "CodeGen/AsmPrinter.hpp"
2 
#include "CodeGen/FrameLowering.hpp"
3 
#include "CodeGen/MEISel.hpp"
4 
#include "CodeGen/SimpleRA.hpp"
5 
#include "Parser/Parser.hpp"
6 
#include "Pass/ConstantFolding.hpp"
7 
#include "Pass/DeadCodeElimination.hpp"
8 
#include "Pass/Dump.hpp"
9 
#include "Pass/IRLegalize.hpp"
10 
#include "Pass/InstCombine.hpp"
11 
#include "Pass/MachineDCE.hpp"
12 
#include "Pass/PassCollection.hpp"
13 
#include "Pass/PeepHole.hpp"
14 
#include "Pass/SimplifyCFG.hpp"
15 
#include "Transformer/Transformer.hpp"
16 
#include "Tree/Tree.hpp"
17 
#include "Util/OptionParser.hpp"
18 
#include "Util/RuntimeStackUtil.hpp"
19 
#include "Util/TrivialValueVector.hpp"
20 
#include <fmt/core.h>
21 
#include <fstream>
22 
#include <string_view>
23
24 
using namespace std;
25
26 
int main(int argc, char *argv[]) {
27 
  RuntimeStack::set_max_size(RuntimeStack::hard_max_size());
28 
  OptionParser optParser;
29 
  optParser.add(Option<bool>("--help", "-h").setDefault("false"),
30 
                Option<bool>("--version", "-v").setDefault("false"),
31 
                Option<bool>("--debug-opt-parser").setDefault("false"),
32 
                Option<bool>("--implicit-runtime").setDefault("true"),
33 
                Option<bool>("-S").setDefault("true"),
34 
                Option<bool>("-O2").setDefault("true"),
35 
                Option<std::string_view>("--output", "-o"),
36 
                Option<std::string_view>("filename"));
37
38 
  optParser.parse(argc, argv);
39
40 
  if (optParser["--debug-opt-parser"].as<bool>()) {
41 
    if (optParser.has("-h"))
42 
      fmt::print("-h: {}\n", optParser["-h"].as<bool>());
43 
    if (optParser.has("-v"))
44 
      fmt::print("-v: {}\n", optParser["-v"].as<bool>());
45 
    if (optParser.has("filename"))
46 
      fmt::print("p[0]: {}\n", optParser["filename"].as<std::string_view>());
47 
    if (optParser.has("-o"))
48 
      fmt::print("-o: {}\n", optParser["-o"].as<std::string_view>());
49 
    return 0;
50 
  }
51
52 
  if (optParser["-h"].as<bool>()) {
53 
    fmt::print("usage: syoc [filename] [(-o | --output) filename]  [-h | "
54 
               "--help] [-v | --version]\n");
55 
    return 0;
56 
  }
57
58 
  if (optParser["-v"].as<bool>()) {
59 
    fmt::print("syoc version 0.1.0\n");
60 
    return 0;
61 
  }
62
63 
  string fileName;
64 
  string fileContent;
65 
  fileName = optParser["filename"].as<std::string_view>();
66 
  if (fileName.empty())
67 
    getline(ifstream("current.txt"), fileName, '\0');
68
69 
  ifstream inputStream(fileName);
70 
  assert(inputStream.good());
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
getline(inputStream, fileContent, '\0'); if (optParser["--implicit-runtime"].as<bool>()) { fileContent = R"( int getint(),getch(),getarray(int a[]); float getfloat(); int getfarray(float a[]); void putint(int a),putch(int a),putarray(int n,int a[]); void putfloat(float a); void putfarray(int n, float a[]); void starttime(); void stoptime(); int __aeabi_idivmod(int a, int b); int __aeabi_idiv(int a, int b); void _sysy_starttime(int lineno); void _sysy_stoptime(int lineno); )" + fileContent; } SyOC::Parser parser(fileContent); parser.tokenize(); auto *tree = parser.parse(); SyOC::Transformer transformer(tree); transformer .doTreeTransformation<SyOC::ConstantInitializerFold, SyOC::TypeCheck>(); // from tree gen ssa ir transformer.doTree2SSATransformation<SyOC::Tree2SSA>(); // opt passes transformer.doSSATransformation< SyOC::SimplifyCFG, SyOC::SimpleAllocationElimination, SyOC::IRDump, // SyOC::PromoteMem2Reg, SyOC::InstCombine, SyOC::ConstantFolding, SyOC::SimplifyCFG, SyOC::SimpleAllocationElimination, SyOC::DeadCodeElimination, SyOC::IRLegalize, SyOC::FixTimeMeasurement, SyOC::IRDump, SyOC::CFGDump>(); // instruction selection SyOC::ARMv7a::AsmPrinter out; transformer.doSSA2MInstTransformation<SyOC::MEISel>(); out.print("mir.s", transformer.getMIR()); transformer.doMInstTransformation< SyOC::ARMv7a::SimpleRA, SyOC::ARMv7a::PeepHole, SyOC::ARMv7a::MachineDCE, SyOC::ARMv7a::FrameLowering>(); std::string asmFileName; if (optParser.has("-o")) { asmFileName = optParser["-o"].as<std::string_view>(); } else { asmFileName = fileName + ".s"; } out.print(asmFileName, transformer.getMIR()); return 0; }