public ITree Infer()
{
fsScope defaultScope = new fsScope(null);
defaultScope.AddFunction("printf", fsType.GetFunctionType(new List<IfsType>() {
fsType.GetStringType(),
new fsTypeVar(),
fsType.GetFunctionType(null)
}));
defaultScope.SetVarInfo("printf", ScopePositionType.functionClass);
functionsInfos.Add("printf", new fsDerFuncInfo("printf", new List<string>(), 0, 0, ""));
Analyse(tree, defaultScope);
return tree;
}
private IfsType InferFuncCallType(ITree node, fsScope scope)
{
List<IfsType> factualArgsTypes = new List<IfsType>();
string callFunctionName = GetChildByType(node, fsharp_ssParser.NAME).GetChild(0).Text;
ITree args = GetChildByType(node, fsharp_ssParser.ARGS);
for (int i = 0; i < args.ChildCount; i++)
{
ITree currentArg = args.GetChild(i);
IfsType argType = Analyse(currentArg, scope);
//if (argType is fsTypeVar)
//{
// if (!(currentArg is fsTreeNode))
// {
// currentArg = new fsTreeNode(argType);
// }
// fsScope.ScopeVarOrFuncTypeChanged += (args.GetChild(i) as fsTreeNode).ScopeVarOrFuncTypeChangedHandler;
//}
factualArgsTypes.Add(argType);
}
if (scope.GetVarInfo(callFunctionName, false).PositionInScopeType == ScopePositionType.functionArg)
{
fsTreeNode typeFANode = new fsTreeNode(scope.GetVarInfo(callFunctionName, false));
node.AddChild(typeFANode);
List<string> argsNames = new List<string>();
for (int i = 0; i < args.ChildCount; i++)
{
argsNames.Add(args.GetChild(i).Text);
}
fsDerFuncInfo faDerFuncInfo = new fsDerFuncInfo(callFunctionName, argsNames,
0, argsNames.Count, null);
node.AddChild(new fsTreeNode(faDerFuncInfo));
return scope.GetVarType(callFunctionName);
}
if (callFunctionName.Length >= "printf".Length &&
callFunctionName.Substring(0, "printf".Length) == "printf")
{
callFunctionName += $"_{uniqueFuctionId++}";
functionsInfos.Add(callFunctionName, functionsInfos["printf"]);
scope.AddFunction(callFunctionName, fsType.GetFunctionType(factualArgsTypes));
scope.SetVarInfo(callFunctionName, ScopePositionType.functionClass);
}
IfsType formalFunctionType = scope.GetFunctionType(callFunctionName);
if (formalFunctionType == null)
{
formalFunctionType = scope.GetVarType(callFunctionName);
}
if (formalFunctionType == null)
{
throw new Exception($"Undeclared function: {callFunctionName}");
}
IfsType returningType = (formalFunctionType as fsType).Types[(formalFunctionType as fsType).Types.Count - 1];
factualArgsTypes.Add(returningType);
IfsType factualFunctionType = fsType.GetFunctionType(factualArgsTypes);
UnifyWrapper(ref factualFunctionType, ref formalFunctionType, scope);
returningType = (formalFunctionType as fsType).Types[(formalFunctionType as fsType).Types.Count - 1];
fsDerFuncInfo oldFuncInfo;
if (functionsInfos.ContainsKey(callFunctionName))
{
oldFuncInfo = functionsInfos[callFunctionName];
}
else
{
oldFuncInfo = varDerFuncTable[callFunctionName];
}
derFunction = new fsDerFuncInfo(oldFuncInfo.Name,
new List<string>(oldFuncInfo.ArgsNames),
oldFuncInfo.BeforePassedArgsNum,
oldFuncInfo.AfterPassedArgsNum,
oldFuncInfo.ContextFuncName);
if (callFunctionName.Length < "printf".Length ||
callFunctionName.Substring(0, "printf".Length) != "printf")
{
derFunction.BeforePassedArgsNum = derFunction.AfterPassedArgsNum;
derFunction.AfterPassedArgsNum += args.ChildCount;
}
node.AddChild(new fsTreeNode(derFunction));
fsTreeNode typeNode = new fsTreeNode(scope.GetVarInfo(callFunctionName, false));
node.AddChild(typeNode);
return returningType;
}
private IfsType InferFunctionDefnType(ITree node, fsScope scope)
{
fsScope innerScope = new fsScope(scope);
List<IfsType> functionTypes = new List<IfsType>();
string funcName = GetChildByType(node, fsharp_ssParser.NAME).GetChild(0).Text;
ITree args = GetChildByType(node, fsharp_ssParser.ARGS);
List<string> argsNames = new List<string>();
for (int i = 0; i < args.ChildCount; i++)
{
ITree arg = args.GetChild(i);
IfsType argType;
if (arg.ChildCount > 0)
{
ITree annotatedArgTypeNode = arg.GetChild(0);
argType = new fsType(annotatedArgTypeNode.Text, null);
}
else
{
argType = new fsTypeVar();
}
argsNames.Add(arg.Text);
innerScope.AddVar(arg.Text, argType);
innerScope.SetVarInfo(arg.Text, ScopePositionType.functionArg);
}
fsDerFuncInfo funcInfo = new fsDerFuncInfo(funcName, argsNames, 0, 0,
enteredFunctionsNames[enteredFunctionsNames.Count - 1]);
if (funcName != "main")
{
enteredFunctionsNames.Add(funcName);
}
functionsInfos.Add(funcName, funcInfo);
node.AddChild(new fsTreeNode(funcInfo));
if (GetChildByType(node, fsharp_ssParser.REC) != null)
{
innerScope.AddFunction(funcName, fsType.GetIdentityType(innerScope, argsNames));
innerScope.SetVarInfo(funcName, ScopePositionType.functionClass);
}
IfsType bodyType = Analyse(GetChildByType(node, fsharp_ssParser.BODY), innerScope);
ITree annotatedReturningTypeNode = GetChildByType(node, fsharp_ssParser.TYPE);
if (annotatedReturningTypeNode != null && annotatedReturningTypeNode.ChildCount > 0)
{
IfsType annotatedReturningType = new fsType(annotatedReturningTypeNode.GetChild(0).Text, null);
UnifyWrapper(ref bodyType, ref annotatedReturningType, innerScope);
}
for (int i = 0; i < argsNames.Count; i++)
{
functionTypes.Add(innerScope.GetVarType(argsNames[i]));
}
functionTypes.Add(bodyType);
fsType functionType = fsType.GetFunctionType(functionTypes);
enteredFunctionsNames.RemoveAt(enteredFunctionsNames.Count - 1);
return functionType;
}
private IfsType InferBodyType(ITree node, fsScope scope)
{
IfsType exprType = null;
for (int i = 0; i < node.GetChild(0).ChildCount; i++)
{
ITree childNode = node.GetChild(0).Text == "elif" ? node.GetChild(0) : node.GetChild(0).GetChild(i);
ITree funcNameNode = null;
bool isLambda = false;
if (childNode.Type == fsharp_ssParser.FUNCTION_DEFN)
{
funcNameNode = GetChildByType(childNode, fsharp_ssParser.NAME);
if (funcNameNode == null)
{
isLambda = true;
funcNameNode = new fsTreeNode("NAME", fsharp_ssParser.NAME);
funcNameNode.AddChild(new fsTreeNode(GetUniqueFunctionName()));
childNode.AddChild(funcNameNode);
}
}
exprType = Analyse(childNode, scope);
if (childNode.Type == fsharp_ssParser.FUNCTION_DEFN)
{
scope.AddFunction(funcNameNode.GetChild(0).Text, exprType);
scope.SetVarInfo(funcNameNode.GetChild(0).Text, ScopePositionType.functionClass);
if (isLambda)
{
fsTreeNode lambdaCall = new fsTreeNode("FUNCTION_CALL", fsharp_ssParser.FUNCTION_CALL);
fsTreeNode lambdaCallArgs = new fsTreeNode("ARGS", fsharp_ssParser.ARGS);
lambdaCall.AddChild(funcNameNode);
lambdaCall.AddChild(lambdaCallArgs);
node.GetChild(0).AddChild(lambdaCall);
}
}
else if (childNode.Type == fsharp_ssParser.VALUE_DEFN)
{
string varName = GetChildByType(childNode, fsharp_ssParser.NAME).GetChild(0).Text;
scope.AddVar(varName, exprType);
scope.SetVarInfo(varName, ScopePositionType.local);
}
}
if (exprType == null)
{
throw new Exception("Cannot recognize body type");
}
return exprType;
}
