#include "ir_control.h"

void initIRControl() {
    // 配置RMT发送通道
    rmt_config_t rmt_tx_config = RMT_DEFAULT_CONFIG_TX((gpio_num_t)IR_SEND_PIN, RMT_TX_CHANNEL);
    rmt_tx_config.clk_div = RMT_CLK_DIV;
    rmt_tx_config.mem_block_num = RMT_MEM_BLOCK_NUM;
    rmt_tx_config.tx_config.carrier_en = RMT_TX_CARRIER_EN;
    rmt_tx_config.tx_config.carrier_freq_hz = RMT_TX_CARRIER_FREQ_HZ;
    rmt_tx_config.tx_config.carrier_duty_percent = RMT_TX_CARRIER_DUTY_PERCENT;
    rmt_tx_config.tx_config.carrier_level = RMT_TX_CARRIER_LEVEL;
    rmt_tx_config.tx_config.idle_level = RMT_IDLE_LEVEL;
    rmt_tx_config.tx_config.idle_output_en = true;
    
    esp_err_t err = rmt_config(&rmt_tx_config);
    if (err != ESP_OK) {
        Serial.printf("RMT发送通道配置失败: %s\n", esp_err_to_name(err));
        return;
    }
    
    err = rmt_driver_install(RMT_TX_CHANNEL, 0, 0);
    if (err != ESP_OK) {
        Serial.printf("RMT发送驱动安装失败: %s\n", esp_err_to_name(err));
        return;
    }
    
    // 配置RMT接收通道
    rmt_config_t rmt_rx_config = RMT_DEFAULT_CONFIG_RX((gpio_num_t)IR_RECEIVE_PIN, RMT_RX_CHANNEL);
    rmt_rx_config.clk_div = RMT_CLK_DIV;
    rmt_rx_config.mem_block_num = RMT_MEM_BLOCK_NUM;
    rmt_rx_config.rx_config.filter_en = RMT_RX_FILTER_EN;
    rmt_rx_config.rx_config.filter_ticks_thresh = RMT_RX_FILTER_THRESH_US;
    rmt_rx_config.rx_config.idle_threshold = RMT_RX_IDLE_THRESH_US;
    
    err = rmt_config(&rmt_rx_config);
    if (err != ESP_OK) {
        Serial.printf("RMT接收通道配置失败: %s\n", esp_err_to_name(err));
        return;
    }
    
    err = rmt_driver_install(RMT_RX_CHANNEL, 1000, 0);
    if (err != ESP_OK) {
        Serial.printf("RMT接收驱动安装失败: %s\n", esp_err_to_name(err));
        return;
    }
    
    Serial.println("红外控制模块初始化完成 (使用RMT)");
    Serial.printf("发送引脚: %d (RMT通道: %d)\n", IR_SEND_PIN, RMT_TX_CHANNEL);
    Serial.printf("接收引脚: %d (RMT通道: %d)\n", IR_RECEIVE_PIN, RMT_RX_CHANNEL);
    Serial.printf("载波频率: %d Hz\n", RMT_TX_CARRIER_FREQ_HZ);
    Serial.printf("时钟分辨率: %d us\n", RMT_CLK_DIV);
}

bool checkIRSignalStart() {
    // 使用RMT检查是否有信号开始
    // 启动接收并检查是否立即有数据
    rmt_rx_start(RMT_RX_CHANNEL, true);
    
    // 等待短时间检查是否有信号
    vTaskDelay(pdMS_TO_TICKS(1));
    
    // 获取RMT环形缓冲区句柄
    RingbufHandle_t rb = NULL;
    esp_err_t err = rmt_get_ringbuf_handle(RMT_RX_CHANNEL, &rb);
    if (err != ESP_OK) {
        rmt_rx_stop(RMT_RX_CHANNEL);
        return false;
    }
    
    // 检查RMT接收缓冲区是否有数据
    size_t rx_size = 0;
    rmt_item32_t* items = (rmt_item32_t*)xRingbufferReceive(rb, &rx_size, 0);
    
    bool hasSignal = (items != nullptr && rx_size > 0);
    
    if (items) {
        vRingbufferReturnItem(rb, (void*)items);
    }
    
    rmt_rx_stop(RMT_RX_CHANNEL);
    
    return hasSignal;
}

IRSignal receiveIRSignal() {
    IRSignal signal;
    signal.markTimes = nullptr;
    signal.spaceTimes = nullptr;
    signal.markCount = 0;
    signal.spaceCount = 0;
    signal.carrierFreq = IR_CARRIER_FREQ;
    signal.isValid = false;
    
    Serial.println("开始接收红外信号 (使用RMT)...");
    
    // 启动RMT接收
    rmt_rx_start(RMT_RX_CHANNEL, true);
    
    // 获取RMT环形缓冲区句柄
    RingbufHandle_t rb = NULL;
    esp_err_t err = rmt_get_ringbuf_handle(RMT_RX_CHANNEL, &rb);
    if (err != ESP_OK) {
        Serial.printf("获取RMT缓冲区句柄失败: %s\n", esp_err_to_name(err));
        rmt_rx_stop(RMT_RX_CHANNEL);
        return signal;
    }
    
    // 等待接收数据
    size_t rx_size = 0;
    rmt_item32_t* items = (rmt_item32_t*)xRingbufferReceive(rb, &rx_size, pdMS_TO_TICKS(RECEIVE_TIMEOUT_US / 1000));
    
    if (items == nullptr || rx_size == 0) {
        Serial.println("RMT接收超时或无数据");
        rmt_rx_stop(RMT_RX_CHANNEL);
        return signal;
    }
    
    // 计算接收到的项目数量
    size_t item_count = rx_size / sizeof(rmt_item32_t);
    Serial.printf("接收到 %d 个RMT项目\n", item_count);
    
    if (item_count == 0) {
        Serial.println("接收到的数据为空");
        vRingbufferReturnItem(rb, (void*)items);
        rmt_rx_stop(RMT_RX_CHANNEL);
        return signal;
    }
    
    // 分配临时数组存储时序数据
    unsigned int* tempData = (unsigned int*)malloc(item_count * 2 * sizeof(unsigned int));
    if (tempData == nullptr) {
        Serial.println("临时数据内存分配失败");
        vRingbufferReturnItem(rb, (void*)items);
        rmt_rx_stop(RMT_RX_CHANNEL);
        return signal;
    }
    
    int tempLength = 0;
    
    // 解析RMT数据项
    for (size_t i = 0; i < item_count; i++) {
        // 每个RMT项目包含两个时间段：level0的持续时间和level1的持续时间
        if (items[i].duration0 > 0 && tempLength < item_count * 2) {
            tempData[tempLength++] = items[i].duration0;
        }
        if (items[i].duration1 > 0 && tempLength < item_count * 2) {
            tempData[tempLength++] = items[i].duration1;
        }
    }
    
    // 释放RMT缓冲区
    vRingbufferReturnItem(rb, (void*)items);
    rmt_rx_stop(RMT_RX_CHANNEL);
    
    if (tempLength > 0) {
        // 分离mark和space时间
        signal.markCount = (tempLength + 1) / 2;  // 奇数索引是mark
        signal.spaceCount = tempLength / 2;       // 偶数索引是space
        
        // 分配mark时间数组
        if (signal.markCount > 0) {
            signal.markTimes = (unsigned int*)malloc(signal.markCount * sizeof(unsigned int));
            if (signal.markTimes == nullptr) {
                Serial.println("mark时间数组内存分配失败");
                free(tempData);
                return signal;
            }
            
            // 复制mark时间（从索引0开始，每隔2个取一个）
            for (int i = 0; i < signal.markCount; i++) {
                signal.markTimes[i] = tempData[i * 2];
            }
        }
        
        // 分配space时间数组
        if (signal.spaceCount > 0) {
            signal.spaceTimes = (unsigned int*)malloc(signal.spaceCount * sizeof(unsigned int));
            if (signal.spaceTimes == nullptr) {
                Serial.println("space时间数组内存分配失败");
                if (signal.markTimes != nullptr) {
                    free(signal.markTimes);
                    signal.markTimes = nullptr;
                }
                free(tempData);
                return signal;
            }
            
            // 复制space时间（从索引1开始，每隔2个取一个）
            for (int i = 0; i < signal.spaceCount; i++) {
                signal.spaceTimes[i] = tempData[i * 2 + 1];
            }
        }
        
        signal.isValid = true;
        Serial.printf("RMT信号接收成功，mark数量: %d, space数量: %d\n", signal.markCount, signal.spaceCount);
    } else {
        Serial.println("未接收到有效的RMT信号数据");
    }
    
    free(tempData);
    return signal;
}

// generateCarrier函数已被RMT模块替代，不再需要

bool sendIRSignal(const IRSignal& signal) {
    if (!isValidIRSignal(signal)) {
        Serial.println("信号无效，无法发送");
        return false;
    }
    
    Serial.println("开始发送红外信号 (使用RMT)...");
    Serial.printf("载波频率: %d Hz\n", signal.carrierFreq);
    Serial.printf("mark数量: %d, space数量: %d\n", signal.markCount, signal.spaceCount);
    
    // 计算需要的RMT项目数量
    int maxCount = max(signal.markCount, signal.spaceCount);
    size_t item_count = maxCount;
    
    // 分配RMT项目数组
    rmt_item32_t* items = (rmt_item32_t*)malloc(item_count * sizeof(rmt_item32_t));
    if (items == nullptr) {
        Serial.println("RMT项目内存分配失败");
        return false;
    }
    
    // 构建RMT数据项
    for (int i = 0; i < maxCount; i++) {
        items[i].level0 = 1;  // mark期间为高电平（载波调制）
        items[i].duration0 = (i < signal.markCount) ? signal.markTimes[i] : 0;
        
        items[i].level1 = 0;  // space期间为低电平（无载波）
        items[i].duration1 = (i < signal.spaceCount) ? signal.spaceTimes[i] : 0;
    }
    
    // 发送RMT数据
    esp_err_t err = rmt_write_items(RMT_TX_CHANNEL, items, item_count, true);
    
    if (err != ESP_OK) {
        Serial.printf("RMT发送失败: %s\n", esp_err_to_name(err));
        free(items);
        return false;
    }
    
    // 等待发送完成
    err = rmt_wait_tx_done(RMT_TX_CHANNEL, pdMS_TO_TICKS(1000));
    
    if (err != ESP_OK) {
        Serial.printf("RMT发送等待超时: %s\n", esp_err_to_name(err));
        free(items);
        return false;
    }
    
    free(items);
    Serial.println("RMT信号发送完成");
    return true;
}

void freeIRSignal(IRSignal& signal) {
    if (signal.markTimes != nullptr) {
        free(signal.markTimes);
        signal.markTimes = nullptr;
    }
    if (signal.spaceTimes != nullptr) {
        free(signal.spaceTimes);
        signal.spaceTimes = nullptr;
    }
    signal.markCount = 0;
    signal.spaceCount = 0;
    signal.carrierFreq = 0;
    signal.isValid = false;
}

IRSignal copyIRSignal(const IRSignal& source) {
    IRSignal copy;
    copy.markTimes = nullptr;
    copy.spaceTimes = nullptr;
    copy.markCount = 0;
    copy.spaceCount = 0;
    copy.carrierFreq = 0;
    copy.isValid = false;
    
    if (!isValidIRSignal(source)) {
        return copy;
    }
    
    // 复制mark时间数组
    if (source.markCount > 0 && source.markTimes != nullptr) {
        copy.markTimes = (unsigned int*)malloc(source.markCount * sizeof(unsigned int));
        if (copy.markTimes == nullptr) {
            Serial.println("复制mark时间数组时内存分配失败");
            return copy;
        }
        
        for (int i = 0; i < source.markCount; i++) {
            copy.markTimes[i] = source.markTimes[i];
        }
        copy.markCount = source.markCount;
    }
    
    // 复制space时间数组
    if (source.spaceCount > 0 && source.spaceTimes != nullptr) {
        copy.spaceTimes = (unsigned int*)malloc(source.spaceCount * sizeof(unsigned int));
        if (copy.spaceTimes == nullptr) {
            Serial.println("复制space时间数组时内存分配失败");
            if (copy.markTimes != nullptr) {
                free(copy.markTimes);
                copy.markTimes = nullptr;
                copy.markCount = 0;
            }
            return copy;
        }
        
        for (int i = 0; i < source.spaceCount; i++) {
            copy.spaceTimes[i] = source.spaceTimes[i];
        }
        copy.spaceCount = source.spaceCount;
    }
    
    copy.carrierFreq = source.carrierFreq;
    copy.isValid = true;
    
    return copy;
}

void printIRSignal(const IRSignal& signal, int maxPrint) {
    if (!isValidIRSignal(signal)) {
        Serial.println("信号无效，无法打印");
        return;
    }
    
    Serial.println("=== 红外信号数据 ===");
    Serial.print("载波频率: ");
    Serial.print(signal.carrierFreq);
    Serial.println(" Hz");
    Serial.print("mark数量: ");
    Serial.println(signal.markCount);
    Serial.print("space数量: ");
    Serial.println(signal.spaceCount);
    Serial.print("信号有效: ");
    Serial.println(signal.isValid ? "是" : "否");
    
    // 打印mark时间
    int markPrintCount = (maxPrint == 0) ? signal.markCount : min(maxPrint, signal.markCount);
    Serial.println("Mark时间 (微秒):");
    for (int i = 0; i < markPrintCount; i++) {
        Serial.print("Mark ");
        Serial.print(i);
        Serial.print(": ");
        Serial.print(signal.markTimes[i]);
        Serial.println(" us");
    }
    
    if (markPrintCount < signal.markCount) {
        Serial.print("... 还有 ");
        Serial.print(signal.markCount - markPrintCount);
        Serial.println(" 个mark数据点未显示");
    }
    
    // 打印space时间
    int spacePrintCount = (maxPrint == 0) ? signal.spaceCount : min(maxPrint, signal.spaceCount);
    Serial.println("Space时间 (微秒):");
    for (int i = 0; i < spacePrintCount; i++) {
        Serial.print("Space ");
        Serial.print(i);
        Serial.print(": ");
        Serial.print(signal.spaceTimes[i]);
        Serial.println(" us");
    }
    
    if (spacePrintCount < signal.spaceCount) {
        Serial.print("... 还有 ");
        Serial.print(signal.spaceCount - spacePrintCount);
        Serial.println(" 个space数据点未显示");
    }
    
    Serial.println("==================");
}

bool isValidIRSignal(const IRSignal& signal) {
    return (signal.isValid && 
            ((signal.markTimes != nullptr && signal.markCount > 0) || 
             (signal.spaceTimes != nullptr && signal.spaceCount > 0)) &&
            signal.carrierFreq > 0);
}