USB_Host_Shield_2.0/examples/Bluetooth/WiiIRCamera/WiiIRCamera.ino

126 lines
4.5 KiB
Arduino
Raw Normal View History

/*
Example sketch for the Wii libary showing the IR camera functionality. This example
is for the Bluetooth Wii library developed for the USB shield from Circuits@Home
2013-11-15 19:05:25 +01:00
Created by Allan Glover and Kristian Lauszus.
Contact Kristian: http://blog.tkjelectronics.dk/ or send an email at kristianl@tkjelectronics.com.
Contact Allan at adglover9.81@gmail.com
2013-11-15 19:05:25 +01:00
To test the Wiimote IR camera, you will need access to an IR source. Sunlight will work but is not ideal.
The simpleist solution is to use the Wii sensor bar, i.e. emitter bar, supplied by the Wii system.
Otherwise, wire up a IR LED yourself.
*/
#include <Wii.h>
2013-08-13 21:31:52 +02:00
#include <usbhub.h>
2013-01-17 00:47:38 +01:00
#ifndef WIICAMERA // Used to check if WIICAMERA is defined
#error "Uncomment WIICAMERA in Wii.h to use this example"
#endif
2013-08-13 21:31:52 +02:00
USB Usb;
2013-11-24 15:17:19 +01:00
//USBHub Hub1(&Usb); // Some dongles have a hub inside
BTD Btd(&Usb); // You have to create the Bluetooth Dongle instance like so
/* You can create the instance of the class in two ways */
2013-11-15 19:05:25 +01:00
WII Wii(&Btd, PAIR); // This will start an inquiry and then pair with your Wiimote - you only have to do this once
//WII Wii(&Btd); // After the Wiimote pairs once with the line of code above, you can simply create the instance like so and re upload and then press any button on the Wiimote
bool printAngle;
uint8_t printObjects;
void setup() {
Serial.begin(115200);
while (!Serial); // Wait for serial port to connect - used on Leonardo, Teensy and other boards with built-in USB CDC serial connection
if (Usb.Init() == -1) {
Serial.print(F("\r\nOSC did not start"));
2013-11-15 19:05:25 +01:00
while (1); //halt
}
Serial.print(F("\r\nWiimote Bluetooth Library Started"));
}
void loop() {
Usb.Task();
2013-11-15 19:05:25 +01:00
if (Wii.wiimoteConnected) {
if (Wii.getButtonClick(HOME)) { // You can use getButtonPress to see if the button is held down
Serial.print(F("\r\nHOME"));
2013-01-23 22:47:47 +01:00
Wii.disconnect();
}
else {
2013-11-15 19:05:25 +01:00
if (Wii.getButtonClick(ONE))
Wii.IRinitialize(); // Run the initialisation sequence
2013-11-15 19:05:25 +01:00
if (Wii.getButtonClick(MINUS) || Wii.getButtonClick(PLUS)) {
if (!Wii.isIRCameraEnabled())
Serial.print(F("\r\nEnable IR camera first"));
else {
2013-11-15 19:05:25 +01:00
if (Wii.getButtonPress(MINUS)) { // getButtonClick will only return true once
if (printObjects > 0)
printObjects--;
}
else {
2013-11-15 19:05:25 +01:00
if (printObjects < 4)
printObjects++;
}
Serial.print(F("\r\nTracking "));
Serial.print(printObjects);
Serial.print(F(" objects"));
}
}
2013-11-15 19:05:25 +01:00
if (Wii.getButtonClick(A)) {
printAngle = !printAngle;
Serial.print(F("\r\nA"));
}
2013-11-15 19:05:25 +01:00
if (Wii.getButtonClick(B)) {
Serial.print(F("\r\nBattery level: "));
Serial.print(Wii.getBatteryLevel()); // You can get the battery level as well
}
}
2013-11-15 19:05:25 +01:00
if (printObjects > 0) {
if (Wii.getIRx1() != 0x3FF || Wii.getIRy1() != 0x3FF || Wii.getIRs1() != 0) { // Only print if the IR camera is actually seeing something
Serial.print(F("\r\nx1: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRx1());
Serial.print(F("\ty1: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRy1());
Serial.print(F("\ts1:"));
Serial.print(Wii.getIRs1());
}
2013-11-15 19:05:25 +01:00
if (printObjects > 1) {
if (Wii.getIRx2() != 0x3FF || Wii.getIRy2() != 0x3FF || Wii.getIRs2() != 0) {
Serial.print(F("\r\nx2: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRx2());
Serial.print(F("\ty2: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRy2());
Serial.print(F("\ts2:"));
Serial.print(Wii.getIRs2());
}
2013-11-15 19:05:25 +01:00
if (printObjects > 2) {
if (Wii.getIRx3() != 0x3FF || Wii.getIRy3() != 0x3FF || Wii.getIRs3() != 0) {
Serial.print(F("\r\nx3: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRx3());
Serial.print(F("\ty3: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRy3());
Serial.print(F("\ts3:"));
Serial.print(Wii.getIRs3());
}
2013-11-15 19:05:25 +01:00
if (printObjects > 3) {
if (Wii.getIRx4() != 0x3FF || Wii.getIRy4() != 0x3FF || Wii.getIRs4() != 0) {
Serial.print(F("\r\nx4: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRx4());
Serial.print(F("\ty4: "));
2013-11-15 19:05:25 +01:00
Serial.print(Wii.getIRy4());
Serial.print(F("\ts4:"));
Serial.print(Wii.getIRs4());
}
}
}
}
2013-11-15 19:05:25 +01:00
}
if (printAngle) { // There is no extension bytes available, so the MotionPlus or Nunchuck can't be read
Serial.print(F("\r\nPitch: "));
Serial.print(Wii.getPitch());
Serial.print(F("\tRoll: "));
Serial.print(Wii.getRoll());
}
}
}