#include "scope.h"

Scope::~Scope()
{
    delete this->topASTNode;
    for(auto i:this->functionList)
        delete i.second;
    for(auto i:this->variableList)
        delete i.second;
    for(Scope* i:this->sonScope)
        delete i;
    if(fatherScope)
        for(auto i = fatherScope->sonScope.begin(); i < fatherScope->sonScope.end(); i++)
        {
            if(*i == this)
                fatherScope->sonScope.erase(i);
        }
}

Variable* Scope::addVariable(string name)
{
    Variable* var=new Variable();
#ifdef READABLEGEN
    var->NAME=name;
#endif
    this->variableList[name]=var;
    return var;
}

void Scope::addVariable(string name, Variable *var)
{
#ifdef READABLEGEN
    var->NAME=name;
#endif
    this->variableList[name]=var;
}

void Scope::addFunction(string name, Function *fun)
{
#ifdef READABLEGEN
    fun->NAME=name;
#endif
    this->functionList[name]=fun;
}

/*void Scope::deleteFunction(string name)
{
    delete this->functionList[name];
    this->functionList.erase(name);
}

void Scope::deleteVariable(string name)
{
    delete this->variableList[name];
    this->variableList.erase(name);
}*/

void Scope::deleteVariable(Variable *var)
{
    auto p = variableList.begin();
    while(p != variableList.end())
    {
        if(p->second == var)
            variableList.erase(p++);
        else
            p++;
    }
}

void Scope::deleteFunction(Function *fun)
{
    auto p = functionList.begin();
    while(p != functionList.end())
    {
        if(p->second == fun)
            functionList.erase(p++);
        else
            p++;
    }
}

Variable* Scope::findVariable(string name,bool thisScope)
{
    //冷漠：variableList[name]会在variablList里面创建一个空的name
    Variable* result = nullptr;

    if(variableList.count(name))
        result = variableList[name];

    if(result != nullptr)
        return result;

    if(thisScope)
        return result;

    if(fatherScope == nullptr)
        return nullptr;

    return this->fatherScope->findVariable(name,false);
}

Function* Scope::findFunction(string name,bool thisScope)
{
    Function* result = nullptr;

    if(functionList.count(name))
        result = functionList[name];

    if(result != nullptr)
        return result;

    if(thisScope)
        return result;
    if(fatherScope == nullptr)
        return nullptr;

    return this->fatherScope->findFunction(name,false);
}