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 fa338c8e0bc05915dd7288951555287972abc7d0..3f14f1a8867e43f340d379cdcba9dee958d58b32 100644
--- a/ui/aura/mus/window_tree_client.h
+++ b/ui/aura/mus/window_tree_client.h
@@ -491,6 +491,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_;