WLED/wled00/wled05_init.ino

333 lines
8.1 KiB
Arduino
Raw Normal View History

/*
* Setup code
*/
void wledInit()
2019-08-17 12:27:06 +02:00
{
EEPROM.begin(EEPSIZE);
2019-08-17 12:27:06 +02:00
ledCount = EEPROM.read(229) + ((EEPROM.read(398) << 8) & 0xFF00);
2019-10-26 00:00:44 +02:00
if (ledCount > MAX_LEDS || ledCount == 0) ledCount = 30;
#ifdef ESP8266
#if LEDPIN == 3
2019-10-26 00:00:44 +02:00
if (ledCount > MAX_LEDS_DMA) ledCount = MAX_LEDS_DMA; //DMA method uses too much ram
#endif
#endif
Serial.begin(115200);
Serial.setTimeout(50);
DEBUG_PRINTLN();
DEBUG_PRINT("---WLED "); DEBUG_PRINT(versionString); DEBUG_PRINT(" "); DEBUG_PRINT(VERSION); DEBUG_PRINTLN(" INIT---");
#ifdef ARDUINO_ARCH_ESP32
DEBUG_PRINT("esp32 "); DEBUG_PRINTLN(ESP.getSdkVersion());
#else
DEBUG_PRINT("esp8266 "); DEBUG_PRINTLN(ESP.getCoreVersion());
#endif
int heapPreAlloc = ESP.getFreeHeap();
DEBUG_PRINT("heap ");
DEBUG_PRINTLN(ESP.getFreeHeap());
2019-08-17 12:27:06 +02:00
strip.init(EEPROM.read(372),ledCount,EEPROM.read(2204)); //init LEDs quickly
strip.setBrightness(0);
DEBUG_PRINT("LEDs inited. heap usage ~");
DEBUG_PRINTLN(heapPreAlloc - ESP.getFreeHeap());
2019-03-16 02:09:37 +01:00
#ifndef WLED_DISABLE_FILESYSTEM
#ifdef ARDUINO_ARCH_ESP32
SPIFFS.begin(true);
#endif
SPIFFS.begin();
#endif
2019-08-17 12:27:06 +02:00
DEBUG_PRINTLN("Load EEPROM");
loadSettingsFromEEPROM(true);
beginStrip();
2019-10-20 17:38:25 +02:00
userSetup();
if (strcmp(clientSSID,"Your_Network") == 0) showWelcomePage = true;
WiFi.persistent(false);
if (macroBoot>0) applyMacro(macroBoot);
Serial.println("Ada");
2016-11-27 22:37:51 +01:00
//generate module IDs
escapedMac = WiFi.macAddress();
escapedMac.replace(":", "");
escapedMac.toLowerCase();
2019-02-05 19:40:24 +01:00
if (strcmp(cmDNS,"x") == 0) //fill in unique mdns default
{
strcpy(cmDNS, "wled-");
sprintf(cmDNS+5, "%*s", 6, escapedMac.c_str()+6);
2019-02-05 19:40:24 +01:00
}
if (mqttDeviceTopic[0] == 0)
{
strcpy(mqttDeviceTopic, "wled/");
sprintf(mqttDeviceTopic+5, "%*s", 6, escapedMac.c_str()+6);
}
2019-08-17 12:27:06 +02:00
if (mqttClientID[0] == 0)
{
strcpy(mqttClientID, "WLED-");
sprintf(mqttClientID+5, "%*s", 6, escapedMac.c_str()+6);
}
strip.service();
2018-02-20 22:29:48 +01:00
2019-10-20 17:38:25 +02:00
#ifndef WLED_DISABLE_OTA
2018-10-04 18:17:01 +02:00
if (aOtaEnabled)
{
ArduinoOTA.onStart([]() {
2019-10-20 17:38:25 +02:00
#ifdef ESP8266
2018-10-04 18:17:01 +02:00
wifi_set_sleep_type(NONE_SLEEP_T);
#endif
DEBUG_PRINTLN("Start ArduinoOTA");
});
if (strlen(cmDNS) > 0) ArduinoOTA.setHostname(cmDNS);
}
2019-10-20 17:38:25 +02:00
#endif
//HTTP server page init
initServer();
}
2018-02-25 14:46:11 +01:00
2018-11-24 11:52:23 +01:00
void beginStrip()
{
// Initialize NeoPixel Strip and button
2020-01-02 22:10:59 +01:00
strip.setShowCallback(handleOverlayDraw);
#ifdef BTNPIN
2018-11-18 00:31:45 +01:00
pinMode(BTNPIN, INPUT_PULLUP);
#endif
if (bootPreset>0) applyPreset(bootPreset, turnOnAtBoot, true, true);
colorUpdated(0);
//init relay pin
#if RLYPIN >= 0
pinMode(RLYPIN, OUTPUT);
#if RLYMDE
digitalWrite(RLYPIN, bri);
#else
digitalWrite(RLYPIN, !bri);
#endif
#endif
//disable button if it is "pressed" unintentionally
#ifdef BTNPIN
2018-11-18 00:31:45 +01:00
if(digitalRead(BTNPIN) == LOW) buttonEnabled = false;
#else
buttonEnabled = false;
#endif
}
2018-11-24 11:52:23 +01:00
2019-10-18 12:19:52 +02:00
void initAP(bool resetAP=false){
2019-10-25 00:14:58 +02:00
if (apBehavior == 3 && !resetAP) return;
if (!apSSID[0] || resetAP) strcpy(apSSID, "WLED-AP");
if (resetAP) strcpy(apPass,"wled1234");
DEBUG_PRINT("Opening access point ");
DEBUG_PRINTLN(apSSID);
2019-09-05 22:45:59 +02:00
WiFi.softAPConfig(IPAddress(4, 3, 2, 1), IPAddress(4, 3, 2, 1), IPAddress(255,255,255,0));
2018-07-21 23:21:07 +02:00
WiFi.softAP(apSSID, apPass, apChannel, apHide);
if (!apActive) //start captive portal if AP active
{
2019-10-20 17:38:25 +02:00
DEBUG_PRINTLN("Init AP interfaces");
server.begin();
if (udpPort > 0 && udpPort != ntpLocalPort)
{
udpConnected = notifierUdp.begin(udpPort);
}
if (udpRgbPort > 0 && udpRgbPort != ntpLocalPort && udpRgbPort != udpPort)
{
udpRgbConnected = rgbUdp.begin(udpRgbPort);
2019-10-20 17:38:25 +02:00
}
dnsServer.setErrorReplyCode(DNSReplyCode::NoError);
dnsServer.start(53, "*", WiFi.softAPIP());
}
apActive = true;
}
void initConnection()
{
WiFi.disconnect(); //close old connections
2020-02-09 11:04:30 +01:00
#ifdef ESP8266
2020-02-09 10:35:32 +01:00
WiFi.setPhyMode(WIFI_PHY_MODE_11N);
2020-02-09 11:04:30 +01:00
#endif
if (staticIP[0] != 0 && staticGateway[0] != 0)
{
2019-03-09 21:41:23 +01:00
WiFi.config(staticIP, staticGateway, staticSubnet, IPAddress(8,8,8,8));
} else
{
WiFi.config(0U, 0U, 0U);
}
lastReconnectAttempt = millis();
2019-10-20 17:38:25 +02:00
if (!WLED_WIFI_CONFIGURED)
{
DEBUG_PRINT("No connection configured. ");
if (!apActive) initAP(); //instantly go to ap mode
return;
} else if (!apActive) {
2019-10-25 00:14:58 +02:00
if (apBehavior == 2)
2019-10-20 12:48:29 +02:00
{
initAP();
} else
{
DEBUG_PRINTLN("Access point disabled.");
WiFi.softAPdisconnect(true);
}
}
showWelcomePage = false;
DEBUG_PRINT("Connecting to ");
DEBUG_PRINT(clientSSID);
DEBUG_PRINTLN("...");
2019-10-18 14:06:07 +02:00
2019-10-20 17:38:25 +02:00
#ifdef ESP8266
WiFi.hostname(serverDescription);
#endif
WiFi.begin(clientSSID, clientPass);
#ifdef ARDUINO_ARCH_ESP32
WiFi.setHostname(serverDescription);
#endif
}
void initInterfaces() {
2019-10-20 17:38:25 +02:00
DEBUG_PRINTLN("Init STA interfaces");
if (hueIP[0] == 0)
{
hueIP[0] = WiFi.localIP()[0];
hueIP[1] = WiFi.localIP()[1];
hueIP[2] = WiFi.localIP()[2];
}
//init Alexa hue emulation
if (alexaEnabled) alexaInit();
#ifndef WLED_DISABLE_OTA
2019-10-25 15:32:09 +02:00
if (aOtaEnabled) ArduinoOTA.begin();
#endif
strip.service();
// Set up mDNS responder:
if (strlen(cmDNS) > 0)
{
2019-10-25 15:32:09 +02:00
if (!aOtaEnabled) MDNS.begin(cmDNS);
DEBUG_PRINTLN("mDNS started");
MDNS.addService("http", "tcp", 80);
MDNS.addService("wled", "tcp", 80);
MDNS.addServiceTxt("wled", "tcp", "mac", escapedMac.c_str());
}
2019-10-25 00:14:58 +02:00
server.begin();
2019-10-25 00:14:58 +02:00
if (udpPort > 0 && udpPort != ntpLocalPort)
{
udpConnected = notifierUdp.begin(udpPort);
if (udpConnected && udpRgbPort != udpPort) udpRgbConnected = rgbUdp.begin(udpRgbPort);
}
if (ntpEnabled) ntpConnected = ntpUdp.begin(ntpLocalPort);
2019-10-20 17:38:25 +02:00
initBlynk(blynkApiKey);
2019-12-06 01:44:45 +01:00
e131.begin((e131Multicast) ? E131_MULTICAST : E131_UNICAST , e131Universe, E131_MAX_UNIVERSE_COUNT);
reconnectHue();
2019-10-20 17:38:25 +02:00
initMqtt();
interfacesInited = true;
2019-10-26 00:00:44 +02:00
wasConnected = true;
}
byte stacO = 0;
2019-12-06 01:44:45 +01:00
uint32_t lastHeap;
unsigned long heapTime = 0;
2019-10-18 14:06:07 +02:00
void handleConnection() {
2019-11-25 00:20:00 +01:00
if (millis() < 2000 && (!WLED_WIFI_CONFIGURED || apBehavior == 2)) return;
if (lastReconnectAttempt == 0) initConnection();
2019-12-06 01:44:45 +01:00
//reconnect WiFi to clear stale allocations if heap gets too low
if (millis() - heapTime > 5000)
{
uint32_t heap = ESP.getFreeHeap();
if (heap < 9000 && lastHeap < 9000) {
DEBUG_PRINT("Heap too low! ");
DEBUG_PRINTLN(heap);
forceReconnect = true;
}
lastHeap = heap;
heapTime = millis();
}
2019-11-25 00:20:00 +01:00
2019-10-20 17:38:25 +02:00
byte stac = 0;
if (apActive) {
#ifdef ESP8266
stac = wifi_softap_get_station_num();
#else
wifi_sta_list_t stationList;
esp_wifi_ap_get_sta_list(&stationList);
stac = stationList.num;
#endif
if (stac != stacO)
{
stacO = stac;
DEBUG_PRINT("Connected AP clients: ");
DEBUG_PRINTLN(stac);
if (!WLED_CONNECTED && WLED_WIFI_CONFIGURED) { //trying to connect, but not connected
if (stac) WiFi.disconnect(); //disable search so that AP can work
else initConnection(); //restart search
}
}
}
if (forceReconnect) {
DEBUG_PRINTLN("Forcing reconnect.");
initConnection();
interfacesInited = false;
forceReconnect = false;
2019-10-26 00:00:44 +02:00
wasConnected = false;
return;
}
2019-10-18 14:06:07 +02:00
if (!WLED_CONNECTED) {
if (interfacesInited) {
DEBUG_PRINTLN("Disconnected!");
interfacesInited = false;
initConnection();
}
if (millis() - lastReconnectAttempt > ((stac) ? 300000 : 20000) && WLED_WIFI_CONFIGURED) initConnection();
2019-10-26 00:00:44 +02:00
if (!apActive && millis() - lastReconnectAttempt > 12000 && (!wasConnected || apBehavior == 1)) initAP();
} else if (!interfacesInited) { //newly connected
DEBUG_PRINTLN("");
DEBUG_PRINT("Connected! IP address: ");
DEBUG_PRINTLN(WiFi.localIP());
initInterfaces();
2019-10-20 17:38:25 +02:00
userConnected();
//shut down AP
2019-10-25 00:14:58 +02:00
if (apBehavior != 2 && apActive)
{
dnsServer.stop();
WiFi.softAPdisconnect(true);
apActive = false;
2019-10-25 00:14:58 +02:00
DEBUG_PRINTLN("Access point disabled.");
}
2019-10-18 14:06:07 +02:00
}
}
//by https://github.com/tzapu/WiFiManager/blob/master/WiFiManager.cpp
int getSignalQuality(int rssi)
{
int quality = 0;
if (rssi <= -100) {
quality = 0;
} else if (rssi >= -50) {
quality = 100;
} else {
quality = 2 * (rssi + 100);
}
return quality;
}