Support multiple displays in EventGenerator

ui/aura/test/event_generator.h

 // Copyright (c) 2012 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.
 
 #ifndef UI_AURA_TEST_EVENT_GENERATOR_H_
 #define UI_AURA_TEST_EVENT_GENERATOR_H_
 
 #include "base/basictypes.h"
+#include "base/memory/scoped_ptr.h"
 #include "ui/base/keycodes/keyboard_codes.h"
 #include "ui/gfx/point.h"
 
 namespace base {
 class TimeDelta;
 }
 
 namespace ui {
 class Event;
 }
 
 namespace aura {
 class RootWindow;
 class Window;
 
+namespace client {
+class ScreenPositionClient;
+}
+
 namespace test {
 
-// EventGenerator is a tool that generates and dispatch events.
+// A delegate interface for EventGenerator that provides a way to
+// locate aura root window for given point.
+class EventGeneratorDelegate {
+ public:
+  virtual ~EventGeneratorDelegate() {}
+
+  // Returns a root window for given point.
+  virtual RootWindow* GetRootWindowAt(const gfx::Point& point) const = 0;
+
+  // Returns the screen position client that determines the
+  // coordinates used in EventGenerator. EventGenerator uses
+  // RootWindow's coordinate if this retruns NULL.
+  virtual client::ScreenPositionClient* GetScreenPositionClient(
+      const aura::Window* window) const = 0;
+};
+
+// EventGenerator is a tool that generates and dispatch events. The
+// coordinates of the points in API is determined by the
+// EventGeneratorDelegate.
 class EventGenerator {
  public:
-  // Creates an EventGenerator with the mouse/touch location (0,0).
+  // Creates an EventGenerator with the mouse/touch location (0,0),
+  // which uses the |root_window|'s coordinates.
   explicit EventGenerator(RootWindow* root_window);
 
+  // Create an EventGenerator with EventGeneratorDelegate,
+  // which uses the coordinates used by |delegate|.
+  explicit EventGenerator(EventGeneratorDelegate* delegate);
+
   // Creates an EventGenerator with the mouse/touch location
-  // at |initial_location|.
+  // at |initial_location|, which uses the |root_window|'s coordinates.
   EventGenerator(RootWindow* root_window, const gfx::Point& initial_location);
 
   // Creates an EventGenerator with the mouse/touch location
-  // centered over |window|.
+  // centered over |window|, which uses the |root_window|'s coordinates.
   EventGenerator(RootWindow* root_window, Window* window);
 
   virtual ~EventGenerator();
 
   // Explicitly sets the location used by mouse/touch events. This is set by the
   // various methods that take a location but can be manipulated directly,
   // typically for touch.
   void set_current_location(const gfx::Point& location) {
     current_location_ = location;
   }
@@ skipping 121 matching lines @@
 
   // Generates a key release event. On platforms except Windows and X11, a key
   // event without native_event() is generated. Note that ui::EF_ flags should
   // be passed as |flags|, not the native ones like 'ShiftMask' in <X11/X.h>.
   // TODO(yusukes): Support native_event() on all platforms.
   void ReleaseKey(ui::KeyboardCode key_code, int flags);
 
   // Dispatch the |event| to the RootWindow.
   void Dispatch(ui::Event& event);
 
+  void set_current_root_window(RootWindow* root_window) {
+    current_root_window_ = root_window;
+  }
+
  private:
   // Dispatch a key event to the RootWindow.
   void DispatchKeyEvent(bool is_press, ui::KeyboardCode key_code, int flags);
 
-  RootWindow* root_window_;
+  void UpdateCurrentRootWindow(const gfx::Point& point);
+  void PressButton(int flag);
+  void ReleaseButton(int flag);
+
+  // Convert a point between API's coordinates and
+  // |target|'s coordinates.
+  void ConvertPointFromTarget(const aura::Window* target,
+                              gfx::Point* point) const;
+  void ConvertPointToTarget(const aura::Window* target,
+                            gfx::Point* point) const;
+
+  gfx::Point GetLocationInCurrentRoot() const;
+  gfx::Point CenterOfWindow(const Window* window) const;
+
+  scoped_ptr<EventGeneratorDelegate> delegate_;
+  gfx::Point current_location_;
+  RootWindow* current_root_window_;
   int flags_;
-  gfx::Point current_location_;
+  bool grab_;
 
   DISALLOW_COPY_AND_ASSIGN(EventGenerator);
 };
 
 }  // namespace test
 }  // namespace aura
 
 #endif  // UI_AURA_TEST_EVENT_GENERATOR_H_