Add GetDisplayWithDisplayId to display::Screen.

headless/lib/browser/headless_screen.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "headless/lib/browser/headless_screen.h"
 
 #include <stdint.h>
 
 #include "base/logging.h"
 #include "ui/aura/env.h"
@@ skipping 122 matching lines @@
 }
 
 int HeadlessScreen::GetNumDisplays() const {
   return 1;
 }
 
 std::vector<display::Display> HeadlessScreen::GetAllDisplays() const {
   return std::vector<display::Display>(1, display_);
 }
 
+bool HeadlessScreen::GetDisplayWithDisplayId(int64_t display_id,
+                                             display::Display* display) const {
+  if (display_.id() == display_id) {
+    *display = display_;
+    return true;
+  }
+  return false;
+}
+
 display::Display HeadlessScreen::GetDisplayNearestWindow(
     gfx::NativeWindow window) const {
   return display_;
 }
 
 display::Display HeadlessScreen::GetDisplayNearestPoint(
     const gfx::Point& point) const {
   return display_;
 }
 
@@ skipping 11 matching lines @@
 void HeadlessScreen::RemoveObserver(display::DisplayObserver* observer) {}
 
 HeadlessScreen::HeadlessScreen(const gfx::Rect& screen_bounds)
     : host_(NULL), ui_scale_(1.0f) {
   static int64_t synthesized_display_id = 2000;
   display_.set_id(synthesized_display_id++);
   display_.SetScaleAndBounds(1.0f, screen_bounds);
 }
 
 }  // namespace headless