X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.scl.compiler%2Fsrc%2Forg%2Fsimantics%2Fscl%2Fcompiler%2Fconstants%2FStringInterpolation.java;h=ab4bafbc4af7d9eaaeecc14d1b608c9f269994e2;hb=dce0962e9a3220b7cc5e4e7d782dd565ab817104;hp=55bba560472aaced6f5b74925dd953d436c8cb29;hpb=969bd23cab98a79ca9101af33334000879fb60c5;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/constants/StringInterpolation.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/constants/StringInterpolation.java
index 55bba5604..ab4bafbc4 100644
--- a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/constants/StringInterpolation.java
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/constants/StringInterpolation.java
@@ -1,64 +1,169 @@
-package org.simantics.scl.compiler.constants;
-
-import org.cojen.classfile.TypeDesc;
-import org.simantics.scl.compiler.internal.codegen.references.Val;
-import org.simantics.scl.compiler.internal.codegen.utils.Constants;
-import org.simantics.scl.compiler.internal.codegen.utils.MethodBuilder;
-import org.simantics.scl.compiler.types.TVar;
-import org.simantics.scl.compiler.types.Type;
-import org.simantics.scl.compiler.types.Types;
-
-public class StringInterpolation extends FunctionValue {
-
-    public static final TypeDesc STRING_BUILDER = TypeDesc.forClass(StringBuilder.class);
-    
-    private final String[] textParts;
-    
-    public StringInterpolation(Type[] parameterTypes, String[] textParts) {
-        super(TVar.EMPTY_ARRAY, Types.NO_EFFECTS, Types.STRING, parameterTypes);
-        this.textParts = textParts;
-    }
-    
-    public StringInterpolation(String[] textParts) {
-        this(stringTypeArray(textParts.length-1), textParts);
-    }
-
-    private static Type[] stringTypeArray(int length) {
-        Type[] result = new Type[length];
-        for(int i=0;i<length;++i)
-            result[i] = Types.STRING;
-        return result;
-    }
-
-    @Override
-    public Type applyExact(MethodBuilder mb, Val[] parameters) {
-        mb.newObject(STRING_BUILDER);
-        mb.dup();
-        mb.invokeConstructor(STRING_BUILDER, Constants.EMPTY_TYPEDESC_ARRAY);
-        
-        for(int i=0;i<parameters.length;++i) {
-            String textPart = textParts[i]; 
-            if(!textPart.isEmpty()) {
-                mb.loadConstant(textPart);
-                mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER, new TypeDesc[] {TypeDesc.STRING});
-            }
-            
-            mb.push(parameters[i], parameterTypes[i]);
-            mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER,
-                    new TypeDesc[] {mb.getJavaTypeTranslator().toTypeDesc(parameterTypes[i])});
-        }
-        
-        {
-            String textPart = textParts[parameters.length]; 
-            if(!textPart.isEmpty()) {
-                mb.loadConstant(textPart);
-                mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER, new TypeDesc[] {TypeDesc.STRING});
-            }
-        }
-        
-        mb.invokeVirtual(STRING_BUILDER, "toString", TypeDesc.STRING, Constants.EMPTY_TYPEDESC_ARRAY);
-        
-        return Types.STRING;
-    }
-
-}
+package org.simantics.scl.compiler.constants;
+
+import java.util.Arrays;
+
+import org.cojen.classfile.TypeDesc;
+import org.objectweb.asm.Label;
+import org.objectweb.asm.Opcodes;
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
+import org.simantics.scl.compiler.internal.codegen.continuations.Cont;
+import org.simantics.scl.compiler.internal.codegen.references.IVal;
+import org.simantics.scl.compiler.internal.codegen.references.Val;
+import org.simantics.scl.compiler.internal.codegen.utils.Constants;
+import org.simantics.scl.compiler.internal.codegen.utils.LocalVariable;
+import org.simantics.scl.compiler.internal.codegen.utils.MethodBuilder;
+import org.simantics.scl.compiler.types.TVar;
+import org.simantics.scl.compiler.types.Type;
+import org.simantics.scl.compiler.types.Types;
+
+public class StringInterpolation extends FunctionValue {
+
+    public static final TypeDesc STRING_BUILDER = TypeDesc.forClass(StringBuilder.class);
+    
+    private final String[] textParts;
+    
+    public StringInterpolation(Type[] parameterTypes, String[] textParts) {
+        super(TVar.EMPTY_ARRAY, Types.NO_EFFECTS, Types.STRING, parameterTypes);
+        this.textParts = textParts;
+    }
+    
+    public StringInterpolation(String[] textParts) {
+        this(stringTypeArray(textParts.length-1), textParts);
+    }
+    
+    @Override
+    public String toString() {
+        StringBuilder b = new StringBuilder();
+        b.append('"');
+        boolean first = true;
+        for(String textPart : textParts) {
+            if(first)
+                first = false;
+            else
+                b.append("\\(.)");
+            b.append(textPart);
+        }
+        b.append('"');
+        return b.toString();
+    }
+
+    private static Type[] stringTypeArray(int length) {
+        Type[] result = new Type[length];
+        for(int i=0;i<length;++i)
+            result[i] = Types.STRING;
+        return result;
+    }
+
+    @Override
+    public Type applyExact(MethodBuilder mb, Val[] parameters) {
+        if(textParts.length==1) {
+            mb.loadConstant(textParts[0]);
+        }
+        else if(textParts.length==2) {
+            if(parameters[0].getType() == Types.STRING) {
+                // Optimized special cases "asd" + x, x + "asd"
+                if(textParts[0].isEmpty()) {
+                    mb.push(parameters[0], Types.STRING);
+                    if(!textParts[1].isEmpty()) {
+                        mb.loadConstant(textParts[1]);
+                        mb.invokeVirtual("java/lang/String", "concat", TypeDesc.STRING, new TypeDesc[] {TypeDesc.STRING});
+                    }
+                    return Types.STRING;
+                }
+                else if(textParts[1].isEmpty()) {
+                    mb.loadConstant(textParts[0]);
+                    mb.push(parameters[0], Types.STRING);
+                    mb.invokeVirtual("java/lang/String", "concat", TypeDesc.STRING, new TypeDesc[] {TypeDesc.STRING});
+                    return Types.STRING;
+                }
+            }
+        }
+        else if(textParts.length==3) {
+            if(parameters[0].getType() == Types.STRING && parameters[1].getType() == Types.STRING
+                    && textParts[0].isEmpty() && textParts[1].isEmpty() && textParts[2].isEmpty()) {
+                mb.push(parameters[0], Types.STRING);
+                mb.push(parameters[1], Types.STRING);
+                mb.invokeVirtual("java/lang/String", "concat", TypeDesc.STRING, new TypeDesc[] {TypeDesc.STRING});
+                return Types.STRING;
+            }
+        }
+        mb.newObject(STRING_BUILDER);
+        mb.dup();
+        mb.invokeConstructor(STRING_BUILDER, Constants.EMPTY_TYPEDESC_ARRAY);
+        
+        for(int i=0;i<parameters.length;++i) {
+            String textPart = textParts[i]; 
+            if(!textPart.isEmpty()) {
+                mb.loadConstant(textPart);
+                mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER, new TypeDesc[] {TypeDesc.STRING});
+            }
+            
+            mb.push(parameters[i], parameterTypes[i]);
+            mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER,
+                    new TypeDesc[] {mb.getJavaTypeTranslator().toTypeDesc(parameterTypes[i])});
+        }
+        
+        {
+            String textPart = textParts[parameters.length]; 
+            if(!textPart.isEmpty()) {
+                mb.loadConstant(textPart);
+                mb.invokeVirtual(STRING_BUILDER, "append", STRING_BUILDER, new TypeDesc[] {TypeDesc.STRING});
+            }
+        }
+        
+        mb.invokeVirtual(STRING_BUILDER, "toString", TypeDesc.STRING, Constants.EMPTY_TYPEDESC_ARRAY);
+        
+        return Types.STRING;
+    }
+    
+    @Override
+    public int hashCode() {
+        return Arrays.hashCode(textParts) ^ Arrays.hashCode(parameterTypes);
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        if(this == obj)
+            return true;
+        if(obj == null || obj.getClass() != getClass())
+            return false;
+        StringInterpolation other = (StringInterpolation)obj;
+        return Arrays.equals(textParts, other.textParts) && Arrays.equals(parameterTypes, other.parameterTypes);
+    }
+    
+    @Override
+    public int constructorTag() {
+        return textParts.length == 2 && parameterTypes[0] == Types.STRING ? 0 : -1;
+    }
+    
+    @Override
+    public void deconstruct(MethodBuilder mb, IVal parameter, Cont success, Label failure) {
+        if(textParts.length != 2)
+            throw new InternalCompilerError("String interpolation with more than one open string parts cannot be used as a pattern.");
+        mb.push(parameter, Types.STRING);
+        LocalVariable temp = mb.createLocalVariable(null, TypeDesc.STRING);
+        mb.storeLocal(temp);
+        if(!textParts[0].isEmpty()) {
+            mb.loadLocal(temp);
+            mb.loadConstant(textParts[0]);
+            mb.invokeVirtual("java/lang/String", "startsWith", TypeDesc.BOOLEAN, new TypeDesc[] {TypeDesc.STRING});
+            mb.ifZeroComparisonBranch(failure, "==");
+        }
+        if(!textParts[1].isEmpty()) {
+            mb.loadLocal(temp);
+            mb.loadConstant(textParts[1]);
+            mb.invokeVirtual("java/lang/String", "endsWith", TypeDesc.BOOLEAN, new TypeDesc[] {TypeDesc.STRING});
+            mb.ifZeroComparisonBranch(failure, "==");
+        }
+        mb.loadLocal(temp);
+        mb.loadConstant(textParts[0].length());
+        mb.loadLocal(temp);
+        mb.invokeVirtual("java/lang/String", "length", TypeDesc.INT, Constants.EMPTY_TYPEDESC_ARRAY);
+        mb.loadConstant(textParts[1].length());
+        mb.math(Opcodes.ISUB);
+        mb.invokeVirtual("java/lang/String", "substring", TypeDesc.STRING, new TypeDesc[] {TypeDesc.INT, TypeDesc.INT});
+        mb.storeLocal(temp);
+        mb.jump(success, new LocalVariableConstant(Types.STRING, temp));
+    }
+    
+}