public override IPExpr VisitFunCallExpr(PParser.FunCallExprContext context)
{
string funName = context.fun.GetText();
if (!table.Lookup(funName, out Function function))
{
throw handler.MissingDeclaration(context.fun, "function", funName);
}
// Check the arguments
IPExpr[] arguments = TypeCheckingUtils.VisitRvalueList(context.rvalueList(), this).ToArray();
ISet <Variable> linearVariables = new HashSet <Variable>();
if (function.Signature.Parameters.Count != arguments.Length)
{
throw handler.IncorrectArgumentCount(context, arguments.Length, function.Signature.Parameters.Count);
}
for (int i = 0; i < arguments.Length; i++)
{
IPExpr argument = arguments[i];
PLanguageType paramType = function.Signature.Parameters[i].Type;
if (!paramType.IsAssignableFrom(argument.Type))
{
throw handler.TypeMismatch(context.rvalueList().rvalue(i), argument.Type, paramType);
}
if (argument is ILinearRef linearRef)
{
if (linearRef.LinearType == LinearType.Swap && !linearRef.Type.IsSameTypeAs(paramType))
{
throw handler.TypeMismatch(context, linearRef.Type, paramType);
}
if (linearVariables.Contains(linearRef.Variable))
{
throw handler.RelinquishedWithoutOwnership(linearRef);
}
linearVariables.Add(linearRef.Variable);
}
}
method.AddCallee(function);
return(new FunCallExpr(context, function, arguments));
}
public override IPExpr VisitFunCallExpr(PParser.FunCallExprContext context)
{
string funName = context.fun.GetText();
if (!table.Lookup(funName, out Function function))
{
throw handler.MissingDeclaration(context.fun, "function", funName);
}
// Check the arguments
// TODO: linearly typed arguments
var arguments = (context.rvalueList()?.rvalue().Select(Visit) ?? Enumerable.Empty <IPExpr>()).ToArray();
ISet <Variable> linearVariables = new HashSet <Variable>();
for (var i = 0; i < arguments.Length; i++)
{
IPExpr argument = arguments[i];
PLanguageType paramType = function.Signature.Parameters[i].Type;
if (!paramType.IsAssignableFrom(argument.Type))
{
throw handler.TypeMismatch(context.rvalueList().rvalue(i), argument.Type, paramType);
}
if (argument is ILinearRef linearRef)
{
if (linearRef.LinearType == LinearType.Swap && !linearRef.Type.IsSameTypeAs(paramType))
{
throw handler.TypeMismatch(context, linearRef.Type, paramType);
}
if (linearVariables.Contains(linearRef.Variable))
{
throw handler.RelinquishedWithoutOwnership(linearRef);
}
linearVariables.Add(linearRef.Variable);
}
}
return(new FunCallExpr(context, function, arguments));
}
public override object VisitFunCallExpr(PParser.FunCallExprContext context)
{
return(null);
}
