[Truffle] Adding Rational long specialization for Fixnum#-.

jruby · Apr 7, 2015 · 73bbd14 · 73bbd14
1 parent bd21747
commit 73bbd14
Showing 1 changed file with 13 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
@@ -248,6 +248,19 @@ public Object sub(VirtualFrame frame, int a, RubyBasicObject b) {
             return rationalSubNode.call(frame, aRational, "-", null, b);
         }
 
+        @Specialization(guards = "isRational(arguments[1])")
+        public Object sub(VirtualFrame frame, long a, RubyBasicObject b) {
+            if (rationalConvertNode == null) {
+                CompilerDirectives.transferToInterpreter();
+                rationalConvertNode = insert(DispatchHeadNodeFactory.createMethodCall(getContext(), true));
+                rationalSubNode = insert(DispatchHeadNodeFactory.createMethodCall(getContext()));
+            }
+
+            final Object aRational = rationalConvertNode.call(frame, getContext().getCoreLibrary().getRationalClass(), "convert", null, a, 1);
+
+            return rationalSubNode.call(frame, aRational, "-", null, b);
+        }
+
         @Specialization(rewriteOn = ArithmeticException.class)
         public long sub(long a, int b) {
             return ExactMath.subtractExact(a, b);