(refs #7746) Fixed applications with intermediate effects

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / types / Types.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/types/Types.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/types/Types.java
index a3db2df52a556f2e523c9e936a0040d53c00f1ea..e1b89eece28ab21f14e87e3e2fd20745ec85e7c5 100644 (file)
--- a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/types/Types.java
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/types/Types.java
@@ -6,6 +6,7 @@ import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
 import org.simantics.scl.compiler.errors.Locations;
 import org.simantics.scl.compiler.internal.parsing.exceptions.SCLSyntaxErrorException;
 import org.simantics.scl.compiler.internal.parsing.parser.SCLParserImpl;
@@ -57,15 +58,6 @@ public class Types {
     public static final TCon LONG = con(BUILTIN, "Long");
     public static final TCon FLOAT = con(BUILTIN, "Float");
     public static final TCon DOUBLE = con(BUILTIN, "Double");
-    
-    public static final TCon BOOLEAN_ARRAY = con(BUILTIN, "BooleanArray");
-    public static final TCon BYTE_ARRAY = con(BUILTIN, "ByteArray");
-    public static final TCon CHARACTER_ARRAY = con(BUILTIN, "CharacterArray");
-    public static final TCon SHORT_ARRAY = con(BUILTIN, "ShortArray");
-    public static final TCon INTEGER_ARRAY = con(BUILTIN, "IntegerArray");
-    public static final TCon LONG_ARRAY = con(BUILTIN, "LongArray");
-    public static final TCon FLOAT_ARRAY = con(BUILTIN, "FloatArray");
-    public static final TCon DOUBLE_ARRAY = con(BUILTIN, "DoubleArray");
 
     public static final TCon STRING = con(BUILTIN, "String");
     public static final TCon ARROW = con(BUILTIN, "->");
@@ -84,6 +76,7 @@ public class Types {
     public static final TCon TYPEABLE = con(BUILTIN, "Typeable");
     public static final TCon SERIALIZABLE = con(BUILTIN, "Serializable");
     public static final TCon VEC_COMP = con(BUILTIN, "VecComp");
+    public static final TCon CLASS = con(BUILTIN, "Class");
     public static final TCon BINDING = con(BUILTIN, "Binding");
 
     public static final TCon TYPE = con(BUILTIN, "Type");
@@ -93,6 +86,7 @@ public class Types {
     
     public static final TCon ADDITIVE = con("Prelude", "Additive");
     public static final TCon MONAD = con("Prelude", "Monad");
+    public static final TCon MONAD_E = con("Prelude", "MonadE");
     public static final TCon INTEGRAL = con("Prelude", "Integral");
     public static final TCon RING = con("Prelude", "Ring");
     public static final TCon ORDERED_RING = con("Prelude", "OrderedRing");
@@ -115,7 +109,15 @@ public class Types {
     public static final TCon BRANCH_POINT = con(BUILTIN, "BranchPoint");
     
     public static final TCon CHRContext = con(BUILTIN, "CHRContext");
-    
+
+    public static final Type BOOLEAN_ARRAY = vector(BOOLEAN);
+    public static final Type BYTE_ARRAY = vector(BYTE);
+    public static final Type CHARACTER_ARRAY = vector(CHARACTER);
+    public static final Type SHORT_ARRAY = vector(SHORT);
+    public static final Type INTEGER_ARRAY = vector(INTEGER);
+    public static final Type LONG_ARRAY = vector(LONG);
+    public static final Type FLOAT_ARRAY = vector(FLOAT);
+    public static final Type DOUBLE_ARRAY = vector(DOUBLE);
 
     private volatile static TCon[] tupleCache = new TCon[] {
         UNIT, null
@@ -641,7 +643,52 @@ public class Types {
                 effect,
                 type);
     }
+    
+    /**
+     * This function always success, but may return a multi function
+     * with arity smaller than given parameter
+     */
+    public static MultiFunction unifyFunction2(Type type, int arity) {
+        type = canonical(type);
 
+        Type[] parameterTypes = new Type[arity];
+        Type effect = Types.NO_EFFECTS;
+        int i;
+        for(i=0;i<arity;++i) {
+            if(type instanceof TFun) {
+                TFun fun = (TFun)type;
+                parameterTypes[i] = fun.getCanonicalDomain();
+                type = fun.getCanonicalRange();
+                effect = fun.getCanonicalEffect();
+                if(effect != Types.NO_EFFECTS) {
+                    ++i;
+                    break;
+                }
+            }
+            else if(type instanceof TMetaVar) {
+                Type domain = metaVar(Kinds.STAR);
+                parameterTypes[i] = domain;
+                Type range = metaVar(Kinds.STAR);
+                effect = metaVar(Kinds.EFFECT);
+                try {
+                    ((TMetaVar) type).setRef(functionE(domain, effect, range));
+                } catch (UnificationException e) {
+                    // Should never happen, if we have checked maximum arity before calling this function
+                    throw new InternalCompilerError(e);
+                }
+                type = range;
+                ++i;
+                break;
+            }
+            else
+                break;
+        }
+        
+        if(i < arity)
+            parameterTypes = Arrays.copyOf(parameterTypes, i);
+        return new MultiFunction(parameterTypes, effect, type);
+    }
+    
     public static MultiFunction unifyFunction(Type type, int arity) throws UnificationException {
         Type[] parameterTypes = new Type[arity];
         for(int i=0;i<arity;++i)
@@ -691,6 +738,23 @@ public class Types {
         }
         return arity;
     }
+    
+    public static int getMaxArity(Type type) {
+        type = Skeletons.canonicalSkeleton(type);
+        int arity = 0;
+        while(true) {
+            if(type instanceof TFun) {
+                ++arity;
+                type = Skeletons.canonicalSkeleton(((TFun) type).getCanonicalRange());
+            }
+            else if(type instanceof TMetaVar) {
+                return Integer.MAX_VALUE;
+            }
+            else
+                break;
+        }
+        return arity;
+    }
 
     public static TMetaVar metaVar(Kind kind) {
         return new TMetaVar(kind);
@@ -1049,6 +1113,10 @@ public class Types {
     public static Type union(Type effect1, Type effect2) {
         return new TUnion(effect1, effect2);
     }
+    
+    public static Type union(Type effect1, Type effect2, Type effect3) {
+        return new TUnion(effect1, effect2, effect3);
+    }
 
     public static Type union(List<Type> effects) {
         if(effects.size() == 0)
@@ -1119,17 +1187,9 @@ public class Types {
         return parseType(new TypeElaborationContext(environment), text);
     }
 
-    public static Type parseType(ITypeEnvironment environment, THashMap<String, TVar> localTypeVars, String text) throws SCLTypeParseException {
-        return parseType(new TypeElaborationContext(localTypeVars, environment), text);
-    }
-
     public static Type parseType(String text) throws SCLTypeParseException {
         return parseType(new TypeElaborationContext(DUMMY_TYPE_ENVIRONMENT), text);
     }
-
-    public static Type parseType(THashMap<String, TVar> localTypeVars, String text) throws SCLTypeParseException {
-        return parseType(new TypeElaborationContext(localTypeVars, DUMMY_TYPE_ENVIRONMENT), text);
-    }
     
     private static Type parseType(TypeElaborationContext context, String text) throws SCLTypeParseException {
         SCLParserImpl parser = new SCLParserImpl(new StringReader(text));