Function ResolveGlobalCustomFunction

source/core/codegen_raze.cpp:168–327

bool CheckArgSize(FName fname, FArgumentList &args, int min, int max, FScriptPosition &sc);

static FxExpression *ResolveGlobalCustomFunction(FxFunctionCall *func, FCompileContext &ctx)
{
	auto& ScriptPosition = func->ScriptPosition;
	if (func->MethodName == NAME_GetDefaultByType)
	{
		if (CheckArgSize(NAME_GetDefaultByType, func->ArgList, 1, 1, ScriptPosition))
		{
			auto newfunc = new FxGetDefaultByType(func->ArgList[0]);
			func->ArgList[0] = nullptr;
			delete func;
			return newfunc->Resolve(ctx);
		}
	}
	// the following 3 are just for compile time argument validation, they do not affect code generation.
	else if (func->MethodName == NAME_SetAction && isDukeEngine())
	{
		auto cls = ctx.Function->Variants[0].SelfClass;
		if (cls == nullptr || !cls->isClass()) return func;
		auto clas = static_cast<PClassType*>(cls)->Descriptor;
		if (!clas->IsDescendantOf(RUNTIME_CLASS(Duke3d::DDukeActor))) return func;
		if (CheckArgSize(NAME_SetAction, func->ArgList, 1, 1, ScriptPosition))
		{
			FxExpression* x;
			x = func->ArgList[0] = func->ArgList[0]->Resolve(ctx);
			if (x && x->isConstant() && (x->ValueType == TypeName || x->ValueType == TypeString))
			{
				auto c = static_cast<FxConstant*>(x);
				auto nm = c->GetValue().GetName();
				if (nm == NAME_None) return func;
				int pos = Duke3d::LookupAction(clas, nm);
				if (pos == 0)
				{
					ScriptPosition.Message(MSG_ERROR, "Invalid action '%s'", nm.GetChars());
					delete func;
					return nullptr;
				}
			}
		}
	}
	else if (func->MethodName == NAME_SetAI && isDukeEngine())
	{
		auto cls = ctx.Function->Variants[0].SelfClass;
		if (cls == nullptr || !cls->isClass()) return func;
		auto clas = static_cast<PClassType*>(cls)->Descriptor;
		if (!clas->IsDescendantOf(RUNTIME_CLASS(Duke3d::DDukeActor))) return func;
		if (CheckArgSize(NAME_SetAI, func->ArgList, 1, 1, ScriptPosition))
		{
			FxExpression* x;
			x = func->ArgList[0] = func->ArgList[0]->Resolve(ctx);
			if (x && x->isConstant() && (x->ValueType == TypeName || x->ValueType == TypeString))
			{
				auto c = static_cast<FxConstant*>(x);
				auto nm = c->GetValue().GetName();
				if (nm == NAME_None) return func;
				int pos = Duke3d::LookupAI(clas, nm);
				if (pos == 0)
				{
					ScriptPosition.Message(MSG_ERROR, "Invalid AI '%s'", nm.GetChars());