573 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			573 lines
		
	
	
		
			22 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /* Copyright (C) 2012 Kristian Lauszus, TKJ Electronics. All rights reserved.
 | |
| 
 | |
|  This software may be distributed and modified under the terms of the GNU
 | |
|  General Public License version 2 (GPL2) as published by the Free Software
 | |
|  Foundation and appearing in the file GPL2.TXT included in the packaging of
 | |
|  this file. Please note that GPL2 Section 2[b] requires that all works based
 | |
|  on this software must also be made publicly available under the terms of
 | |
|  the GPL2 ("Copyleft").
 | |
| 
 | |
|  Contact information
 | |
|  -------------------
 | |
| 
 | |
|  Kristian Lauszus, TKJ Electronics
 | |
|  Web      :  http://www.tkjelectronics.com
 | |
|  e-mail   :  kristianl@tkjelectronics.com
 | |
|  */
 | |
| 
 | |
| #include "PS3USB.h"
 | |
| // To enable serial debugging see "settings.h"
 | |
| //#define EXTRADEBUG // Uncomment to get even more debugging data
 | |
| //#define PRINTREPORT // Uncomment to print the report send by the PS3 Controllers
 | |
| 
 | |
| PS3USB::PS3USB(USB *p, uint8_t btadr5, uint8_t btadr4, uint8_t btadr3, uint8_t btadr2, uint8_t btadr1, uint8_t btadr0) :
 | |
| pUsb(p), // pointer to USB class instance - mandatory
 | |
| bAddress(0), // device address - mandatory
 | |
| bPollEnable(false) // don't start polling before dongle is connected
 | |
| {
 | |
|         for(uint8_t i = 0; i < PS3_MAX_ENDPOINTS; i++) {
 | |
|                 epInfo[i].epAddr = 0;
 | |
|                 epInfo[i].maxPktSize = (i) ? 0 : 8;
 | |
|                 epInfo[i].epAttribs = 0;
 | |
|                 epInfo[i].bmNakPower = (i) ? USB_NAK_NOWAIT : USB_NAK_MAX_POWER;
 | |
|         }
 | |
| 
 | |
|         if(pUsb) // register in USB subsystem
 | |
|                 pUsb->RegisterDeviceClass(this); //set devConfig[] entry
 | |
| 
 | |
|         my_bdaddr[5] = btadr5; // Change to your dongle's Bluetooth address instead
 | |
|         my_bdaddr[4] = btadr4;
 | |
|         my_bdaddr[3] = btadr3;
 | |
|         my_bdaddr[2] = btadr2;
 | |
|         my_bdaddr[1] = btadr1;
 | |
|         my_bdaddr[0] = btadr0;
 | |
| }
 | |
| 
 | |
| uint8_t PS3USB::Init(uint8_t parent, uint8_t port, bool lowspeed) {
 | |
|         uint8_t buf[sizeof (USB_DEVICE_DESCRIPTOR)];
 | |
|         USB_DEVICE_DESCRIPTOR * udd = reinterpret_cast<USB_DEVICE_DESCRIPTOR*>(buf);
 | |
|         uint8_t rcode;
 | |
|         UsbDevice *p = NULL;
 | |
|         EpInfo *oldep_ptr = NULL;
 | |
|         uint16_t PID;
 | |
|         uint16_t VID;
 | |
| 
 | |
|         // get memory address of USB device address pool
 | |
|         AddressPool &addrPool = pUsb->GetAddressPool();
 | |
| #ifdef EXTRADEBUG
 | |
|         Notify(PSTR("\r\nPS3USB Init"), 0x80);
 | |
| #endif
 | |
|         // check if address has already been assigned to an instance
 | |
|         if(bAddress) {
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nAddress in use"), 0x80);
 | |
| #endif
 | |
|                 return USB_ERROR_CLASS_INSTANCE_ALREADY_IN_USE;
 | |
|         }
 | |
| 
 | |
|         // Get pointer to pseudo device with address 0 assigned
 | |
|         p = addrPool.GetUsbDevicePtr(0);
 | |
| 
 | |
|         if(!p) {
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nAddress not found"), 0x80);
 | |
| #endif
 | |
|                 return USB_ERROR_ADDRESS_NOT_FOUND_IN_POOL;
 | |
|         }
 | |
| 
 | |
|         if(!p->epinfo) {
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nepinfo is null"), 0x80);
 | |
| #endif
 | |
|                 return USB_ERROR_EPINFO_IS_NULL;
 | |
|         }
 | |
| 
 | |
|         // Save old pointer to EP_RECORD of address 0
 | |
|         oldep_ptr = p->epinfo;
 | |
| 
 | |
|         // Temporary assign new pointer to epInfo to p->epinfo in order to avoid toggle inconsistence
 | |
|         p->epinfo = epInfo;
 | |
| 
 | |
|         p->lowspeed = lowspeed;
 | |
| 
 | |
|         // Get device descriptor
 | |
|         rcode = pUsb->getDevDescr(0, 0, sizeof (USB_DEVICE_DESCRIPTOR), (uint8_t*)buf); // Get device descriptor - addr, ep, nbytes, data
 | |
|         // Restore p->epinfo
 | |
|         p->epinfo = oldep_ptr;
 | |
| 
 | |
|         if(rcode)
 | |
|                 goto FailGetDevDescr;
 | |
| 
 | |
|         VID = udd->idVendor;
 | |
|         PID = udd->idProduct;
 | |
| 
 | |
|         if(VID != PS3_VID || (PID != PS3_PID && PID != PS3NAVIGATION_PID && PID != PS3MOVE_PID))
 | |
|                 goto FailUnknownDevice;
 | |
| 
 | |
|         // Allocate new address according to device class
 | |
|         bAddress = addrPool.AllocAddress(parent, false, port);
 | |
| 
 | |
|         if(!bAddress)
 | |
|                 return USB_ERROR_OUT_OF_ADDRESS_SPACE_IN_POOL;
 | |
| 
 | |
|         // Extract Max Packet Size from device descriptor
 | |
|         epInfo[0].maxPktSize = udd->bMaxPacketSize0;
 | |
| 
 | |
|         // Assign new address to the device
 | |
|         rcode = pUsb->setAddr(0, 0, bAddress);
 | |
|         if(rcode) {
 | |
|                 p->lowspeed = false;
 | |
|                 addrPool.FreeAddress(bAddress);
 | |
|                 bAddress = 0;
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nsetAddr: "), 0x80);
 | |
|                 D_PrintHex<uint8_t > (rcode, 0x80);
 | |
| #endif
 | |
|                 return rcode;
 | |
|         }
 | |
| #ifdef EXTRADEBUG
 | |
|         Notify(PSTR("\r\nAddr: "), 0x80);
 | |
|         D_PrintHex<uint8_t > (bAddress, 0x80);
 | |
| #endif
 | |
|         //delay(300); // Spec says you should wait at least 200ms
 | |
| 
 | |
|         p->lowspeed = false;
 | |
| 
 | |
|         //get pointer to assigned address record
 | |
|         p = addrPool.GetUsbDevicePtr(bAddress);
 | |
|         if(!p)
 | |
|                 return USB_ERROR_ADDRESS_NOT_FOUND_IN_POOL;
 | |
| 
 | |
|         p->lowspeed = lowspeed;
 | |
| 
 | |
|         // Assign epInfo to epinfo pointer - only EP0 is known
 | |
|         rcode = pUsb->setEpInfoEntry(bAddress, 1, epInfo);
 | |
|         if(rcode)
 | |
|                 goto FailSetDevTblEntry;
 | |
| 
 | |
| 
 | |
|         /* The application will work in reduced host mode, so we can save program and data
 | |
|            memory space. After verifying the PID and VID we will use known values for the
 | |
|            configuration values for device, interface, endpoints and HID for the PS3 Controllers */
 | |
| 
 | |
|         /* Initialize data structures for endpoints of device */
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].epAddr = 0x02; // PS3 output endpoint
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].epAttribs = USB_TRANSFER_TYPE_INTERRUPT;
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].bmNakPower = USB_NAK_NOWAIT; // Only poll once for interrupt endpoints
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].maxPktSize = EP_MAXPKTSIZE;
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].bmSndToggle = 0;
 | |
|         epInfo[ PS3_OUTPUT_PIPE ].bmRcvToggle = 0;
 | |
|         epInfo[ PS3_INPUT_PIPE ].epAddr = 0x01; // PS3 report endpoint
 | |
|         epInfo[ PS3_INPUT_PIPE ].epAttribs = USB_TRANSFER_TYPE_INTERRUPT;
 | |
|         epInfo[ PS3_INPUT_PIPE ].bmNakPower = USB_NAK_NOWAIT; // Only poll once for interrupt endpoints
 | |
|         epInfo[ PS3_INPUT_PIPE ].maxPktSize = EP_MAXPKTSIZE;
 | |
|         epInfo[ PS3_INPUT_PIPE ].bmSndToggle = 0;
 | |
|         epInfo[ PS3_INPUT_PIPE ].bmRcvToggle = 0;
 | |
| 
 | |
|         rcode = pUsb->setEpInfoEntry(bAddress, 3, epInfo);
 | |
|         if(rcode)
 | |
|                 goto FailSetDevTblEntry;
 | |
| 
 | |
|         delay(200); //Give time for address change
 | |
| 
 | |
|         rcode = pUsb->setConf(bAddress, epInfo[ PS3_CONTROL_PIPE ].epAddr, 1);
 | |
|         if(rcode)
 | |
|                 goto FailSetConfDescr;
 | |
| 
 | |
|         if(PID == PS3_PID || PID == PS3NAVIGATION_PID) {
 | |
|                 if(PID == PS3_PID) {
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                         Notify(PSTR("\r\nDualshock 3 Controller Connected"), 0x80);
 | |
| #endif
 | |
|                         PS3Connected = true;
 | |
|                 } else { // must be a navigation controller
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                         Notify(PSTR("\r\nNavigation Controller Connected"), 0x80);
 | |
| #endif
 | |
|                         PS3NavigationConnected = true;
 | |
|                 }
 | |
|                 enable_sixaxis(); // The PS3 controller needs a special command before it starts sending data
 | |
| 
 | |
|                 // Needed for PS3 Dualshock and Navigation commands to work
 | |
|                 for(uint8_t i = 0; i < PS3_REPORT_BUFFER_SIZE; i++)
 | |
|                         writeBuf[i] = pgm_read_byte(&PS3_REPORT_BUFFER[i]);
 | |
| 
 | |
|                 for(uint8_t i = 6; i < 10; i++)
 | |
|                         readBuf[i] = 0x7F; // Set the analog joystick values to center position
 | |
|         } else { // must be a Motion controller
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nMotion Controller Connected"), 0x80);
 | |
| #endif
 | |
|                 PS3MoveConnected = true;
 | |
|                 writeBuf[0] = 0x02; // Set report ID, this is needed for Move commands to work
 | |
|         }
 | |
|         if(my_bdaddr[0] != 0x00 || my_bdaddr[1] != 0x00 || my_bdaddr[2] != 0x00 || my_bdaddr[3] != 0x00 || my_bdaddr[4] != 0x00 || my_bdaddr[5] != 0x00) {
 | |
|                 if(PS3MoveConnected)
 | |
|                         setMoveBdaddr(my_bdaddr); // Set internal Bluetooth address
 | |
|                 else
 | |
|                         setBdaddr(my_bdaddr); // Set internal Bluetooth address
 | |
| 
 | |
| #ifdef DEBUG_USB_HOST
 | |
|                 Notify(PSTR("\r\nBluetooth Address was set to: "), 0x80);
 | |
|                 for(int8_t i = 5; i > 0; i--) {
 | |
|                         D_PrintHex<uint8_t > (my_bdaddr[i], 0x80);
 | |
|                         Notify(PSTR(":"), 0x80);
 | |
|                 }
 | |
|                 D_PrintHex<uint8_t > (my_bdaddr[0], 0x80);
 | |
| #endif
 | |
|         }
 | |
|         onInit();
 | |
| 
 | |
|         bPollEnable = true;
 | |
|         Notify(PSTR("\r\n"), 0x80);
 | |
|         timer = millis();
 | |
|         return 0; // Successful configuration
 | |
| 
 | |
|         /* Diagnostic messages */
 | |
| FailGetDevDescr:
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         NotifyFailGetDevDescr();
 | |
|         goto Fail;
 | |
| #endif
 | |
| 
 | |
| FailSetDevTblEntry:
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         NotifyFailSetDevTblEntry();
 | |
|         goto Fail;
 | |
| #endif
 | |
| 
 | |
| FailSetConfDescr:
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         NotifyFailSetConfDescr();
 | |
| #endif
 | |
|         goto Fail;
 | |
| 
 | |
| FailUnknownDevice:
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         NotifyFailUnknownDevice(VID, PID);
 | |
| #endif
 | |
|         rcode = USB_DEV_CONFIG_ERROR_DEVICE_NOT_SUPPORTED;
 | |
| 
 | |
| Fail:
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         Notify(PSTR("\r\nPS3 Init Failed, error code: "), 0x80);
 | |
|         NotifyFail(rcode);
 | |
| #endif
 | |
|         Release();
 | |
|         return rcode;
 | |
| }
 | |
| 
 | |
| /* Performs a cleanup after failed Init() attempt */
 | |
| uint8_t PS3USB::Release() {
 | |
|         PS3Connected = false;
 | |
|         PS3MoveConnected = false;
 | |
|         PS3NavigationConnected = false;
 | |
|         pUsb->GetAddressPool().FreeAddress(bAddress);
 | |
|         bAddress = 0;
 | |
|         bPollEnable = false;
 | |
|         return 0;
 | |
| }
 | |
| 
 | |
| uint8_t PS3USB::Poll() {
 | |
|         if(!bPollEnable)
 | |
|                 return 0;
 | |
| 
 | |
|         if(PS3Connected || PS3NavigationConnected) {
 | |
|                 uint16_t BUFFER_SIZE = EP_MAXPKTSIZE;
 | |
|                 pUsb->inTransfer(bAddress, epInfo[ PS3_INPUT_PIPE ].epAddr, &BUFFER_SIZE, readBuf); // input on endpoint 1
 | |
|                 if(millis() - timer > 100) { // Loop 100ms before processing data
 | |
|                         readReport();
 | |
| #ifdef PRINTREPORT
 | |
|                         printReport(); // Uncomment "#define PRINTREPORT" to print the report send by the PS3 Controllers
 | |
| #endif
 | |
|                 }
 | |
|         } else if(PS3MoveConnected) { // One can only set the color of the bulb, set the rumble, set and get the bluetooth address and calibrate the magnetometer via USB
 | |
|                 if(millis() - timer > 4000) { // Send at least every 4th second
 | |
|                         Move_Command(writeBuf, MOVE_REPORT_BUFFER_SIZE); // The Bulb and rumble values, has to be written again and again, for it to stay turned on
 | |
|                         timer = millis();
 | |
|                 }
 | |
|         }
 | |
|         return 0;
 | |
| }
 | |
| 
 | |
| void PS3USB::readReport() {
 | |
|         ButtonState = (uint32_t)(readBuf[2] | ((uint16_t)readBuf[3] << 8) | ((uint32_t)readBuf[4] << 16));
 | |
| 
 | |
|         //Notify(PSTR("\r\nButtonState", 0x80);
 | |
|         //PrintHex<uint32_t>(ButtonState, 0x80);
 | |
| 
 | |
|         if(ButtonState != OldButtonState) {
 | |
|                 ButtonClickState = ButtonState & ~OldButtonState; // Update click state variable
 | |
|                 OldButtonState = ButtonState;
 | |
|         }
 | |
| }
 | |
| 
 | |
| void PS3USB::printReport() { // Uncomment "#define PRINTREPORT" to print the report send by the PS3 Controllers
 | |
| #ifdef PRINTREPORT
 | |
|         for(uint8_t i = 0; i < PS3_REPORT_BUFFER_SIZE; i++) {
 | |
|                 D_PrintHex<uint8_t > (readBuf[i], 0x80);
 | |
|                 Notify(PSTR(" "), 0x80);
 | |
|         }
 | |
|         Notify(PSTR("\r\n"), 0x80);
 | |
| #endif
 | |
| }
 | |
| 
 | |
| bool PS3USB::getButtonPress(ButtonEnum b) {
 | |
|         return (ButtonState & pgm_read_dword(&PS3_BUTTONS[(uint8_t)b]));
 | |
| }
 | |
| 
 | |
| bool PS3USB::getButtonClick(ButtonEnum b) {
 | |
|         uint32_t button = pgm_read_dword(&PS3_BUTTONS[(uint8_t)b]);
 | |
|         bool click = (ButtonClickState & button);
 | |
|         ButtonClickState &= ~button; // Clear "click" event
 | |
|         return click;
 | |
| }
 | |
| 
 | |
| uint8_t PS3USB::getAnalogButton(ButtonEnum a) {
 | |
|         return (uint8_t)(readBuf[(pgm_read_byte(&PS3_ANALOG_BUTTONS[(uint8_t)a])) - 9]);
 | |
| }
 | |
| 
 | |
| uint8_t PS3USB::getAnalogHat(AnalogHatEnum a) {
 | |
|         return (uint8_t)(readBuf[((uint8_t)a + 6)]);
 | |
| }
 | |
| 
 | |
| uint16_t PS3USB::getSensor(SensorEnum a) {
 | |
|         return ((readBuf[((uint16_t)a) - 9] << 8) | readBuf[((uint16_t)a + 1) - 9]);
 | |
| }
 | |
| 
 | |
| double PS3USB::getAngle(AngleEnum a) {
 | |
|         if(PS3Connected) {
 | |
|                 double accXval;
 | |
|                 double accYval;
 | |
|                 double accZval;
 | |
| 
 | |
|                 // Data for the Kionix KXPC4 used in the DualShock 3
 | |
|                 const double zeroG = 511.5; // 1.65/3.3*1023 (1,65V)
 | |
|                 accXval = -((double)getSensor(aX) - zeroG);
 | |
|                 accYval = -((double)getSensor(aY) - zeroG);
 | |
|                 accZval = -((double)getSensor(aZ) - zeroG);
 | |
| 
 | |
|                 // Convert to 360 degrees resolution
 | |
|                 // atan2 outputs the value of -π to π (radians)
 | |
|                 // We are then converting it to 0 to 2π and then to degrees
 | |
|                 if(a == Pitch)
 | |
|                         return (atan2(accYval, accZval) + PI) * RAD_TO_DEG;
 | |
|                 else
 | |
|                         return (atan2(accXval, accZval) + PI) * RAD_TO_DEG;
 | |
|         } else
 | |
|                 return 0;
 | |
| }
 | |
| 
 | |
| bool PS3USB::getStatus(StatusEnum c) {
 | |
|         return (readBuf[((uint16_t)c >> 8) - 9] == ((uint8_t)c & 0xff));
 | |
| }
 | |
| 
 | |
| void PS3USB::printStatusString() {
 | |
|         char statusOutput[100]; // Max string length plus null character
 | |
|         if(PS3Connected || PS3NavigationConnected) {
 | |
|                 strcpy_P(statusOutput, PSTR("ConnectionStatus: "));
 | |
| 
 | |
|                 if(getStatus(Plugged)) strcat_P(statusOutput, PSTR("Plugged"));
 | |
|                 else if(getStatus(Unplugged)) strcat_P(statusOutput, PSTR("Unplugged"));
 | |
|                 else strcat_P(statusOutput, PSTR("Error"));
 | |
| 
 | |
|                 strcat_P(statusOutput, PSTR(" - PowerRating: "));
 | |
| 
 | |
|                 if(getStatus(Charging)) strcat_P(statusOutput, PSTR("Charging"));
 | |
|                 else if(getStatus(NotCharging)) strcat_P(statusOutput, PSTR("Not Charging"));
 | |
|                 else if(getStatus(Shutdown)) strcat_P(statusOutput, PSTR("Shutdown"));
 | |
|                 else if(getStatus(Dying)) strcat_P(statusOutput, PSTR("Dying"));
 | |
|                 else if(getStatus(Low)) strcat_P(statusOutput, PSTR("Low"));
 | |
|                 else if(getStatus(High)) strcat_P(statusOutput, PSTR("High"));
 | |
|                 else if(getStatus(Full)) strcat_P(statusOutput, PSTR("Full"));
 | |
|                 else strcat_P(statusOutput, PSTR("Error"));
 | |
| 
 | |
|                 strcat_P(statusOutput, PSTR(" - WirelessStatus: "));
 | |
| 
 | |
|                 if(getStatus(CableRumble)) strcat_P(statusOutput, PSTR("Cable - Rumble is on"));
 | |
|                 else if(getStatus(Cable)) strcat_P(statusOutput, PSTR("Cable - Rumble is off"));
 | |
|                 else if(getStatus(BluetoothRumble)) strcat_P(statusOutput, PSTR("Bluetooth - Rumble is on"));
 | |
|                 else if(getStatus(Bluetooth)) strcat_P(statusOutput, PSTR("Bluetooth - Rumble is off"));
 | |
|                 else strcat_P(statusOutput, PSTR("Error"));
 | |
|         } else
 | |
|                 strcpy_P(statusOutput, PSTR("Error"));
 | |
| 
 | |
|         USB_HOST_SERIAL.write(statusOutput);
 | |
| }
 | |
| 
 | |
| /* Playstation Sixaxis Dualshock and Navigation Controller commands */
 | |
| void PS3USB::PS3_Command(uint8_t *data, uint16_t nbytes) {
 | |
|         // bmRequest = Host to device (0x00) | Class (0x20) | Interface (0x01) = 0x21, bRequest = Set Report (0x09), Report ID (0x01), Report Type (Output 0x02), interface (0x00), datalength, datalength, data)
 | |
|         pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_OUT, HID_REQUEST_SET_REPORT, 0x01, 0x02, 0x00, nbytes, nbytes, data, NULL);
 | |
| }
 | |
| 
 | |
| void PS3USB::setAllOff() {
 | |
|         for(uint8_t i = 0; i < PS3_REPORT_BUFFER_SIZE; i++)
 | |
|                 writeBuf[i] = pgm_read_byte(&PS3_REPORT_BUFFER[i]); // Reset buffer
 | |
| 
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setRumbleOff() {
 | |
|         writeBuf[1] = 0x00;
 | |
|         writeBuf[2] = 0x00; // Low mode off
 | |
|         writeBuf[3] = 0x00;
 | |
|         writeBuf[4] = 0x00; // High mode off
 | |
| 
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setRumbleOn(RumbleEnum mode) {
 | |
|         if((mode & 0x30) > 0x00) {
 | |
|                 uint8_t power[2] = {0xff, 0x00}; // Defaults to RumbleLow
 | |
|                 if(mode == RumbleHigh) {
 | |
|                         power[0] = 0x00;
 | |
|                         power[1] = 0xff;
 | |
|                 }
 | |
|                 setRumbleOn(0xfe, power[0], 0xfe, power[1]);
 | |
|         }
 | |
| }
 | |
| 
 | |
| void PS3USB::setRumbleOn(uint8_t rightDuration, uint8_t rightPower, uint8_t leftDuration, uint8_t leftPower) {
 | |
|         writeBuf[1] = rightDuration;
 | |
|         writeBuf[2] = rightPower;
 | |
|         writeBuf[3] = leftDuration;
 | |
|         writeBuf[4] = leftPower;
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setLedRaw(uint8_t value) {
 | |
|         writeBuf[9] = value << 1;
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setLedOff(LEDEnum a) {
 | |
|         writeBuf[9] &= ~((uint8_t)((pgm_read_byte(&PS3_LEDS[(uint8_t)a]) & 0x0f) << 1));
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setLedOn(LEDEnum a) {
 | |
|         if(a == OFF)
 | |
|                 setLedRaw(0);
 | |
|         else {
 | |
|                 writeBuf[9] |= (uint8_t)((pgm_read_byte(&PS3_LEDS[(uint8_t)a]) & 0x0f) << 1);
 | |
|                 PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
|         }
 | |
| }
 | |
| 
 | |
| void PS3USB::setLedToggle(LEDEnum a) {
 | |
|         writeBuf[9] ^= (uint8_t)((pgm_read_byte(&PS3_LEDS[(uint8_t)a]) & 0x0f) << 1);
 | |
|         PS3_Command(writeBuf, PS3_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setBdaddr(uint8_t *bdaddr) {
 | |
|         /* Set the internal Bluetooth address */
 | |
|         uint8_t buf[8];
 | |
|         buf[0] = 0x01;
 | |
|         buf[1] = 0x00;
 | |
| 
 | |
|         for(uint8_t i = 0; i < 6; i++)
 | |
|                 buf[i + 2] = bdaddr[5 - i]; // Copy into buffer, has to be written reversed, so it is MSB first
 | |
| 
 | |
|         // bmRequest = Host to device (0x00) | Class (0x20) | Interface (0x01) = 0x21, bRequest = Set Report (0x09), Report ID (0xF5), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data
 | |
|         pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_OUT, HID_REQUEST_SET_REPORT, 0xF5, 0x03, 0x00, 8, 8, buf, NULL);
 | |
| }
 | |
| 
 | |
| void PS3USB::getBdaddr(uint8_t *bdaddr) {
 | |
|         uint8_t buf[8];
 | |
| 
 | |
|         // bmRequest = Device to host (0x80) | Class (0x20) | Interface (0x01) = 0xA1, bRequest = Get Report (0x01), Report ID (0xF5), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data
 | |
|         pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_IN, HID_REQUEST_GET_REPORT, 0xF5, 0x03, 0x00, 8, 8, buf, NULL);
 | |
| 
 | |
|         for(uint8_t i = 0; i < 6; i++)
 | |
|                 bdaddr[5 - i] = buf[i + 2]; // Copy into buffer reversed, so it is LSB first
 | |
| }
 | |
| 
 | |
| void PS3USB::enable_sixaxis() { // Command used to enable the Dualshock 3 and Navigation controller to send data via USB
 | |
|         uint8_t cmd_buf[4];
 | |
|         cmd_buf[0] = 0x42; // Special PS3 Controller enable commands
 | |
|         cmd_buf[1] = 0x0c;
 | |
|         cmd_buf[2] = 0x00;
 | |
|         cmd_buf[3] = 0x00;
 | |
| 
 | |
|         // bmRequest = Host to device (0x00) | Class (0x20) | Interface (0x01) = 0x21, bRequest = Set Report (0x09), Report ID (0xF4), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data)
 | |
|         pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_OUT, HID_REQUEST_SET_REPORT, 0xF4, 0x03, 0x00, 4, 4, cmd_buf, NULL);
 | |
| }
 | |
| 
 | |
| /* Playstation Move Controller commands */
 | |
| void PS3USB::Move_Command(uint8_t *data, uint16_t nbytes) {
 | |
|         pUsb->outTransfer(bAddress, epInfo[ PS3_OUTPUT_PIPE ].epAddr, nbytes, data);
 | |
| }
 | |
| 
 | |
| void PS3USB::moveSetBulb(uint8_t r, uint8_t g, uint8_t b) { // Use this to set the Color using RGB values
 | |
|         // Set the Bulb's values into the write buffer
 | |
|         writeBuf[2] = r;
 | |
|         writeBuf[3] = g;
 | |
|         writeBuf[4] = b;
 | |
| 
 | |
|         Move_Command(writeBuf, MOVE_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::moveSetBulb(ColorsEnum color) { // Use this to set the Color using the predefined colors in "enums.h"
 | |
|         moveSetBulb((uint8_t)(color >> 16), (uint8_t)(color >> 8), (uint8_t)(color));
 | |
| }
 | |
| 
 | |
| void PS3USB::moveSetRumble(uint8_t rumble) {
 | |
| #ifdef DEBUG_USB_HOST
 | |
|         if(rumble < 64 && rumble != 0) // The rumble value has to at least 64, or approximately 25% (64/255*100)
 | |
|                 Notify(PSTR("\r\nThe rumble value has to at least 64, or approximately 25%"), 0x80);
 | |
| #endif
 | |
|         writeBuf[6] = rumble; // Set the rumble value into the write buffer
 | |
| 
 | |
|         Move_Command(writeBuf, MOVE_REPORT_BUFFER_SIZE);
 | |
| }
 | |
| 
 | |
| void PS3USB::setMoveBdaddr(uint8_t *bdaddr) {
 | |
|         /* Set the internal Bluetooth address */
 | |
|         uint8_t buf[11];
 | |
|         buf[0] = 0x05;
 | |
|         buf[7] = 0x10;
 | |
|         buf[8] = 0x01;
 | |
|         buf[9] = 0x02;
 | |
|         buf[10] = 0x12;
 | |
| 
 | |
|         for(uint8_t i = 0; i < 6; i++)
 | |
|                 buf[i + 1] = bdaddr[i];
 | |
| 
 | |
|         // 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::getMoveBdaddr(uint8_t *bdaddr) {
 | |
|         uint8_t buf[16];
 | |
| 
 | |
|         // bmRequest = Device to host (0x80) | Class (0x20) | Interface (0x01) = 0xA1, bRequest = Get Report (0x01), Report ID (0x04), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data
 | |
|         pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_IN, HID_REQUEST_GET_REPORT, 0x04, 0x03, 0x00, 16, 16, buf, NULL);
 | |
| 
 | |
|         for(uint8_t i = 0; i < 6; i++)
 | |
|                 bdaddr[i] = buf[10 + i];
 | |
| }
 | |
| 
 | |
| void PS3USB::getMoveCalibration(uint8_t *data) {
 | |
|         uint8_t buf[49];
 | |
| 
 | |
|         for(uint8_t i = 0; i < 3; i++) {
 | |
|                 // bmRequest = Device to host (0x80) | Class (0x20) | Interface (0x01) = 0xA1, bRequest = Get Report (0x01), Report ID (0x10), Report Type (Feature 0x03), interface (0x00), datalength, datalength, data
 | |
|                 pUsb->ctrlReq(bAddress, epInfo[PS3_CONTROL_PIPE].epAddr, bmREQ_HID_IN, HID_REQUEST_GET_REPORT, 0x10, 0x03, 0x00, 49, 49, buf, NULL);
 | |
| 
 | |
|                 for(byte j = 0; j < 49; j++)
 | |
|                         data[49 * i + j] = buf[j];
 | |
|         }
 | |
| }
 | |
| 
 | |
| void PS3USB::onInit() {
 | |
|         if(pFuncOnInit)
 | |
|                 pFuncOnInit(); // Call the user function
 | |
|         else {
 | |
|                 if(PS3MoveConnected)
 | |
|                         moveSetBulb(Red);
 | |
|                 else // Dualshock 3 or Navigation controller
 | |
|                         setLedOn(LED1);
 | |
|         }
 | |
| }
 |