#include "leak_detector.h"
#include "VarUsages.h"

using namespace std;

void VarUsages::extend(const VarUsages& to_insert)
{
    undefined |= to_insert.undefined;

    reads.insert(to_insert.reads.begin(), to_insert.reads.end());
    writes.insert(to_insert.writes.begin(), to_insert.writes.end());
}

set<SgSymbol*> VarUsages::get_reads(const set<VAR_TYPE> var_type) const
{
    if (undefined)
        printInternalError(convertFileName(__FILE__).c_str(), __LINE__);

    return filter(reads, var_type);
}

set<SgSymbol*> VarUsages::get_writes(const set<VAR_TYPE> var_type) const
{
    if (undefined)
        printInternalError(convertFileName(__FILE__).c_str(), __LINE__);

    return filter(writes, var_type);
}

set<SgSymbol*> VarUsages::get_all(const set<VAR_TYPE> var_type) const
{
    set<TypedSymbol> all_usages;
    all_usages.insert(reads.begin(), reads.end());
    all_usages.insert(writes.begin(), writes.end());

    return filter(all_usages, var_type);
}

set<SgSymbol*> VarUsages::get_reads() const { return get_reads({ VAR_TYPE::VAR_ANY }); }
set<SgSymbol*> VarUsages::get_writes() const { return get_writes({ VAR_TYPE::VAR_ANY }); }
set<SgSymbol*> VarUsages::get_all() const { return get_all({ VAR_TYPE::VAR_ANY }); }

set<SgSymbol*> VarUsages::filter(const set<TypedSymbol> &symbols, const set<VAR_TYPE> var_type)
{
    set<SgSymbol*> filtered;
    if (var_type.find(VAR_TYPE::VAR_ANY) != var_type.end())
        for (auto& s : symbols)
            filtered.insert(s.orig);
    else
        for (auto& s : symbols)
            if (var_type.find(s.type) != var_type.end())
                filtered.insert(s.orig);
    return filtered;
}

void VarUsages::insert_undefined(const TypedSymbol& s)
{
    undefined = true;

    reads.insert(s);
    writes.insert(s);
}

void VarUsages::insert_read(const TypedSymbol& s) { reads.insert(s); }
void VarUsages::insert_write(const TypedSymbol& s) { writes.insert(s); }

void VarUsages::print() const
{
    printf("reads: ");
    if (undefined)
        printf("not_defined");
    else
    {
        for (auto& s : reads)
            printf("%s ", s.orig->identifier());
    }
    printf("\n");

    printf("writes: ");
    if (undefined)
        printf("not_defined");
    else
    {
        for (auto& s : writes)
            printf("%s ", s.orig->identifier());
    }
    printf("\n");

    printf("all: ");
    for (auto& s : get_all({ VAR_TYPE::VAR_ANY }))
        printf("%s ", s->identifier());
    printf("\n");

    printf("***\n");
}

bool VarUsages::is_undefined() const { return undefined; }