cdc_to_uart.hpp

#pragma once
 
#include "cdc_uart.hpp"
#include "libxr_cb.hpp"
#include "libxr_rw.hpp"
#include "uart.hpp"
 
namespace LibXR::USB
{
 
class CDCToUart : public CDCUart
{
 public:
  CDCToUart(LibXR::UART& uart, size_t rx_buffer_size = 128, size_t tx_buffer_size = 128,
            size_t tx_queue_size = 5,
            Endpoint::EPNumber data_in_ep_num = Endpoint::EPNumber::EP_AUTO,
            Endpoint::EPNumber data_out_ep_num = Endpoint::EPNumber::EP_AUTO,
            Endpoint::EPNumber comm_ep_num = Endpoint::EPNumber::EP_AUTO)
      : CDCUart(rx_buffer_size, tx_buffer_size, tx_queue_size, data_in_ep_num,
                data_out_ep_num, comm_ep_num),
        rx_buffer_(new uint8_t[rx_buffer_size], rx_buffer_size),
        tx_buffer_(new uint8_t[tx_buffer_size], tx_buffer_size),
        uart_(uart)
  {
    // CDC->UART：CDC RX 缓冲必须能写入 UART 的 data 队列。
    // CDC->UART: the UART TX data queue must accept at least rx_buffer_size bytes.
    ASSERT(uart_.write_port_->queue_data_->MaxSize() >= rx_buffer_size);
 
    // 1) CDC 读完成回调：从 CDC 读一段数据并写入 UART。
    // 1) CDC read callback: read one chunk from CDC and write it into UART.
    // Guarded to flatten the bridge pump chain.
    cb_read_cdc_ = Callback<ErrorCode>::CreateGuarded(
        [](bool in_isr, CDCToUart* cdc_to_uart, ErrorCode)
        {
          auto size =
              LibXR::min(cdc_to_uart->read_port_->Size(), cdc_to_uart->rx_buffer_.size_);
 
          static ReadOperation op_read_cdc_noblock;
          auto ans = cdc_to_uart->Read({cdc_to_uart->rx_buffer_.addr_, size},
                                       op_read_cdc_noblock, in_isr);
          ASSERT(ans == ErrorCode::OK);
 
          ans = cdc_to_uart->uart_.Write({cdc_to_uart->rx_buffer_.addr_, size},
                                         cdc_to_uart->op_write_uart_, in_isr);
          ASSERT(ans == ErrorCode::OK);
        },
        this);
 
    op_read_cdc_ = ReadOperation(cb_read_cdc_);
 
    // 2) UART 写完成回调：继续触发下一次 CDC 读。
    // 2) UART write-complete callback: trigger the next CDC read.
    // Guarded to flatten the bridge pump chain.
    cb_uart_write_ = Callback<ErrorCode>::CreateGuarded(
        [](bool in_isr, CDCToUart* cdc_to_uart, ErrorCode)
        {
          auto ans = cdc_to_uart->Read({nullptr, 0}, cdc_to_uart->op_read_cdc_, in_isr);
          ASSERT(ans == ErrorCode::OK);
        },
        this);
 
    op_write_uart_ = WriteOperation(cb_uart_write_);
 
    // 3) UART 读完成回调：从 UART 读一段数据并写入 CDC。
    // 3) UART read callback: read one chunk from UART and write it into CDC.
    // Guarded to flatten the bridge pump chain.
    cb_uart_read_ = Callback<ErrorCode>::CreateGuarded(
        [](bool in_isr, CDCToUart* cdc_to_uart, ErrorCode)
        {
          auto size = LibXR::min(cdc_to_uart->uart_.read_port_->Size(),
                                 cdc_to_uart->tx_buffer_.size_);
          static ReadOperation op_read_uart_noblock;
 
          auto ans = cdc_to_uart->uart_.Read({cdc_to_uart->tx_buffer_.addr_, size},
                                             op_read_uart_noblock, in_isr);
          ASSERT(ans == ErrorCode::OK);
 
          ans = cdc_to_uart->Write({cdc_to_uart->tx_buffer_.addr_, size},
                                   cdc_to_uart->op_write_cdc_, in_isr);
          ASSERT(ans == ErrorCode::OK);
        },
        this);
 
    op_read_uart_ = ReadOperation(cb_uart_read_);
 
    // 4) CDC 写完成回调：继续触发下一次 UART 读。
    // 4) CDC write-complete callback: trigger the next UART read.
    // Guarded to flatten the bridge pump chain.
    cb_cdc_write_ = Callback<ErrorCode>::CreateGuarded(
        [](bool in_isr, CDCToUart* cdc_to_uart, ErrorCode)
        {
          auto ans_uart_read =
              cdc_to_uart->uart_.Read({nullptr, 0}, cdc_to_uart->op_read_uart_, in_isr);
          ASSERT(ans_uart_read == ErrorCode::OK);
        },
        this);
 
    op_write_cdc_ = WriteOperation(cb_cdc_write_);
 
    // 5) LineCoding 回调：USB CDC 侧配置变化同步到 UART。
    // 5) LineCoding callback: forward CDC line-coding changes to UART configuration.
    set_line_coding_cb_ = LibXR::Callback<LibXR::UART::Configuration>::Create(
        [](bool, CDCToUart* self, LibXR::UART::Configuration cfg)
        { self->uart_.SetConfig(cfg); }, this);
 
    SetOnSetLineCodingCallback(set_line_coding_cb_);
 
    // 启动搬运：挂起一次 CDC 读与 UART 读，以进入回调链。
    // Kick the pump: schedule one CDC read and one UART read to enter the callback chain.
    this->Read({nullptr, 0}, op_read_cdc_, false);
    uart_.Read({nullptr, 0}, op_read_uart_, false);
  }
 
  RawData rx_buffer_;  
  RawData tx_buffer_;  
 
  LibXR::Callback<ErrorCode> cb_read_cdc_;  
  LibXR::Callback<ErrorCode>
      cb_uart_write_;  
  LibXR::Callback<ErrorCode> cb_uart_read_;  
  LibXR::Callback<ErrorCode>
      cb_cdc_write_;  
 
  LibXR::Callback<LibXR::UART::Configuration>
      set_line_coding_cb_;  
 
  LibXR::WriteOperation op_write_cdc_;   
  LibXR::WriteOperation op_write_uart_;  
  LibXR::ReadOperation op_read_cdc_;     
  LibXR::ReadOperation op_read_uart_;    
 
  LibXR::UART& uart_;  
};
 
}  // namespace LibXR::USB