Refactored gamepad polling to support dynamic sources

device/gamepad/gamepad_provider.h

 // Copyright (c) 2012 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.
 
 #ifndef DEVICE_GAMEPAD_GAMEPAD_PROVIDER_H_
 #define DEVICE_GAMEPAD_GAMEPAD_PROVIDER_H_
 
+#include <stdint.h>
+
 #include <memory>
 #include <utility>
 #include <vector>
 
 #include "base/callback_forward.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/shared_memory.h"
-#include "base/memory/weak_ptr.h"
 #include "base/synchronization/lock.h"
 #include "base/system_monitor/system_monitor.h"
 #include "device/gamepad/gamepad_export.h"
 #include "device/gamepad/gamepad_shared_buffer.h"
+#include "device/gamepad/gamepad_standard_mappings.h"
 #include "third_party/WebKit/public/platform/WebGamepads.h"
 
 namespace base {
 class SingleThreadTaskRunner;
 class Thread;
 }
 
 namespace device {
 
 class GamepadDataFetcher;
 
+enum GamepadSource {
+  GAMEPAD_SOURCE_NONE = 0,
+  GAMEPAD_SOURCE_ANDROID,
+  GAMEPAD_SOURCE_LINUX_UDEV,
+  GAMEPAD_SOURCE_MAC_HID,
+  GAMEPAD_SOURCE_MAC_XBOX,
+  GAMEPAD_SOURCE_TEST,
+  GAMEPAD_SOURCE_WIN_XINPUT,
+  GAMEPAD_SOURCE_WIN_RAW,
+};
+
+enum GamepadActiveState {
+  GAMEPAD_INACTIVE = 0,
+  GAMEPAD_ACTIVE,
+  GAMEPAD_NEWLY_ACTIVE,
+};
+
+struct PadState {
+  // Which data fetcher provided this gamepad's data.
+  GamepadSource source;
+  // Data fetcher-specific identifier for this gamepad.
+  int source_id;
+
+  // Indicates whether or not the gamepad is actively being updated
+  GamepadActiveState active_state;
+
+  // Gamepad data, unmapped.
+  blink::WebGamepad data;
+
+  // Functions to map from device data to standard layout, if available. May
+  // be null if no mapping is available or needed.
+  GamepadStandardMappingFunction mapper;
+
+  // Sanitization masks
+  // axis_mask and button_mask are bitfields that represent the reset state of
+  // each input. If a button or axis has ever reported 0 in the past the
+  // corresponding bit will be set to 1.
+
+  // If we ever increase the max axis count this will need to be updated.
+  static_assert(blink::WebGamepad::axesLengthCap <=
+                    std::numeric_limits<uint32_t>::digits,
+                "axis_mask is not large enough");
+  uint32_t axis_mask;
+
+  // If we ever increase the max button count this will need to be updated.
+  static_assert(blink::WebGamepad::buttonsLengthCap <=
+                    std::numeric_limits<uint32_t>::digits,
+                "button_mask is not large enough");
+  uint32_t button_mask;
+};
+
 class DEVICE_GAMEPAD_EXPORT GamepadConnectionChangeClient {
  public:
   virtual void OnGamepadConnectionChange(bool connected,
                                          int index,
                                          const blink::WebGamepad& pad) = 0;
 };
 
 class DEVICE_GAMEPAD_EXPORT GamepadProvider
     : public base::SystemMonitor::DevicesChangedObserver {
  public:
   explicit GamepadProvider(
       std::unique_ptr<GamepadSharedBuffer> buffer,
       GamepadConnectionChangeClient* connection_change_client);
 
   // Manually specifies the data fetcher. Used for testing.
   explicit GamepadProvider(
       std::unique_ptr<GamepadSharedBuffer> buffer,
       GamepadConnectionChangeClient* connection_change_client,
       std::unique_ptr<GamepadDataFetcher> fetcher);
 
   ~GamepadProvider() override;
 
   // Returns the shared memory handle of the gamepad data duplicated into the
   // given process.
   base::SharedMemoryHandle GetSharedMemoryHandleForProcess(
       base::ProcessHandle renderer_process);
 
+  void AddGamepadDataFetcher(GamepadDataFetcher* fetcher);
+  void RemoveGamepadDataFetcher(GamepadDataFetcher* fetcher);
+
   void GetCurrentGamepadData(blink::WebGamepads* data);
 
   // Pause and resume the background polling thread. Can be called from any
   // thread.
   void Pause();
   void Resume();
 
   // Registers the given closure for calling when the user has interacted with
   // the device. This callback will only be issued once.
   void RegisterForUserGesture(const base::Closure& closure);
 
   // base::SystemMonitor::DevicesChangedObserver implementation.
   void OnDevicesChanged(base::SystemMonitor::DeviceType type) override;
 
+  // Add a gamepad data fetcher. Takes ownership of |fetcher|.
+  void AddGamepadDataFetcher(std::unique_ptr<GamepadDataFetcher> fetcher);
+
+  // Remove gamepad data fetchers with the given source.
+  void RemoveSourceGamepadDataFetcher(GamepadSource source);
+
+  // Gets a PadState object for the given source and id. If the device hasn't
+  // been encountered before one of the remaining slots will be reserved for it.
+  // If no slots are available will return NULL.
+  PadState* GetPadState(GamepadSource source, int source_id);
+
+  void SetSanitizationEnabled(bool sanitize) { sanitize_ = sanitize; }
+
  private:
   void Initialize(std::unique_ptr<GamepadDataFetcher> fetcher);
 
   // Method for setting up the platform-specific data fetcher. Takes ownership
   // of |fetcher|.
-  void DoInitializePollingThread(std::unique_ptr<GamepadDataFetcher> fetcher);
+  void DoAddGamepadDataFetcher(std::unique_ptr<GamepadDataFetcher> fetcher);
+  void DoRemoveSourceGamepadDataFetcher(GamepadSource source);
 
   // Method for sending pause hints to the low-level data fetcher. Runs on
   // polling_thread_.
   void SendPauseHint(bool paused);
 
   // Method for polling a GamepadDataFetcher. Runs on the polling_thread_.
   void DoPoll();
   void ScheduleDoPoll();
 
   void OnGamepadConnectionChange(bool connected,
                                  int index,
                                  const blink::WebGamepad& pad);
 
   // Checks the gamepad state to see if the user has interacted with it.
   void CheckForUserGesture();
+  void ClearPadState(PadState& state);
+
+  void MapAndSanitizeGamepadData(PadState* pad_state, blink::WebGamepad* pad);
 
   enum { kDesiredSamplingIntervalMs = 16 };
 
   // Keeps track of when the background thread is paused. Access to is_paused_
   // must be guarded by is_paused_lock_.
   base::Lock is_paused_lock_;
   bool is_paused_;
 
   // Keep track of when a polling task is schedlued, so as to prevent us from
   // accidentally scheduling more than one at any time, when rapidly toggling
@@ skipping 20 matching lines @@
 
   // Updated based on notification from SystemMonitor when the system devices
   // have been updated, and this notification is passed on to the data fetcher
   // to enable it to avoid redundant (and possibly expensive) is-connected
   // tests. Access to devices_changed_ must be guarded by
   // devices_changed_lock_.
   base::Lock devices_changed_lock_;
   bool devices_changed_;
 
   bool ever_had_user_gesture_;
+  bool sanitize_;
 
-  class PadState {
-   public:
-    PadState() { SetDisconnected(); }
-
-    bool Match(const blink::WebGamepad& pad) const;
-    void SetPad(const blink::WebGamepad& pad);
-    void SetDisconnected();
-    void AsWebGamepad(blink::WebGamepad* pad);
-
-    bool connected() const { return connected_; }
-
-   private:
-    bool connected_;
-    unsigned axes_length_;
-    unsigned buttons_length_;
-    blink::WebUChar id_[blink::WebGamepad::idLengthCap];
-    blink::WebUChar mapping_[blink::WebGamepad::mappingLengthCap];
-  };
-
-  // Used to detect connections and disconnections.
+  // Tracks the state of each gamepad slot.
   std::unique_ptr<PadState[]> pad_states_;
 
   // Only used on the polling thread.
-  std::unique_ptr<GamepadDataFetcher> data_fetcher_;
+  typedef std::vector<std::unique_ptr<GamepadDataFetcher>> GamepadFetcherVector;
+  GamepadFetcherVector data_fetchers_;
 
   base::Lock shared_memory_lock_;
   std::unique_ptr<GamepadSharedBuffer> gamepad_shared_buffer_;
 
   // Polling is done on this background thread.
   std::unique_ptr<base::Thread> polling_thread_;
 
   GamepadConnectionChangeClient* connection_change_client_;
 
-  static GamepadProvider* instance_;
-
   DISALLOW_COPY_AND_ASSIGN(GamepadProvider);
 };
 
 }  // namespace device
 
 #endif  // DEVICE_GAMEPAD_GAMEPAD_PROVIDER_H_