001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.codegen;
018
019 import kotlin.Function0;
020 import org.jetbrains.annotations.NotNull;
021 import org.jetbrains.jet.codegen.context.CodegenContext;
022 import org.jetbrains.jet.codegen.context.MethodContext;
023 import org.jetbrains.jet.codegen.context.ScriptContext;
024 import org.jetbrains.jet.codegen.state.GenerationState;
025 import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
026 import org.jetbrains.jet.lang.descriptors.PropertyDescriptor;
027 import org.jetbrains.jet.lang.descriptors.ScriptDescriptor;
028 import org.jetbrains.jet.lang.descriptors.ValueParameterDescriptor;
029 import org.jetbrains.jet.lang.psi.JetDeclaration;
030 import org.jetbrains.jet.lang.psi.JetScript;
031 import org.jetbrains.jet.lang.resolve.BindingContext;
032 import org.jetbrains.jet.lang.resolve.java.jvmSignature.JvmMethodSignature;
033 import org.jetbrains.org.objectweb.asm.MethodVisitor;
034 import org.jetbrains.org.objectweb.asm.Type;
035 import org.jetbrains.org.objectweb.asm.commons.InstructionAdapter;
036
037 import java.util.Collections;
038 import java.util.List;
039
040 import static org.jetbrains.jet.lang.resolve.java.diagnostics.DiagnosticsPackage.OtherOrigin;
041 import static org.jetbrains.jet.lang.resolve.java.diagnostics.JvmDeclarationOrigin.NO_ORIGIN;
042 import static org.jetbrains.jet.codegen.binding.CodegenBinding.*;
043 import static org.jetbrains.jet.lang.resolve.java.AsmTypeConstants.OBJECT_TYPE;
044 import static org.jetbrains.org.objectweb.asm.Opcodes.*;
045
046 // SCRIPT: script code generator
047 public class ScriptCodegen extends MemberCodegen<JetScript> {
048
049 public static ScriptCodegen createScriptCodegen(
050 @NotNull JetScript declaration,
051 @NotNull GenerationState state,
052 @NotNull CodegenContext parentContext
053 ) {
054 ScriptDescriptor scriptDescriptor = state.getBindingContext().get(BindingContext.SCRIPT, declaration);
055 assert scriptDescriptor != null;
056
057 ClassDescriptor classDescriptorForScript = state.getBindingContext().get(CLASS_FOR_SCRIPT, scriptDescriptor);
058 assert classDescriptorForScript != null;
059
060 Type className = state.getBindingContext().get(ASM_TYPE, classDescriptorForScript);
061 assert className != null;
062
063 ClassBuilder builder = state.getFactory().newVisitor(OtherOrigin(declaration, classDescriptorForScript),
064 className, declaration.getContainingFile());
065 List<ScriptDescriptor> earlierScripts = state.getEarlierScriptsForReplInterpreter();
066 ScriptContext scriptContext = parentContext.intoScript(
067 scriptDescriptor,
068 earlierScripts == null ? Collections.<ScriptDescriptor>emptyList() : earlierScripts,
069 classDescriptorForScript
070 );
071 return new ScriptCodegen(declaration, state, scriptContext, builder);
072 }
073
074 private final JetScript scriptDeclaration;
075 private final ScriptContext context;
076 private final ScriptDescriptor scriptDescriptor;
077
078 private ScriptCodegen(
079 @NotNull JetScript scriptDeclaration,
080 @NotNull GenerationState state,
081 @NotNull ScriptContext context,
082 @NotNull ClassBuilder builder
083 ) {
084 super(state, null, context, scriptDeclaration, builder);
085 this.scriptDeclaration = scriptDeclaration;
086 this.context = context;
087 this.scriptDescriptor = context.getScriptDescriptor();
088 }
089
090 @Override
091 protected void generateDeclaration() {
092 Type classType = bindingContext.get(ASM_TYPE, context.getContextDescriptor());
093 assert classType != null;
094
095 v.defineClass(scriptDeclaration,
096 V1_6,
097 ACC_PUBLIC,
098 classType.getInternalName(),
099 null,
100 "java/lang/Object",
101 new String[0]);
102 }
103
104 @Override
105 protected void generateBody() {
106 genMembers();
107 genFieldsForParameters(scriptDescriptor, v);
108 genConstructor(scriptDescriptor, context.getContextDescriptor(), v,
109 context.intoFunction(scriptDescriptor.getScriptCodeDescriptor()));
110 }
111
112 @Override
113 protected void generateKotlinAnnotation() {
114 // TODO
115 }
116
117 private void genConstructor(
118 @NotNull ScriptDescriptor scriptDescriptor,
119 @NotNull ClassDescriptor classDescriptorForScript,
120 @NotNull ClassBuilder classBuilder,
121 @NotNull final MethodContext methodContext
122 ) {
123 Type blockType = typeMapper.mapType(scriptDescriptor.getScriptCodeDescriptor().getReturnType());
124
125 PropertyDescriptor scriptResultProperty = scriptDescriptor.getScriptResultProperty();
126 classBuilder.newField(OtherOrigin(scriptResultProperty),
127 ACC_PUBLIC | ACC_FINAL, scriptResultProperty.getName().asString(),
128 blockType.getDescriptor(), null, null);
129
130 JvmMethodSignature jvmSignature = typeMapper.mapScriptSignature(scriptDescriptor, context.getEarlierScripts());
131
132 MethodVisitor mv = classBuilder.newMethod(
133 OtherOrigin(scriptDeclaration, scriptDescriptor.getClassDescriptor().getUnsubstitutedPrimaryConstructor()),
134 ACC_PUBLIC, jvmSignature.getAsmMethod().getName(), jvmSignature.getAsmMethod().getDescriptor(),
135 null, null);
136
137 mv.visitCode();
138
139 final InstructionAdapter instructionAdapter = new InstructionAdapter(mv);
140
141 Type classType = bindingContext.get(ASM_TYPE, classDescriptorForScript);
142 assert classType != null;
143
144 instructionAdapter.load(0, classType);
145 instructionAdapter.invokespecial("java/lang/Object", "<init>", "()V");
146
147 instructionAdapter.load(0, classType);
148
149 final FrameMap frameMap = methodContext.prepareFrame(typeMapper);
150
151 for (ScriptDescriptor importedScript : context.getEarlierScripts()) {
152 frameMap.enter(importedScript, OBJECT_TYPE);
153 }
154
155 Type[] argTypes = jvmSignature.getAsmMethod().getArgumentTypes();
156 int add = 0;
157
158 for (int i = 0; i < scriptDescriptor.getScriptCodeDescriptor().getValueParameters().size(); i++) {
159 ValueParameterDescriptor parameter = scriptDescriptor.getScriptCodeDescriptor().getValueParameters().get(i);
160 frameMap.enter(parameter, argTypes[i + add]);
161 }
162
163 generateInitializers(new Function0<ExpressionCodegen>() {
164 @Override
165 public ExpressionCodegen invoke() {
166 return new ExpressionCodegen(instructionAdapter, frameMap, Type.VOID_TYPE, methodContext, state, ScriptCodegen.this);
167 }
168 });
169
170 int offset = 1;
171
172 for (ScriptDescriptor earlierScript : context.getEarlierScripts()) {
173 Type earlierClassType = asmTypeForScriptDescriptor(bindingContext, earlierScript);
174 instructionAdapter.load(0, classType);
175 instructionAdapter.load(offset, earlierClassType);
176 offset += earlierClassType.getSize();
177 instructionAdapter.putfield(classType.getInternalName(), context.getScriptFieldName(earlierScript), earlierClassType.getDescriptor());
178 }
179
180 for (ValueParameterDescriptor parameter : scriptDescriptor.getScriptCodeDescriptor().getValueParameters()) {
181 Type parameterType = typeMapper.mapType(parameter.getType());
182 instructionAdapter.load(0, classType);
183 instructionAdapter.load(offset, parameterType);
184 offset += parameterType.getSize();
185 instructionAdapter.putfield(classType.getInternalName(), parameter.getName().getIdentifier(), parameterType.getDescriptor());
186 }
187
188 StackValue stackValue =
189 new ExpressionCodegen(mv, frameMap, Type.VOID_TYPE, methodContext, state, this).gen(scriptDeclaration.getBlockExpression());
190 if (stackValue.type != Type.VOID_TYPE) {
191 stackValue.put(stackValue.type, instructionAdapter);
192 instructionAdapter.putfield(classType.getInternalName(), ScriptDescriptor.LAST_EXPRESSION_VALUE_FIELD_NAME,
193 blockType.getDescriptor());
194 }
195
196 instructionAdapter.areturn(Type.VOID_TYPE);
197 mv.visitMaxs(-1, -1);
198 mv.visitEnd();
199 }
200
201 private void genFieldsForParameters(@NotNull ScriptDescriptor script, @NotNull ClassBuilder classBuilder) {
202 for (ScriptDescriptor earlierScript : context.getEarlierScripts()) {
203 Type earlierClassName = asmTypeForScriptDescriptor(bindingContext, earlierScript);
204 int access = ACC_PRIVATE | ACC_FINAL;
205 classBuilder.newField(NO_ORIGIN, access, context.getScriptFieldName(earlierScript), earlierClassName.getDescriptor(), null, null);
206 }
207
208 for (ValueParameterDescriptor parameter : script.getScriptCodeDescriptor().getValueParameters()) {
209 Type parameterType = typeMapper.mapType(parameter);
210 int access = ACC_PUBLIC | ACC_FINAL;
211 classBuilder.newField(OtherOrigin(parameter), access, parameter.getName().getIdentifier(), parameterType.getDescriptor(), null, null);
212 }
213 }
214
215 private void genMembers() {
216 for (JetDeclaration declaration : scriptDeclaration.getDeclarations()) {
217 genFunctionOrProperty(declaration);
218 }
219 }
220 }