PorygonLang/src/Evaluator/Evaluator.cpp

#include <utility>
#include <memory>
#include "Evaluator.hpp"
#include "EvaluationException.hpp"
#include "../Script.hpp"
#include "EvaluationScope/EvaluationScope.hpp"
#include "EvalValues/ScriptFunctionEvalValue.hpp"
#include "EvalValues/TableEvalValue.hpp"
#include "../Binder/BoundExpressions/BoundTableExpression.hpp"
#include "../TableScriptType.hpp"

using namespace std;

void Evaluator::Evaluate(BoundScriptStatement *statement) {
    this->_evaluationScope = make_shared<EvaluationScope>(this->_scriptData->_scriptVariables, statement->GetDeepestScope());
    EvaluateBlockStatement(statement, false);
}

void Evaluator::EvaluateStatement(BoundStatement *statement) {
    if (this->_hasReturned)
        return;
    switch (statement->GetKind()){
        case BoundStatementKind ::Script: throw; // Should never happen
        case BoundStatementKind ::Block: return this -> EvaluateBlockStatement((BoundBlockStatement*)statement, true);
        case BoundStatementKind ::Expression: return this -> EvaluateExpressionStatement((BoundExpressionStatement*)statement);
        case BoundStatementKind ::Assignment: return this -> EvaluateAssignmentStatement((BoundAssignmentStatement*)statement);
        case BoundStatementKind ::FunctionDeclaration: return this->EvaluateFunctionDeclarationStatement((BoundFunctionDeclarationStatement*)statement);
        case BoundStatementKind::Return: return this -> EvaluateReturnStatement((BoundReturnStatement*)statement);
        case BoundStatementKind::Conditional: return this -> EvaluateConditionalStatement((BoundConditionalStatement*)statement);

        case BoundStatementKind::Bad:
            throw;
    }
}

void Evaluator::EvaluateBlockStatement(BoundBlockStatement* statement, bool clearScope) {
    this->_evaluationScope->OuterScope();
    for (auto s: statement->GetStatements()){
        this -> EvaluateStatement(s);
        if (this->_hasReturned)
            break;
    }
    this->_evaluationScope->InnerScope(clearScope);
}

void Evaluator::EvaluateExpressionStatement(BoundExpressionStatement *statement) {
    // Save new value
    this->_lastValue = this -> EvaluateExpression(statement->GetExpression());
}

void Evaluator::EvaluateAssignmentStatement(BoundAssignmentStatement *statement) {
    auto value = this -> EvaluateExpression(statement->GetExpression());
    auto key = statement->GetKey();
    if (key->IsCreation()){
        this->_evaluationScope->CreateVariable(key->GetScopeId(), key->GetIdentifier(), value);
    } else{
        this->_evaluationScope->SetVariable(key->GetScopeId(), key->GetIdentifier(), value);
    }
}

void Evaluator::EvaluateFunctionDeclarationStatement(BoundFunctionDeclarationStatement *statement) {
    auto type = statement->GetType();
    auto key = statement->GetKey();
    auto block = statement->GetBlock();
    auto evaluator = shared_ptr<EvaluationScope>(this->_evaluationScope->CreateBranchingScope(this->_evaluationScope->GetCurrentScope() + 1));
    auto value = make_shared<ScriptFunctionEvalValue>(block, evaluator, type);
    if (key->IsCreation()){
        this->_evaluationScope->CreateVariable(key->GetScopeId(), key->GetIdentifier(), value);
    } else{
        this->_evaluationScope->SetVariable(key->GetScopeId(), key->GetIdentifier(), value);
    }
}

void Evaluator::EvaluateReturnStatement(BoundReturnStatement* statement){
    auto expression = statement->GetExpression();
    this->_hasReturned = true;
    if (expression == nullptr){
        return;
    }
    auto value = this -> EvaluateExpression(expression);
    this -> _returnValue = value;
}

void Evaluator::EvaluateConditionalStatement(BoundConditionalStatement *statement) {
    auto condition = statement->GetCondition();
    if (EvaluateBoolExpression(condition) -> EvaluateBool()){
        this -> EvaluateStatement(statement->GetBlock());
    } else{
        auto elseStatement = statement -> GetElseStatement();
        if (elseStatement != nullptr){
            this->EvaluateStatement(elseStatement);
        }
    }
}

shared_ptr<EvalValue> Evaluator::EvaluateExpression(BoundExpression *expression) {
    auto type = expression -> GetType();
    switch (type->GetClass()){
        case TypeClass ::Number: return this -> EvaluateIntegerExpression(expression);
        case TypeClass ::Bool: return this -> EvaluateBoolExpression(expression);
        case TypeClass ::String: return this -> EvaluateStringExpression(expression);
        case TypeClass ::Function: return this->EvaluateFunctionExpression(expression);
        case TypeClass ::Nil: return this->EvaluateNilExpression(expression);
        case TypeClass ::Table: return this-> EvaluateTableExpression(expression);
        default: throw;
    }
}

shared_ptr<EvalValue> Evaluator::GetVariable(BoundVariableExpression* expression){
    auto variable = this->_evaluationScope->GetVariable(expression->GetScope(), expression->GetId());
    return variable->Clone();
}

shared_ptr<NumericEvalValue> Evaluator::EvaluateIntegerExpression(BoundExpression *expression) {
    switch (expression->GetKind()){
        case BoundExpressionKind ::LiteralInteger: return make_shared<IntegerEvalValue>(((BoundLiteralIntegerExpression*)expression)->GetValue());
        case BoundExpressionKind ::LiteralFloat: return make_shared<FloatEvalValue>(((BoundLiteralFloatExpression*)expression)->GetValue());
        case BoundExpressionKind::Unary: return this -> EvaluateIntegerUnary((BoundUnaryExpression*)expression);
        case BoundExpressionKind ::Binary: return this -> EvaluateIntegerBinary((BoundBinaryExpression*)expression);
        case BoundExpressionKind::Variable: return dynamic_pointer_cast<NumericEvalValue>(this->GetVariable((BoundVariableExpression*)expression));
        case BoundExpressionKind ::FunctionCall: return dynamic_pointer_cast<NumericEvalValue>(this->EvaluateFunctionCallExpression(expression));
        case BoundExpressionKind ::Index: return dynamic_pointer_cast<NumericEvalValue>(this->EvaluateIndexExpression(expression));

        case BoundExpressionKind ::LiteralString:
        case BoundExpressionKind ::LiteralBool:
        case BoundExpressionKind ::Bad:
        case BoundExpressionKind::NumericalTable:
        case BoundExpressionKind::Table:
            throw;
    }
}

shared_ptr<BooleanEvalValue> Evaluator::EvaluateBoolExpression(BoundExpression *expression) {
    switch (expression->GetKind()) {
        case BoundExpressionKind ::LiteralBool: return make_shared<BooleanEvalValue>(((BoundLiteralBoolExpression*)expression)->GetValue());
        case BoundExpressionKind::Unary: return this -> EvaluateBooleanUnary((BoundUnaryExpression*)expression);
        case BoundExpressionKind::Binary: return this -> EvaluateBooleanBinary((BoundBinaryExpression*)expression);
        case BoundExpressionKind::Variable: return dynamic_pointer_cast<BooleanEvalValue>(this->GetVariable((BoundVariableExpression*)expression));
        case BoundExpressionKind ::FunctionCall: return dynamic_pointer_cast<BooleanEvalValue>(this->EvaluateFunctionCallExpression(expression));
        case BoundExpressionKind ::Index: return dynamic_pointer_cast<BooleanEvalValue>(this->EvaluateIndexExpression(expression));

        case BoundExpressionKind::Bad:
        case BoundExpressionKind::LiteralInteger:
        case BoundExpressionKind::LiteralFloat:
        case BoundExpressionKind::LiteralString:
        case BoundExpressionKind::NumericalTable:
        case BoundExpressionKind::Table:
            throw;

    }
}

shared_ptr<StringEvalValue> Evaluator::EvaluateStringExpression(BoundExpression *expression) {
    switch (expression->GetKind()) {
        case BoundExpressionKind ::LiteralString:
            return make_shared<StringEvalValue>(((BoundLiteralStringExpression*)expression)->GetValue());
        case BoundExpressionKind::Binary:
            return this -> EvaluateStringBinary((BoundBinaryExpression*)expression);
        case BoundExpressionKind::Variable: return dynamic_pointer_cast<StringEvalValue>(this->GetVariable((BoundVariableExpression*)expression));
        case BoundExpressionKind ::FunctionCall: return dynamic_pointer_cast<StringEvalValue>(this->EvaluateFunctionCallExpression(expression));
        case BoundExpressionKind ::Index: return dynamic_pointer_cast<StringEvalValue>(this->EvaluateIndexExpression(expression));

        case BoundExpressionKind::Bad:
        case BoundExpressionKind::LiteralInteger:
        case BoundExpressionKind::LiteralFloat:
        case BoundExpressionKind::LiteralBool:
        case BoundExpressionKind::Unary:
        case BoundExpressionKind::NumericalTable:
        case BoundExpressionKind::Table:
            throw;

    }
}

shared_ptr<EvalValue> Evaluator::EvaluateFunctionExpression(BoundExpression * expression){
    switch (expression->GetKind()){
        case BoundExpressionKind ::Variable: return this->GetVariable((BoundVariableExpression*)expression);
        case BoundExpressionKind ::Index: return this->EvaluateIndexExpression(expression);
        default: throw;
    }
}
shared_ptr<EvalValue> Evaluator::EvaluateNilExpression(BoundExpression * expression){
    switch (expression->GetKind()){
        case BoundExpressionKind ::FunctionCall:
            return this->EvaluateFunctionCallExpression(expression);
        default:
            return nullptr;
    }
}
shared_ptr<EvalValue> Evaluator::EvaluateTableExpression(BoundExpression * expression){
    switch (expression->GetKind()){
        case BoundExpressionKind ::FunctionCall:
            return this->EvaluateFunctionCallExpression(expression);
        case BoundExpressionKind ::Variable: return this->GetVariable((BoundVariableExpression*)expression);
        case BoundExpressionKind ::Index: return this->EvaluateIndexExpression(expression);
        case BoundExpressionKind ::NumericalTable: return this-> EvaluateNumericTableExpression(expression);
        case BoundExpressionKind ::Table: return this -> EvaluateComplexTableExpression(expression);
        default:
            throw;
    }
}




shared_ptr<EvalValue> Evaluator::EvaluateFunctionCallExpression(BoundExpression* expression){
    auto functionCall = (BoundFunctionCallExpression*)expression;
    auto function = dynamic_pointer_cast<ScriptFunctionEvalValue>(this->EvaluateExpression(functionCall->GetFunctionExpression()));

    auto boundParameters = functionCall->GetParameters();
    auto parameters = vector<shared_ptr<EvalValue>>(boundParameters.size());
    for (int i = 0; i < boundParameters.size(); i++){
        parameters[i] = this->EvaluateExpression(boundParameters[i]);
    }

    auto type = std::dynamic_pointer_cast<FunctionScriptType>(function->GetType());
    auto parameterTypes = type->GetParameterTypes();
    auto parameterKeys = type->GetParameterKeys();
    auto scope = type -> GetScopeIndex();
    auto originalScope = this->_evaluationScope;
    this->_evaluationScope = shared_ptr<EvaluationScope>(function -> GetScope() -> CreateBranchingScope(scope));

    for (int i = 0; i < parameterTypes.size() && i < parameterKeys.size() && i < parameters.size(); i++){
        auto parameter = parameters[i];
        auto requiredType = parameterTypes.at(i);
        if (*parameter->GetType() != requiredType.get()){
            throw EvaluationException("Passed wrong type to function.");
        }
        auto key = parameterKeys.at(i);
        this->_evaluationScope->CreateVariable(key->GetScopeId(), key->GetIdentifier(), parameter->Clone());
    }
    this->EvaluateBlockStatement(function->GetInnerBlock().get(), true);

    this->_evaluationScope = originalScope;
    this->_hasReturned = false;
    auto r = this -> _returnValue;
    this -> _returnValue = nullptr;
    return r;
}

shared_ptr<EvalValue> Evaluator::EvaluateFunction(ScriptFunctionEvalValue *function, vector<EvalValue *> parameters) {
    auto type = std::dynamic_pointer_cast<FunctionScriptType>(function->GetType());
    auto parameterTypes = type->GetParameterTypes();
    auto parameterKeys = type->GetParameterKeys();

    auto scope = type -> GetScopeIndex();
    auto originalScope = this->_evaluationScope;
    this->_evaluationScope = shared_ptr<EvaluationScope>(function -> GetScope() -> CreateBranchingScope(scope));

    for (int i = 0; i < parameterTypes.size() && i < parameterKeys.size() && i < parameters.size(); i++){
        auto parameter = parameters[i];
        auto requiredType = parameterTypes.at(i);
        if (*parameter->GetType() != requiredType.get()){
            throw EvaluationException("Passed wrong type to function.");
        }
        auto key = parameterKeys.at(i);
        this->_evaluationScope->CreateVariable(key->GetScopeId(), key->GetIdentifier(), parameter->Clone());
    }
    this->EvaluateBlockStatement(function->GetInnerBlock().get(), true);
    this->_evaluationScope = originalScope;
    this->_hasReturned = false;
    auto r = this -> _returnValue;
    this -> _returnValue = nullptr;
    return r;
}

shared_ptr<EvalValue> Evaluator::EvaluateIndexExpression(BoundExpression *expression) {
    auto indexExpression = (BoundIndexExpression*)expression;
    auto index = this -> EvaluateExpression(indexExpression->GetIndexExpression());
    auto indexable = this -> EvaluateExpression(indexExpression->GetIndexableExpression());
    return indexable -> IndexValue(index.get()) -> Clone();
}

shared_ptr<EvalValue> Evaluator::EvaluateNumericTableExpression(BoundExpression *expression) {
    auto tableExpression = (BoundNumericalTableExpression*)expression;
    auto valueExpressions = tableExpression->GetExpressions();
    auto values = new unordered_map<size_t, shared_ptr<EvalValue>>(valueExpressions.size());
    for (int i = 0; i < valueExpressions.size(); i++){
        auto val = this -> EvaluateExpression(valueExpressions[i]);
        values -> insert({i + 1, val});
    }
    auto valuesPointer = shared_ptr<unordered_map<size_t, shared_ptr<EvalValue>>>(values);
    return make_shared<TableEvalValue>(valuesPointer, tableExpression->GetType());
}

shared_ptr<EvalValue> Evaluator::EvaluateComplexTableExpression(BoundExpression *expression) {
    auto tableExpression = (BoundTableExpression*)expression;
    auto type = dynamic_pointer_cast<TableScriptType>(tableExpression->GetType());
    auto declaredVars = type -> GetValues();
    auto variables = make_shared<unordered_map<size_t, shared_ptr<EvalValue>>>(declaredVars->size());
    for (auto i : *declaredVars){
        variables->insert({i.first, nullptr});
    }
    auto evaluator = make_shared<EvaluationScope>(variables.get(), type -> GetDeepestScope());
    auto currentEvaluator = this -> _evaluationScope;
    this -> _evaluationScope = evaluator;
    this -> EvaluateBlockStatement(tableExpression->GetBlock(), false);
    this -> _evaluationScope = currentEvaluator;
    return shared_ptr<TableEvalValue>(new TableEvalValue(variables, tableExpression->GetType()));
}