- added named RNG support to FxFunctionCall.

This looks simpler than duplicating all that stuff and avoids a lot code duplication.
This commit is contained in:
Christoph Oelckers 2016-10-16 01:08:02 +02:00
parent 55f6661c87
commit d6ce60f63a
4 changed files with 29 additions and 9 deletions

View file

@ -3643,11 +3643,30 @@ ExpEmit FxArrayElement::Emit(VMFunctionBuilder *build)
// //
//========================================================================== //==========================================================================
FxFunctionCall::FxFunctionCall(FName methodname, FArgumentList *args, const FScriptPosition &pos) FxFunctionCall::FxFunctionCall(FName methodname, FName rngname, FArgumentList *args, const FScriptPosition &pos)
: FxExpression(pos) : FxExpression(pos)
{ {
MethodName = methodname; MethodName = methodname;
RNG = nullptr;
ArgList = args; ArgList = args;
if (rngname != NAME_None)
{
switch (MethodName)
{
case NAME_Random:
case NAME_FRandom:
case NAME_RandomPick:
case NAME_FRandomPick:
case NAME_Random2:
RNG = FRandom::StaticFindRNG(rngname.GetChars());
break;
default:
pos.Message(MSG_ERROR, "Cannot use named RNGs with %s", MethodName.GetChars());
break;
}
}
} }
//========================================================================== //==========================================================================
@ -3749,7 +3768,7 @@ FxExpression *FxFunctionCall::Resolve(FCompileContext& ctx)
return x->Resolve(ctx); return x->Resolve(ctx);
} }
// Last but not least: Check builtins. The random functions will have to be duplicated for the case where an RNG is specified. // Last but not least: Check builtins. The random functions can take a named RNG if specified.
// Note that for all builtins the used arguments have to be nulled in the ArgList so that they won't get deleted before they get used. // Note that for all builtins the used arguments have to be nulled in the ArgList so that they won't get deleted before they get used.
FxExpression *func = nullptr; FxExpression *func = nullptr;
@ -3759,7 +3778,7 @@ FxExpression *FxFunctionCall::Resolve(FCompileContext& ctx)
case NAME_Random: case NAME_Random:
if (CheckArgSize(NAME_Random, ArgList, 2, 2, ScriptPosition)) if (CheckArgSize(NAME_Random, ArgList, 2, 2, ScriptPosition))
{ {
func = new FxRandom(nullptr, (*ArgList)[0], (*ArgList)[1], ScriptPosition, ctx.FromDecorate); func = new FxRandom(RNG, (*ArgList)[0], (*ArgList)[1], ScriptPosition, ctx.FromDecorate);
(*ArgList)[0] = (*ArgList)[1] = nullptr; (*ArgList)[0] = (*ArgList)[1] = nullptr;
} }
break; break;
@ -3767,7 +3786,7 @@ FxExpression *FxFunctionCall::Resolve(FCompileContext& ctx)
case NAME_FRandom: case NAME_FRandom:
if (CheckArgSize(NAME_FRandom, ArgList, 2, 2, ScriptPosition)) if (CheckArgSize(NAME_FRandom, ArgList, 2, 2, ScriptPosition))
{ {
func = new FxFRandom(nullptr, (*ArgList)[0], (*ArgList)[1], ScriptPosition); func = new FxFRandom(RNG, (*ArgList)[0], (*ArgList)[1], ScriptPosition);
(*ArgList)[0] = (*ArgList)[1] = nullptr; (*ArgList)[0] = (*ArgList)[1] = nullptr;
} }
break; break;
@ -3776,7 +3795,7 @@ FxExpression *FxFunctionCall::Resolve(FCompileContext& ctx)
case NAME_FRandomPick: case NAME_FRandomPick:
if (CheckArgSize(MethodName, ArgList, 1, -1, ScriptPosition)) if (CheckArgSize(MethodName, ArgList, 1, -1, ScriptPosition))
{ {
func = new FxRandomPick(nullptr, *ArgList, MethodName == NAME_FRandomPick, ScriptPosition, ctx.FromDecorate); func = new FxRandomPick(RNG, *ArgList, MethodName == NAME_FRandomPick, ScriptPosition, ctx.FromDecorate);
for (auto &i : *ArgList) i = nullptr; // Ownership of items is transferred to FxMinMax but not ownership of the array itself, so Clear cannot be called here. for (auto &i : *ArgList) i = nullptr; // Ownership of items is transferred to FxMinMax but not ownership of the array itself, so Clear cannot be called here.
} }
break; break;
@ -3784,7 +3803,7 @@ FxExpression *FxFunctionCall::Resolve(FCompileContext& ctx)
case NAME_Random2: case NAME_Random2:
if (CheckArgSize(NAME_Random2, ArgList, 1, 1, ScriptPosition)) if (CheckArgSize(NAME_Random2, ArgList, 1, 1, ScriptPosition))
{ {
func = new FxRandom2(nullptr, (*ArgList)[0], ScriptPosition, ctx.FromDecorate); func = new FxRandom2(RNG, (*ArgList)[0], ScriptPosition, ctx.FromDecorate);
(*ArgList)[0] = nullptr; (*ArgList)[0] = nullptr;
} }
break; break;

View file

@ -894,11 +894,12 @@ typedef TDeletingArray<FxExpression*> FArgumentList;
class FxFunctionCall : public FxExpression class FxFunctionCall : public FxExpression
{ {
FName MethodName; FName MethodName;
FRandom *RNG;
FArgumentList *ArgList; FArgumentList *ArgList;
public: public:
FxFunctionCall(FName methodname, FArgumentList *args, const FScriptPosition &pos); FxFunctionCall(FName methodname, FName rngname, FArgumentList *args, const FScriptPosition &pos);
~FxFunctionCall(); ~FxFunctionCall();
FxExpression *Resolve(FCompileContext&); FxExpression *Resolve(FCompileContext&);
}; };

View file

@ -548,7 +548,7 @@ static FxExpression *ParseExpression0 (FScanner &sc, PClassActor *cls)
while (sc.CheckToken(',')); while (sc.CheckToken(','));
sc.MustGetToken(')'); sc.MustGetToken(')');
} }
return new FxFunctionCall(identifier, args, sc); return new FxFunctionCall(identifier, NAME_None, args, sc);
} }
catch (...) catch (...)
{ {

View file

@ -2290,7 +2290,7 @@ FxExpression *ZCCCompiler::ConvertNode(ZCC_TreeNode *ast)
auto fcall = static_cast<ZCC_ExprFuncCall *>(ast); auto fcall = static_cast<ZCC_ExprFuncCall *>(ast);
assert(fcall->Function->NodeType == AST_ExprID); // of course this cannot remain. Right now nothing more complex can come along but later this will have to be decomposed into 'self' and the actual function name. assert(fcall->Function->NodeType == AST_ExprID); // of course this cannot remain. Right now nothing more complex can come along but later this will have to be decomposed into 'self' and the actual function name.
auto fname = static_cast<ZCC_ExprID *>(fcall->Function)->Identifier; auto fname = static_cast<ZCC_ExprID *>(fcall->Function)->Identifier;
return new FxFunctionCall(fname, ConvertNodeList(fcall->Parameters), *ast); return new FxFunctionCall(fname, NAME_None, ConvertNodeList(fcall->Parameters), *ast);
//return ConvertFunctionCall(fcall->Function, ConvertNodeList(fcall->Parameters), ConvertClass, *ast); //return ConvertFunctionCall(fcall->Function, ConvertNodeList(fcall->Parameters), ConvertClass, *ast);
} }