|
@@ -9,6 +9,7 @@ using namespace std;
|
|
|
|
|
|
|
|
void *shared_region = NULL;
|
|
void *shared_region = NULL;
|
|
|
uint32_t allocatedBytesForSTBI = 0;
|
|
uint32_t allocatedBytesForSTBI = 0;
|
|
|
|
|
+std::string sharedMemoryInUseFor = "";
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Reserve a large block in the PSRAM which will be shared between the different steps.
|
|
/** Reserve a large block in the PSRAM which will be shared between the different steps.
|
|
@@ -33,9 +34,24 @@ bool reserve_psram_shared_region(void) {
|
|
|
/*******************************************************************
|
|
/*******************************************************************
|
|
|
* Memory used in Take Image (STBI)
|
|
* Memory used in Take Image (STBI)
|
|
|
*******************************************************************/
|
|
*******************************************************************/
|
|
|
-void psram_init_shared_memory_for_take_image_step(void) {
|
|
|
|
|
- LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Init shared memory for step 'Take Image' (STBI buffers)");
|
|
|
|
|
|
|
+bool psram_init_shared_memory_for_take_image_step(void) {
|
|
|
|
|
+ if (sharedMemoryInUseFor != "") {
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM already in use for " + sharedMemoryInUseFor + "!");
|
|
|
|
|
+ return false;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Init shared memory for step 'Take Image' (STBI buffers)");
|
|
|
allocatedBytesForSTBI = 0;
|
|
allocatedBytesForSTBI = 0;
|
|
|
|
|
+ sharedMemoryInUseFor = "TakeImage";
|
|
|
|
|
+
|
|
|
|
|
+ return true;
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+
|
|
|
|
|
+void psram_deinit_shared_memory_for_take_image_step(void) {
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Deinit shared memory for step 'Take Image' (STBI buffers)");
|
|
|
|
|
+ allocatedBytesForSTBI = 0;
|
|
|
|
|
+ sharedMemoryInUseFor = "";
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@@ -45,7 +61,8 @@ void *psram_reserve_shared_stbi_memory(size_t size) {
|
|
|
if (size >= 100000) {
|
|
if (size >= 100000) {
|
|
|
if ((allocatedBytesForSTBI + size) > TENSOR_ARENA_SIZE + MAX_MODEL_SIZE) { // Check if it still fits in the shared region
|
|
if ((allocatedBytesForSTBI + size) > TENSOR_ARENA_SIZE + MAX_MODEL_SIZE) { // Check if it still fits in the shared region
|
|
|
LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM too small (STBI) to fit additional " +
|
|
LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM too small (STBI) to fit additional " +
|
|
|
- std::to_string(size) + "bytes! Available: " + std::to_string(TENSOR_ARENA_SIZE + MAX_MODEL_SIZE - allocatedBytesForSTBI) + " bytes!");
|
|
|
|
|
|
|
+ std::to_string(size) + " bytes! Available: " + std::to_string(TENSOR_ARENA_SIZE + MAX_MODEL_SIZE - allocatedBytesForSTBI) + " bytes!");
|
|
|
|
|
+
|
|
|
return NULL;
|
|
return NULL;
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -83,14 +100,21 @@ void psram_free_shared_stbi_memory(void *p) {
|
|
|
* During this step we only use the shared part of the PSRAM
|
|
* During this step we only use the shared part of the PSRAM
|
|
|
* for the tmpImage.
|
|
* for the tmpImage.
|
|
|
*******************************************************************/
|
|
*******************************************************************/
|
|
|
-void *psram_reserve_shared_tmp_image_memory(void) {
|
|
|
|
|
|
|
+void *psram_reserve_shared_tmp_image_memory(void) {
|
|
|
|
|
+ if (sharedMemoryInUseFor != "") {
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM already in use for " + sharedMemoryInUseFor + "!");
|
|
|
|
|
+ return NULL;
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating tmpImage (" + std::to_string(IMAGE_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating tmpImage (" + std::to_string(IMAGE_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
|
|
|
+ sharedMemoryInUseFor = "Aligning";
|
|
|
return shared_region; // Use 1th part of the shared memory for the tmpImage (only user)
|
|
return shared_region; // Use 1th part of the shared memory for the tmpImage (only user)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
void psram_free_shared_temp_image_memory(void) {
|
|
void psram_free_shared_temp_image_memory(void) {
|
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Shared memory used for tmpImage (PSRAM, part of shared memory) is free again");
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Shared memory used for tmpImage (PSRAM, part of shared memory) is free again");
|
|
|
|
|
+ sharedMemoryInUseFor = "";
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
@@ -103,18 +127,33 @@ void psram_free_shared_temp_image_memory(void) {
|
|
|
* Tensor Arena. Therefore we do not need to monitor the usage.
|
|
* Tensor Arena. Therefore we do not need to monitor the usage.
|
|
|
*******************************************************************/
|
|
*******************************************************************/
|
|
|
void *psram_get_shared_tensor_arena_memory(void) {
|
|
void *psram_get_shared_tensor_arena_memory(void) {
|
|
|
- LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating Tensor Arena (" + std::to_string(TENSOR_ARENA_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
|
|
|
- return shared_region; // Use 1th part of the shared memory for Tensor
|
|
|
|
|
|
|
+ if ((sharedMemoryInUseFor == "") || (sharedMemoryInUseFor == "Digitalization_Model")) {
|
|
|
|
|
+ sharedMemoryInUseFor = "Digitalization_Tensor";
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating Tensor Arena (" + std::to_string(TENSOR_ARENA_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
|
|
|
+ return shared_region; // Use 1th part of the shared memory for Tensor
|
|
|
|
|
+ }
|
|
|
|
|
+ else {
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM already in use for " + sharedMemoryInUseFor + "!");
|
|
|
|
|
+ return NULL;
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
void *psram_get_shared_model_memory(void) {
|
|
void *psram_get_shared_model_memory(void) {
|
|
|
- LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating Model memory (" + std::to_string(MAX_MODEL_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
|
|
|
- return (uint8_t *)shared_region + TENSOR_ARENA_SIZE; // Use 2nd part of the shared memory (after Tensor Arena) for the model
|
|
|
|
|
|
|
+ if ((sharedMemoryInUseFor == "") || (sharedMemoryInUseFor == "Digitalization_Tensor")) {
|
|
|
|
|
+ sharedMemoryInUseFor = "Digitalization_Model";
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Allocating Model memory (" + std::to_string(MAX_MODEL_SIZE) + " bytes, use shared memory in PSRAM)...");
|
|
|
|
|
+ return (uint8_t *)shared_region + TENSOR_ARENA_SIZE; // Use 2nd part of the shared memory (after Tensor Arena) for the model
|
|
|
|
|
+ }
|
|
|
|
|
+ else {
|
|
|
|
|
+ LogFile.WriteToFile(ESP_LOG_ERROR, TAG, "Shared memory in PSRAM already in use for " + sharedMemoryInUseFor + "!");
|
|
|
|
|
+ return NULL;
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
void psram_free_shared_tensor_arena_and_model_memory(void) {
|
|
void psram_free_shared_tensor_arena_and_model_memory(void) {
|
|
|
|
|
+ sharedMemoryInUseFor = "";
|
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Shared memory used for Tensor Arena and model (PSRAM, part of shared memory) is free again");
|
|
LogFile.WriteToFile(ESP_LOG_DEBUG, TAG, "Shared memory used for Tensor Arena and model (PSRAM, part of shared memory) is free again");
|
|
|
}
|
|
}
|
|
|
|
|
|