koop.cc

#include "koop.hh"
using namespace std;

int name_counter = 0;


Value::Value(string instx, string return_v, int num)
{
    inst = instx;
    return_value = return_v;
    space_num = num;
}

Value::Value(string instx, string target, string left, string right, int num)
{
    inst = instx;
    target_reg = target;
    left_reg = left;
    right_reg = right;
    space_num = num;
}


void Value::Dump(string &s)
{
    for (int i=0;i<4*space_num;i++)
        s+=" ";
    if (inst == "ret")
        s += (inst+" "+return_value+'\n');
    else
        s += (target_reg+" = "+inst+" "+left_reg+", "+right_reg+'\n');
}



void BasicBlock::addValue(BlockAST* block)
{
    //BlockAST* blockAST = dynamic_cast<BlockAST*>(block.get());
    StmtAST* stmt = dynamic_cast<StmtAST*>(block->stmt.get());
    auto return_value = addValue(stmt);
    Values.emplace_back(Value("ret",return_value,space_num + 1));
}

string BasicBlock::addValue(StmtAST* stmt)
{
    ExpAST* exp = dynamic_cast<ExpAST*>(stmt->exp.get());
    return addValue(exp);
}

string BasicBlock::addValue(ExpAST* exp)
{
    AddExpAST* addexp = dynamic_cast<AddExpAST*>(exp->addexp.get());
    return addValue(addexp);
}

string BasicBlock::addValue(UnaryExpAST* unaryexp)
{
    if (unaryexp->type == UnaryExpAST::Type::Primary)
    {
        PrimaryExpAST* primaryexp = dynamic_cast<PrimaryExpAST*>(unaryexp->primary_exp.get());
        return addValue(primaryexp);
    }
    else
    {
        UnaryExpAST* unary_exp = dynamic_cast<UnaryExpAST*>(unaryexp->unary_exp.get());
        string op = unaryexp->unary_op;
        auto res = addValue(unary_exp);
        if (op == "+")
            return res;
        string target = "";