pico hid: ps/2 keyboard support finished (#146)

1 files changed, 114 insertions, 21 deletions

diff --git a/hid/pico/src/ph_ps2.c b/hid/pico/src/ph_ps2.c
index 4aca54ee..587500ac 100644
--- a/hid/pico/src/ph_ps2.c
+++ b/hid/pico/src/ph_ps2.c
@@ -30,9 +30,18 @@
 u8 ph_g_ps2_kbd_leds = 0;
 bool ph_g_ps2_kbd_online = 0;
 bool ph_g_ps2_mouse_online = 0;
+
 ph_ps2_phy ph_ps2_kbd;
 ph_ps2_phy ph_ps2_mouse;
 
+bool ph_ps2_kbd_scanning;
+u32 ph_ps2_kbd_repeat_us;
+u16 ph_ps2_kbd_delay_ms;
+u8 ph_ps2_kbd_repeat = 0;
+bool ph_ps2_kbd_repeatmod = false;
+alarm_id_t ph_ps2_kbd_repeater;
+s8 ph_ps2_is_ctrl = 0;
+
 u8 const ph_ps2_led2ps2[] = { 0, 4, 1, 5, 2, 6, 3, 7 };
 u8 const ph_ps2_mod2ps2[] = { 0x14, 0x12, 0x11, 0x1f, 0x14, 0x59, 0x11, 0x27 };
 u8 const ph_ps2_hid2ps2[] = {
@@ -46,6 +55,13 @@ u8 const ph_ps2_hid2ps2[] = {
 	0x48, 0x50, 0x57, 0x5f
 };
 u8 const ph_ps2_maparray = sizeof(ph_ps2_hid2ps2);
+u32 const ph_ps2_repeats[] = {
+	33333, 37453, 41667, 45872, 48309, 54054, 58480, 62500,
+	66667, 75188, 83333, 91743, 100000, 108696, 116279, 125000,
+	133333, 149254, 166667, 181818, 200000, 217391, 232558, 250000,
+	270270, 303030, 333333, 370370, 400000, 434783, 476190, 500000
+};
+u16 const ph_ps2_delays[] = { 250, 500, 750, 1000 };
 
 void ph_ps2_kbd_send(u8 byte) {
 	queue_try_add(&ph_ps2_kbd.qbytes, &byte);
@@ -61,24 +77,59 @@ void ph_ps2_kbd_maybe_send_e0(u8 byte) {
 	}
 }
 
+int64_t ph_ps2_repeat_callback() {
+	if (ph_ps2_kbd_repeat) {
+		if (ph_ps2_kbd_repeatmod) {
+			
+			if (ph_ps2_kbd_repeat > 3 && ph_ps2_kbd_repeat != 6) ph_ps2_kbd_send(0xe0);
+			ph_ps2_kbd_send(ph_ps2_mod2ps2[ph_ps2_kbd_repeat - 1]);
+			
+		} else {
+			
+			ph_ps2_kbd_maybe_send_e0(ph_ps2_kbd_repeat);
+			ph_ps2_kbd_send(ph_ps2_hid2ps2[ph_ps2_kbd_repeat]);
+			
+		}
+		
+		return ph_ps2_kbd_repeat_us;
+	}
+	
+	ph_ps2_kbd_repeater = 0;
+	return 0;
+}
+
+int64_t ph_ps2_blink_callback() {
+	ph_g_ps2_kbd_leds = 0;
+	ph_ps2_kbd_send(0xaa);
+	return 0;
+}
+
+void ph_ps2_kbd_reset() {
+	ph_ps2_kbd_scanning = true;
+	ph_ps2_kbd_repeat_us = 91743;
+	ph_ps2_kbd_delay_ms = 500;
+	ph_ps2_kbd_repeat = 0;
+	ph_g_ps2_kbd_leds = 7;
+	add_alarm_in_ms(500, ph_ps2_blink_callback, NULL, false);
+}
+
 void ph_ps2_kbd_receive(u8 byte, u8 prev_byte) {
-	switch(prev_byte) {
+	switch (prev_byte) {
 		case 0xed: // CMD: Set LEDs
+		  if (byte > 7) byte = 0;
 			ph_g_ps2_kbd_leds = ph_ps2_led2ps2[byte];
 		break;
 		
 		case 0xf3: // CMD: Set typematic rate and delay
-			
+			ph_ps2_kbd_repeat_us = ph_ps2_repeats[byte & 0x1f];
+			ph_ps2_kbd_delay_ms = ph_ps2_delays[(byte & 0x60) >> 5];
 		break;
 		
 		default:
-			switch(byte) {
+			switch (byte) {
 				case 0xff: // CMD: Reset
-					ph_g_ps2_kbd_online = true;
-					ph_g_ps2_kbd_leds = 0;
-					ph_ps2_kbd_send(0xfa);
-					ph_ps2_kbd_send(0xaa);
-				return;
+					ph_ps2_kbd_reset();
+				break;
 				
 				case 0xee: // CMD: Echo
 					ph_ps2_kbd_send(0xee);
@@ -91,12 +142,15 @@ void ph_ps2_kbd_receive(u8 byte, u8 prev_byte) {
 				return;
 				
 				case 0xf4: // CMD: Enable scanning
-					ph_g_ps2_kbd_online = true;
+					ph_ps2_kbd_scanning = true;
 				break;
 				
 				case 0xf5: // CMD: Disable scanning, restore default parameters
 				case 0xf6: // CMD: Set default parameters
-					ph_g_ps2_kbd_online = byte == 0xf6;
+					ph_ps2_kbd_scanning = byte == 0xf6;
+					ph_ps2_kbd_repeat_us = 91743;
+					ph_ps2_kbd_delay_ms = 500;
+					ph_ps2_kbd_repeat = 0;
 					ph_g_ps2_kbd_leds = 0;
 				break;
 			}
@@ -107,7 +161,7 @@ void ph_ps2_kbd_receive(u8 byte, u8 prev_byte) {
 }
 
 void ph_ps2_mouse_receive(u8 byte) {
-	switch(byte) {
+	switch (byte) {
 		
 	}
 }
@@ -121,35 +175,61 @@ void ph_ps2_init(void) {
 	
 	if (PH_O_IS_KBD_PS2) {
 		ph_ps2_phy_init(&ph_ps2_kbd, pio0, 11, &ph_ps2_kbd_receive); // keyboard: GPIO11=data, GPIO12=clock
-		ph_ps2_kbd_send(0xaa);
-		ph_g_ps2_kbd_online = true;
+		ph_ps2_kbd_reset();
 	}
 	
 	if (PH_O_IS_MOUSE_PS2) {
-		ph_ps2_phy_init(&ph_ps2_mouse, pio1, 14, &ph_ps2_mouse_receive); // mouse: GPIO14=data, GPIO15=clock
+		//ph_ps2_phy_init(&ph_ps2_mouse, pio1, 14, &ph_ps2_mouse_receive); // mouse: GPIO14=data, GPIO15=clock
 	}
 }
 
 void ph_ps2_task(void) {
 	if (PH_O_IS_KBD_PS2) {
 		ph_ps2_phy_task(&ph_ps2_kbd);
+		ph_g_ps2_kbd_online = ph_ps2_kbd_scanning && ph_ps2_kbd.idle;
 	}
 	
 	if (PH_O_IS_MOUSE_PS2) {
-		ph_ps2_phy_task(&ph_ps2_mouse);
+		//ph_ps2_phy_task(&ph_ps2_mouse);
 	}
 }
 
 void ph_ps2_kbd_send_key(u8 key, bool state) {
-	if (PH_O_IS_KBD_PS2 && ph_g_ps2_kbd_online) {
+	if (PH_O_IS_KBD_PS2 && ph_ps2_kbd_scanning) {
 		if (key >= 0xe0 && key <= 0xe7) {
+			
+			if (key == 0xe0 || key == 0xe4) {
+				if (state) {
+					ph_ps2_is_ctrl++;
+				} else {
+					ph_ps2_is_ctrl--;
+				}
+				
+				if (ph_ps2_is_ctrl < 0 || ph_ps2_is_ctrl > 2) {
+					ph_ps2_is_ctrl = 0;
+				}
+			}
+			
 			key -= 0xe0;
 			
 			if (key > 2 && key != 5) {
 				ph_ps2_kbd_send(0xe0);
 			}
 			
-			if (!state) {
+			if (state) {
+				ph_ps2_kbd_repeat = key + 1;
+				ph_ps2_kbd_repeatmod = true;
+				
+				if (ph_ps2_kbd_repeater) {
+					cancel_alarm(ph_ps2_kbd_repeater);
+				}
+				
+				ph_ps2_kbd_repeater = add_alarm_in_ms(ph_ps2_kbd_delay_ms, ph_ps2_repeat_callback, NULL, false);
+			} else {
+				if (ph_ps2_kbd_repeat == key + 1 && ph_ps2_kbd_repeatmod) {
+					ph_ps2_kbd_repeat = 0;
+				}
+				
 				ph_ps2_kbd_send(0xf0);
 			}
 			
@@ -158,20 +238,33 @@ void ph_ps2_kbd_send_key(u8 key, bool state) {
 		} else if (key < ph_ps2_maparray) {
 			
 			if (key == 0x48) {
+				ph_ps2_kbd_repeat = 0;
+				
 				if (state) {
-					
-					if (false) { // TODO: is shift
+					if (ph_ps2_is_ctrl) {
 						ph_ps2_kbd_send(0xe0); ph_ps2_kbd_send(0x7e); ph_ps2_kbd_send(0xe0); ph_ps2_kbd_send(0xf0); ph_ps2_kbd_send(0x7e);
 					} else {
 						ph_ps2_kbd_send(0xe1); ph_ps2_kbd_send(0x14); ph_ps2_kbd_send(0x77); ph_ps2_kbd_send(0xe1);
 						ph_ps2_kbd_send(0xf0); ph_ps2_kbd_send(0x14); ph_ps2_kbd_send(0xf0); ph_ps2_kbd_send(0x77);
 					}
-					
 				}
 			} else {
 				ph_ps2_kbd_maybe_send_e0(key);
 				
-				if (!state) {
+				if (state) {
+					ph_ps2_kbd_repeat = key;
+					ph_ps2_kbd_repeatmod = false;
+					
+					if (ph_ps2_kbd_repeater) {
+						cancel_alarm(ph_ps2_kbd_repeater);
+					}
+					
+					ph_ps2_kbd_repeater = add_alarm_in_ms(ph_ps2_kbd_delay_ms, ph_ps2_repeat_callback, NULL, false);
+				} else {
+					if (ph_ps2_kbd_repeat == key && !ph_ps2_kbd_repeatmod) {
+						ph_ps2_kbd_repeat = 0;
+					}
+					
 					ph_ps2_kbd_send(0xf0);
 				}