[Truffle] Add Thread::Backtrace::Location#lineno.

jruby · May 1, 2015 · 1457605 · 1457605
1 parent fc0e290
commit 1457605
Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/spec/truffle/tags/core/thread/backtrace/location/absolute_path_tags.txt b/spec/truffle/tags/core/thread/backtrace/location/absolute_path_tags.txt
diff --git a/spec/truffle/tags/core/thread/backtrace/location/lineno_tags.txt b/spec/truffle/tags/core/thread/backtrace/location/lineno_tags.txt
diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/core/KernelNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/core/KernelNodes.java
@@ -297,7 +297,7 @@ public RubySymbol calleeName(VirtualFrame frame) {
         }
     }
 
-    @CoreMethod(names = "caller_locations", isModuleFunction = true, optional = 2)
+    @CoreMethod(names = "caller_locations", isModuleFunction = true, optional = 2, lowerFixnumParameters = {0, 1})
     public abstract static class CallerLocationsNode extends CoreMethodArrayArgumentsNode {
 
         public CallerLocationsNode(RubyContext context, SourceSection sourceSection) {

diff --git a/truffle/src/main/java/org/jruby/truffle/nodes/core/ThreadBacktraceLocationNodes.java b/truffle/src/main/java/org/jruby/truffle/nodes/core/ThreadBacktraceLocationNodes.java
@@ -88,6 +88,25 @@ public RubyString absolutePath(RubyBasicObject threadBacktraceLocation) {
 
     }
 
+    @CoreMethod(names = "lineno")
+    public abstract static class LinenoNode extends UnaryCoreMethodNode {
+
+        public LinenoNode(RubyContext context, SourceSection sourceSection) {
+            super(context, sourceSection);
+        }
+
+        @CompilerDirectives.TruffleBoundary
+        @Specialization
+        public int lineno(RubyBasicObject threadBacktraceLocation) {
+            final Activation activation = getActivation(threadBacktraceLocation);
+
+            final SourceSection sourceSection = activation.getCallNode().getEncapsulatingSourceSection();
+
+            return sourceSection.getStartLine();
+        }
+
+    }
+
     @CoreMethod(names = {"to_s", "inspect"})
     public abstract static class ToSNode extends UnaryCoreMethodNode {