mtb_ml_stream.h

/***************************************************************************/
#if !defined(__MTB_ML_STREAM_H__)
#define __MTB_ML_STREAM_H__

#include "mtb_ml_model.h"
#include "mtb_ml_dataset.h"


#if defined(__cplusplus)
extern "C" {
#endif

#if !defined(COMPONENT_MTB_HAL)
#include "cyhal.h"
#else
#include "mtb_hal.h"
#endif

/******************************************************************************
* Compile-time flags
*****************************************************************************/

/******************************************************************************
* Enums
*****************************************************************************/

/******************************************************************************
* Macros
*****************************************************************************/

/*******************************************************************************
* extern variables
******************************************************************************/

/*******************************************************************************
* Structures and enumerations
******************************************************************************/
typedef struct
{
    void* placeholder[3]; 
} mtb_data_streaming_vcontext_t;

typedef cy_rslt_t (* mtb_data_streaming_send_t)(mtb_data_streaming_vcontext_t* context,
                                                void *data, size_t count, void* tag);

typedef cy_rslt_t (* mtb_data_streaming_receive_t)(mtb_data_streaming_vcontext_t* context,
                                                   uint8_t* data, size_t count, void* tag);

typedef void (* mtb_data_streaming_xfer_done_t)(void* tag, cy_rslt_t rslt);

typedef struct
{
#if !defined(COMPONENT_MTB_HAL)
    cyhal_uart_t*                       uart; 
#else
    mtb_hal_uart_t*                     uart; 
#endif
} mtb_data_streaming_obj_t;

typedef struct
{
    mtb_data_streaming_obj_t        obj_inst; 
    mtb_data_streaming_xfer_done_t  callback; 
    void*                           call_tag; 
} mtb_data_streaming_context_t;


typedef struct
{
    mtb_data_streaming_send_t send;         
    mtb_data_streaming_receive_t receive;   
    mtb_data_streaming_vcontext_t context;  
} mtb_data_streaming_interface_t;


typedef struct
{
    bool stream_done;           
    cy_rslt_t stream_status;    
} mtb_ml_stream_tag_t;


typedef struct
{
    mtb_data_streaming_interface_t *interface_obj;  
    mtb_ml_stream_tag_t *stream_tag;                
    mtb_ml_x_file_header_t x_data_info;             
    size_t input_size;                              
    size_t output_size;                             
} mtb_ml_stream_interface_t;

/*******************************************************************************
* Function prototypes
******************************************************************************/
cy_rslt_t mtb_ml_stream_init(mtb_ml_stream_interface_t *interface,
                            const mtb_ml_model_t *model_object);
cy_rslt_t mtb_ml_stream_output_data(mtb_ml_stream_interface_t *interface,
                                    void *tx_buf,
                                    uint32_t timeout_ms);
cy_rslt_t mtb_ml_stream_input_data( mtb_ml_stream_interface_t *interface,
                                    void *rx_buf,
                                    uint32_t timeout_ms);
cy_rslt_t mtb_ml_inform_host_done( mtb_ml_stream_interface_t *interface, uint32_t timeout_ms);
void mtb_ml_stream_cb(void* tag, cy_rslt_t rslt);

#if defined(__cplusplus)
}
#endif

#endif /* __MTB_ML_STREAM_H__ */