]> gerrit.simantics Code Review - simantics/platform.git/blob
d740de293824b33e9241c6addd9bc66eebad46a6
[simantics/platform.git] /
1 package org.simantics.scl.compiler.internal.codegen.utils;
2
3 import org.cojen.classfile.TypeDesc;
4 import org.objectweb.asm.Label;
5 import org.objectweb.asm.MethodVisitor;
6 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
7 import org.simantics.scl.compiler.internal.codegen.continuations.Cont;
8 import org.simantics.scl.compiler.internal.codegen.continuations.ContRef;
9 import org.simantics.scl.compiler.internal.codegen.continuations.ReturnCont;
10 import org.simantics.scl.compiler.internal.codegen.references.BoundVar;
11 import org.simantics.scl.compiler.internal.codegen.references.IVal;
12 import org.simantics.scl.compiler.internal.codegen.references.Val;
13 import org.simantics.scl.compiler.internal.codegen.ssa.SSABlock;
14 import org.simantics.scl.compiler.internal.codegen.types.JavaTypeTranslator;
15 import org.simantics.scl.compiler.top.SCLCompilerConfiguration;
16 import org.simantics.scl.compiler.types.TVar;
17 import org.simantics.scl.compiler.types.Type;
18 import org.simantics.scl.compiler.types.Types;
19
20 import gnu.trove.map.hash.THashMap;
21 import gnu.trove.set.hash.THashSet;
22
23 public class MethodBuilder extends MethodBuilderBase {
24     ModuleBuilder moduleBuilder;
25     THashMap<Cont, Label> labels = new THashMap<Cont, Label>();
26     THashMap<BoundVar, LocalVariable> localVariables = new THashMap<BoundVar, LocalVariable>();
27     THashSet<Cont> generatedConts = new THashSet<Cont>();
28     THashMap<Object,Object> preparationSteps = new THashMap<Object,Object>(4); 
29     
30     public MethodBuilder(ClassBuilder classBuilder, ModuleBuilder moduleBuilder, boolean isStatic, MethodVisitor methodVisitor, TypeDesc[] parameterTypes) {
31         super(classBuilder, isStatic, methodVisitor, parameterTypes);
32         this.moduleBuilder = moduleBuilder;
33     }
34     
35     public JavaTypeTranslator getJavaTypeTranslator() {
36         return getModuleBuilder().getJavaTypeTranslator();
37     }
38     
39     public ModuleBuilder getModuleBuilder() {
40         return moduleBuilder;
41     }
42     
43     public void push(IVal val, Type type) {
44         val.push(this);
45         if(Types.canonical(type) instanceof TVar)
46             box(val.getType());
47     }
48             
49     public void push(Val[] vals, Type[] types) {
50         assert(vals.length == types.length);
51         for(int i=0;i<vals.length;++i)
52             push(vals[i], types[i]);
53     }
54         
55     public void pushBoxed(Val val) {
56         val.push(this);
57         box(val.getType());
58     }
59     
60     public void pushBoxed(Val[] vals) {
61         for(Val val : vals)
62             pushBoxed(val);
63     }
64
65     public void genericApply(int arity) {
66         invokeInterface(Constants.FUNCTION, "apply", TypeDesc.OBJECT, Constants.OBJECTS[arity]);
67     }
68     
69     public void pushTuple0() {
70         loadStaticField(Constants.TUPLE0, "INSTANCE", Constants.TUPLE0);
71     }
72     
73     private void boxPrimitiveType(TypeDesc primitiveType) {
74         if(primitiveType.equals(TypeDesc.VOID)) {            
75             pushTuple0();
76         }
77         else {
78             TypeDesc objectType = primitiveType.toObjectType();
79             invokeStatic(getClassName(objectType), "valueOf", 
80                     objectType, new TypeDesc[] {primitiveType});
81         }
82     }
83     
84     /**
85      * Boxes the top of the stack.
86      */
87     public void box(Type type) {
88         TypeDesc typeDesc = getJavaTypeTranslator().toTypeDesc(type);
89         if(typeDesc.isPrimitive()) {
90             boxPrimitiveType(typeDesc);
91         }
92         //codeBuilder.convert(typeDesc, TypeDesc.OBJECT);
93     }
94
95     /**
96      * Unboxes the top of the stack.
97      */
98     public void unbox(Type type) {
99         TypeDesc td = getJavaTypeTranslator().toTypeDesc(type);
100         if(td.equals(TypeDesc.VOID)) {
101             pop();
102             return;
103         }
104         if(td.isPrimitive()) {
105             TypeDesc objectType = td.toObjectType();
106             checkCast(objectType);
107             convert(objectType, td);
108         }
109         else {
110             if(td != TypeDesc.OBJECT)
111                 checkCast(td);
112         }
113     }
114
115     public void setLocation(Cont continuation) {
116         setLocation(getLabel(continuation));   
117         if(!generatedConts.add(continuation))
118             throw new InternalCompilerError("Label location is set multiple times");
119     }
120     
121     public Label getLabel(Cont continuation) {
122         Label label = labels.get(continuation);
123         if(label == null) {
124             label = createLabel();
125             labels.put(continuation, label);
126         }
127         return label;
128     }
129     
130     public void setLocalVariable(BoundVar var, LocalVariable localVariable) {
131         localVariables.put(var, localVariable);
132     }
133
134     public LocalVariable getLocalVariable(BoundVar var) {
135         LocalVariable localVariable = localVariables.get(var);
136         if(localVariable == null) {
137             TypeDesc typeDesc = getJavaTypeTranslator().getTypeDesc(var);
138             if(SCLCompilerConfiguration.DEBUG) {
139                 if(typeDesc.equals(TypeDesc.VOID))
140                     throw new InternalCompilerError();
141             }
142             localVariable = createLocalVariable(null, typeDesc);
143             localVariables.put(var, localVariable);
144         }
145         return localVariable;
146     }
147
148     public void jump(ContRef target) {
149         jump(target.getBinding());
150     }
151
152     public void jump(ContRef target, Val ... parameters) {
153         jump(target.getBinding(), parameters);
154     }
155     
156     public void jump(Cont cont) {
157         if(cont instanceof SSABlock) {
158             SSABlock block = (SSABlock)cont;
159             if(generatedConts.contains(block))
160                 branch(getLabel(block));
161             else            
162                 block.generateCode(this);
163         }
164         else
165             throw new InternalCompilerError();
166     }
167             
168     public void jump(Cont cont, IVal ... parameters) {
169         if(cont instanceof ReturnCont) {
170             ReturnCont returnCont = (ReturnCont)cont;
171             if(parameters.length != 1)
172                 throw new InternalCompilerError();
173             push(parameters[0], returnCont.getType());
174             //parameters[0].push(this);
175             returnValue(getJavaTypeTranslator().toTypeDesc(returnCont.getType()));
176         }
177         else if(cont instanceof SSABlock) {
178             SSABlock block = (SSABlock)cont;
179             BoundVar[] parameterVars = block.getParameters();
180             if(parameters.length != parameterVars.length)
181                 throw new InternalCompilerError();
182             // First calculate all parameters and push them into stack
183             for(int i=0;i<parameters.length;++i)
184                 if(parameters[i] != parameterVars[i]) {
185                     push(parameters[i], parameterVars[i].getType());
186                     //parameters[i].push(this);
187                 }
188             // Then store them to the local variables
189             // NOTE: some computations inside pushs may depend on these variables
190             for(int i=parameters.length-1;i>=0;--i)
191                 if(parameters[i] != parameterVars[i])
192                     store(parameterVars[i]);
193             if(generatedConts.contains(block))
194                 branch(getLabel(block));
195             else            
196                 block.generateCode(this);
197         }
198         else
199             throw new InternalCompilerError();
200     }
201
202     public void store(BoundVar var) {
203         Type type = var.getType();
204         TypeDesc typeDesc = getJavaTypeTranslator().toTypeDesc(type);
205         if(typeDesc.equals(TypeDesc.VOID))
206             return;
207         LocalVariable lv = getLocalVariable(var);
208         storeLocal(lv);
209     }
210
211     public void ensureExists(Cont continuation) {
212         if(!generatedConts.contains(continuation))
213             ((SSABlock)continuation).generateCode(this);
214     }
215     
216     @SuppressWarnings("unchecked")
217     public <T> T getPreparation(PreparationStep<T> step) {
218         return (T)preparationSteps.get(step);
219     }
220     
221     public <T> void addPreparation(PreparationStep<T> step, T result) {
222         preparationSteps.put(step, result);
223     }
224 }