X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=ROM2%2Fmain_basic.c;h=ae25a79390b385c49d07c30991d4fa9535368522;hb=b3523cf99eff476642609ad6537e460595f84e72;hp=94680c806a792839c42921df5bdc436bdf99e4b2;hpb=33b09b7ac579f186f8e53813017629e5031952e9;p=uccvend-snackrom.git

diff --git a/ROM2/main_basic.c b/ROM2/main_basic.c
index 94680c8..ae25a79 100644
--- a/ROM2/main_basic.c
+++ b/ROM2/main_basic.c
@@ -234,11 +234,29 @@ int main() {
 				cur_motor[0] = 0xff;
 			} else {
 				if (cur_motor[0]) {
+					u8 motor_num;
 					cur_motor[1] = last_key%10;
 					display_buf[1] = cur_motor[1]+'0';
 					set_msg(display_buf);
-					dispense_motor((cur_motor[0]%10)*10 + cur_motor[1]);
-					set_msg("THANK  YOU");
+
+					motor_num = cur_motor[0]%10;
+					motor_num *= 10;
+					motor_num += cur_motor[1];
+					switch (dispense_motor(motor_num)) {
+						case MOTOR_HOME_FAIL:
+							set_msg(" HOME FAIL ");
+							break;
+						case MOTOR_CURRENT_FAIL:
+							set_msg(" OVER CRNT ");
+							break;
+						case MOTOR_SUCCESS:
+							set_msg("THANK  YOU");
+							break;
+						default:
+							set_msg("ERRRRRRRR?");
+							break;
+					}
+
 					display_buf[0] = ' ';
 					display_buf[1] = ' ';
 					cur_motor[0] = 0xff;