Skip to content

Commit

Permalink
- implicitly clear local dynamic arrays
Browse files Browse the repository at this point in the history
  • Loading branch information
alexey-lysiuk authored and coelckers committed Oct 20, 2019
1 parent 1dd08a7 commit ce8b235
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 15 deletions.
38 changes: 24 additions & 14 deletions src/scripting/backend/codegen.cpp
Expand Up @@ -11272,6 +11272,7 @@ FxLocalVariableDeclaration::FxLocalVariableDeclaration(PType *type, FName name,
Name = name;
RegCount = type == TypeVector2 ? 2 : type == TypeVector3 ? 3 : 1;
Init = initval;
clearExpr = nullptr;
}

FxLocalVariableDeclaration::~FxLocalVariableDeclaration()
Expand All @@ -11282,6 +11283,15 @@ FxLocalVariableDeclaration::~FxLocalVariableDeclaration()
FxExpression *FxLocalVariableDeclaration::Resolve(FCompileContext &ctx)
{
CHECKRESOLVED();

if (IsDynamicArray())
{
auto stackVar = new FxStackVariable(ValueType, StackOffset, ScriptPosition);
FArgumentList argsList;
clearExpr = new FxMemberFunctionCall(stackVar, "Clear", argsList, ScriptPosition);
SAFE_RESOLVE(clearExpr, ctx);
}

if (ctx.Block == nullptr)
{
ScriptPosition.Message(MSG_ERROR, "Variable declaration outside compound statement");
Expand Down Expand Up @@ -11476,6 +11486,10 @@ ExpEmit FxLocalVariableDeclaration::Emit(VMFunctionBuilder *build)
}
if (pstr->mDestructor != nullptr) build->ConstructedStructs.Push(this);
}
else if (ValueType->isDynArray())
{
ClearDynamicArray(build);
}
}
return ExpEmit();
}
Expand Down Expand Up @@ -11507,6 +11521,11 @@ void FxLocalVariableDeclaration::Release(VMFunctionBuilder *build)
// For that all local stack variables need to live for the entire execution of a function.
}

void FxLocalVariableDeclaration::ClearDynamicArray(VMFunctionBuilder *build)
{
assert(clearExpr != nullptr);
clearExpr->Emit(build);
}

FxStaticArray::FxStaticArray(PType *type, FName name, FArgumentList &args, const FScriptPosition &pos)
: FxLocalVariableDeclaration(NewArray(type, args.Size()), name, nullptr, VARF_Static|VARF_ReadOnly, pos)
Expand Down Expand Up @@ -11591,7 +11610,6 @@ FxLocalArrayDeclaration::FxLocalArrayDeclaration(PType *type, FName name, FArgum
{
ExprType = EFX_LocalArrayDeclaration;
values = std::move(args);
clearExpr = nullptr;
}

FxExpression *FxLocalArrayDeclaration::Resolve(FCompileContext &ctx)
Expand All @@ -11607,16 +11625,6 @@ FxExpression *FxLocalArrayDeclaration::Resolve(FCompileContext &ctx)
auto elementType = (static_cast<PArray *> (ValueType))->ElementType;
auto elementCount = (static_cast<PArray *> (ValueType))->ElementCount;

// We HAVE to clear dynamic arrays before initializing them
if (IsDynamicArray())
{
FArgumentList argsList;
argsList.Clear();

clearExpr = new FxMemberFunctionCall(stackVar, "Clear", argsList, (const FScriptPosition) ScriptPosition);
SAFE_RESOLVE(clearExpr, ctx);
}

if (values.Size() > elementCount)
{
ScriptPosition.Message(MSG_ERROR, "Initializer contains more elements than the array can contain");
Expand Down Expand Up @@ -11673,9 +11681,11 @@ ExpEmit FxLocalArrayDeclaration::Emit(VMFunctionBuilder *build)
{
assert(!(VarFlags & VARF_Out)); // 'out' variables should never be initialized, they can only exist as function parameters.

if (IsDynamicArray() && clearExpr != nullptr)
const bool isDynamicArray = IsDynamicArray();

if (isDynamicArray)
{
clearExpr->Emit(build);
ClearDynamicArray(build);
}

auto elementSizeConst = build->GetConstantInt(static_cast<PArray *>(ValueType)->ElementSize);
Expand All @@ -11686,7 +11696,7 @@ ExpEmit FxLocalArrayDeclaration::Emit(VMFunctionBuilder *build)
{
ExpEmit emitval = v->Emit(build);

if (IsDynamicArray())
if (isDynamicArray)
{
continue;
}
Expand Down
7 changes: 6 additions & 1 deletion src/scripting/backend/codegen.h
Expand Up @@ -2131,6 +2131,12 @@ class FxLocalVariableDeclaration : public FxExpression
FxExpression *Init;
int VarFlags;
int RegCount;

protected:
FxExpression *clearExpr;

void ClearDynamicArray(VMFunctionBuilder *build);

public:
int StackOffset = -1;
int RegNum = -1;
Expand Down Expand Up @@ -2194,7 +2200,6 @@ class FxLocalArrayDeclaration : public FxLocalVariableDeclaration
{
PType *ElementType;
FArgumentList values;
FxExpression *clearExpr;

public:

Expand Down

0 comments on commit ce8b235

Please sign in to comment.