213 lines
6.2 KiB
C++
213 lines
6.2 KiB
C++
#include <Arduino.h>
|
|
#line 1 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
#include <WiFi.h>
|
|
#include <PubSubClient.h>
|
|
#include <driver/i2s.h>
|
|
#include <Freenove_WS2812_Lib_for_ESP32.h>
|
|
|
|
// --- CONNECTIVITY ---
|
|
const char* WIFI_SSID = "Aussie Broadband 8729";
|
|
const char* WIFI_PASS = "Ffdfmunfca";
|
|
|
|
const char* MQTT_BROKER = "192.168.20.13";
|
|
const int MQTT_PORT = 1883;
|
|
const char* MQTT_USER = "mqtt-user";
|
|
const char* MQTT_PASS = "sam4jo";
|
|
|
|
const char* TOPIC_AUDIO = "voice/audio_stream";
|
|
const char* TOPIC_ACK = "voice/status";
|
|
|
|
// --- HARDWARE ---
|
|
#define I2S_SCK 4
|
|
#define I2S_WS 5
|
|
#define I2S_SD 6
|
|
#define I2S_PORT I2S_NUM_0
|
|
#define LED_PIN 1
|
|
#define NUM_LEDS 12
|
|
|
|
// --- TUNING ---
|
|
#define VAD_THRESHOLD 1000
|
|
const int BLOCK_SIZE = 512;
|
|
int16_t sBuffer[BLOCK_SIZE];
|
|
|
|
// --- OBJECTS ---
|
|
WiFiClient espClient;
|
|
PubSubClient mqtt(espClient);
|
|
Freenove_ESP32_WS2812 strip(NUM_LEDS, LED_PIN, 0, TYPE_GRB);
|
|
|
|
enum DeviceState { STATE_IDLE, STATE_RECORDING, STATE_TRANSMITTING, STATE_ACK_RECEIVED };
|
|
DeviceState currentState = STATE_IDLE;
|
|
unsigned long stateTimer = 0;
|
|
|
|
const i2s_config_t i2s_config = {
|
|
.mode = (i2s_mode_t)(I2S_MODE_MASTER | I2S_MODE_RX),
|
|
.sample_rate = 16000,
|
|
.bits_per_sample = I2S_BITS_PER_SAMPLE_16BIT,
|
|
.channel_format = I2S_CHANNEL_FMT_ONLY_LEFT,
|
|
.communication_format = I2S_COMM_FORMAT_STAND_I2S,
|
|
.intr_alloc_flags = ESP_INTR_FLAG_LEVEL1,
|
|
.dma_buf_count = 8,
|
|
.dma_buf_len = 64,
|
|
.use_apll = true
|
|
};
|
|
|
|
const i2s_pin_config_t pin_config = {
|
|
.bck_io_num = I2S_SCK,
|
|
.ws_io_num = I2S_WS,
|
|
.data_out_num = I2S_PIN_NO_CHANGE,
|
|
.data_in_num = I2S_SD
|
|
};
|
|
|
|
#line 59 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void setRingColor(uint8_t r, uint8_t g, uint8_t b);
|
|
#line 64 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void updateLEDs();
|
|
#line 79 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void mqttCallback(char* topic, byte* payload, unsigned int length);
|
|
#line 84 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void setup();
|
|
#line 132 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void loop();
|
|
#line 59 "/media/sam/8294CD2994CD2111/Users/Dell/Documents/Arduino/voice_assistant/voice_assistant.ino"
|
|
void setRingColor(uint8_t r, uint8_t g, uint8_t b) {
|
|
strip.setAllLedsColor(r, g, b);
|
|
strip.show();
|
|
}
|
|
|
|
void updateLEDs() {
|
|
if (currentState == STATE_ACK_RECEIVED) {
|
|
for (int j = 0; j < 255; j += 20) {
|
|
for (int i = 0; i < NUM_LEDS; i++) {
|
|
byte pos = (i * 256 / NUM_LEDS + j) & 255;
|
|
strip.setLedColorData(i, pos, 255-pos, 0); // Simple fast rainbow
|
|
}
|
|
strip.show();
|
|
delay(5);
|
|
}
|
|
currentState = STATE_IDLE;
|
|
setRingColor(0,0,0);
|
|
}
|
|
}
|
|
|
|
void mqttCallback(char* topic, byte* payload, unsigned int length) {
|
|
Serial.print("MSG Recv: "); Serial.println(topic);
|
|
currentState = STATE_ACK_RECEIVED;
|
|
}
|
|
|
|
void setup() {
|
|
delay(3000); // Wait for Serial
|
|
Serial.begin(115200);
|
|
Serial.println("\n\n=== BOOTING DEBUG MODE ===");
|
|
|
|
// 1. LEDs
|
|
strip.begin();
|
|
strip.setBrightness(30);
|
|
setRingColor(50, 50, 50); // White
|
|
Serial.println("1. LEDs Ready");
|
|
|
|
// 2. I2S
|
|
if (i2s_driver_install(I2S_PORT, &i2s_config, 0, NULL) != ESP_OK) {
|
|
Serial.println("ERR: I2S Driver Fail");
|
|
} else {
|
|
i2s_set_pin(I2S_PORT, &pin_config);
|
|
Serial.println("2. I2S Ready");
|
|
}
|
|
|
|
// 3. WiFi
|
|
Serial.print("3. WiFi Connecting: ");
|
|
Serial.println(WIFI_SSID);
|
|
WiFi.begin(WIFI_SSID, WIFI_PASS);
|
|
|
|
int wifi_timeout = 0;
|
|
while (WiFi.status() != WL_CONNECTED) {
|
|
delay(500);
|
|
Serial.print(".");
|
|
setRingColor(0, 0, 255); delay(100); setRingColor(0,0,0);
|
|
wifi_timeout++;
|
|
if(wifi_timeout > 20) {
|
|
Serial.println("\nERR: WiFi Timeout! Checking Creds...");
|
|
wifi_timeout = 0; // Keep trying or reset?
|
|
}
|
|
}
|
|
Serial.print("\n WiFi Connected! IP: ");
|
|
Serial.println(WiFi.localIP());
|
|
|
|
// 4. MQTT
|
|
mqtt.setServer(MQTT_BROKER, MQTT_PORT);
|
|
mqtt.setCallback(mqttCallback);
|
|
mqtt.setBufferSize(2048); // Vital for audio
|
|
Serial.println("4. MQTT Configured");
|
|
|
|
setRingColor(0,0,0);
|
|
Serial.println("=== SYSTEM READY ===");
|
|
}
|
|
|
|
void loop() {
|
|
// MQTT Reconnect Handler
|
|
if (!mqtt.connected()) {
|
|
Serial.print("MQTT Connecting to "); Serial.print(MQTT_BROKER); Serial.print(" ... ");
|
|
if (mqtt.connect("ESP32Mic", MQTT_USER, MQTT_PASS)) {
|
|
Serial.println("CONNECTED");
|
|
mqtt.subscribe(TOPIC_ACK);
|
|
setRingColor(0, 255, 0); delay(500); setRingColor(0,0,0); // Flash Green on Connect
|
|
} else {
|
|
Serial.print("FAIL rc="); Serial.println(mqtt.state());
|
|
delay(2000);
|
|
return;
|
|
}
|
|
}
|
|
mqtt.loop();
|
|
updateLEDs();
|
|
|
|
size_t bytesRead;
|
|
i2s_read(I2S_PORT, &sBuffer, sizeof(sBuffer), &bytesRead, portMAX_DELAY);
|
|
|
|
// Gain Boost (x6)
|
|
for (int i = 0; i < bytesRead / 2; i++) {
|
|
int32_t boosted = sBuffer[i] * 6;
|
|
if (boosted > 32767) boosted = 32767;
|
|
if (boosted < -32768) boosted = -32768;
|
|
sBuffer[i] = (int16_t)boosted;
|
|
}
|
|
|
|
long sum = 0;
|
|
for (int i = 0; i < bytesRead / 2; i++) {
|
|
sum += abs(sBuffer[i]);
|
|
}
|
|
int avg = sum / (bytesRead / 2);
|
|
|
|
// Uncomment to debug sensitivity
|
|
// Serial.printf("Vol: %d\n", avg);
|
|
|
|
// Trigger Logic
|
|
if (currentState == STATE_IDLE && avg > VAD_THRESHOLD) {
|
|
Serial.println("TRIGGERED");
|
|
currentState = STATE_RECORDING;
|
|
stateTimer = millis();
|
|
setRingColor(0, 255, 0); // Green
|
|
|
|
// START OF SENTENCE FIX: Send the trigger buffer immediately
|
|
mqtt.publish(TOPIC_AUDIO, (const uint8_t*)sBuffer, bytesRead);
|
|
}
|
|
|
|
if (currentState == STATE_RECORDING) {
|
|
mqtt.publish(TOPIC_AUDIO, (const uint8_t*)sBuffer, bytesRead);
|
|
|
|
if (avg > VAD_THRESHOLD) stateTimer = millis();
|
|
|
|
if (millis() - stateTimer > 1500) {
|
|
Serial.println("SILENCE DETECTED");
|
|
currentState = STATE_TRANSMITTING;
|
|
mqtt.publish("voice/status", "processing");
|
|
setRingColor(0, 0, 255); // Blue
|
|
}
|
|
}
|
|
|
|
if (currentState == STATE_TRANSMITTING && millis() - stateTimer > 4000) {
|
|
Serial.println("TIMEOUT WAIT");
|
|
currentState = STATE_IDLE;
|
|
setRingColor(0,0,0);
|
|
}
|
|
}
|
|
|