esphome/components/esp32_presense/esp32_presense.cpp

153 lines
5.6 KiB
C++
Raw Normal View History

2023-12-07 23:51:43 +03:00
#include "esp32_presense.h"
2023-12-10 10:29:34 +03:00
#include "defaults.h"
2023-12-07 23:51:43 +03:00
namespace esphome
{
2023-12-10 09:08:23 +03:00
namespace esp32_presense
2023-12-07 23:51:43 +03:00
{
unsigned int totalSeen = 0;
unsigned int totalFpSeen = 0;
unsigned int totalFpQueried = 0;
unsigned int totalFpReported = 0;
2023-12-10 10:29:34 +03:00
TimerHandle_t reconnectTimer;
TaskHandle_t scanTaskHandle;
2023-12-07 23:51:43 +03:00
2023-12-10 10:29:34 +03:00
unsigned long updateStartedMillis = 0;
unsigned long lastTeleMillis = 0;
int reconnectTries = 0;
int teleFails = 0;
int reportFailed = 0;
bool online = false; // Have we successfully sent status=online
bool sentDiscovery = false; // Have we successfully sent discovery
UBaseType_t bleStack = 0;
2023-12-07 23:51:43 +03:00
2023-12-10 10:29:34 +03:00
bool discovery, publishTele, publishRooms, publishDevices;
class MyAdvertisedDeviceCallbacks : public BLEAdvertisedDeviceCallbacks {
void onResult(BLEAdvertisedDevice *advertisedDevice) {
bleStack = uxTaskGetStackHighWaterMark(nullptr);
BleFingerprintCollection::Seen(advertisedDevice);
}
};
void scanTask(void *parameter) {
NimBLEDevice::init("ESPresense");
// Enrollment::Setup();
NimBLEDevice::setMTU(23);
auto pBLEScan = NimBLEDevice::getScan();
pBLEScan->setInterval(BLE_SCAN_INTERVAL);
pBLEScan->setWindow(BLE_SCAN_WINDOW);
pBLEScan->setAdvertisedDeviceCallbacks(new MyAdvertisedDeviceCallbacks(), true);
pBLEScan->setActiveScan(false);
pBLEScan->setDuplicateFilter(false);
pBLEScan->setMaxResults(0);
if (!pBLEScan->start(0, nullptr, false))
ESP_LOGE(TAG, "Error starting continuous ble scan");
while (true) {
for (auto &f : BleFingerprintCollection::fingerprints)
if (f->query())
totalFpQueried++;
// Enrollment::Loop();
if (!pBLEScan->isScanning()) {
if (!pBLEScan->start(0, nullptr, true))
ESP_LOGE(TAG, "Error re-starting continuous ble scan");
delay(3000); // If we stopped scanning, don't query for 3 seconds in order for us to catch any missed broadcasts
} else {
delay(100);
}
}
}
void ESP32Presense::setup()
{
2023-12-10 11:18:04 +03:00
ESP_LOGD(TAG, "SETUP1")
2023-12-10 10:29:34 +03:00
BleFingerprintCollection::Setup();
2023-12-10 11:18:04 +03:00
ESP_LOGD(TAG, "SETUP2")
2023-12-10 10:29:34 +03:00
xTaskCreatePinnedToCore(scanTask, "scanTask", SCAN_TASK_STACK_SIZE, nullptr, 1, &scanTaskHandle, CONFIG_BT_NIMBLE_PINNED_TO_CORE);
2023-12-10 11:18:04 +03:00
ESP_LOGD(TAG, "SETUP3")
2023-12-10 10:29:34 +03:00
}
void ESP32Presense::loop()
{
2023-12-10 11:18:04 +03:00
ESP_LOGD(TAG, "LOOP1")
2023-12-10 10:29:34 +03:00
reportLoop();
2023-12-10 11:18:04 +03:00
ESP_LOGD(TAG, "LOOP2")
2023-12-10 10:29:34 +03:00
}
2023-12-07 23:51:43 +03:00
void ESP32Presense::reportLoop()
{
auto copy = BleFingerprintCollection::GetCopy();
unsigned int count = 0;
2023-12-10 10:29:34 +03:00
2023-12-07 23:51:43 +03:00
for (auto &i : copy)
if (i->shouldCount())
count++;
2023-12-10 10:29:34 +03:00
yield();
2023-12-07 23:51:43 +03:00
sendTelemetry(totalSeen, totalFpSeen, totalFpQueried, totalFpReported, count);
2023-12-10 10:29:34 +03:00
yield();
2023-12-07 23:51:43 +03:00
auto reported = 0;
for (auto &f : copy) {
auto seen = f->getSeenCount();
if (seen) {
totalSeen += seen;
totalFpSeen++;
}
/*
if (f->hasReport()) {
if (reportBuffer(f))
f->clearReport();
}
if (reportDevice(f)) {
totalFpReported++;
reported++;
}
*/
}
}
bool ESP32Presense::sendTelemetry(
unsigned int totalSeen,
unsigned int totalFpSeen,
unsigned int totalFpQueried,
unsigned int totalFpReported,
unsigned int count
) {
this->publish(this->room_ + "/status", "online");
this->publish(this->room_ + "/max_distance", BleFingerprintCollection::maxDistance);
this->publish(this->room_ + "/absorption", BleFingerprintCollection::absorption);
this->publish(this->room_ + "/tx_ref_rssi", BleFingerprintCollection::txRefRssi);
this->publish(this->room_ + "/rx_adj_rssi", BleFingerprintCollection::rxAdjRssi);
this->publish(this->room_ + "/query", BleFingerprintCollection::query);
this->publish(this->room_ + "/include", BleFingerprintCollection::include);
this->publish(this->room_ + "/exclude", BleFingerprintCollection::exclude);
this->publish(this->room_ + "/known_macs", BleFingerprintCollection::knownMacs);
this->publish(this->room_ + "/known_irks", BleFingerprintCollection::knownIrks);
this->publish(this->room_ + "/count_ids", BleFingerprintCollection::countIds);
return true;
}
/*
void ESP32NimbleMQTTRoom::on_result(nimble_distance_custom::NimbleDistanceCustomResult& result)
{
auto address = result.address.toString();
this->publish_json(
this->base_topic_ + "/devices/" + address + "/" + this->room_,
[=](ArduinoJson::JsonObject root) -> void {
root["id"] = address;
root["distance"] = result.distance;
}
);
};
*/
} // namespace esp32_nimble_tracker
} // namespace esphome