/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <set>
#include "emit.h"

namespace maplebe {
using namespace maple;

uint32 VregInfo::virtualRegCount = kBaseVirtualRegNO;
uint32 VregInfo::maxRegCount = 0;
std::vector<VirtualRegNode> VregInfo::vRegTable;
std::unordered_map<regno_t, RegOperand *> VregInfo::vRegOperandTable;
/*  There are two builders, cgfunc builder (original code selector) and irbuilder (abstract).
 *  This is to prevent conflict between the two for VregInfo as for arm64 both co-exists.
 *  When switching to irbuilder completely, then this bool can go away.
 */
bool VregInfo::initialized = false;

void Globals::SetTarget(CG &target)
{
    cg = &target;
}
const CG *Globals::GetTarget() const
{
    DEBUG_ASSERT(cg, " set target info please ");
    return cg;
}

CGFunc *CG::currentCGFunction = nullptr;
std::map<MIRFunction *, std::pair<LabelIdx, LabelIdx>> CG::funcWrapLabels;

CG::~CG()
{
    Emit([](Emitter *emitter) {
        emitter->CloseOutput();
    });
    delete memPool;
    memPool = nullptr;
    mirModule = nullptr;
    emitters.clear();
    currentCGFunction = nullptr;
    instrumentationFunction = nullptr;
    dbgTraceEnter = nullptr;
    dbgTraceExit = nullptr;
    dbgFuncProfile = nullptr;
}
/* This function intends to be a more general form of GenFieldOffsetmap. */
void CG::GenExtraTypeMetadata(const std::string &classListFileName, const std::string &outputBaseName)
{
    const std::string &cMacroDefSuffix = ".macros.def";
    BECommon *beCommon = Globals::GetInstance()->GetBECommon();
    std::vector<MIRClassType *> classesToGenerate;

    if (classListFileName.empty()) {
        /*
         * Class list not specified. Visit all classes.
         */
        std::set<std::string> visited;

        for (const auto &tyId : mirModule->GetClassList()) {
            MIRType *mirType = GlobalTables::GetTypeTable().GetTypeFromTyIdx(tyId);
            if ((mirType->GetKind() != kTypeClass) && (mirType->GetKind() != kTypeClassIncomplete)) {
                continue; /* Skip non-class. Too paranoid. We just enumerated classlist_! */
            }
            MIRClassType *classType = static_cast<MIRClassType *>(mirType);
            const std::string &name = classType->GetName();

            if (visited.find(name) != visited.end()) {
                continue; /* Skip duplicated class definitions. */
            }

            (void)visited.insert(name);
            classesToGenerate.emplace_back(classType);
        }
    } else {
        /* Visit listed classes. */
        std::ifstream inFile(classListFileName);
        CHECK_FATAL(inFile.is_open(), "Failed to open file: %s", classListFileName.c_str());
        std::string str;

        /* check each class name first and expose all unknown classes */
        while (inFile >> str) {
            MIRType *type = GlobalTables::GetTypeTable().GetOrCreateClassType(str, *mirModule);
            MIRClassType *classType = static_cast<MIRClassType *>(type);
            if (classType == nullptr) {
                LogInfo::MapleLogger() << " >>>>>>>> unknown class: " << str.c_str() << "\n";
                return;
            }

            classesToGenerate.emplace_back(classType);
        }
    }

    if (cgOption.GenDef()) {
        const std::string &outputFileName = outputBaseName + cMacroDefSuffix;
        FILE *outputFile = fopen(outputFileName.c_str(), "w");
        if (outputFile == nullptr) {
            FATAL(kLncFatal, "open file failed in CG::GenExtraTypeMetadata");
        }

        for (auto classType : classesToGenerate) {
            beCommon->GenObjSize(*classType, *outputFile);
            beCommon->GenFieldOffsetMap(*classType, *outputFile);
        }
        fclose(outputFile);
    }

    if (cgOption.GenGctib()) {
        maple::LogInfo::MapleLogger(kLlErr) << "--gen-gctib-file option not implemented";
    }
}

void CG::GenPrimordialObjectList(const std::string &outputBaseName)
{
    const std::string &kPrimorListSuffix = ".primordials.txt";
    if (!cgOption.GenPrimorList()) {
        return;
    }

    const std::string &outputFileName = outputBaseName + kPrimorListSuffix;
    FILE *outputFile = fopen(outputFileName.c_str(), "w");
    if (outputFile == nullptr) {
        FATAL(kLncFatal, "open file failed in CG::GenPrimordialObjectList");
    }

    for (StIdx stIdx : mirModule->GetSymbolSet()) {
        MIRSymbol *symbol = GlobalTables::GetGsymTable().GetSymbolFromStidx(stIdx.Idx());
        DEBUG_ASSERT(symbol != nullptr, "get symbol from st idx failed");
        if (symbol->IsPrimordialObject()) {
            const std::string &name = symbol->GetName();
            fprintf(outputFile, "%s\n", name.c_str());
        }
    }

    fclose(outputFile);
}

void CG::AddStackGuardvar()
{
    MIRSymbol *chkGuard = GlobalTables::GetGsymTable().CreateSymbol(kScopeGlobal);
    chkGuard->SetNameStrIdx(std::string("__stack_chk_guard"));
    chkGuard->SetStorageClass(kScExtern);
    chkGuard->SetSKind(kStVar);
    CHECK_FATAL(GlobalTables::GetTypeTable().GetTypeTable().size() > PTY_u64, "out of vector range");
    chkGuard->SetTyIdx(GlobalTables::GetTypeTable().GetTypeTable()[PTY_u64]->GetTypeIndex());
    GlobalTables::GetGsymTable().AddToStringSymbolMap(*chkGuard);

    MIRFunction *func = GetMIRModule()->GetMIRBuilder()->GetOrCreateFunction(
        "__stack_chk_fail", GlobalTables::GetTypeTable().GetVoid()->GetTypeIndex());
    MIRSymbol *chkFuncSym = func->GetFuncSymbol();
    DEBUG_ASSERT(chkFuncSym != nullptr, "nullptr check");
    chkFuncSym->SetAppearsInCode(true);
    chkFuncSym->SetStorageClass(kScExtern);
    GlobalTables::GetGsymTable().AddToStringSymbolMap(*chkFuncSym);
}

void CG::SetInstrumentationFunction(const std::string &name)
{
    instrumentationFunction = GlobalTables::GetGsymTable().CreateSymbol(kScopeGlobal);
    instrumentationFunction->SetNameStrIdx(std::string("__").append(name).append("__"));
    instrumentationFunction->SetStorageClass(kScText);
    instrumentationFunction->SetSKind(kStFunc);
}

#define DBG_TRACE_ENTER MplDtEnter
#define DBG_TRACE_EXIT MplDtExit
#define XSTR(s) str(s)
#define str(s) #s

void CG::DefineDebugTraceFunctions()
{
    dbgTraceEnter = GlobalTables::GetGsymTable().CreateSymbol(kScopeGlobal);
    dbgTraceEnter->SetNameStrIdx(std::string("__" XSTR(DBG_TRACE_ENTER) "__"));
    dbgTraceEnter->SetStorageClass(kScText);
    dbgTraceEnter->SetSKind(kStFunc);

    dbgTraceExit = GlobalTables::GetGsymTable().CreateSymbol(kScopeGlobal);
    dbgTraceExit->SetNameStrIdx(std::string("__" XSTR(DBG_TRACE_EXIT) "__"));
    dbgTraceExit->SetStorageClass(kScText);
    dbgTraceExit->SetSKind(kStFunc);

    dbgFuncProfile = GlobalTables::GetGsymTable().CreateSymbol(kScopeGlobal);
    dbgFuncProfile->SetNameStrIdx(std::string("__" XSTR(MplFuncProfile) "__"));
    dbgFuncProfile->SetStorageClass(kScText);
    dbgFuncProfile->SetSKind(kStFunc);
}

/*
 * Add the fields of curStructType to the result. Used to handle recursive
 * structures.
 */
static void AppendReferenceOffsets64(const BECommon &beCommon, MIRStructType &curStructType, int64 &curOffset,
                                     std::vector<int64> &result)
{
    /*
     * We are going to reimplement BECommon::GetFieldOffset so that we can do
     * this in one pass through all fields.
     *
     * The tricky part is to make sure the object layout described here is
     * compatible with the rest of the system. This implies that we need
     * something like a "Maple ABI" documented for each platform.
     */
    if (curStructType.GetKind() == kTypeClass) {
        MIRClassType &curClassTy = static_cast<MIRClassType &>(curStructType);
        auto maybeParent = GlobalTables::GetTypeTable().GetTypeFromTyIdx(curClassTy.GetParentTyIdx());
        if (maybeParent != nullptr) {
            if (maybeParent->GetKind() == kTypeClass) {
                auto parentClassType = static_cast<MIRClassType *>(maybeParent);
                AppendReferenceOffsets64(beCommon, *parentClassType, curOffset, result);
            } else {
                LogInfo::MapleLogger() << "WARNING:: generating objmap for incomplete class\n";
            }
        }
    }

    for (const auto &fieldPair : curStructType.GetFields()) {
        auto fieldNameIdx = fieldPair.first;
        auto fieldTypeIdx = fieldPair.second.first;

        auto &fieldName = GlobalTables::GetStrTable().GetStringFromStrIdx(fieldNameIdx);
        auto fieldType = GlobalTables::GetTypeTable().GetTypeFromTyIdx(fieldTypeIdx);
        auto &fieldTypeName = GlobalTables::GetStrTable().GetStringFromStrIdx(fieldType->GetNameStrIdx());
        auto fieldTypeKind = fieldType->GetKind();

        auto fieldSize = beCommon.GetTypeSize(fieldTypeIdx);
        auto fieldAlign = beCommon.GetTypeAlign(fieldTypeIdx);
        int64 myOffset = static_cast<int64>(RoundUp(curOffset, fieldAlign));
        int64 nextOffset = myOffset + static_cast<int64>(fieldSize);

        if (!CGOptions::IsQuiet()) {
            LogInfo::MapleLogger() << "    field: " << fieldName << "\n";
            LogInfo::MapleLogger() << "      type: " << fieldTypeIdx << ": " << fieldTypeName << "\n";
            LogInfo::MapleLogger() << "      type kind: " << fieldTypeKind << "\n";
            LogInfo::MapleLogger() << "      size: " << fieldSize << "\n";                        /* int64 */
            LogInfo::MapleLogger() << "      align: " << static_cast<uint32>(fieldAlign) << "\n"; /* int8_t */
            LogInfo::MapleLogger() << "      field offset:" << myOffset << "\n";                  /* int64 */
        }

        if (fieldTypeKind == kTypePointer) {
            if (!CGOptions::IsQuiet()) {
                LogInfo::MapleLogger() << "      ** Is a pointer field.\n";
            }
            result.emplace_back(myOffset);
        }

        if ((fieldTypeKind == kTypeArray) || (fieldTypeKind == kTypeStruct) || (fieldTypeKind == kTypeClass) ||
            (fieldTypeKind == kTypeInterface)) {
            if (!CGOptions::IsQuiet()) {
                LogInfo::MapleLogger() << "    ** ERROR: We are not expecting nested aggregate type. ";
                LogInfo::MapleLogger() << "All J classes are flat -- no nested structs. ";
                LogInfo::MapleLogger() << "Please extend me if we are going to work with non-j languages.\n";
            }
        }

        curOffset = nextOffset;
    }
}

/* Return a list of offsets of reference fields. */
std::vector<int64> CG::GetReferenceOffsets64(const BECommon &beCommon, MIRStructType &structType)
{
    std::vector<int64> result;
    if (structType.GetKind() == kTypeClass) {
        for (auto fieldInfo : beCommon.GetJClassLayout(static_cast<MIRClassType &>(structType))) {
            if (fieldInfo.IsRef()) {
                result.emplace_back(static_cast<int64>(fieldInfo.GetOffset()));
            }
        }
    } else if (structType.GetKind() != kTypeInterface) { /* interface doesn't have reference fields */
        int64 curOffset = 0;
        AppendReferenceOffsets64(beCommon, structType, curOffset, result);
    }

    return result;
}

const std::string CG::ExtractFuncName(const std::string &str)
{
    /* 3: length of "_7C" */
    size_t offset = 3;
    size_t pos1 = str.find("_7C");
    if (pos1 == std::string::npos) {
        return str;
    }
    size_t pos2 = str.find("_7C", pos1 + offset);
    if (pos2 == std::string::npos) {
        return str;
    }
    std::string funcName = str.substr(pos1 + offset, pos2 - pos1 - offset);
    /* avoid funcName like __LINE__ and __FILE__ which will be resolved by assembler */
    if (funcName.find("__") != std::string::npos) {
        return str;
    }
    if (funcName == "_3Cinit_3E") {
        return "init";
    }
    if (funcName == "_3Cclinit_3E") {
        return "clinit";
    }
    return funcName;
}

void CG::EmitAllEmitters(const std::function<void(Emitter *)>& cb) const
{
    DEBUG_ASSERT(!emitters.empty(), "Emitter were not set");
    DEBUG_ASSERT(emitters.size() <= 2U, "Emitters number isn't correct");
    for (auto emitter: emitters) {
        cb(emitter);
    }
}

void CG::EmitAsmEmitters(const std::function<void(Emitter *)>& cb) const
{
    if (emitters.size() == 2U) {
        cb(emitters[1]);
    }
}

void CG::EmitObjEmitters(const std::function<void(Emitter *)>& cb) const
{
    DEBUG_ASSERT(!emitters.empty(), "ObjEmmiter wasn't set");
    cb(emitters[0]);
}

} /* namespace maplebe */