From d76474bc13971983d5898993c6974d73746105ef Mon Sep 17 00:00:00 2001
From: nifi <nifi>
Date: Mon, 2 Jun 2008 13:12:07 +0000
Subject: [PATCH] Changed energest_type_time() to first update the total time,
 if the device is currently active, before returning the total time.

Added energest_flush() that updates the total time for all currently
active devices. It should be called periodically to avoid the time to
overflow for devices that are active for long periods of time.
---
 core/lib/energest.c | 32 ++++++++++++++++++++++++++++++--
 core/lib/energest.h |  7 ++++++-
 2 files changed, 36 insertions(+), 3 deletions(-)

diff --git a/core/lib/energest.c b/core/lib/energest.c
index a6d084d5d..394cd7a4c 100644
--- a/core/lib/energest.c
+++ b/core/lib/energest.c
@@ -28,7 +28,7 @@
  *
  * This file is part of the Contiki operating system.
  *
- * $Id: energest.c,v 1.6 2008/01/14 16:18:39 thiemovoigt Exp $
+ * $Id: energest.c,v 1.7 2008/06/02 13:12:07 nifi Exp $
  */
 
 /**
@@ -49,6 +49,7 @@ unsigned short energest_current_time[ENERGEST_TYPE_MAX];
 #ifdef ENERGEST_CONF_LEVELDEVICE_LEVELS
 energest_t energest_leveldevice_current_leveltime[ENERGEST_CONF_LEVELDEVICE_LEVELS];
 #endif
+unsigned char energest_current_mode[ENERGEST_TYPE_MAX];
 
 /*---------------------------------------------------------------------------*/
 void
@@ -57,8 +58,9 @@ energest_init(void)
   int i;
   for(i = 0; i < ENERGEST_TYPE_MAX; ++i) {
     energest_total_time[i].current = energest_current_time[i] = 0;
+    energest_current_mode[i] = 0;
   }
-#ifdef ENERGEST_CONF_LEVELDEVICE_LEVELS 
+#ifdef ENERGEST_CONF_LEVELDEVICE_LEVELS
   for(i = 0; i < ENERGEST_CONF_LEVELDEVICE_LEVELS; ++i) {
     energest_leveldevice_current_leveltime[i].current = 0;
   }
@@ -68,6 +70,15 @@ energest_init(void)
 unsigned long
 energest_type_time(int type)
 {
+  /* Note: does not support ENERGEST_CONF_LEVELDEVICE_LEVELS! */
+#ifndef ENERGEST_CONF_LEVELDEVICE_LEVELS
+  if(energest_current_mode[type]) {
+    rtimer_clock_t now = RTIMER_NOW();
+    energest_total_time[type].current += (unsigned long)
+      ((signed short)now - (signed short)energest_current_time[type]);
+    energest_current_time[type] = now;
+  }
+#endif /* ENERGEST_CONF_LEVELDEVICE_LEVELS */
   return energest_total_time[type].current;
 }
 /*---------------------------------------------------------------------------*/
@@ -87,8 +98,25 @@ energest_type_set(int type, unsigned long val)
   energest_total_time[type].current = val;
 }
 /*---------------------------------------------------------------------------*/
+/* Note: does not support ENERGEST_CONF_LEVELDEVICE_LEVELS! */
+void
+energest_flush(void)
+{
+  rtimer_clock_t now;
+  int i;
+  for(i = 0; i < ENERGEST_TYPE_MAX; i++) {
+    if(energest_current_mode[i]) {
+      now = RTIMER_NOW();
+      energest_total_time[i].current += (unsigned long)
+	((signed short)now - (signed short)energest_current_time[i]);
+      energest_current_time[i] = now;
+    }
+  }
+}
+/*---------------------------------------------------------------------------*/
 #else /* ENERGEST_CONF_ON */
 void energest_type_set(int type, unsigned long val) {}
 void energest_init(void) {}
 unsigned long energest_type_time(int type) { return 0; }
+void energest_flush(void) {}
 #endif /* ENERGEST_CONF_ON */
diff --git a/core/lib/energest.h b/core/lib/energest.h
index 6b139e177..54189a370 100644
--- a/core/lib/energest.h
+++ b/core/lib/energest.h
@@ -28,7 +28,7 @@
  *
  * This file is part of the Contiki operating system.
  *
- * $Id: energest.h,v 1.9 2008/01/21 12:36:17 adamdunkels Exp $
+ * $Id: energest.h,v 1.10 2008/06/02 13:12:07 nifi Exp $
  */
 
 /**
@@ -71,11 +71,13 @@ unsigned long energest_type_time(int type);
 unsigned long energest_leveldevice_leveltime(int powerlevel);
 #endif
 void energest_type_set(int type, unsigned long value);
+void energest_flush(void);
 
 #if ENERGEST_CONF_ON
 /*extern int energest_total_count;*/
 extern energest_t energest_total_time[ENERGEST_TYPE_MAX];
 extern unsigned short energest_current_time[ENERGEST_TYPE_MAX];
+extern unsigned char energest_current_mode[ENERGEST_TYPE_MAX];
 
 #ifdef ENERGEST_CONF_LEVELDEVICE_LEVELS
 extern energest_t energest_leveldevice_current_leveltime[ENERGEST_CONF_LEVELDEVICE_LEVELS];
@@ -84,16 +86,19 @@ extern energest_t energest_leveldevice_current_leveltime[ENERGEST_CONF_LEVELDEVI
 #define ENERGEST_ON(type)  do { \
                            /*++energest_total_count;*/ \
                            energest_current_time[type] = RTIMER_NOW(); \
+			   energest_current_mode[type] = 1; \
                            } while(0)
 
 #define ENERGEST_OFF(type) do { \
                            energest_total_time[type].current += (unsigned long)((signed short)RTIMER_NOW() - \
                            (signed short)energest_current_time[type]); \
+			   energest_current_mode[type] = 0; \
                            } while(0) 
 
 #define ENERGEST_OFF_LEVEL(type,level) do { \
                                         energest_leveldevice_current_leveltime[level].current += (unsigned long)((signed short)RTIMER_NOW() - \
 			                (signed short)energest_current_time[type]); \
+			   energest_current_mode[type] = 0; \
                                         } while(0)