From 56da6399b846874789955562efca8dae6b5feca7 Mon Sep 17 00:00:00 2001
From: Ben Ogle & Nathan Sobo <benogle+nathan@github.com>
Date: Thu, 12 Jun 2014 14:33:14 -0600
Subject: [PATCH] Add startScreenRow and endScreenRow to
 DisplayBuffer::findMarkers

---
 spec/display-buffer-spec.coffee | 6 ++++++
 src/display-buffer.coffee       | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/spec/display-buffer-spec.coffee b/spec/display-buffer-spec.coffee
index 40c5f1384..0d8c42a72 100644
--- a/spec/display-buffer-spec.coffee
+++ b/spec/display-buffer-spec.coffee
@@ -912,6 +912,12 @@ describe "DisplayBuffer", ->
         expect(displayBuffer.findMarkers(class: 'a', startBufferRow: 0, endBufferRow: 3)).toEqual [marker1]
         expect(displayBuffer.findMarkers(endBufferRow: 10)).toEqual [marker3]
 
+      it "allows the startScreenRow and endScreenRow to be specified", ->
+        marker1 = displayBuffer.markBufferRange([[6, 0], [7, 0]], class: 'a')
+        marker2 = displayBuffer.markBufferRange([[9, 0], [10, 0]], class: 'a')
+        displayBuffer.createFold(4, 7)
+        expect(displayBuffer.findMarkers(class: 'a', startScreenRow: 6, endScreenRow: 7)).toEqual [marker2]
+
     describe "marker destruction", ->
       it "allows markers to be destroyed", ->
         marker = displayBuffer.markScreenRange([[5, 4], [5, 10]])
diff --git a/src/display-buffer.coffee b/src/display-buffer.coffee
index 9911848f1..a062696a2 100644
--- a/src/display-buffer.coffee
+++ b/src/display-buffer.coffee
@@ -912,6 +912,12 @@ class DisplayBuffer extends Model
           key = 'startRow'
         when 'endBufferRow'
           key = 'endRow'
+        when 'startScreenRow'
+          key = 'startRow'
+          value = @bufferRowForScreenRow(value)
+        when 'endScreenRow'
+          key = 'endRow'
+          value = @bufferRowForScreenRow(value)
         when 'containsBufferRange'
           key = 'containsRange'
         when 'containsBufferPosition'