work on trajectory, update cobboard and ballboard too
[aversive.git] / projects / microb2010 / cobboard / state.c
index f9aefa6..9d082ba 100644 (file)
@@ -1,6 +1,6 @@
-/*  
+/*
  *  Copyright Droids Corporation (2009)
- * 
+ *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
  *  the Free Software Foundation; either version 2 of the License, or
 #include "cmdline.h"
 #include "sensor.h"
 #include "actuator.h"
+#include "spickle.h"
+#include "shovel.h"
 #include "state.h"
 
 #define STMCH_DEBUG(args...) DEBUG(E_USER_ST_MACH, args)
 #define STMCH_NOTICE(args...) NOTICE(E_USER_ST_MACH, args)
 #define STMCH_ERROR(args...) ERROR(E_USER_ST_MACH, args)
 
-/* shorter aliases for this file */
-#define INIT               I2C_COBBOARD_MODE_INIT
-#define MANUAL             I2C_COBBOARD_MODE_MANUAL
-#define HARVEST            I2C_COBBOARD_MODE_HARVEST
-#define EXIT               I2C_COBBOARD_MODE_EXIT
-
-static struct i2c_cmd_cobboard_set_mode mainboard_command;
 static struct vt100 local_vt100;
-static volatile uint8_t prev_state;
-static volatile uint8_t changed = 0;
+static volatile uint8_t state_mode;
+static uint8_t cob_count;
+
+/* short aliases */
+#define L_DEPLOY(mode)   (!!((mode) & I2C_COBBOARD_MODE_L_DEPLOY))
+#define R_DEPLOY(mode)   (!!((mode) & I2C_COBBOARD_MODE_R_DEPLOY))
+#define DEPLOY(side, mode) ((side) == I2C_LEFT_SIDE ? L_DEPLOY(mode) : R_DEPLOY(mode))
+#define L_HARVEST(mode)  (!!((mode) & I2C_COBBOARD_MODE_L_HARVEST))
+#define R_HARVEST(mode)  (!!((mode) & I2C_COBBOARD_MODE_R_HARVEST))
+#define HARVEST(side, mode) ((side) == I2C_LEFT_SIDE ? L_HARVEST(mode) : R_HARVEST(mode))
+#define EJECT(mode)      (!!((mode) & I2C_COBBOARD_MODE_EJECT))
+#define INIT(mode)       (!!((mode) & I2C_COBBOARD_MODE_INIT))
 
 uint8_t state_debug = 0;
 
-void state_dump_sensors(void)
+#if 0
+static void state_dump_sensors(void)
 {
        STMCH_DEBUG("TODO\n");
 }
+#endif
 
-void state_debug_wait_key_pressed(void)
+uint8_t state_get_cob_count(void)
+{
+       return cob_count;
+}
+
+static void state_debug_wait_key_pressed(void)
 {
        if (!state_debug)
                return;
@@ -80,84 +92,217 @@ void state_debug_wait_key_pressed(void)
        while(!cmdline_keypressed());
 }
 
+/* return true if cob is present */
+static uint8_t state_cob_present(uint8_t side)
+{
+       if (side == I2C_LEFT_SIDE)
+               return sensor_get(S_LCOB);
+       else
+               return sensor_get(S_RCOB);
+}
+
+/* return the detected color of the cob (only valid if present) */
+static uint8_t state_cob_color(uint8_t side)
+{
+       /* XXX no color sensor for now */
+       if (side == I2C_LEFT_SIDE)
+               return I2C_COB_WHITE;
+       else
+               return I2C_COB_WHITE;
+}
+
+/* return true if the cob is correctly inside */
+static uint8_t state_cob_inside(void)
+{
+       return sensor_get(S_COB_INSIDE_L) &&
+               sensor_get(S_COB_INSIDE_R);
+}
+
+/* return true if there is no cob correctly inside */
+static uint8_t state_no_cob_inside(void)
+{
+       return !sensor_get(S_COB_INSIDE_L) &&
+               !sensor_get(S_COB_INSIDE_R);
+}
+
 /* set a new state, return 0 on success */
-int8_t state_set_mode(struct i2c_cmd_cobboard_set_mode *cmd)
+int8_t state_set_mode(uint8_t mode)
 {
-       changed = 1;
-       prev_state = mainboard_command.mode;
-       memcpy(&mainboard_command, cmd, sizeof(mainboard_command));
-       STMCH_DEBUG("%s mode=%d", __FUNCTION__, mainboard_command.mode);
+       state_mode = mode;
+
+/*     STMCH_DEBUG("%s(): l_deploy=%d l_harvest=%d " */
+/*                 "r_deploy=%d r_harvest=%d eject=%d", */
+/*                 __FUNCTION__, L_DEPLOY(mode), L_HARVEST(mode), */
+/*                 R_DEPLOY(mode), R_HARVEST(mode), EJECT(mode)); */
+
        return 0;
 }
 
 /* check that state is the one in parameter and that state did not
  * changed */
-uint8_t state_check(uint8_t mode)
+static uint8_t state_want_exit(void)
 {
        int16_t c;
-       if (mode != mainboard_command.mode)
-               return 0;
-
-       if (changed)
-               return 0;
 
        /* force quit when CTRL-C is typed */
        c = cmdline_getchar();
        if (c == -1)
-               return 1;
-       if (vt100_parser(&local_vt100, c) == KEY_CTRL_C) {
-               mainboard_command.mode = EXIT;
                return 0;
-       }
-       return 1;
+       if (vt100_parser(&local_vt100, c) == KEY_CTRL_C)
+               return 1;
+       printf_P(PSTR("CTRL-C\r\n"));
+       return 0;
 }
 
 uint8_t state_get_mode(void)
 {
-       return mainboard_command.mode;
+       return state_mode;
 }
 
-/* manual mode, arm position is sent from mainboard */
-static void state_do_manual(void)
+/* harvest cobs from area */
+static void state_do_harvest(uint8_t side)
 {
-       if (!state_check(MANUAL))
+       /* if there is no cob, return */
+       if (state_cob_present(side))
                return;
-       STMCH_DEBUG("%s mode=%d", __FUNCTION__, state_get_mode());
-       while (state_check(MANUAL));
-}
 
-/* init mode */
-static void state_do_init(void)
-{
-       if (!state_check(INIT))
+       /* if it is black, nothing to do */
+       if (state_cob_color(side) == I2C_COB_BLACK)
+               return;
+
+       STMCH_DEBUG("start");
+
+       /* eat the cob */
+       spickle_pack(side);
+
+       time_wait_ms(250);
+       cobroller_on(side);
+
+       if (WAIT_COND_OR_TIMEOUT(state_cob_inside(), 750) == 0) {
+               if (state_no_cob_inside()) {
+                       printf_P(PSTR("NO COB\r\n"));
+                       return;
+               }
+               printf_P(PSTR("BAD COB - press a key\r\n"));
+               while(!cmdline_keypressed());
+       }
+
+       /* cob is inside, switch off roller */
+       cobroller_off(side);
+       cob_count ++;
+
+       /* last cob, nothing to do */
+       if (cob_count == 5)
                return;
-       state_init();
-       STMCH_DEBUG("%s mode=%d", __FUNCTION__, state_get_mode());
-       while (state_check(INIT));
+
+       /* redeploy the spickle */
+       spickle_deploy(side);
+       state_debug_wait_key_pressed();
+
+       /* let the loaded cobs go */
+       servo_door_block();
+       servo_carry_open();
+       time_wait_ms(100);
+       state_debug_wait_key_pressed();
+
+       /* store it */
+       shovel_up();
+
+       if (WAIT_COND_OR_TIMEOUT(shovel_is_up(), 400) == 0) {
+               BRAKE_ON();
+               printf_P(PSTR("BLOCKED\r\n"));
+               while(!cmdline_keypressed());
+       }
+
+       state_debug_wait_key_pressed();
+
+       /* close the carry servos */
+       servo_carry_close();
+       servo_door_close();
+       time_wait_ms(200);
+       state_debug_wait_key_pressed();
+
+       shovel_down();
+
+       if (WAIT_COND_OR_TIMEOUT(shovel_is_down(), 400) == 0) {
+               BRAKE_ON();
+               printf_P(PSTR("BLOCKED\r\n"));
+               while(!cmdline_keypressed());
+       }
+
+       STMCH_DEBUG("end");
 }
 
-/* harvest columns elts from area */
-static void state_do_harvest(void)
+/* eject cobs */
+static void state_do_eject(void)
 {
-       if (!state_check(HARVEST))
-               return;
-       STMCH_DEBUG("%s mode=%d", __FUNCTION__, state_get_mode());
-       while (state_check(HARVEST));
+       cob_count = 0;
+       shovel_mid();
+       servo_carry_open();
+       servo_door_open();
+       time_wait_ms(2000);
+       shovel_down();
+       servo_door_close();
+       servo_carry_close();
 }
+
 /* main state machine */
 void state_machine(void)
 {
-       while (state_get_mode() != EXIT) {
-               changed = 0;
-               state_do_init();
-               state_do_manual();
-               state_do_harvest();
+       while (state_want_exit() == 0) {
+
+               /* init */
+               if (INIT(state_mode)) {
+                       state_mode &= (~I2C_COBBOARD_MODE_INIT);
+                       state_init();
+               }
+
+               cobroller_off(I2C_LEFT_SIDE);
+               cobroller_off(I2C_RIGHT_SIDE);
+
+               /* pack/deply spickles, enable/disable roller */
+               if (cob_count >= 5)
+                       spickle_pack(I2C_LEFT_SIDE);
+               else if (L_DEPLOY(state_mode) && !L_HARVEST(state_mode))
+                       spickle_mid(I2C_LEFT_SIDE);
+               else if (L_DEPLOY(state_mode) && L_HARVEST(state_mode))
+                       spickle_deploy(I2C_LEFT_SIDE);
+               else
+                       spickle_pack(I2C_LEFT_SIDE);
+
+               if (cob_count >= 5)
+                       spickle_pack(I2C_RIGHT_SIDE);
+               else if (R_DEPLOY(state_mode) && !R_HARVEST(state_mode))
+                       spickle_mid(I2C_RIGHT_SIDE);
+               else if (R_DEPLOY(state_mode) && R_HARVEST(state_mode))
+                       spickle_deploy(I2C_RIGHT_SIDE);
+               else
+                       spickle_pack(I2C_RIGHT_SIDE);
+
+               /* harvest */
+               if (cob_count < 5) {
+                       if (L_DEPLOY(state_mode) && L_HARVEST(state_mode))
+                               state_do_harvest(I2C_LEFT_SIDE);
+                       if (R_DEPLOY(state_mode) && R_HARVEST(state_mode))
+                               state_do_harvest(I2C_RIGHT_SIDE);
+               }
+
+               /* eject */
+               if (EJECT(state_mode)) {
+                       state_mode &= (~I2C_COBBOARD_MODE_EJECT);
+                       state_do_eject();
+               }
        }
 }
 
 void state_init(void)
 {
        vt100_init(&local_vt100);
-       mainboard_command.mode = HARVEST;
-       cobboard.cob_count = 0;
+       shovel_down();
+       servo_door_close();
+       servo_carry_close();
+       spickle_pack(I2C_LEFT_SIDE);
+       spickle_pack(I2C_RIGHT_SIDE);
+       state_mode = 0;
+       cob_count = 0;
 }