[Truffle] Adding int long specialization to Fixnum#>>.

jruby · Apr 19, 2015 · 787bdf0 · 787bdf0
1 parent 0902cd3
commit 787bdf0
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/core/FixnumNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/core/FixnumNodes.java
@@ -1328,6 +1328,28 @@ public Object rightShift(VirtualFrame frame, int a, int b) {
             }
         }
 
+        @Specialization
+        public Object rightShift(VirtualFrame frame, int a, long b) {
+            if (b > 0) {
+                if (b >= BITS - 1) {
+                    if (a < 0) {
+                        return -1;
+                    } else {
+                        return 0;
+                    }
+                } else {
+                    return a >> b;
+                }
+            } else {
+                if (leftShiftNode == null) {
+                    CompilerDirectives.transferToInterpreter();
+                    leftShiftNode = insert(FixnumNodesFactory.LeftShiftNodeFactory.create(getContext(), getSourceSection(), new RubyNode[]{null, null}));
+                }
+
+                return leftShiftNode.executeLeftShift(frame, a, -b);
+            }
+        }
+
         @Specialization
         public Object rightShift(VirtualFrame frame, long a, int b) {
             if (b > 0) {