Add pid parameter instead of EVENT_ZCM1_ADDED/EVENT_ZCM2_ADDED

Author: wmww

src/daemon/moved.cpp

@@ -94,13 +94,12 @@ static void
 on_monitor_update_moved(enum MonitorEvent event,
         enum MonitorEventDeviceType device_type,
         const char *path, const wchar_t *serial,
-        void *user_data)
+        unsigned short pid, void *user_data)
 {
     move_daemon *moved = static_cast<move_daemon *>(user_data);
 
-    if (event == EVENT_ZCM1_ADDED || event == EVENT_ZCM2_ADDED) {
+    if (event == EVENT_DEVICE_ADDED) {
         if (device_type == EVENT_DEVICE_TYPE_USB) {
-            unsigned short pid = event == EVENT_ZCM1_ADDED ? PSMOVE_PID : PSMOVE_PS4_PID;
             PSMove *move = psmove_connect_internal(serial, path, -1, pid);
             if (psmove_pair(move)) {
                 // Indicate to the user that pairing was successful

src/daemon/moved_monitor.h

@@ -37,8 +37,7 @@ extern "C" {
 #include <wchar.h>
 
 enum MonitorEvent {
-    EVENT_ZCM1_ADDED,
-    EVENT_ZCM2_ADDED,
+    EVENT_DEVICE_ADDED,
     EVENT_DEVICE_REMOVED,
 };
 
@@ -48,9 +47,10 @@ enum MonitorEventDeviceType {
     EVENT_DEVICE_TYPE_UNKNOWN,
 };
 
+// pid is the product ID on connect, it may be 0 on disconnect
 typedef void (*moved_event_callback)(enum MonitorEvent event,
         enum MonitorEventDeviceType device_type, const char *path,
-        const wchar_t *serial, void *user_data);
+        const wchar_t *serial, unsigned short pid, void *user_data);
 
 typedef struct _moved_monitor moved_monitor;
 

src/daemon/moved_monitor_linux.c

@@ -158,8 +158,8 @@ _moved_monitor_handle_device(moved_monitor *monitor, struct udev_device *dev)
                     device_type = EVENT_DEVICE_TYPE_USB;
                 }
 
-                monitor->event_callback(product_id == PSMOVE_PID ? EVENT_ZCM1_ADDED : EVENT_ZCM2_ADDED, device_type, path,
-                        serial_number, monitor->event_callback_user_data);
+                monitor->event_callback(EVENT_DEVICE_ADDED, device_type, path,
+                        serial_number, product_id, monitor->event_callback_user_data);
             }
 
             free(serial_number);
@@ -170,7 +170,7 @@ _moved_monitor_handle_device(moved_monitor *monitor, struct udev_device *dev)
         free(uevent);
     } else if (strcmp(action, "remove") == 0) {
         monitor->event_callback(EVENT_DEVICE_REMOVED, device_type, path,
-                NULL, monitor->event_callback_user_data);
+                NULL, 0, monitor->event_callback_user_data);
     }
 }
 

src/daemon/moved_monitor_osx.mm

@@ -49,6 +49,7 @@
     enum MonitorEventDeviceType device_type;
     char path[256];
     wchar_t serial_number[256];
+    unsigned short pid;
 };
 
 struct _moved_monitor {
@@ -85,7 +86,7 @@ void pump_loop()
             auto event = events.front();
             event_callback(event.event, event.device_type,
                            event.path, event.serial_number,
-                           event_callback_user_data);
+                           event.pid, event_callback_user_data);
             events.pop();
         }
     }
@@ -94,10 +95,12 @@ void pump_loop()
     void
     make_event(enum MonitorEvent event, IOHIDDeviceRef device)
     {
-        if (get_vendor_id(device) == PSMOVE_VID && (get_product_id(device) == PSMOVE_PID || get_product_id(device) == PSMOVE_PS4_PID)) {
+        unsigned short pid = get_product_id(device);
+        if (get_vendor_id(device) == PSMOVE_VID && (pid == PSMOVE_PID || pid == PSMOVE_PS4_PID)) {
             ASyncDeviceEvent ade;
             ade.event = event;
             ade.device_type = EVENT_DEVICE_TYPE_UNKNOWN;
+            ade.pid = pid;
 
             // Example paths:
             // e.g. "USB_054c_03d5_14100000"
@@ -119,7 +122,7 @@ void pump_loop()
     static void on_device_matching(void *context, IOReturn result, void *sender, IOHIDDeviceRef device)
     {
         _moved_monitor *monitor = (_moved_monitor *)(context);
-        monitor->make_event(get_product_id(device) == PSMOVE_PID ? EVENT_ZCM1_ADDED : EVENT_ZCM2_ADDED, device);
+        monitor->make_event(EVENT_DEVICE_ADDED, device);
     }
 
     static void on_device_removal(void *context, IOReturn result, void *sender, IOHIDDeviceRef device)

src/psmoveapi.cpp

@@ -70,7 +70,7 @@ struct PSMoveAPI {
 
     void update();
 
-    static void on_monitor_event(enum MonitorEvent event, enum MonitorEventDeviceType device_type, const char *path, const wchar_t *serial, void *user_data);
+    static void on_monitor_event(enum MonitorEvent event, enum MonitorEventDeviceType device_type, const char *path, const wchar_t *serial, unsigned short pid, void *user_data);
 
     EventReceiver *receiver;
     void *user_data;
@@ -290,16 +290,14 @@ PSMoveAPI::update()
 }
 
 void
-PSMoveAPI::on_monitor_event(enum MonitorEvent event, enum MonitorEventDeviceType device_type, const char *path, const wchar_t *serial, void *user_data)
+PSMoveAPI::on_monitor_event(enum MonitorEvent event, enum MonitorEventDeviceType device_type, const char *path, const wchar_t *serial, unsigned short pid, void *user_data)
 {
     auto self = static_cast<PSMoveAPI *>(user_data);
 
     switch (event) {
-        case EVENT_ZCM1_ADDED:
-        case EVENT_ZCM2_ADDED:
+        case EVENT_DEVICE_ADDED:
             {
-                PSMOVE_DEBUG("on_monitor_event(event=%s, device_type=0x%08x, path=\"%s\", serial=%p)",
-                       event == EVENT_ZCM1_ADDED ? "EVENT_ZCM1_ADDED" : "EVENT_ZCM2_ADDED",
+                PSMOVE_DEBUG("on_monitor_event(event=EVENT_DEVICE_ADDED, device_type=0x%08x, path=\"%s\", serial=%p)",
                        device_type, path, serial);
 
                 for (auto &c: self->controllers) {
@@ -310,7 +308,6 @@ PSMoveAPI::on_monitor_event(enum MonitorEvent event, enum MonitorEventDeviceType
                     }
                 }
 
-                unsigned short pid = event == EVENT_ZCM1_ADDED ? PSMOVE_PID : PSMOVE_PS4_PID;
                 PSMove *move = psmove_connect_internal(serial, path, -1, pid);
                 if (move == nullptr) {
                     PSMOVE_ERROR("Cannot open move for retrieving serial!");

src/utils/psmovepair.c

@@ -95,9 +95,9 @@ static void
 on_monitor_update_pair(enum MonitorEvent event,
         enum MonitorEventDeviceType device_type,
         const char *path, const wchar_t *serial,
-        void *user_data)
+        unsigned short pid, void *user_data)
 {
-    if (event == EVENT_ZCM1_ADDED || event == EVENT_ZCM2_ADDED) {
+    if (event == EVENT_DEVICE_ADDED) {
         if (device_type == EVENT_DEVICE_TYPE_USB) {
             pair(NULL);
         }