@@ -28,6 +28,9 @@

 #include "display.h"

 #include "button.h"

+#include <EEPROM.h>

+#define EEPROM_SIZE 1 // To save BeaconType upon restart

+

 BLEAdvertising *pAdvertising;

@@ -80,6 +83,7 @@ void setup() {

   Heltec.display->clear();

   #endif //_HELTEC_H_

   Serial.begin(115200);

+  EEPROM.begin(EEPROM_SIZE);

   Serial.println("------------------------------");

@@ -117,7 +121,32 @@ void setup() {

   }

   sprintf(macStr,"%02x:%02x:%02x:%02x:%02x:%02x", new_mac[0], new_mac[1], new_mac[2], new_mac[3], new_mac[4], new_mac[5]);

-  beaconTypeStr = "Init(Safe)  ";

+  beacon_type = EEPROM.read(0);

+  if( ( 0 == beacon_type ) || ( 0xff == beacon_type ) ) {

+    beacon_type = EN_BEACON_TYPE_safe;

+    beaconTypeStr = "Init(Safe)  ";  

+  } 

+  switch( beacon_type ) {

+  case EN_BEACON_TYPE_entry:

+    beaconTypeStr="Init(Entry) ";

+    break;

+  case EN_BEACON_TYPE_hazard:

+    beaconTypeStr="Init(Hazard)";

+    break;

+  case EN_BEACON_TYPE_safe:

+    beaconTypeStr="Init(Safe) ";

+    break;

+  case EN_BEACON_TYPE_exit:

+    beaconTypeStr="Init(Exit) ";

+    break;

+  case EN_BEACON_TYPE_other:

+    beaconTypeStr="Init(Other)";

+    break;

+  default:

+    beacon_type = EN_BEACON_TYPE_safe;

+    beaconTypeStr="Init(Safe)  ";

+    break;

+  } 

   // Create the BLE Device

   BLEDevice::init(BEACON_NAME);

@@ -1,6 +1,7 @@

 #ifndef _BUTTON_H

 #define _BUTTON_H

+#include <EEPROM.h>

 #include "beacon.h"

 int buttonState = 0;         // variable for reading the pushbutton status

@@ -39,6 +40,8 @@ void pollPrg()

       break;

     }

     Serial.println("Switching to "+beaconTypeStr);

+    EEPROM.write(0, beacon_type);

+    EEPROM.commit();

   }

   DisplayButtonStatus();

 }