src/main/fc/stats.c

   1 /*
   2  * This file is part of Cleanflight and Betaflight.
   3  *
   4  * Cleanflight and Betaflight are free software. You can redistribute
   5  * this software and/or modify this software under the terms of the
   6  * GNU General Public License as published by the Free Software
   7  * Foundation, either version 3 of the License, or (at your option)
   8  * any later version.
   9  *
  10  * Cleanflight and Betaflight are distributed in the hope that they
  11  * will be useful, but WITHOUT ANY WARRANTY; without even the implied
  12  * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * See the GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this software.
  17  *
  18  * If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <math.h>
  22
  23 #include "platform.h"
  24
  25 #ifdef USE_PERSISTENT_STATS
  26
  27 #include "config/config.h"
  28
  29 #include "drivers/time.h"
  30
  31 #include "fc/dispatch.h"
  32 #include "fc/runtime_config.h"
  33
  34 #include "io/beeper.h"
  35 #include "io/gps.h"
  36
  37 #include "pg/stats.h"
  38
  39 #ifdef USE_BATTERY_CONTINUE
  40 #include "sensors/battery.h"
  41 #endif
  42
  43 #include "sensors/gyro.h"
  44
  45 #include "stats.h"
  46
  47 #define STATS_SAVE_DELAY_US 500000 // Let disarming complete and save stats after this time
  48
  49 static timeMs_t arm_millis;
  50 static uint32_t arm_distance_cm;
  51
  52 static bool saveRequired = false;
  53
  54 static void writeStats(dispatchEntry_t *self);
  55 dispatchEntry_t writeStatsEntry = { writeStats, 0, NULL, false };
  56
  57 #ifdef USE_GPS
  58     #define DISTANCE_FLOWN_CM (GPS_distanceFlownInCm)
  59 #else
  60     #define DISTANCE_FLOWN_CM (0)
  61 #endif
  62
  63 void statsInit(void)
  64 {
  65     dispatchEnable();
  66 }
  67
  68 static void writeStats(dispatchEntry_t *self)
  69 {
  70     UNUSED(self);
  71
  72     if (!ARMING_FLAG(ARMED)) {
  73         // Don't save if the user made config changes that have not yet been saved.
  74         if (!isConfigDirty()) {
  75
  76             const bool gyroIsStill = fabsf(gyro.gyroADCf[FD_ROLL]) < statsConfig()->statsSaveMoveLimit &&
  77                 fabsf(gyro.gyroADCf[FD_PITCH]) < statsConfig()->statsSaveMoveLimit &&
  78                 fabsf(gyro.gyroADCf[FD_YAW]) < statsConfig()->statsSaveMoveLimit;
  79
  80             if (gyroIsStill || statsConfig()->statsSaveMoveLimit == 0) {
  81                 writeEEPROM();
  82                 // Repeat disarming beep indicating the stats save is complete
  83                 beeper(BEEPER_DISARMING);
  84             } else {
  85                 dispatchAdd(&writeStatsEntry, STATS_SAVE_DELAY_US);
  86             }
  87         }
  88
  89         saveRequired = false;
  90     }
  91 }
  92
  93 void statsOnArm(void)
  94 {
  95     arm_millis      = millis();
  96     arm_distance_cm = DISTANCE_FLOWN_CM;
  97 }
  98
  99 void statsOnDisarm(void)
 100 {
 101     int8_t minArmedTimeS = statsConfig()->stats_min_armed_time_s;
 102     if (minArmedTimeS >= 0) {
 103         uint32_t dtS = (millis() - arm_millis) / 1000;
 104         if (dtS >= (uint8_t)minArmedTimeS) {
 105             statsConfigMutable()->stats_total_flights += 1;    // arm / flight counter
 106             statsConfigMutable()->stats_total_time_s += dtS;
 107             statsConfigMutable()->stats_total_dist_m += (DISTANCE_FLOWN_CM - arm_distance_cm) / 100;
 108 #ifdef USE_BATTERY_CONTINUE
 109             statsConfigMutable()->stats_mah_used = getMAhDrawn();
 110 #endif
 111
 112             saveRequired = true;
 113         }
 114
 115         if (saveRequired) {
 116             /* signal that stats need to be saved but don't execute time consuming flash operation
 117                now - let the disarming process complete and then execute the actual save */
 118             dispatchAdd(&writeStatsEntry, STATS_SAVE_DELAY_US);
 119         }
 120     }
 121 }
 122 #endif