src/main/fc/stats.c

   1
   2 #include "platform.h"
   3
   4 #ifdef USE_STATS
   5
   6 #include "fc/settings.h"
   7 #include "fc/stats.h"
   8
   9 #include "sensors/battery.h"
  10
  11 #include "drivers/time.h"
  12 #include "navigation/navigation.h"
  13
  14 #include "fc/config.h"
  15 #include "config/parameter_group.h"
  16 #include "config/parameter_group_ids.h"
  17
  18 #define MIN_FLIGHT_TIME_TO_RECORD_STATS_S 10    //prevent recording stats for that short "flights" [s]
  19
  20
  21 PG_REGISTER_WITH_RESET_TEMPLATE(statsConfig_t, statsConfig, PG_STATS_CONFIG, 1);
  22
  23 PG_RESET_TEMPLATE(statsConfig_t, statsConfig,
  24     .stats_enabled = SETTING_STATS_DEFAULT,
  25     .stats_total_time = SETTING_STATS_TOTAL_TIME_DEFAULT,
  26     .stats_total_dist = SETTING_STATS_TOTAL_DIST_DEFAULT,
  27 #ifdef USE_ADC
  28     .stats_total_energy = SETTING_STATS_TOTAL_ENERGY_DEFAULT
  29 #endif
  30 );
  31
  32 static uint32_t arm_millis;
  33 static uint32_t arm_distance_cm;
  34
  35 #ifdef USE_ADC
  36 static uint32_t arm_mWhDrawn;
  37 static uint32_t flyingEnergy; // energy drawn during flying up to last disarm (ARMED) mWh
  38
  39 uint32_t getFlyingEnergy(void) {
  40     return flyingEnergy;
  41 }
  42 #endif
  43
  44 void statsOnArm(void)
  45 {
  46     arm_millis      = millis();
  47     arm_distance_cm = getTotalTravelDistance();
  48 #ifdef USE_ADC
  49     arm_mWhDrawn    = getMWhDrawn();
  50 #endif
  51 }
  52
  53 void statsOnDisarm(void)
  54 {
  55     if (statsConfig()->stats_enabled) {
  56         uint32_t dt = (millis() - arm_millis) / 1000;
  57         if (dt >= MIN_FLIGHT_TIME_TO_RECORD_STATS_S) {
  58             statsConfigMutable()->stats_total_time += dt;   //[s]
  59             statsConfigMutable()->stats_total_dist += (getTotalTravelDistance() - arm_distance_cm) / 100;   //[m]
  60 #ifdef USE_ADC
  61             if (feature(FEATURE_VBAT) && isAmperageConfigured()) {
  62                 const uint32_t energy = getMWhDrawn() - arm_mWhDrawn;
  63                 statsConfigMutable()->stats_total_energy += energy;
  64                 flyingEnergy += energy;
  65             }
  66 #endif
  67             saveConfigAndNotify();
  68         }
  69     }
  70 }
  71
  72 #endif