ArDrone SDK 1.8 added
[mardrone] / mardrone / ARDrone_SDK_Version_1_8_20110726 / Examples / Linux / Navigation / Sources / navdata_client / navdata_ihm.c
diff --git a/mardrone/ARDrone_SDK_Version_1_8_20110726/Examples/Linux/Navigation/Sources/navdata_client/navdata_ihm.c b/mardrone/ARDrone_SDK_Version_1_8_20110726/Examples/Linux/Navigation/Sources/navdata_client/navdata_ihm.c
new file mode 100644 (file)
index 0000000..3a1c3ae
--- /dev/null
@@ -0,0 +1,329 @@
+#include <string.h>
+#include <VP_Os/vp_os_malloc.h>
+#include <VP_Os/vp_os_print.h>
+
+#include <config.h>
+
+#include "common/common.h"
+#include "ihm/ihm.h"
+#include "ihm/ihm_vision.h"
+#include "ihm/ihm_stages_o_gtk.h"
+#include "ihm/view_drone_attitude.h"
+#include "navdata_client/navdata_ihm.h"
+#include <ardrone_api.h>
+
+#ifdef PC_USE_VISION
+#     include <Vision/vision_tracker_engine.h>
+#endif
+
+#define CTRL_STATES_STRING
+#include "control_states.h"
+
+#define SIGN(x) ( x>=0 ? 1: -1)
+
+#ifdef EXTERNAL_CONTROL
+
+#include <mykonos.h>
+#include <mykonos_config.h>
+#include <ATcodec/ATcodec_api.h>
+#include <Acquisition/acq_fsm.h>
+#include <Control/control_fsm.h>
+
+extern C_RESULT set_ui_input(bool_t force_landing);
+extern bool_t is_ui_ignored(void);
+extern int32_t get_ui_misc0( void );
+
+extern mykonos_t mykonos;
+extern int32_t iphone_acceleros_enable;
+static uint32_t input_check_cpt = COM_INPUT_CHECK_CPT_THRESHOLD; // Make sure we have no input at startup
+
+static uint32_t ctrl_start, ctrl_current;
+static int32_t ctrl_watchdog;
+static int32_t cpt_50hz;
+static int32_t cpt_10hz;
+static acq_t acq;
+static ctrl_t ctrl = { 0 };
+
+#endif // ! EXTERNAL_CONTROL
+
+drone_angular_rate_references_t wref;//, wtrim;
+drone_euler_angles_references_t earef;//, eatrim;
+int32_t rc_ref_gaz;
+double tab_trims[3];
+int32_t ihm_val_idx = 0;
+
+#ifdef PC_USE_VISION
+static vp_stages_draw_trackers_config_t draw_trackers_cfg = { 0 };
+#endif
+
+
+
+/*============================================================================*/
+/**
+  * Builds a string describing the drone control state.
+  * @param ctrl_state Integer containing the control state from the unpacked navdata.
+  * @return A pointer to the built string. The returned address always points to the last built string.
+  */
+const char* ctrl_state_str(uint32_t ctrl_state)
+{
+  static char str_ctrl_state[MAX_STR_CTRL_STATE];
+
+  ctrl_string_t* ctrl_string;
+  uint32_t major = ctrl_state >> 16;
+  uint32_t minor = ctrl_state & 0xFFFF;
+
+  if( strlen(ctrl_states[major]) < MAX_STR_CTRL_STATE )
+  {
+    vp_os_memset(str_ctrl_state, 0, sizeof(str_ctrl_state));
+
+    strcat(str_ctrl_state, ctrl_states[major]);
+    ctrl_string = control_states_link[major];
+
+    if( ctrl_string != NULL && (strlen(ctrl_states[major]) + strlen(ctrl_string[minor]) < MAX_STR_CTRL_STATE) )
+    {
+      strcat( str_ctrl_state, " | " );
+      strcat( str_ctrl_state, ctrl_string[minor] );
+    }
+  }
+  else
+  {
+    vp_os_memset( str_ctrl_state, '#', sizeof(str_ctrl_state) );
+  }
+
+  return str_ctrl_state;
+}
+
+#ifdef EXTERNAL_CONTROL
+  static vision_params_t vision_params;
+#endif // ! EXTERNAL_CONTROL
+
+
+
+/*============================================================================*/
+/**
+*/
+C_RESULT navdata_ihm_init( mobile_config_t* cfg )
+{
+
+#ifdef EXTERNAL_CONTROL
+  // Watchdog init
+  set_mykonos_state(ARDRONE_CTRL_WATCHDOG_MASK, FALSE);
+  set_mykonos_state(ARDRONE_ADC_WATCHDOG_MASK,  FALSE);
+  set_mykonos_state(ARDRONE_COM_WATCHDOG_MASK,  TRUE);  // No input at startup
+  set_mykonos_state(ARDRONE_COMMAND_MASK,       FALSE);
+  set_mykonos_state(ARDRONE_TIMER_ELAPSED,      FALSE);
+
+  mykonos.vision = &vision_params;
+
+  //C_RESULT res;
+  //adc_t adc;
+
+  //adc.running_fsm     = &acq_fsm;
+
+  acq.n_accs          = mykonos.accs;
+  acq.n_gyros         = mykonos.gyros;
+
+  ctrl.euler_angles     = (euler_angles_t*) &mykonos.orientation;
+  ctrl.angular_rates    = (angular_rates_t*) mykonos.gyros;
+
+  //fsm_set_param( &adc_fsm, (void*)&adc );
+  fsm_set_param( &acq_fsm, (void*)&acq );
+  fsm_set_param( &ctrl_fsm, (void*)&ctrl );
+
+  set_mykonos_state(ARDRONE_ACQ_THREAD_ON, TRUE);
+  set_mykonos_state(ARDRONE_CONTROL_MASK, FALSE); // EULER ANGLE CONTROL
+  set_mykonos_state(ARDRONE_ALTITUDE_MASK, FALSE); // NO ALTITUDE CONTROL
+
+  ctrl_watchdog = 0;
+  ctrl_start    = 0;
+  ctrl_current  = 0;
+
+  cpt_50hz      = 0;
+  cpt_10hz      = 0;
+#endif // ! EXTERNAL_CONTROL
+
+  while (NULL != cfg && !cfg->ihm_curve_alloc_OK) sleep(1);
+
+  return C_OK;
+}
+
+
+/*============================================================================*/
+/**
+  * Navdata handling callback
+  * This callback is registered to ARDroneTool in the BEGIN_NAVDATA_HANDLER_TABLE in navdata_client.c
+  * @param pnd Pointer to the ready-to-use navdata structure
+  * @return Always returns C_OK.
+  */
+C_RESULT navdata_ihm_process( const navdata_unpacked_t* const pnd )
+{
+  //char text_buffer[1024];
+  const navdata_vision_detect_t* pndvision = &pnd->navdata_vision_detect;
+
+  gdk_threads_enter();
+
+  // States' display update for IHM
+  vp_os_memset( label_mykonos_state_value, '\0', sizeof(label_mykonos_state_value) );
+  sprintf(label_mykonos_state_value, "%08x", pnd->ardrone_state);
+
+  strcpy(label_ctrl_state_value, ctrl_state_str( pnd->navdata_demo.ctrl_state ));
+  set_control_state(pnd->navdata_demo.ctrl_state);
+
+  ihm_val_idx--;
+  if (ihm_val_idx < 0) ihm_val_idx = KIHM_N_PT2PLOT-1;
+
+  // Angular rates received and displayed in deg/s
+
+  ihm_CA[KIHM_CURVE_GYRO].tval[0][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_gyros[GYRO_X]);
+  ihm_CA[KIHM_CURVE_GYRO].tval[1][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_gyros[GYRO_Y]);
+  ihm_CA[KIHM_CURVE_GYRO].tval[2][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_gyros[GYRO_Z]);
+
+  // Accelerations are received in mg
+  // They are displayed in g: conversion gain is 1/1000
+  ihm_CA[KIHM_CURVE_ACC].tval[0][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_accs[ACC_X]) / KIHM_MIL_F;
+  ihm_CA[KIHM_CURVE_ACC].tval[1][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_accs[ACC_Y]) / KIHM_MIL_F;
+  ihm_CA[KIHM_CURVE_ACC].tval[2][ihm_val_idx] = ((double) pnd->navdata_phys_measures.phys_accs[ACC_Z]) / KIHM_MIL_F;
+
+  // Filtered  temperature is drawn in the Vbat window
+  ihm_CA[KIHM_CURVE_VBAT].tval[0][ihm_val_idx] = ((double) (pnd->navdata_phys_measures.accs_temp));   // Filtered temperature
+
+  ihm_CA[KIHM_CURVE_VBAT].tval[1][ihm_val_idx] = ((double) (pnd->navdata_demo.vbat_flying_percentage)) ; // Filtered vbat in %
+
+  ihm_CA[KIHM_CURVE_THETA].tval[2][ihm_val_idx] = ((double) pnd->navdata_euler_angles.theta_a) / KIHM_MIL_F; // theta accelerometer value
+  ihm_CA[KIHM_CURVE_PHI  ].tval[2][ihm_val_idx] = ((double) pnd->navdata_euler_angles.phi_a) / KIHM_MIL_F; // phi accelerometer value
+
+  ihm_CA[KIHM_CURVE_THETA].tval[0][ihm_val_idx] = ((double) pnd->navdata_demo.theta) / KIHM_MIL_F; // theta fused value
+  ihm_CA[KIHM_CURVE_PHI  ].tval[0][ihm_val_idx] = ((double) pnd->navdata_demo.phi) / KIHM_MIL_F; // phi fused value
+  ihm_psi                                       = ((double) pnd->navdata_demo.psi) / KIHM_MIL_F;
+
+  earef.theta = pnd->navdata_references.ref_theta;
+  ihm_CA[KIHM_CURVE_THETA].tval[1][ihm_val_idx] = ((double) earef.theta) / KIHM_MIL_F; // theta reference value
+
+  earef.phi = pnd->navdata_references.ref_phi;
+  ihm_CA[KIHM_CURVE_PHI  ].tval[1][ihm_val_idx] = ((double) earef.phi) / KIHM_MIL_F; // phi reference value
+
+  earef.psi = pnd->navdata_references.ref_psi;
+
+  wref.p = pnd->navdata_references.ref_roll;
+  wref.q = pnd->navdata_references.ref_pitch;
+  wref.r = pnd->navdata_references.ref_yaw;
+
+  rc_ref_gaz = pnd->navdata_rc_references.rc_ref_gaz;
+
+  ihm_CA[KIHM_CURVE_ALTITUDE].tval[0][ihm_val_idx] = ((double) pnd->navdata_demo.altitude);
+
+  ihm_CA[KIHM_CURVE_ALTITUDE].tval[1][ihm_val_idx] = ((double) pnd->navdata_altitude.altitude_vision);
+  ihm_CA[KIHM_CURVE_ALTITUDE].tval[2][ihm_val_idx] = ((double) pnd->navdata_altitude.altitude_ref);
+  ihm_CA[KIHM_CURVE_ALTITUDE].tval[3][ihm_val_idx] = ((double) pnd->navdata_altitude.altitude_raw);
+
+  ihm_CA[KIHM_CURVE_VZ].tval[1][ihm_val_idx] = (double) (pnd->navdata_altitude.altitude_vz);      // Fusion Vz
+
+  tab_trims[0] = (double)pnd->navdata_trims.euler_angles_trim_theta;
+  tab_trims[1] = (double)pnd->navdata_trims.euler_angles_trim_phi;
+  tab_trims[2] = (double)pnd->navdata_trims.angular_rates_trim_r;
+
+  ihm_CA[KIHM_CURVE_PWM].tval[0][ihm_val_idx] = (double) (pnd->navdata_pwm.motor1); // PWM motor 1
+  ihm_CA[KIHM_CURVE_PWM].tval[1][ihm_val_idx] = (double) (pnd->navdata_pwm.motor2); // PWM motor 2
+  ihm_CA[KIHM_CURVE_PWM].tval[2][ihm_val_idx] = (double) (pnd->navdata_pwm.motor3); // PWM motor 3
+  ihm_CA[KIHM_CURVE_PWM].tval[3][ihm_val_idx] = (double) (pnd->navdata_pwm.motor4); // PWM motor 4
+  ihm_CA[KIHM_CURVE_CURRENT].tval[0][ihm_val_idx] = (double) (pnd->navdata_pwm.current_motor1); // current motor 1
+  ihm_CA[KIHM_CURVE_CURRENT].tval[1][ihm_val_idx] = (double) (pnd->navdata_pwm.current_motor2); // current motor 2
+  ihm_CA[KIHM_CURVE_CURRENT].tval[2][ihm_val_idx] = (double) (pnd->navdata_pwm.current_motor3); // current motor 3
+  ihm_CA[KIHM_CURVE_CURRENT].tval[3][ihm_val_idx] = (double) (pnd->navdata_pwm.current_motor4); // current motor 4
+
+  ihm_CA[KIHM_CURVE_VZ].tval[3][ihm_val_idx] = (double) (pnd->navdata_pwm.vz_ref);  // VZ reference
+
+  ihm_CA[KIHM_CURVE_VX].tval[0][ihm_val_idx] = (double) (pnd->navdata_vision_raw.vision_tx_raw);      // VISION trans. en x
+  ihm_CA[KIHM_CURVE_VY].tval[0][ihm_val_idx] = (double) (pnd->navdata_vision_raw.vision_ty_raw);      // VISION trans. en y
+  ihm_CA[KIHM_CURVE_VZ].tval[0][ihm_val_idx] = (double) (pnd->navdata_vision_raw.vision_tz_raw);      // VISION trans. en z
+
+
+  if( pnd->vision_defined )
+  {
+    if( fabsf(pnd->navdata_vision_raw.vision_tx_raw) <= 10.0f && fabsf(pnd->navdata_vision_raw.vision_ty_raw) <= 10.0f )
+    {
+      ihm_dir = HOVER_VAL;
+    }
+    else
+    {
+      ihm_dir = (double) atan2f(pnd->navdata_vision_raw.vision_ty_raw, pnd->navdata_vision_raw.vision_tx_raw);
+    }
+  }
+  else
+  {
+    ihm_dir = NOT_DEF_VAL;
+  }
+
+  //Vision States' display update for IHM
+  vp_os_memset( label_vision_state_value, '\0', sizeof(label_vision_state_value) );
+  sprintf(label_vision_state_value, "%08x", pnd->navdata_vision.vision_state);
+
+  ihm_CA[KIHM_CURVE_VX].tval[1][ihm_val_idx] = (double) (pnd->navdata_demo.vx);      // Fusion Vx
+  ihm_CA[KIHM_CURVE_VY].tval[1][ihm_val_idx] = (double) (pnd->navdata_demo.vy);      // Fusion Vy
+
+#ifdef PC_USE_VISION
+  draw_trackers_cfg.num_points = NUM_MAX_SCREEN_POINTS;
+  vp_os_memset(&draw_trackers_cfg.locked[0], 0, NUM_MAX_SCREEN_POINTS * sizeof(uint32_t));
+  vp_os_memcpy(&draw_trackers_cfg.locked[0], pnd->navdata_trackers_send.locked, NUM_MAX_SCREEN_POINTS * sizeof(uint32_t));
+  vp_os_memset(&draw_trackers_cfg.points[0], 0, NUM_MAX_SCREEN_POINTS * sizeof(screen_point_t));
+  vp_os_memcpy(&draw_trackers_cfg.points[0], pnd->navdata_trackers_send.point, NUM_MAX_SCREEN_POINTS * sizeof(screen_point_t));
+
+  int i;
+  draw_trackers_cfg.detected = pnd->navdata_vision_detect.nb_detected;
+  for(i = 0 ; i < pnd->navdata_vision_detect.nb_detected ; i++)
+  {
+    draw_trackers_cfg.type[i]           = pnd->navdata_vision_detect.type[i];
+    draw_trackers_cfg.patch_center[i].x = pnd->navdata_vision_detect.xc[i];
+    draw_trackers_cfg.patch_center[i].y = pnd->navdata_vision_detect.yc[i];
+    draw_trackers_cfg.width[i]          = pnd->navdata_vision_detect.width[i];
+    draw_trackers_cfg.height[i]         = pnd->navdata_vision_detect.height[i];
+  }
+  set_draw_trackers_config(&draw_trackers_cfg);
+#endif
+
+  /*Stephane*/
+
+         extern char label_detection_state_value[32];
+         switch(pnd->navdata_demo.detection_camera_type)
+         {
+                 case CAD_TYPE_NONE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Detection disabled"); break;
+                 case CAD_TYPE_VISION:   snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching 2D tags"); break;
+                 case CAD_TYPE_COCARDE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching roundels"); break;
+                 case CAD_TYPE_ORIENTED_COCARDE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching roundels"); break;
+                 case CAD_TYPE_H_COCARDE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching H roundels"); break;
+                 case CAD_TYPE_H_ORIENTED_COCARDE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching H roundels"); break;
+                 case CAD_TYPE_STRIPE:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching ground stripe (15Hz)"); break;
+                 case CAD_TYPE_STRIPE_V:  snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Searching ground stripe (60Hz)"); break;
+                 default : snprintf(label_detection_state_value,sizeof(label_detection_state_value),"Navdata error");
+         }
+
+         char nbDetectedTags_label_buffer[1024];
+
+         if (pndvision->nb_detected==1 && ( pnd->navdata_demo.detection_camera_type==CAD_TYPE_ORIENTED_COCARDE ||  pnd->navdata_demo.detection_camera_type==CAD_TYPE_H_ORIENTED_COCARDE ) )
+         {  snprintf(nbDetectedTags_label_buffer,sizeof(nbDetectedTags_label_buffer),"Found : %i (%4.2f deg)",pndvision->nb_detected,pndvision->orientation_angle[0]); }
+         else if (pndvision->nb_detected==1 && (pnd->navdata_demo.detection_camera_type==CAD_TYPE_STRIPE || pnd->navdata_demo.detection_camera_type==CAD_TYPE_STRIPE_V))
+         {  snprintf(nbDetectedTags_label_buffer,sizeof(nbDetectedTags_label_buffer),"Found : %i (%4.2f deg - w: %i)",
+                         pndvision->nb_detected,
+                         pndvision->orientation_angle[0],
+                         pndvision->width[0]);
+         }
+         else
+         {  snprintf(nbDetectedTags_label_buffer,sizeof(nbDetectedTags_label_buffer),"Found : %i",pndvision->nb_detected);       }
+
+         if(nbDetectedTags_label!=NULL)
+                 gtk_label_set_text((GtkLabel*)nbDetectedTags_label,(const gchar*)nbDetectedTags_label_buffer);
+
+
+
+
+
+  gdk_threads_leave();
+
+  return C_OK;
+}
+
+C_RESULT navdata_ihm_release( void )
+{
+  return C_OK;
+}
+