Change OnWindowInputEvent to use display_id to find the host and update event root_location in WS.

ui/aura/mus/window_tree_client.h

Index: ui/aura/mus/window_tree_client.h
diff --git a/ui/aura/mus/window_tree_client.h b/ui/aura/mus/window_tree_client.h
index 63073714c818f062032f27e0558111c8cbff5323..8c87d938710fbe10dee87007eb4de5cf2cac471d 100644
--- a/ui/aura/mus/window_tree_client.h
+++ b/ui/aura/mus/window_tree_client.h
@@ -478,6 +478,9 @@ class AURA_EXPORT WindowTreeClient
         cursor_location_mapping_.get());
   }
 
+  // Helper function to get the WindowTreeHostMus with the |display_id|.
+  WindowTreeHostMus* GetWindowTreeHostMusWithDisplayId(int64_t display_id);
+
   // This may be null in tests.
   service_manager::Connector* connector_;