X-Git-Url: https://defiant.homedns.org/gitweb/?p=ros_wild_thumper.git;a=blobdiff_plain;f=avr%2Fmotor_ctrl%2Fmain.c;h=fbdc1cecd5ed095a1cec4eb3f6f5186c3dfc2283;hp=2713674b951e8a8f2cd6b0c0fe28812a3bbc43a7;hb=c79c0e44c7fdd7603a1910bc92fd80b6eb60ee78;hpb=a4e84ce5a9d3245334956b761e319f9f52ee162f

diff --git a/avr/motor_ctrl/main.c b/avr/motor_ctrl/main.c
index 2713674..fbdc1ce 100644
--- a/avr/motor_ctrl/main.c
+++ b/avr/motor_ctrl/main.c
@@ -87,6 +87,8 @@
  * 0x91 Motor 2 switch
  * 0x92 Motor 3 switch
  * 0x93 Motor 4 switch
+ * 0x94 Front Handicap backward
+ * 0x95 Aft Handicap forward
  * free
  * 0xA0 Reset reason
  * 0xA1 TLE Error status
@@ -104,7 +106,7 @@
 #define KI 0.051429
 #define KD 0.000378
 #define PID_T 0.01
-#define STEP_PER_M 3376.1 // wheel diameter=12cm, encoder=48cpr, gear ratio=1:34
+#define STEP_PER_M 4171.4 // wheel diameter=12cm, encoder=48cpr, gear ratio=1:34
 #define WHEEL_DIST 0.252
 
 enum mode {
@@ -156,6 +158,8 @@ static volatile ufloat_t angle={0.0};
 static volatile float cur_speed_lin=0;
 static volatile float cur_speed_rot=0;
 static volatile uint8_t count_test=0;
+static volatile uint8_t front_handicap_bwd=0;
+static volatile uint8_t aft_handicap_fwd=0;
 
 ISR(TWI_vect)
 {
@@ -340,6 +344,16 @@ ISR(TWI_vect)
 					motor4_switch = TWDR;
 					TWI_ACK;
 					break;
+				case 0x94: // Front Handicap backward
+					front_handicap_bwd = TWDR;
+					cmd_vel.bUpdate = 1;
+					TWI_ACK;
+					break;
+				case 0x95: // Aft Handicap forward
+					aft_handicap_fwd = TWDR;
+					cmd_vel.bUpdate = 1;
+					TWI_ACK;
+					break;
 				case 0xff: // bootloader
 					bootloader = TWDR;
 				default:
@@ -533,19 +547,19 @@ ISR(TWI_vect)
 					TWI_ACK;
 					break;
 				case 0x48: // Position angle MSB
-					TWDR = pos_y.i>>24;
+					TWDR = angle.i>>24;
 					TWI_ACK;
 					break;
 				case 0x49: // Position angle
-					TWDR = pos_y.i>>16;
+					TWDR = angle.i>>16;
 					TWI_ACK;
 					break;
 				case 0x4A: // Position angle
-					TWDR = pos_y.i>>8;
+					TWDR = angle.i>>8;
 					TWI_ACK;
 					break;
 				case 0x4B: // Position angle LSB
-					TWDR = pos_y.i;
+					TWDR = angle.i;
 					TWI_ACK;
 					break;
 				case 0xA0: // Reset reason
@@ -764,7 +778,7 @@ static void update_pos(void) {
 
 	angle_new = angle.f + angle_diff;
 	if (angle_new > 2*M_PI) angle_new-=2*M_PI;
-	else if (angle_new < 2*M_PI) angle_new+=2*M_PI;
+	else if (angle_new < -2*M_PI) angle_new+=2*M_PI;
 
 	translation = (diff_left_m + diff_right_m)/2.0;
 	pos_x_new = pos_x.f + cos(angle_new)*translation;
@@ -883,6 +897,9 @@ int main(void) {
 	DDRB = (1 << 3);
 	DDRC = (1 << 7) | (1 << 6) | (1 << 5) | (1 << 4) | (1 << 3) | (1 << 2);
 	DDRD = (1 << 7) | (1 << 5) | (1 << 4) | (1 << 3) | (1 << 2);
+	// Pullup TLEs EF
+	PORTB = (1 << 0) | (1 << 1) | (1 << 2);
+	PORTD = (1 << 6);
 
 	bootloader = 0x00;
 	setup_uart(9600);
@@ -948,10 +965,26 @@ int main(void) {
 			speed_wish_left*=STEP_PER_M;
 			speed_wish_right*=STEP_PER_M;
 
-			speed1_wish = speed_wish_left;
-			speed2_wish = speed_wish_left;
-			speed3_wish = speed_wish_right;
-			speed4_wish = speed_wish_right;
+			if (speed_wish_left > 0 && aft_handicap_fwd > 0) {
+				speed1_wish = speed_wish_left * (100-aft_handicap_fwd)/100.0;
+			} else {
+				speed1_wish = speed_wish_left;
+			}
+			if (speed_wish_left < 0 && front_handicap_bwd > 0) {
+				speed2_wish = speed_wish_left * (100-front_handicap_bwd)/100.0;
+			} else {
+				speed2_wish = speed_wish_left;
+			}
+			if (speed_wish_right < 0 && front_handicap_bwd > 0) {
+				speed3_wish = speed_wish_right * (100-front_handicap_bwd)/100.0;
+			} else {
+				speed3_wish = speed_wish_right;
+			}
+			if (speed_wish_right > 0 && aft_handicap_fwd > 0) {
+				speed4_wish = speed_wish_right * (100-aft_handicap_fwd)/100.0;
+			} else {
+				speed4_wish = speed_wish_right;
+			}
 			motor1_mode = MOTOR_PID;
 			motor2_mode = MOTOR_PID;
 			motor3_mode = MOTOR_PID;