/*
 * SPDX-FileCopyrightText: 2015-2025 Espressif Systems (Shanghai) CO LTD
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef __STATS__H
#define __STATS__H

#include "port_esp_hosted_host_config.h"

#ifdef __cplusplus
extern "C" {
#endif

/* Stats CONFIG:
 *
 * 1. TEST_RAW_TP
 *    These are debug stats which show the raw throughput
 *    performance of transport like SPI or SDIO
 *    (a) TEST_RAW_TP__ESP_TO_HOST
 *    When this enabled, throughput will be measured from ESP to Host
 *
 *    (b) TEST_RAW_TP__HOST_TO_ESP
 *    This is opposite of TEST_RAW_TP__ESP_TO_HOST. when (a) TEST_RAW_TP__ESP_TO_HOST
 *    is disabled, it will automatically mean throughput to be measured from host to ESP
 */
#define TEST_RAW_TP                    H_TEST_RAW_TP

/* TEST_RAW_TP is disabled on production.
 * This is only to test the throughout over transport
 * like SPI or SDIO. In this testing, dummy task will
 * push the packets over transport.
 * Currently this testing is possible on one direction
 * at a time
 */

#if TEST_RAW_TP

#define TEST_RAW_TP__TIMEOUT         H_RAW_TP_REPORT_INTERVAL

void update_test_raw_tp_rx_len(uint16_t len);
void process_test_capabilities(uint8_t cap);

/* Please note, this size is to assess transport speed,
 * so kept maximum possible for that transport
 *
 * If you want to compare maximum network throughput and
 * relevance with max transport speed, Plz lower this value to
 * UDP: 1460 - H_ESP_PAYLOAD_HEADER_OFFSET = 1460-12=1448
 * TCP: Find MSS in nodes
 * H_ESP_PAYLOAD_HEADER_OFFSET is header size, which is not included in calcs
 */
#define TEST_RAW_TP__BUF_SIZE    H_RAW_TP_PKT_LEN


#endif

#if H_MEM_STATS
struct mempool_stats
{
	uint32_t num_fresh_alloc;
	uint32_t num_reuse;
	uint32_t num_free;
};

struct spi_stats
{
	int rx_alloc;
	int rx_freed;
	int tx_alloc;
	int tx_dummy_alloc;
	int tx_freed;
};

struct nw_stats
{
	int tx_alloc;
	int tx_freed;
};

struct others_stats {
	int tx_others_freed;
};

struct mem_stats {
	struct mempool_stats mp_stats;
	struct spi_stats spi_mem_stats;
	struct nw_stats nw_mem_stats;
	struct others_stats others;
};

extern struct mem_stats h_stats_g;
#endif /*H_MEM_STATS*/

#ifdef ESP_PKT_NUM_DEBUG
struct dbg_stats_t {
	uint16_t tx_pkt_num;
	uint16_t exp_rx_pkt_num;
};

extern struct dbg_stats_t dbg_stats;
#define UPDATE_HEADER_TX_PKT_NO(h) h->pkt_num = htole16(dbg_stats.tx_pkt_num++)
#define UPDATE_HEADER_RX_PKT_NO(h)                              \
	do {                                                        \
		uint16_t rcvd_pkt_num = le16toh(h->pkt_num);            \
		if (dbg_stats.exp_rx_pkt_num != rcvd_pkt_num) {         \
			ESP_LOGI(TAG, "exp_pkt_num[%u], rx_pkt_num[%u]",    \
					dbg_stats.exp_rx_pkt_num, rcvd_pkt_num);    \
			dbg_stats.exp_rx_pkt_num = rcvd_pkt_num;            \
		}                                                       \
		dbg_stats.exp_rx_pkt_num++;                             \
	} while(0);

#else /*ESP_PKT_NUM_DEBUG*/

  #define UPDATE_HEADER_TX_PKT_NO(h)
  #define UPDATE_HEADER_RX_PKT_NO(h)

#endif /*ESP_PKT_NUM_DEBUG*/

#if ESP_PKT_STATS
struct pkt_stats_t {
	uint32_t sta_rx_in;
	uint32_t sta_rx_out;
	uint32_t sta_tx_in_pass;
	uint32_t sta_tx_trans_in;
	uint32_t sta_tx_flowctrl_drop;
	uint32_t sta_tx_out;
	uint32_t sta_tx_out_drop;
	uint32_t sta_flow_ctrl_on;
	uint32_t sta_flow_ctrl_off;
};

extern struct pkt_stats_t pkt_stats;
#endif /*ESP_PKT_STATS*/

#ifdef __cplusplus
}
#endif

void create_debugging_tasks(void);

#endif