diff --git a/PS3BT.cpp b/PS3BT.cpp index d80f0b52..34afd011 100644 --- a/PS3BT.cpp +++ b/PS3BT.cpp @@ -516,25 +516,23 @@ void PS3BT::Run() { case L2CAP_HID_PS3_LED: if (millis() - timer > 1000) { // loop 1 second before sending the command if (remote_name[0] == 'P') { // First letter in PLAYSTATION(R)3 Controller ('P') - setLedOn(LED1); #ifdef DEBUG_USB_HOST Notify(PSTR("\r\nDualshock 3 Controller Enabled\r\n"), 0x80); #endif PS3Connected = true; } else if (remote_name[0] == 'N') { // First letter in Navigation Controller ('N') - setLedOn(LED1); // This just turns LED constantly on, on the Navigation controller #ifdef DEBUG_USB_HOST Notify(PSTR("\r\nNavigation Controller Enabled\r\n"), 0x80); #endif PS3NavigationConnected = true; } else if (remote_name[0] == 'M') { // First letter in Motion Controller ('M') - moveSetBulb(Red); timerBulbRumble = millis(); #ifdef DEBUG_USB_HOST Notify(PSTR("\r\nMotion Controller Enabled\r\n"), 0x80); #endif PS3MoveConnected = true; } + onInit(); // Turn on the LED on the controller l2cap_state = L2CAP_DONE; } break; @@ -659,3 +657,14 @@ void PS3BT::moveSetRumble(uint8_t rumble) { HIDMove_Command(HIDMoveBuffer, HID_BUFFERSIZE); } + +void PS3BT::onInit() { + if (pFuncOnInit) + pFuncOnInit(); // Call the user function + else { + if (PS3MoveConnected) + moveSetBulb(Red); + else // Dualshock 3 or Navigation controller + setLedOn(LED1); + } +} \ No newline at end of file diff --git a/PS3BT.h b/PS3BT.h index 1336828e..5b00232f 100644 --- a/PS3BT.h +++ b/PS3BT.h @@ -207,19 +207,35 @@ public: * @param rumble The desired value in the range from 64-255. */ void moveSetRumble(uint8_t rumble); + + /** + * Used to call your own function when the controller is successfully initialized. + * @param funcOnInit Function to call. + */ + void attachOnInit(void (*funcOnInit)(void)) { + pFuncOnInit = funcOnInit; + }; /**@}*/ - /** Variable used to indicate if the normal playstation controller is successfully connected. */ + /** Variable used to indicate if the normal Playstation controller is successfully connected. */ bool PS3Connected; - /** Variable used to indicate if the move controller is successfully connected. */ + /** Variable used to indicate if the Move controller is successfully connected. */ bool PS3MoveConnected; - /** Variable used to indicate if the navigation controller is successfully connected. */ + /** Variable used to indicate if the Navigation controller is successfully connected. */ bool PS3NavigationConnected; private: - /* mandatory members */ + /* Mandatory members */ BTD *pBtd; + /** + * Called when the controller is successfully initialized. + * Use attachOnInit(void (*funcOnInit)(void)) to call your own function. + * This is useful for instance if you want to set the LEDs in a specific way. + */ + void onInit(); + void (*pFuncOnInit)(void); // Pointer to function called in onInit() + void L2CAP_task(); // L2CAP state machine /* Variables filled from HCI event management */ @@ -229,7 +245,7 @@ private: /* variables used by high level L2CAP task */ uint8_t l2cap_state; - uint16_t l2cap_event_flag; // L2CAP flags of received bluetooth events + uint16_t l2cap_event_flag; // L2CAP flags of received Bluetooth events unsigned long timer; diff --git a/PS3USB.cpp b/PS3USB.cpp index 7bbcfd9d..7fbb0a46 100644 --- a/PS3USB.cpp +++ b/PS3USB.cpp @@ -183,10 +183,9 @@ uint8_t PS3USB::Init(uint8_t parent, uint8_t port, bool lowspeed) { #endif PS3NavigationConnected = true; } - /* Set internal bluetooth address and request for data */ + /* Set internal Bluetooth address and request for data */ setBdaddr(my_bdaddr); enable_sixaxis(); - setLedOn(LED1); // Needed for PS3 Dualshock and Navigation commands to work for (uint8_t i = 0; i < PS3_REPORT_BUFFER_SIZE; i++) @@ -199,12 +198,12 @@ uint8_t PS3USB::Init(uint8_t parent, uint8_t port, bool lowspeed) { Notify(PSTR("\r\nMotion Controller Connected"), 0x80); #endif PS3MoveConnected = true; - setMoveBdaddr(my_bdaddr); // Set internal bluetooth address - moveSetBulb(Red); - + setMoveBdaddr(my_bdaddr); // Set internal Bluetooth address + writeBuf[0] = 0x02; // Set report ID, this is needed for Move commands to work } - + onInit(); + #ifdef DEBUG_USB_HOST Notify(PSTR("\r\nBluetooth Address was set to: "), 0x80); for (int8_t i = 5; i > 0; i--) { @@ -538,3 +537,14 @@ void PS3USB::setMoveBdaddr(uint8_t* BDADDR) { //bmRequest = Host to device (0x00) | Class (0x20) | Interface (0x01) = 0x21, bRequest = Set Report (0x09), Report ID (0x05), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data) pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_OUT, HID_REQUEST_SET_REPORT, 0x05, 0x03, 0x00, 11, 11, buf, NULL); } + +void PS3USB::onInit() { + if (pFuncOnInit) + pFuncOnInit(); // Call the user function + else { + if (PS3MoveConnected) + moveSetBulb(Red); + else // Dualshock 3 or Navigation controller + setLedOn(LED1); + } +} \ No newline at end of file diff --git a/PS3USB.h b/PS3USB.h index d0b45e47..23ae927c 100644 --- a/PS3USB.h +++ b/PS3USB.h @@ -224,6 +224,14 @@ public: * @param rumble The desired value in the range from 64-255. */ void moveSetRumble(uint8_t rumble); + + /** + * Used to call your own function when the controller is successfully initialized. + * @param funcOnInit Function to call. + */ + void attachOnInit(void (*funcOnInit)(void)) { + pFuncOnInit = funcOnInit; + }; /**@}*/ /** Variable used to indicate if the normal playstation controller is successfully connected. */ @@ -242,6 +250,14 @@ protected: EpInfo epInfo[PS3_MAX_ENDPOINTS]; private: + /** + * Called when the controller is successfully initialized. + * Use attachOnInit(void (*funcOnInit)(void)) to call your own function. + * This is useful for instance if you want to set the LEDs in a specific way. + */ + void onInit(); + void (*pFuncOnInit)(void); // Pointer to function called in onInit() + bool bPollEnable; uint32_t timer; // used to continuously set PS3 Move controller Bulb and rumble values diff --git a/Wii.cpp b/Wii.cpp index 9948354f..751bdd9b 100755 --- a/Wii.cpp +++ b/Wii.cpp @@ -326,7 +326,7 @@ void WII::ACLData(uint8_t* l2capinbuf) { #endif nunchuckConnected = false; // It must be the Nunchuck controller then l2cap_event_flag &= ~WII_FLAG_NUNCHUCK_CONNECTED; - setLedStatus(); + onInit(); setReportMode(false, 0x31); // If there is no extension connected we will read the button and accelerometer } else { setReportMode(false, 0x31); // If there is no extension connected we will read the button and accelerometer @@ -779,7 +779,7 @@ void WII::Run() { case L2CAP_LED_STATE: if (nunchuck_connected_flag) nunchuckConnected = true; - setLedStatus(); + onInit(); l2cap_state = L2CAP_DONE; break; @@ -824,7 +824,7 @@ void WII::Run() { else if (stateCounter == 400) readExtensionType(); // Check if it has been activated else if (stateCounter == 450) { - //setLedStatus(); + onInit(); stateCounter = 0; unknownExtensionConnected = false; } @@ -1062,6 +1062,14 @@ uint16_t WII::getAnalogHat(AnalogHat a) { return output; } } + +void WII::onInit() { + if (pFuncOnInit) + pFuncOnInit(); // Call the user function + else + setLedStatus(); +} + /************************************************************/ /* The following functions are for the IR camera */ /************************************************************/ diff --git a/Wii.h b/Wii.h index 753947ba..60c9f07e 100755 --- a/Wii.h +++ b/Wii.h @@ -233,6 +233,14 @@ public: uint8_t getWiiState() { return wiiState; }; + + /** + * Used to call your own function when the controller is successfully initialized. + * @param funcOnInit Function to call. + */ + void attachOnInit(void (*funcOnInit)(void)) { + pFuncOnInit = funcOnInit; + }; /**@}*/ /**@{*/ @@ -418,8 +426,15 @@ public: #endif private: - /* Mandatory members */ - BTD *pBtd; + BTD *pBtd; // Pointer to BTD instance + + /** + * Called when the controller is successfully initialized. + * Use attachOnInit(void (*funcOnInit)(void)) to call your own function. + * This is useful for instance if you want to set the LEDs in a specific way. + */ + void onInit(); + void (*pFuncOnInit)(void); // Pointer to function called in onInit() void L2CAP_task(); // L2CAP state machine @@ -427,9 +442,9 @@ private: uint16_t hci_handle; bool activeConnection; // Used to indicate if it's already has established a connection - /* variables used by high level L2CAP task */ + /* Variables used by high level L2CAP task */ uint8_t l2cap_state; - uint16_t l2cap_event_flag; // l2cap flags of received bluetooth events + uint16_t l2cap_event_flag; // l2cap flags of received Bluetooth events uint32_t ButtonState; uint32_t OldButtonState; diff --git a/XBOXRECV.cpp b/XBOXRECV.cpp index 56dbe567..1d4d7724 100644 --- a/XBOXRECV.cpp +++ b/XBOXRECV.cpp @@ -310,7 +310,7 @@ void XBOXRECV::readReport(uint8_t controller) { #endif if (Xbox360Connected[controller]) { #ifdef DEBUG_USB_HOST - char* str = 0; + const char* str = 0; switch (readBuf[1]) { case 0x80: str = PSTR(" as controller\r\n"); break; @@ -322,18 +322,7 @@ void XBOXRECV::readReport(uint8_t controller) { Notify(PSTR(": connected"), 0x80); Notify(str, 0x80); #endif - LED led; - switch (controller) { - case 0: led = LED1; - break; - case 1: led = LED2; - break; - case 2: led = LED3; - break; - case 3: led = LED4; - break; - } - setLedOn(led, controller); + onInit(controller); } #ifdef DEBUG_USB_HOST else @@ -529,3 +518,20 @@ void XBOXRECV::setRumbleOn(uint8_t lValue, uint8_t rValue, uint8_t controller) { XboxCommand(controller, writeBuf, 7); } + +void XBOXRECV::onInit(uint8_t controller) { + if (pFuncOnInit) + pFuncOnInit(); // Call the user function + else { + LED led; + if (controller == 0) + led = LED1; + else if (controller == 1) + led = LED2; + if (controller == 2) + led = LED3; + else + led = LED4; + setLedOn(led, controller); + } +} diff --git a/XBOXRECV.h b/XBOXRECV.h index 9ae12aca..63bc8a70 100644 --- a/XBOXRECV.h +++ b/XBOXRECV.h @@ -200,6 +200,14 @@ public: * @return True if a button has changed. */ bool buttonChanged(uint8_t controller = 0); + + /** + * Used to call your own function when the controller is successfully initialized. + * @param funcOnInit Function to call. + */ + void attachOnInit(void (*funcOnInit)(void)) { + pFuncOnInit = funcOnInit; + }; /**@}*/ /** True if a wireless receiver is connected. */ @@ -216,6 +224,15 @@ protected: EpInfo epInfo[XBOX_MAX_ENDPOINTS]; private: + /** + * Called when the controller is successfully initialized. + * Use attachOnInit(void (*funcOnInit)(void)) to call your own function. + * This is useful for instance if you want to set the LEDs in a specific way. + * @param controller The initialized controller. + */ + void onInit(uint8_t controller); + void (*pFuncOnInit)(void); // Pointer to function called in onInit() + bool bPollEnable; /* Variables to store the buttons */ diff --git a/XBOXUSB.cpp b/XBOXUSB.cpp index e79c8a15..a5b76fc0 100644 --- a/XBOXUSB.cpp +++ b/XBOXUSB.cpp @@ -167,7 +167,7 @@ uint8_t XBOXUSB::Init(uint8_t parent, uint8_t port, bool lowspeed) { if (rcode) goto FailSetDevTblEntry; - delay(200); //Give time for address change + delay(200); // Give time for address change rcode = pUsb->setConf(bAddress, epInfo[ XBOX_CONTROL_PIPE ].epAddr, 1); if (rcode) @@ -176,12 +176,12 @@ uint8_t XBOXUSB::Init(uint8_t parent, uint8_t port, bool lowspeed) { #ifdef DEBUG_USB_HOST Notify(PSTR("\r\nXbox 360 Controller Connected\r\n"), 0x80); #endif - setLedOn(LED1); + onInit(); Xbox360Connected = true; bPollEnable = true; - return 0; // successful configuration + return 0; // Successful configuration - /* diagnostic messages */ + /* Diagnostic messages */ FailGetDevDescr: #ifdef DEBUG_USB_HOST NotifyFailGetDevDescr(); @@ -345,3 +345,10 @@ void XBOXUSB::setRumbleOn(uint8_t lValue, uint8_t rValue) { XboxCommand(writeBuf, 8); } + +void XBOXUSB::onInit() { + if (pFuncOnInit) + pFuncOnInit(); // Call the user function + else + setLedOn(LED1); +} diff --git a/XBOXUSB.h b/XBOXUSB.h index c3f7bdd8..07839c48 100644 --- a/XBOXUSB.h +++ b/XBOXUSB.h @@ -168,6 +168,14 @@ public: * @param lm See ::LEDMode. */ void setLedMode(LEDMode lm); + + /** + * Used to call your own function when the controller is successfully initialized. + * @param funcOnInit Function to call. + */ + void attachOnInit(void (*funcOnInit)(void)) { + pFuncOnInit = funcOnInit; + }; /**@}*/ /** True if a Xbox 360 controller is connected. */ @@ -182,6 +190,14 @@ protected: EpInfo epInfo[XBOX_MAX_ENDPOINTS]; private: + /** + * Called when the controller is successfully initialized. + * Use attachOnInit(void (*funcOnInit)(void)) to call your own function. + * This is useful for instance if you want to set the LEDs in a specific way. + */ + void onInit(); + void (*pFuncOnInit)(void); // Pointer to function called in onInit() + bool bPollEnable; /* Variables to store the buttons */ diff --git a/keywords.txt b/keywords.txt index 01dcbdc1..2d713b5c 100644 --- a/keywords.txt +++ b/keywords.txt @@ -56,6 +56,8 @@ setLedToggle KEYWORD2 moveSetBulb KEYWORD2 moveSetRumble KEYWORD2 +attachOnInit KEYWORD2 + PS3Connected KEYWORD2 PS3MoveConnected KEYWORD2 PS3NavigationConnected KEYWORD2