yunTune/main/main.c

/*
 * SPDX-FileCopyrightText: 2021-2024 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Unlicense OR CC0-1.0
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <inttypes.h>
#include "freertos/FreeRTOS.h"
#include "freertos/task.h"
#include "freertos/queue.h"
#include "nvs.h"
#include "nvs_flash.h"
#include "esp_system.h"
#include "esp_log.h"

#include "esp_bt.h"
#include "bt_app_core.h"
#include "bt_app_av.h"
#include "esp_bt_main.h"
#include "esp_bt_device.h"
#include "esp_gap_bt_api.h"
#include "esp_a2dp_api.h"
#include "esp_avrc_api.h"
#include "driver/gpio.h"


/* device name */
#define LOCAL_DEVICE_NAME    "ESP_SPEAKER"

/* event for stack up */
enum {
    BT_APP_EVT_STACK_UP = 0,
};

/* GPIO definitions */
#define GPIO_BTN        GPIO_NUM_2   // 按钮引脚
#define GPIO_OUTPUT     GPIO_NUM_19  // 输出控制引脚


/* Audio mode states */
typedef enum {
    AUDIO_MODE_NORMAL = 0,
    AUDIO_MODE_BASS_BOOST,
    AUDIO_MODE_MAX
} audio_mode_t;

static audio_mode_t current_audio_mode = AUDIO_MODE_NORMAL;
static QueueHandle_t gpio_evt_queue = NULL;

/********************************
 * STATIC FUNCTION DECLARATIONS
 *******************************/

/* Device callback function */
static void bt_app_dev_cb(esp_bt_dev_cb_event_t event, esp_bt_dev_cb_param_t *param);
/* GAP callback function */
static void bt_app_gap_cb(esp_bt_gap_cb_event_t event, esp_bt_gap_cb_param_t *param);
/* handler for bluetooth stack enabled events */
static void bt_av_hdl_stack_evt(uint16_t event, void *p_param);


/* GPIO interrupt handler */
static void IRAM_ATTR gpio_isr_handler(void* arg);
/* GPIO task to handle events */
static void gpio_task(void* arg);
/* Switch audio mode */
static void switch_audio_mode(void);
/* Update GPIO output based on audio mode */
static void update_gpio_output(void);

/*******************************
 * STATIC FUNCTION DEFINITIONS
 ******************************/
static char *bda2str(uint8_t * bda, char *str, size_t size)
{
    if (bda == NULL || str == NULL || size < 18) {
        return NULL;
    }

    uint8_t *p = bda;
    sprintf(str, "%02x:%02x:%02x:%02x:%02x:%02x",
            p[0], p[1], p[2], p[3], p[4], p[5]);
    return str;
}

static void bt_app_dev_cb(esp_bt_dev_cb_event_t event, esp_bt_dev_cb_param_t *param)
{
    switch (event) {
    case ESP_BT_DEV_NAME_RES_EVT: {
        if (param->name_res.status == ESP_BT_STATUS_SUCCESS) {
            ESP_LOGI(BT_AV_TAG, "Get local device name success: %s", param->name_res.name);
        } else {
            ESP_LOGE(BT_AV_TAG, "Get local device name failed, status: %d", param->name_res.status);
        }
        break;
    }
    default: {
        ESP_LOGI(BT_AV_TAG, "event: %d", event);
        break;
    }
    }
}

static void bt_app_gap_cb(esp_bt_gap_cb_event_t event, esp_bt_gap_cb_param_t *param)
{
    uint8_t *bda = NULL;

    switch (event) {
    /* when authentication completed, this event comes */
    case ESP_BT_GAP_AUTH_CMPL_EVT: {
        if (param->auth_cmpl.stat == ESP_BT_STATUS_SUCCESS) {
            ESP_LOGI(BT_AV_TAG, "authentication success: %s", param->auth_cmpl.device_name);
            esp_log_buffer_hex(BT_AV_TAG, param->auth_cmpl.bda, ESP_BD_ADDR_LEN);
        } else {
            ESP_LOGE(BT_AV_TAG, "authentication failed, status: %d", param->auth_cmpl.stat);
        }
        ESP_LOGI(BT_AV_TAG, "link key type of current link is: %d", param->auth_cmpl.lk_type);
        break;
    }
    case ESP_BT_GAP_ENC_CHG_EVT: {
        char *str_enc[3] = {"OFF", "E0", "AES"};
        bda = (uint8_t *)param->enc_chg.bda;
        ESP_LOGI(BT_AV_TAG, "Encryption mode to [%02x:%02x:%02x:%02x:%02x:%02x] changed to %s",
                 bda[0], bda[1], bda[2], bda[3], bda[4], bda[5], str_enc[param->enc_chg.enc_mode]);
        break;
    }

#if (CONFIG_EXAMPLE_A2DP_SINK_SSP_ENABLED == true)
    /* when Security Simple Pairing user confirmation requested, this event comes */
    case ESP_BT_GAP_CFM_REQ_EVT:
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_CFM_REQ_EVT Please compare the numeric value: %"PRIu32, param->cfm_req.num_val);
        esp_bt_gap_ssp_confirm_reply(param->cfm_req.bda, true);
        break;
    /* when Security Simple Pairing passkey notified, this event comes */
    case ESP_BT_GAP_KEY_NOTIF_EVT:
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_KEY_NOTIF_EVT passkey: %"PRIu32, param->key_notif.passkey);
        break;
    /* when Security Simple Pairing passkey requested, this event comes */
    case ESP_BT_GAP_KEY_REQ_EVT:
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_KEY_REQ_EVT Please enter passkey!");
        break;
#endif

    /* when GAP mode changed, this event comes */
    case ESP_BT_GAP_MODE_CHG_EVT:
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_MODE_CHG_EVT mode: %d", param->mode_chg.mode);
        break;
    /* when ACL connection completed, this event comes */
    case ESP_BT_GAP_ACL_CONN_CMPL_STAT_EVT:
        bda = (uint8_t *)param->acl_conn_cmpl_stat.bda;
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_ACL_CONN_CMPL_STAT_EVT Connected to [%02x:%02x:%02x:%02x:%02x:%02x], status: 0x%x",
                 bda[0], bda[1], bda[2], bda[3], bda[4], bda[5], param->acl_conn_cmpl_stat.stat);
        break;
    /* when ACL disconnection completed, this event comes */
    case ESP_BT_GAP_ACL_DISCONN_CMPL_STAT_EVT:
        bda = (uint8_t *)param->acl_disconn_cmpl_stat.bda;
        ESP_LOGI(BT_AV_TAG, "ESP_BT_GAP_ACL_DISC_CMPL_STAT_EVT Disconnected from [%02x:%02x:%02x:%02x:%02x:%02x], reason: 0x%x",
                 bda[0], bda[1], bda[2], bda[3], bda[4], bda[5], param->acl_disconn_cmpl_stat.reason);
        break;
    /* others */
    default: {
        ESP_LOGI(BT_AV_TAG, "event: %d", event);
        break;
    }
    }
}

static void bt_av_hdl_stack_evt(uint16_t event, void *p_param)
{
    ESP_LOGD(BT_AV_TAG, "%s event: %d", __func__, event);

    switch (event) {
    /* when do the stack up, this event comes */
    case BT_APP_EVT_STACK_UP: {
        esp_bt_gap_set_device_name(LOCAL_DEVICE_NAME);
        esp_bt_dev_register_callback(bt_app_dev_cb);
        esp_bt_gap_register_callback(bt_app_gap_cb);

        assert(esp_avrc_ct_init() == ESP_OK);
        esp_avrc_ct_register_callback(bt_app_rc_ct_cb);
        assert(esp_avrc_tg_init() == ESP_OK);
        esp_avrc_tg_register_callback(bt_app_rc_tg_cb);

        esp_avrc_rn_evt_cap_mask_t evt_set = {0};
        esp_avrc_rn_evt_bit_mask_operation(ESP_AVRC_BIT_MASK_OP_SET, &evt_set, ESP_AVRC_RN_VOLUME_CHANGE);
        assert(esp_avrc_tg_set_rn_evt_cap(&evt_set) == ESP_OK);

        assert(esp_a2d_sink_init() == ESP_OK);
        esp_a2d_register_callback(&bt_app_a2d_cb);
        esp_a2d_sink_register_data_callback(bt_app_a2d_data_cb);

        /* Get the default value of the delay value */
        esp_a2d_sink_get_delay_value();
        /* Get local device name */
        esp_bt_gap_get_device_name();

        /* set discoverable and connectable mode, wait to be connected */
        esp_bt_gap_set_scan_mode(ESP_BT_CONNECTABLE, ESP_BT_GENERAL_DISCOVERABLE);
        break;
    }
    /* others */
    default:
        ESP_LOGE(BT_AV_TAG, "%s unhandled event: %d", __func__, event);
        break;
    }
}


static void IRAM_ATTR gpio_isr_handler(void* arg)
{
    uint32_t gpio_num = (uint32_t) arg;
    xQueueSendFromISR(gpio_evt_queue, &gpio_num, NULL);
}

static void gpio_task(void* arg)
{
    uint32_t io_num;
    for(;;) {
        if(xQueueReceive(gpio_evt_queue, &io_num, portMAX_DELAY)) {
            // 消抖处理
            vTaskDelay(50 / portTICK_PERIOD_MS);
            if(gpio_get_level(io_num) == 0) { // 确认仍然是低电平
                ESP_LOGI(BT_AV_TAG, "GPIO%"PRIu32" interrupt, button pressed", io_num);
                switch_audio_mode();
            }
        }
    }
}
static void switch_audio_mode(void)
{
    // 切换音频模式
    current_audio_mode = (current_audio_mode + 1) % AUDIO_MODE_MAX;
    
    const char* mode_names[] = {
        "NORMAL",
        "BASS_BOOST", 
    };
    
    ESP_LOGI(BT_AV_TAG, "Audio mode switched to: %s", mode_names[current_audio_mode]);
    
    // 更新GPIO输出状态
    update_gpio_output();
}
static void update_gpio_output(void)
{
    // 根据音频模式设置GPIO19输出状态
    switch(current_audio_mode) {
        case AUDIO_MODE_NORMAL:
            gpio_set_level(GPIO_OUTPUT, 0); // 低电平
            break;
        case AUDIO_MODE_BASS_BOOST:
            gpio_set_level(GPIO_OUTPUT, 1); // 高电平
            break;
        default:
            gpio_set_level(GPIO_OUTPUT, 0);
            break;
    }
    
    ESP_LOGI(BT_AV_TAG, "GPIO19 set to: %d", gpio_get_level(GPIO_OUTPUT));
}

/*******************************
 * MAIN ENTRY POINT
 ******************************/

void app_main(void)
{
    char bda_str[18] = {0};
    /* initialize NVS — it is used to store PHY calibration data */
    esp_err_t err = nvs_flash_init();
    if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND) {
        ESP_ERROR_CHECK(nvs_flash_erase());
        err = nvs_flash_init();
    }
    ESP_ERROR_CHECK(err);

    // todo 拉低引脚 io32
    // todo 拉低引脚 io32 - 已完善
    // 配置GPIO32为输出模式并拉低电平
    gpio_config_t io_conf = {
        .pin_bit_mask = (1ULL << GPIO_NUM_32),  // GPIO32
        .mode = GPIO_MODE_OUTPUT,               // 输出模式
        .pull_up_en = GPIO_PULLUP_DISABLE,      // 禁用上拉
        .pull_down_en = GPIO_PULLDOWN_DISABLE,  // 禁用下拉
        .intr_type = GPIO_INTR_DISABLE          // 禁用中断
    };
    
    // 配置GPIO
    err = gpio_config(&io_conf);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "GPIO32 config failed: %s", esp_err_to_name(err));
        return;
    }
    
    // 拉低GPIO32
    err = gpio_set_level(GPIO_NUM_32, 0);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "Set GPIO32 low failed: %s", esp_err_to_name(err));
        return;
    }

    // system state led
    gpio_config_t led_io_conf = {
        .pin_bit_mask = (1ULL << GPIO_NUM_15),  // GPIO15
        .mode = GPIO_MODE_OUTPUT,               // 输出模式
        .pull_up_en = GPIO_PULLUP_DISABLE,      // 禁用上拉
        .pull_down_en = GPIO_PULLDOWN_DISABLE,  // 禁用下拉
        .intr_type = GPIO_INTR_DISABLE          // 禁用中断
    };
    
    // 配置GPIO
    err = gpio_config(&led_io_conf);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "GPIO15 config failed: %s", esp_err_to_name(err));
        return;
    }
    
    // 拉低GPIO15
    err = gpio_set_level(GPIO_NUM_15, 0);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "Set GPIO15 low failed: %s", esp_err_to_name(err));
        return;
    }
    
    ESP_LOGI(BT_AV_TAG, "GPIO15 set to low level");

    // 配置GPIO2为输入模式，用于低电平中断
    gpio_config_t btn_conf = {
        .pin_bit_mask = (1ULL << GPIO_BTN),
        .mode = GPIO_MODE_INPUT,
        .pull_up_en = GPIO_PULLUP_ENABLE,       // 启用上拉，按钮按下时拉低
        .pull_down_en = GPIO_PULLDOWN_DISABLE,
        .intr_type = GPIO_INTR_NEGEDGE          // 下降沿触发（按下按钮）
    };
    
    err = gpio_config(&btn_conf);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "GPIO2 config failed: %s", esp_err_to_name(err));
        return;
    }

    // 配置GPIO19为输出模式
    gpio_config_t output_conf = {
        .pin_bit_mask = (1ULL << GPIO_OUTPUT),
        .mode = GPIO_MODE_OUTPUT,
        .pull_up_en = GPIO_PULLUP_DISABLE,
        .pull_down_en = GPIO_PULLDOWN_DISABLE,
        .intr_type = GPIO_INTR_DISABLE
    };
    
    err = gpio_config(&output_conf);
    if (err != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "GPIO19 config failed: %s", esp_err_to_name(err));
        return;
    }

    // 初始化GPIO19为低电平
    gpio_set_level(GPIO_OUTPUT, 0);

    // 创建GPIO事件队列
    gpio_evt_queue = xQueueCreate(10, sizeof(uint32_t));
    
    // 启动GPIO任务
    xTaskCreate(gpio_task, "gpio_task", 2048, NULL, 10, NULL);

    // 安装GPIO ISR服务
    gpio_install_isr_service(0);
    
    // 注册GPIO2的中断处理函数
    gpio_isr_handler_add(GPIO_BTN, gpio_isr_handler, (void*) GPIO_BTN);

    ESP_LOGI(BT_AV_TAG, "GPIO2 interrupt and GPIO19 output initialized");





    /*
     * This example only uses the functions of Classical Bluetooth.
     * So release the controller memory for Bluetooth Low Energy.
     */
    ESP_ERROR_CHECK(esp_bt_controller_mem_release(ESP_BT_MODE_BLE));

    esp_bt_controller_config_t bt_cfg = BT_CONTROLLER_INIT_CONFIG_DEFAULT();
    if ((err = esp_bt_controller_init(&bt_cfg)) != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "%s initialize controller failed: %s", __func__, esp_err_to_name(err));
        return;
    }
    if ((err = esp_bt_controller_enable(ESP_BT_MODE_CLASSIC_BT)) != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "%s enable controller failed: %s", __func__, esp_err_to_name(err));
        return;
    }

    esp_bluedroid_config_t bluedroid_cfg = BT_BLUEDROID_INIT_CONFIG_DEFAULT();
#if (CONFIG_EXAMPLE_A2DP_SINK_SSP_ENABLED == false)
    bluedroid_cfg.ssp_en = false;
#endif
    if ((err = esp_bluedroid_init_with_cfg(&bluedroid_cfg)) != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "%s initialize bluedroid failed: %s", __func__, esp_err_to_name(err));
        return;
    }

    if ((err = esp_bluedroid_enable()) != ESP_OK) {
        ESP_LOGE(BT_AV_TAG, "%s enable bluedroid failed: %s", __func__, esp_err_to_name(err));
        return;
    }

#if (CONFIG_EXAMPLE_A2DP_SINK_SSP_ENABLED == true)
    /* set default parameters for Secure Simple Pairing */
    esp_bt_sp_param_t param_type = ESP_BT_SP_IOCAP_MODE;
    esp_bt_io_cap_t iocap = ESP_BT_IO_CAP_IO;
    esp_bt_gap_set_security_param(param_type, &iocap, sizeof(uint8_t));
#endif

    /* set default parameters for Legacy Pairing (use fixed pin code 1234) */
    esp_bt_pin_type_t pin_type = ESP_BT_PIN_TYPE_FIXED;
    esp_bt_pin_code_t pin_code;
    pin_code[0] = '1';
    pin_code[1] = '2';
    pin_code[2] = '3';
    pin_code[3] = '4';
    esp_bt_gap_set_pin(pin_type, 4, pin_code);

    ESP_LOGI(BT_AV_TAG, "Own address:[%s]", bda2str((uint8_t *)esp_bt_dev_get_address(), bda_str, sizeof(bda_str)));
    bt_app_task_start_up();
    /* bluetooth device name, connection mode and profile set up */
    bt_app_work_dispatch(bt_av_hdl_stack_evt, BT_APP_EVT_STACK_UP, NULL, 0, NULL);
}