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 535fcd3e5f8cb892f64dc02eb6e4175c04344365..6054571336ed101324eeb77f761d3366e385bc01 100644
--- a/ui/aura/mus/window_tree_client.h
+++ b/ui/aura/mus/window_tree_client.h
@@ -475,6 +475,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_;