mirror of
https://github.com/jomjol/AI-on-the-edge-device.git
synced 2025-12-07 20:16:55 +03:00
265 lines
8.5 KiB
C++
265 lines
8.5 KiB
C++
#include <string>
|
|
#include "freertos/FreeRTOS.h"
|
|
#include "freertos/task.h"
|
|
#include "freertos/event_groups.h"
|
|
|
|
#include "driver/gpio.h"
|
|
#include "sdkconfig.h"
|
|
|
|
// SD-Card ////////////////////
|
|
#include "nvs_flash.h"
|
|
#include "esp_vfs_fat.h"
|
|
#include "sdmmc_cmd.h"
|
|
#include "driver/sdmmc_host.h"
|
|
#include "driver/sdmmc_defs.h"
|
|
///////////////////////////////
|
|
|
|
#include "ClassLogFile.h"
|
|
|
|
#include "connect_wlan.h"
|
|
#include "read_wlanini.h"
|
|
|
|
#include "server_main.h"
|
|
#include "server_tflite.h"
|
|
#include "server_file.h"
|
|
#include "server_ota.h"
|
|
#include "time_sntp.h"
|
|
#include "ClassControllCamera.h"
|
|
#include "server_main.h"
|
|
#include "server_camera.h"
|
|
#include "Helper.h"
|
|
|
|
extern const char* GIT_TAG;
|
|
extern const char* GIT_REV;
|
|
extern const char* GIT_BRANCH;
|
|
extern const char* BUILD_TIME;
|
|
|
|
|
|
#define __HIDE_PASSWORD
|
|
|
|
// #include "jomjol_WS2812Slow.h"
|
|
#include "SmartLeds.h"
|
|
|
|
|
|
#define __SD_USE_ONE_LINE_MODE__
|
|
|
|
#include "server_GPIO.h"
|
|
|
|
|
|
#define BLINK_GPIO GPIO_NUM_33
|
|
|
|
static const char *TAGMAIN = "main";
|
|
|
|
//#define FLASH_GPIO GPIO_NUM_4
|
|
|
|
bool Init_NVS_SDCard()
|
|
{
|
|
esp_err_t ret = nvs_flash_init();
|
|
if (ret == ESP_ERR_NVS_NO_FREE_PAGES) {
|
|
ESP_ERROR_CHECK(nvs_flash_erase());
|
|
ret = nvs_flash_init();
|
|
}
|
|
////////////////////////////////////////////////
|
|
|
|
ESP_LOGI(TAGMAIN, "Using SDMMC peripheral");
|
|
sdmmc_host_t host = SDMMC_HOST_DEFAULT();
|
|
|
|
// This initializes the slot without card detect (CD) and write protect (WP) signals.
|
|
// Modify slot_config.gpio_cd and slot_config.gpio_wp if your board has these signals.
|
|
sdmmc_slot_config_t slot_config = SDMMC_SLOT_CONFIG_DEFAULT();
|
|
|
|
// To use 1-line SD mode, uncomment the following line:
|
|
|
|
#ifdef __SD_USE_ONE_LINE_MODE__
|
|
slot_config.width = 1;
|
|
#endif
|
|
|
|
// GPIOs 15, 2, 4, 12, 13 should have external 10k pull-ups.
|
|
// Internal pull-ups are not sufficient. However, enabling internal pull-ups
|
|
// does make a difference some boards, so we do that here.
|
|
gpio_set_pull_mode(GPIO_NUM_15, GPIO_PULLUP_ONLY); // CMD, needed in 4- and 1- line modes
|
|
gpio_set_pull_mode(GPIO_NUM_2, GPIO_PULLUP_ONLY); // D0, needed in 4- and 1-line modes
|
|
#ifndef __SD_USE_ONE_LINE_MODE__
|
|
gpio_set_pull_mode(GPIO_NUM_4, GPIO_PULLUP_ONLY); // D1, needed in 4-line mode only
|
|
gpio_set_pull_mode(GPIO_NUM_12, GPIO_PULLUP_ONLY); // D2, needed in 4-line mode only
|
|
#endif
|
|
gpio_set_pull_mode(GPIO_NUM_13, GPIO_PULLUP_ONLY); // D3, needed in 4- and 1-line modes
|
|
|
|
// Options for mounting the filesystem.
|
|
// If format_if_mount_failed is set to true, SD card will be partitioned and
|
|
// formatted in case when mounting fails.
|
|
esp_vfs_fat_sdmmc_mount_config_t mount_config = {
|
|
.format_if_mount_failed = false,
|
|
.max_files = 7, // anstatt 5 (2022-09-21)
|
|
.allocation_unit_size = 16 * 1024
|
|
};
|
|
|
|
// Use settings defined above to initialize SD card and mount FAT filesystem.
|
|
// Note: esp_vfs_fat_sdmmc_mount is an all-in-one convenience function.
|
|
// Please check its source code and implement error recovery when developing
|
|
// production applications.
|
|
sdmmc_card_t* card;
|
|
ret = esp_vfs_fat_sdmmc_mount("/sdcard", &host, &slot_config, &mount_config, &card);
|
|
|
|
if (ret != ESP_OK) {
|
|
if (ret == ESP_FAIL) {
|
|
ESP_LOGE(TAGMAIN, "Failed to mount filesystem. "
|
|
"If you want the card to be formatted, set format_if_mount_failed = true.");
|
|
} else {
|
|
ESP_LOGE(TAGMAIN, "Failed to initialize the card (%s). "
|
|
"Make sure SD card lines have pull-up resistors in place.", esp_err_to_name(ret));
|
|
}
|
|
return false;
|
|
}
|
|
sdmmc_card_print_info(stdout, card);
|
|
|
|
return true;
|
|
}
|
|
|
|
void task_NoSDBlink(void *pvParameter)
|
|
{
|
|
gpio_pad_select_gpio(BLINK_GPIO);
|
|
gpio_set_direction(BLINK_GPIO, GPIO_MODE_OUTPUT);
|
|
|
|
|
|
TickType_t xDelay;
|
|
xDelay = 100 / portTICK_PERIOD_MS;
|
|
printf("SD-Card could not be inialized - STOP THE PROGRAMM HERE\n");
|
|
|
|
while (1)
|
|
{
|
|
gpio_set_level(BLINK_GPIO, 1);
|
|
vTaskDelay( xDelay );
|
|
gpio_set_level(BLINK_GPIO, 0);
|
|
vTaskDelay( xDelay );
|
|
|
|
}
|
|
vTaskDelete(NULL); //Delete this task if it exits from the loop above
|
|
}
|
|
|
|
|
|
extern "C" void app_main(void)
|
|
{
|
|
TickType_t xDelay;
|
|
string versionFormated = "Branch: '" + std::string(GIT_BRANCH) + "', Tag: '" + std::string(GIT_TAG) + \
|
|
"', Revision: " + std::string(GIT_REV) +", Date/Time: " + std::string(BUILD_TIME);
|
|
|
|
printf("=============================================================================================\n");
|
|
printf("%s\n", versionFormated.c_str());
|
|
printf("=============================================================================================\n");
|
|
|
|
PowerResetCamera();
|
|
esp_err_t cam = Camera.InitCam();
|
|
Camera.LightOnOff(false);
|
|
xDelay = 2000 / portTICK_PERIOD_MS;
|
|
printf("After camera initialization: sleep for : %ldms\n", (long) xDelay);
|
|
vTaskDelay( xDelay );
|
|
|
|
|
|
if (!Init_NVS_SDCard())
|
|
{
|
|
xTaskCreate(&task_NoSDBlink, "task_NoSDBlink", configMINIMAL_STACK_SIZE * 64, NULL, tskIDLE_PRIORITY+1, NULL);
|
|
return;
|
|
};
|
|
|
|
CheckOTAUpdate();
|
|
|
|
char *ssid = NULL, *passwd = NULL, *hostname = NULL, *ip = NULL, *gateway = NULL, *netmask = NULL, *dns = NULL;
|
|
LoadWlanFromFile("/sdcard/wlan.ini", ssid, passwd, hostname, ip, gateway, netmask, dns);
|
|
|
|
if (ssid != NULL && passwd != NULL)
|
|
#ifdef __HIDE_PASSWORD
|
|
printf("\nWLan: %s, XXXXXX\n", ssid);
|
|
#else
|
|
printf("\nWLan: %s, %s\n", ssid, passwd);
|
|
#endif
|
|
|
|
else
|
|
printf("No SSID and PASSWORD set!!!");
|
|
|
|
if (hostname != NULL)
|
|
printf("Hostename: %s\n", hostname);
|
|
else
|
|
printf("Hostname not set.\n");
|
|
|
|
if (ip != NULL && gateway != NULL && netmask != NULL)
|
|
printf("Fixed IP: %s, Gateway %s, Netmask %s\n", ip, gateway, netmask);
|
|
if (dns != NULL)
|
|
printf("DNS IP: %s\n", dns);
|
|
|
|
|
|
wifi_init_sta(ssid, passwd, hostname, ip, gateway, netmask, dns);
|
|
|
|
|
|
xDelay = 2000 / portTICK_PERIOD_MS;
|
|
printf("main: sleep for : %ldms\n", (long) xDelay);
|
|
vTaskDelay( xDelay );
|
|
setup_time();
|
|
setBootTime();
|
|
LogFile.WriteToFile("=============================================================================================");
|
|
LogFile.WriteToFile("=================================== Main Started ============================================");
|
|
LogFile.WriteToFile("=============================================================================================");
|
|
LogFile.WriteToFile(versionFormated);
|
|
LogFile.SwitchOnOff(false);
|
|
|
|
std::string zw = gettimestring("%Y%m%d-%H%M%S");
|
|
printf("time %s\n", zw.c_str());
|
|
|
|
size_t _hsize = getESPHeapSize();
|
|
if (_hsize < 4000000)
|
|
{
|
|
std::string _zws = "Not enough PSRAM available. Expected 4.194.304 MByte - available: " + std::to_string(_hsize);
|
|
_zws = _zws + "\nEither not initialzed, too small (2MByte only) or not present at all. Firmware cannot start!!";
|
|
printf(_zws.c_str());
|
|
LogFile.SwitchOnOff(true);
|
|
LogFile.WriteToFile(_zws);
|
|
LogFile.SwitchOnOff(false);
|
|
} else {
|
|
if (cam != ESP_OK) {
|
|
ESP_LOGE(TAGMAIN, "Failed to initialize camera module. "
|
|
"Check that your camera module is working and connected properly.");
|
|
|
|
LogFile.SwitchOnOff(true);
|
|
LogFile.WriteToFile("Failed to initialize camera module. "
|
|
"Check that your camera module is working and connected properly.");
|
|
LogFile.SwitchOnOff(false);
|
|
} else {
|
|
// Test Camera
|
|
camera_fb_t * fb = esp_camera_fb_get();
|
|
if (!fb) {
|
|
ESP_LOGE(TAGMAIN, "esp_camera_fb_get: Camera Capture Failed");
|
|
LogFile.SwitchOnOff(true);
|
|
LogFile.WriteToFile("Camera cannot be initialzed. "
|
|
"System will reboot.");
|
|
doReboot();
|
|
}
|
|
esp_camera_fb_return(fb);
|
|
Camera.LightOnOff(false);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
xDelay = 2000 / portTICK_PERIOD_MS;
|
|
printf("main: sleep for : %ldms\n", (long) xDelay*10);
|
|
vTaskDelay( xDelay );
|
|
|
|
printf("starting server\n");
|
|
|
|
server = start_webserver();
|
|
register_server_camera_uri(server);
|
|
register_server_tflite_uri(server);
|
|
register_server_file_uri(server, "/sdcard");
|
|
register_server_ota_sdcard_uri(server);
|
|
|
|
gpio_handler_create(server);
|
|
|
|
printf("vor reg server main\n");
|
|
register_server_main_uri(server, "/sdcard");
|
|
|
|
printf("vor dotautostart\n");
|
|
TFliteDoAutoStart();
|
|
|
|
}
|
|
|