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 dade587123f887dfd4d25bdfb9ac611438bea935..8a8090a4b17457caaa81cdc2e60174e62253b2c4 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_;