jruby · Aug 6, 2015 · Aug 6, 2015 · Aug 6, 2015 · Aug 6, 2015 · Aug 6, 2015
diff --git a/core/src/main/java/org/jruby/compiler/JITCompiler.java b/core/src/main/java/org/jruby/compiler/JITCompiler.java
@@ -37,9 +37,9 @@
 import org.jruby.ast.util.SexpMaker;
 import org.jruby.internal.runtime.methods.CompiledIRMethod;
 import org.jruby.internal.runtime.methods.MixedModeIRMethod;
-import org.jruby.ir.IRMethod;
 import org.jruby.ir.interpreter.InterpreterContext;
 import org.jruby.ir.targets.JVMVisitor;
+import org.jruby.ir.targets.JVMVisitorMethodContext;
 import org.jruby.runtime.ThreadContext;
 import org.jruby.runtime.builtin.IRubyObject;
 import org.jruby.threading.DaemonThreadFactory;
@@ -243,7 +243,8 @@ public void run() {
                 JVMVisitor visitor = new JVMVisitor();
                 JITClassGenerator generator = new JITClassGenerator(className, methodName, key, runtime, method, visitor);
 
-                generator.compile();
+                JVMVisitorMethodContext context = new JVMVisitorMethodContext();
+                generator.compile(context);
 
                 // FIXME: reinstate active bytecode size check
                 // At this point we still need to reinstate the bytecode size check, to ensure we're not loading code
@@ -274,8 +275,8 @@ public void run() {
                     log(method.getImplementationClass(), method.getFile(), method.getLine(), className + "." + methodName, "done jitting");
                 }
 
-                Map<Integer, MethodType> signatures = ((IRMethod)method.getIRScope()).getNativeSignatures();
-                String jittedName = ((IRMethod)method.getIRScope()).getJittedName();
+                Map<Integer, MethodType> signatures = context.getNativeSignatures();
+                String jittedName = context.getJittedName();
                 if (signatures.size() == 1) {
                     // only variable-arity
                     method.completeBuild(
@@ -324,10 +325,9 @@ public static String getHashForBytes(byte[] bytes) {
         try {
             MessageDigest sha1 = MessageDigest.getInstance("SHA1");
             sha1.update(bytes);
-            byte[] digest = sha1.digest();
             StringBuilder builder = new StringBuilder();
-            for (int i = 0; i < digest.length; i++) {
-                builder.append(Integer.toString( ( digest[i] & 0xff ) + 0x100, 16).substring( 1 ));
+            for (byte aByte : sha1.digest()) {
+                builder.append(Integer.toString((aByte & 0xff) + 0x100, 16).substring(1));
             }
             return builder.toString().toUpperCase(Locale.ENGLISH);
         } catch (NoSuchAlgorithmException nsae) {
@@ -358,7 +358,7 @@ public JITClassGenerator(String className, String methodName, String key, Ruby r
         }
 
         @SuppressWarnings("unchecked")
-        protected void compile() {
+        protected void compile(JVMVisitorMethodContext context) {
             if (bytecode != null) return;
 
             // Time the compilation
@@ -374,7 +374,7 @@ protected void compile() {
 
             // This may not be ok since we'll end up running passes specific to JIT
             // CON FIXME: Really should clone scope before passes in any case
-            bytecode = visitor.compileToBytecode(method.getIRScope());
+            bytecode = visitor.compileToBytecode(method.getIRScope(), context);
 
             compileTime = System.nanoTime() - start;
         }
@@ -392,8 +392,9 @@ void updateCounters(JITCounts counts) {
             }
         }
 
+        // FIXME: Does anything call this?  If so we should document it.
         public void generate() {
-            compile();
+            compile(new JVMVisitorMethodContext());
         }
 
         public byte[] bytecode() {
@@ -432,8 +433,8 @@ static void log(RubyModule implementationClass, String file, int line, String na
 
         if (reason.length > 0) {
             builder.append(" because of: \"");
-            for (int i = 0; i < reason.length; i++) {
-                builder.append(reason[i]);
+            for (String aReason : reason) {
+                builder.append(aReason);
             }
             builder.append('"');
         }

diff --git a/core/src/main/java/org/jruby/internal/runtime/methods/InterpretedIRMethod.java b/core/src/main/java/org/jruby/internal/runtime/methods/InterpretedIRMethod.java
@@ -60,7 +60,7 @@ public StaticScope getStaticScope() {
 
     public ArgumentDescriptor[] getArgumentDescriptors() {
         ensureInstrsReady(); // Make sure method is minimally built before returning this info
-        return method.getArgumentDescriptors();
+        return ((IRMethod) method).getArgumentDescriptors();
     }
 
     public Signature getSignature() {

diff --git a/core/src/main/java/org/jruby/ir/Compiler.java b/core/src/main/java/org/jruby/ir/Compiler.java
@@ -14,6 +14,7 @@
 import org.jruby.ir.operands.IRException;
 import org.jruby.ir.runtime.IRBreakJump;
 import org.jruby.ir.targets.JVMVisitor;
+import org.jruby.ir.targets.JVMVisitorMethodContext;
 import org.jruby.parser.StaticScope;
 import org.jruby.runtime.Block;
 import org.jruby.runtime.DynamicScope;
@@ -43,15 +44,14 @@ public static Compiler getInstance() {
 
     @Override
     protected ScriptAndCode execute(final Ruby runtime, final IRScriptBody scope, ClassDefiningClassLoader classLoader) {
-        JVMVisitor visitor;
         byte[] bytecode;
-        Class compiled;
-
         MethodHandle _compiledHandle;
+
         try {
-            visitor = new JVMVisitor();
-            bytecode = visitor.compileToBytecode(scope);
-            compiled = visitor.defineFromBytecode(scope, bytecode, classLoader);
+            JVMVisitor visitor = new JVMVisitor();
+            JVMVisitorMethodContext context = new JVMVisitorMethodContext();
+            bytecode = visitor.compileToBytecode(scope, context);
+            Class compiled = visitor.defineFromBytecode(scope, bytecode, classLoader);
 
             Method compiledMethod = compiled.getMethod("RUBY$script", ThreadContext.class,
                     StaticScope.class, IRubyObject.class, IRubyObject[].class, Block.class, RubyModule.class, String.class);

diff --git a/core/src/main/java/org/jruby/ir/IRBuilder.java b/core/src/main/java/org/jruby/ir/IRBuilder.java
@@ -2109,7 +2109,7 @@ public void receiveMethodArgs(final ArgsNode argsNode) {
     public void receiveBlockArgs(final IterNode node) {
         Node args = node.getVarNode();
         if (args instanceof ArgsNode) { // regular blocks
-            scope.setArgumentDescriptors(Helpers.argsNodeToArgumentDescriptors(((ArgsNode) args)));
+            ((IRClosure) scope).setArgumentDescriptors(Helpers.argsNodeToArgumentDescriptors(((ArgsNode) args)));
             receiveArgs((ArgsNode)args);
         } else  {
             // for loops -- reuse code in IRBuilder:buildBlockArgsAssignment

diff --git a/core/src/main/java/org/jruby/ir/IRClosure.java b/core/src/main/java/org/jruby/ir/IRClosure.java
@@ -9,6 +9,7 @@
 import org.jruby.ir.transformations.inlining.CloneInfo;
 import org.jruby.ir.transformations.inlining.SimpleCloneInfo;
 import org.jruby.parser.StaticScope;
+import org.jruby.runtime.ArgumentDescriptor;
 import org.jruby.runtime.BlockBody;
 import org.jruby.runtime.IRBlockBody;
 import org.jruby.runtime.InterpretedIRBlockBody;
@@ -28,6 +29,9 @@ public class IRClosure extends IRScope {
 
     private Signature signature;
 
+    // Argument description
+    protected ArgumentDescriptor[] argDesc = ArgumentDescriptor.EMPTY_ARRAY;
+
     /** Added for interp/JIT purposes */
     private IRBlockBody body;
 
@@ -290,4 +294,16 @@ public void setHandle(Handle handle) {
     public Handle getHandle() {
         return handle;
     }
+
+    public ArgumentDescriptor[] getArgumentDescriptors() {
+        return argDesc;
+    }
+
+
+    /**
+     * Set upon completion of IRBuild of this IRClosure.
+     */
+    public void setArgumentDescriptors(ArgumentDescriptor[] argDesc) {
+        this.argDesc = argDesc;
+    }
 }
diff --git a/core/src/main/java/org/jruby/ir/IRMethod.java b/core/src/main/java/org/jruby/ir/IRMethod.java
@@ -1,23 +1,17 @@
 package org.jruby.ir;
 
-import java.lang.invoke.MethodType;
-import java.util.Collections;
-import java.util.HashMap;
-import java.util.Map;
 import org.jruby.ast.MethodDefNode;
 import org.jruby.ir.interpreter.InterpreterContext;
 import org.jruby.ir.operands.LocalVariable;
 import org.jruby.ir.representations.BasicBlock;
 import org.jruby.parser.StaticScope;
+import org.jruby.runtime.ArgumentDescriptor;
 
 public class IRMethod extends IRScope {
     public final boolean isInstanceMethod;
 
-    // Signatures to the jitted versions of this method
-    private Map<Integer, MethodType> signatures;
-
-    // Method name in the jitted version of this method
-    private String jittedName;
+    // Argument description
+    protected ArgumentDescriptor[] argDesc = ArgumentDescriptor.EMPTY_ARRAY;
 
     private MethodDefNode defn;
 
@@ -27,7 +21,6 @@ public IRMethod(IRManager manager, IRScope lexicalParent, MethodDefNode defn, St
 
         this.defn = defn;
         this.isInstanceMethod = isInstanceMethod;
-        this.signatures = null;
 
         if (!getManager().isDryRun() && staticScope != null) {
             staticScope.setIRScope(this);
@@ -73,20 +66,15 @@ public LocalVariable getLocalVariable(String name, int scopeDepth) {
         return lvar;
     }
 
-    public void addNativeSignature(int arity, MethodType signature) {
-        if (signatures == null) signatures = new HashMap<>(1);
-        signatures.put(arity, signature);
-    }
-
-    public Map<Integer, MethodType> getNativeSignatures() {
-        return Collections.unmodifiableMap(signatures);
+    public ArgumentDescriptor[] getArgumentDescriptors() {
+        return argDesc;
     }
 
-    public String getJittedName() {
-        return jittedName;
-    }
 
-    public void setJittedName(String jittedName) {
-        this.jittedName = jittedName;
+    /**
+     * Set upon completion of IRBuild of this IRMethod.
+     */
+    public void setArgumentDescriptors(ArgumentDescriptor[] argDesc) {
+        this.argDesc = argDesc;
     }
 }
diff --git a/core/src/main/java/org/jruby/ir/IRScope.java b/core/src/main/java/org/jruby/ir/IRScope.java
@@ -22,7 +22,6 @@
 import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 
-import org.jruby.runtime.ArgumentDescriptor;
 import org.jruby.util.log.Logger;
 import org.jruby.util.log.LoggerFactory;
 
@@ -63,8 +62,6 @@ public abstract class IRScope implements ParseResult {
     private static final Collection<IRClosure> NO_CLOSURES = Collections.unmodifiableCollection(new ArrayList<IRClosure>(0));
 
     private static AtomicInteger globalScopeCount = new AtomicInteger();
-    // Argument description
-    protected ArgumentDescriptor[] argDesc = ArgumentDescriptor.EMPTY_ARRAY;
 
     /** Unique global scope id */
     private int scopeId;
@@ -189,7 +186,6 @@ public IRScope(IRManager manager, IRScope lexicalParent, String name,
         this.scopeId = globalScopeCount.getAndIncrement();
 
         setupLexicalContainment();
-        this.argDesc = null;
     }
 
     private void setupLexicalContainment() {
@@ -780,17 +776,6 @@ public LocalVariable lookupExistingLVar(String name) {
         return localVars.get(name);
     }
 
-    public ArgumentDescriptor[] getArgumentDescriptors() {
-        return argDesc;
-    }
-
-    /**
-     * Set upon completion of IRBuild of this IRMethod.
-     */
-    public void setArgumentDescriptors(ArgumentDescriptor[] argDesc) {
-        this.argDesc = argDesc;
-    }
-
     protected LocalVariable findExistingLocalVariable(String name, int depth) {
         return localVars.get(name);
     }