-
Notifications
You must be signed in to change notification settings - Fork 843
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #3626 from tommy-gilligan/rp23-usb-hid-keyboard-ex…
…ample rp23: port usb_hid_keyboard example from rp
- Loading branch information
Showing
1 changed file
with
193 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,193 @@ | ||
#![no_std] | ||
#![no_main] | ||
|
||
use core::sync::atomic::{AtomicBool, Ordering}; | ||
|
||
use defmt::*; | ||
use embassy_executor::Spawner; | ||
use embassy_futures::join::join; | ||
use embassy_rp::bind_interrupts; | ||
use embassy_rp::block::ImageDef; | ||
use embassy_rp::gpio::{Input, Pull}; | ||
use embassy_rp::peripherals::USB; | ||
use embassy_rp::usb::{Driver as UsbDriver, InterruptHandler}; | ||
use embassy_usb::class::hid::{HidReaderWriter, ReportId, RequestHandler, State as HidState}; | ||
use embassy_usb::control::OutResponse; | ||
use embassy_usb::{Builder, Config, Handler}; | ||
use usbd_hid::descriptor::{KeyboardReport, SerializedDescriptor}; | ||
use {defmt_rtt as _, panic_probe as _}; | ||
|
||
#[link_section = ".start_block"] | ||
#[used] | ||
pub static IMAGE_DEF: ImageDef = ImageDef::secure_exe(); | ||
|
||
bind_interrupts!(struct Irqs { | ||
USBCTRL_IRQ => InterruptHandler<USB>; | ||
}); | ||
|
||
#[embassy_executor::main] | ||
async fn main(_spawner: Spawner) { | ||
let p = embassy_rp::init(Default::default()); | ||
// Create the driver, from the HAL. | ||
let driver = UsbDriver::new(p.USB, Irqs); | ||
|
||
// Create embassy-usb Config | ||
let mut config = Config::new(0xc0de, 0xcafe); | ||
config.manufacturer = Some("Embassy"); | ||
config.product = Some("HID keyboard example"); | ||
config.serial_number = Some("12345678"); | ||
config.max_power = 100; | ||
config.max_packet_size_0 = 64; | ||
|
||
// Create embassy-usb DeviceBuilder using the driver and config. | ||
// It needs some buffers for building the descriptors. | ||
let mut config_descriptor = [0; 256]; | ||
let mut bos_descriptor = [0; 256]; | ||
// You can also add a Microsoft OS descriptor. | ||
let mut msos_descriptor = [0; 256]; | ||
let mut control_buf = [0; 64]; | ||
let mut request_handler = MyRequestHandler {}; | ||
let mut device_handler = MyDeviceHandler::new(); | ||
|
||
let mut state = HidState::new(); | ||
|
||
let mut builder = Builder::new( | ||
driver, | ||
config, | ||
&mut config_descriptor, | ||
&mut bos_descriptor, | ||
&mut msos_descriptor, | ||
&mut control_buf, | ||
); | ||
|
||
builder.handler(&mut device_handler); | ||
|
||
// Create classes on the builder. | ||
let config = embassy_usb::class::hid::Config { | ||
report_descriptor: KeyboardReport::desc(), | ||
request_handler: None, | ||
poll_ms: 60, | ||
max_packet_size: 64, | ||
}; | ||
let hid = HidReaderWriter::<_, 1, 8>::new(&mut builder, &mut state, config); | ||
|
||
// Build the builder. | ||
let mut usb = builder.build(); | ||
|
||
// Run the USB device. | ||
let usb_fut = usb.run(); | ||
|
||
// Set up the signal pin that will be used to trigger the keyboard. | ||
let mut signal_pin = Input::new(p.PIN_16, Pull::None); | ||
|
||
// Enable the schmitt trigger to slightly debounce. | ||
signal_pin.set_schmitt(true); | ||
|
||
let (reader, mut writer) = hid.split(); | ||
|
||
// Do stuff with the class! | ||
let in_fut = async { | ||
loop { | ||
info!("Waiting for HIGH on pin 16"); | ||
signal_pin.wait_for_high().await; | ||
info!("HIGH DETECTED"); | ||
// Create a report with the A key pressed. (no shift modifier) | ||
let report = KeyboardReport { | ||
keycodes: [4, 0, 0, 0, 0, 0], | ||
leds: 0, | ||
modifier: 0, | ||
reserved: 0, | ||
}; | ||
// Send the report. | ||
match writer.write_serialize(&report).await { | ||
Ok(()) => {} | ||
Err(e) => warn!("Failed to send report: {:?}", e), | ||
}; | ||
signal_pin.wait_for_low().await; | ||
info!("LOW DETECTED"); | ||
let report = KeyboardReport { | ||
keycodes: [0, 0, 0, 0, 0, 0], | ||
leds: 0, | ||
modifier: 0, | ||
reserved: 0, | ||
}; | ||
match writer.write_serialize(&report).await { | ||
Ok(()) => {} | ||
Err(e) => warn!("Failed to send report: {:?}", e), | ||
}; | ||
} | ||
}; | ||
|
||
let out_fut = async { | ||
reader.run(false, &mut request_handler).await; | ||
}; | ||
|
||
// Run everything concurrently. | ||
// If we had made everything `'static` above instead, we could do this using separate tasks instead. | ||
join(usb_fut, join(in_fut, out_fut)).await; | ||
} | ||
|
||
struct MyRequestHandler {} | ||
|
||
impl RequestHandler for MyRequestHandler { | ||
fn get_report(&mut self, id: ReportId, _buf: &mut [u8]) -> Option<usize> { | ||
info!("Get report for {:?}", id); | ||
None | ||
} | ||
|
||
fn set_report(&mut self, id: ReportId, data: &[u8]) -> OutResponse { | ||
info!("Set report for {:?}: {=[u8]}", id, data); | ||
OutResponse::Accepted | ||
} | ||
|
||
fn set_idle_ms(&mut self, id: Option<ReportId>, dur: u32) { | ||
info!("Set idle rate for {:?} to {:?}", id, dur); | ||
} | ||
|
||
fn get_idle_ms(&mut self, id: Option<ReportId>) -> Option<u32> { | ||
info!("Get idle rate for {:?}", id); | ||
None | ||
} | ||
} | ||
|
||
struct MyDeviceHandler { | ||
configured: AtomicBool, | ||
} | ||
|
||
impl MyDeviceHandler { | ||
fn new() -> Self { | ||
MyDeviceHandler { | ||
configured: AtomicBool::new(false), | ||
} | ||
} | ||
} | ||
|
||
impl Handler for MyDeviceHandler { | ||
fn enabled(&mut self, enabled: bool) { | ||
self.configured.store(false, Ordering::Relaxed); | ||
if enabled { | ||
info!("Device enabled"); | ||
} else { | ||
info!("Device disabled"); | ||
} | ||
} | ||
|
||
fn reset(&mut self) { | ||
self.configured.store(false, Ordering::Relaxed); | ||
info!("Bus reset, the Vbus current limit is 100mA"); | ||
} | ||
|
||
fn addressed(&mut self, addr: u8) { | ||
self.configured.store(false, Ordering::Relaxed); | ||
info!("USB address set to: {}", addr); | ||
} | ||
|
||
fn configured(&mut self, configured: bool) { | ||
self.configured.store(configured, Ordering::Relaxed); | ||
if configured { | ||
info!("Device configured, it may now draw up to the configured current limit from Vbus.") | ||
} else { | ||
info!("Device is no longer configured, the Vbus current limit is 100mA."); | ||
} | ||
} | ||
} |