Fix software mirror mode on Ozone part 2/2

ash/touch/touch_transformer_controller.cc

 // Copyright 2014 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 "ash/touch/touch_transformer_controller.h"
 
 #include "ash/display/display_controller.h"
 #include "ash/display/display_manager.h"
 #include "ash/host/ash_window_tree_host.h"
 #include "ash/root_window_controller.h"
@@ skipping 108 matching lines @@
 TouchTransformerController::TouchTransformerController() {
   Shell::GetInstance()->display_controller()->AddObserver(this);
 }
 
 TouchTransformerController::~TouchTransformerController() {
   Shell::GetInstance()->display_controller()->RemoveObserver(this);
 }
 
 void TouchTransformerController::UpdateTouchTransformer() const {
   ui::DeviceDataManager* device_manager = ui::DeviceDataManager::GetInstance();
-  device_manager->ClearTouchTransformerRecord();
+  device_manager->ClearTouchDeviceAssociations();
 
   // Display IDs and DisplayInfo for mirror or extended mode.
   int64 display1_id = gfx::Display::kInvalidDisplayID;
   int64 display2_id = gfx::Display::kInvalidDisplayID;
   DisplayInfo display1;
   DisplayInfo display2;
   // Display ID and DisplayInfo for single display mode.
   int64 single_display_id = gfx::Display::kInvalidDisplayID;
   DisplayInfo single_display;
 
@@ skipping 31 matching lines @@
         single_display.touch_device_id(),
         GetTouchResolutionScale(
             single_display,
             FindTouchscreenById(single_display.touch_device_id())));
   }
 
   gfx::Size fb_size =
       Shell::GetInstance()->display_configurator()->framebuffer_size();
 
   if (display_state == ui::MULTIPLE_DISPLAY_STATE_DUAL_MIRROR) {
-    // In mirror mode, both displays share the same root window so
-    // both display ids are associated with the root window.
-    aura::Window* root = display_controller->GetPrimaryRootWindow();
-    RootWindowController::ForWindow(root)->ash_host()->UpdateDisplayID(
-        display1_id, display2_id);
+    // In mirror mode, there is just one WindowTreeHost and two displays. Make
+    // the WindowTreeHost accept touch events from both displays.
+    int64 primary_display_id = display_controller->GetPrimaryDisplayId();
     device_manager->UpdateTouchInfoForDisplay(
-        display1_id, display1.touch_device_id(),
+        primary_display_id, display1.touch_device_id(),
         GetTouchTransform(display1, display1,
                           FindTouchscreenById(display1.touch_device_id()),
                           fb_size));
     device_manager->UpdateTouchInfoForDisplay(
-        display2_id, display2.touch_device_id(),
+        primary_display_id, display2.touch_device_id(),
         GetTouchTransform(display2, display2,
                           FindTouchscreenById(display2.touch_device_id()),
                           fb_size));
     return;
   }
 
   if (display_state == ui::MULTIPLE_DISPLAY_STATE_DUAL_EXTENDED) {
-    // In extended but software mirroring mode, ther is only one X root window
-    // that associates with both displays.
+    // In extended but software mirroring mode, there is a WindowTreeHost for
+    // each display, but all touches are forwarded to the primary root window's
+    // WindowTreeHost.
     if (GetDisplayManager()->software_mirroring_enabled())  {
-      aura::Window* root = display_controller->GetPrimaryRootWindow();
-      RootWindowController::ForWindow(root)->ash_host()->UpdateDisplayID(
-          display1_id, display2_id);
-      DisplayInfo source_display =
+      DisplayInfo target_display =
           display_controller->GetPrimaryDisplayId() == display1_id ? display1
                                                                    : display2;
-      // Mapping from framebuffer size to the source display's native
-      // resolution.
       device_manager->UpdateTouchInfoForDisplay(
-          display1_id, display1.touch_device_id(),
-          GetTouchTransform(source_display, display1,
+          target_display.id(), display1.touch_device_id(),
+          GetTouchTransform(target_display, display1,
                             FindTouchscreenById(display1.touch_device_id()),
                             fb_size));
       device_manager->UpdateTouchInfoForDisplay(
-          display2_id, display2.touch_device_id(),
-          GetTouchTransform(source_display, display2,
+          target_display.id(), display2.touch_device_id(),
+          GetTouchTransform(target_display, display2,
                             FindTouchscreenById(display2.touch_device_id()),
                             fb_size));
     } else {
-      // In actual extended mode, each display is associated with one root
-      // window.
-      aura::Window* root1 =
-          display_controller->GetRootWindowForDisplayId(display1_id);
-      aura::Window* root2 =
-          display_controller->GetRootWindowForDisplayId(display2_id);
-      RootWindowController::ForWindow(root1)->ash_host()->UpdateDisplayID(
-          display1_id, gfx::Display::kInvalidDisplayID);
-      RootWindowController::ForWindow(root2)->ash_host()->UpdateDisplayID(
-          display2_id, gfx::Display::kInvalidDisplayID);
-      // Mapping from framebuffer size to each display's native resolution.
+      // In actual extended mode, each display is associated with one
+      // WindowTreeHost.
       device_manager->UpdateTouchInfoForDisplay(
           display1_id, display1.touch_device_id(),
           GetTouchTransform(display1, display1,
                             FindTouchscreenById(display1.touch_device_id()),
                             fb_size));
       device_manager->UpdateTouchInfoForDisplay(
           display2_id, display2.touch_device_id(),
           GetTouchTransform(display2, display2,
                             FindTouchscreenById(display2.touch_device_id()),
                             fb_size));
     }
     return;
   }
 
-  // Single display mode. The root window has one associated display id.
-  aura::Window* root =
-      display_controller->GetRootWindowForDisplayId(single_display.id());
-  RootWindowController::ForWindow(root)->ash_host()->UpdateDisplayID(
-      single_display.id(), gfx::Display::kInvalidDisplayID);
+  // Single display mode. The WindowTreeHost has one associated display id.
   device_manager->UpdateTouchInfoForDisplay(
       single_display_id, single_display.touch_device_id(),
       GetTouchTransform(single_display, single_display,
                         FindTouchscreenById(single_display.touch_device_id()),
                         fb_size));
 }
 
 void TouchTransformerController::OnDisplaysInitialized() {
   UpdateTouchTransformer();
 }
 
 void TouchTransformerController::OnDisplayConfigurationChanged() {
   UpdateTouchTransformer();
 }
 
 }  // namespace ash