OrdinaryMagician/vkdoom_m
OrdinaryMagician/vkdoom_m
1
Fork 0
Code Activity

Multi-Assign Declarations

Browse source
This commit is contained in:
Ricardo Luís Vaz Silva 2022-12-20 10:44:20 -03:00 • committed by Christoph Oelckers
commit c8fe074041
7 changed files with 144 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

69
src/common/scripting/backend/codegen.cpp
View file

@ -2718,6 +2718,75 @@ ExpEmit FxMultiAssign::Emit(VMFunctionBuilder *build)
return LocalVarContainer->Emit(build);
}
//==========================================================================
//
//
//
//==========================================================================
FxMultiAssignDecl::FxMultiAssignDecl(FArgumentList &base, FxExpression *right, const FScriptPosition &pos)
:FxExpression(EFX_MultiAssign, pos)
{
Base = std::move(base);
Right = right;
}
//==========================================================================
//
//
//
//==========================================================================
FxMultiAssignDecl::~FxMultiAssignDecl()
{
SAFE_DELETE(Right);
}
//==========================================================================
//
//
//
//==========================================================================
FxExpression *FxMultiAssignDecl::Resolve(FCompileContext &ctx)
{
CHECKRESOLVED();
SAFE_RESOLVE(Right, ctx);
if (Right->ExprType != EFX_VMFunctionCall)
{
Right->ScriptPosition.Message(MSG_ERROR, "Function call expected on right side of multi-assigment");
delete this;
return nullptr;
}
auto VMRight = static_cast<FxVMFunctionCall *>(Right);
auto rets = VMRight->GetReturnTypes();
if (Base.Size() == 1)
{
Right->ScriptPosition.Message(MSG_ERROR, "Multi-assignment with only one element in function %s", VMRight->Function->SymbolName.GetChars());
delete this;
return nullptr;
}
if (rets.Size() < Base.Size())
{
Right->ScriptPosition.Message(MSG_ERROR, "Insufficient returns in function %s", VMRight->Function->SymbolName.GetChars());
delete this;
return nullptr;
}
FxSequence * DeclAndAssign = new FxSequence(ScriptPosition);
const unsigned int n = Base.Size();
for (unsigned int i = 0; i < n; i++)
{
assert(Base[i]->ExprType == EFX_Identifier);
DeclAndAssign->Add(new FxLocalVariableDeclaration(rets[i], ((FxIdentifier*)Base[i])->Identifier, nullptr, 0, Base[i]->ScriptPosition));
}
DeclAndAssign->Add(new FxMultiAssign(Base, Right, ScriptPosition));
Right = nullptr;
delete this;
return DeclAndAssign->Resolve(ctx);
}
//==========================================================================
//
//