Tools: Testbench: Track and print heap usage for modules

This patch adds to end to sof-testbench4 run print of peak
heap consumption for each module. The information is retrieved
from heap_high_water_mark data that is tracked by module adapter
for each module.

Signed-off-by: Seppo Ingalsuo <seppo.ingalsuo@linux.intel.com>

Author: singalsu

tools/testbench/include/testbench/utils.h

@@ -148,6 +148,11 @@ struct testbench_prm {
 #endif
 };
 
+struct tb_heap_usage_record {
+	char *module_name;
+	int heap_max;
+};
+
 extern int debug;
 
 int tb_decode_enum(struct snd_soc_tplg_enum_control *enum_ctl, char *token);

tools/testbench/testbench.c

@@ -240,7 +240,48 @@ static int parse_input_args(int argc, char **argv, struct testbench_prm *tp)
 	return ret;
 }
 
-static void test_pipeline_stats(struct testbench_prm *tp, long long delta_t)
+#if CONFIG_IPC_MAJOR_4
+static int tb_collect_heap_usage(struct testbench_prm *tp, struct tb_heap_usage_record *records,
+				 int *count_out)
+{
+	struct list_item *item;
+	int count = 0;
+
+	list_for_item(item, &tp->widget_list)
+	{
+		struct tplg_comp_info *info = container_of(item, struct tplg_comp_info, item);
+		uint32_t comp_id = IPC4_COMP_ID(info->module_id, info->instance_id);
+		struct comp_dev *dev = ipc4_get_comp_dev(comp_id);
+
+		if (!dev || !dev->mod)
+			continue;
+
+		/* In testbench environment, skip AIF/DAI because they are not real components. */
+		if (info->type == SND_SOC_TPLG_DAPM_AIF_IN ||
+		    info->type == SND_SOC_TPLG_DAPM_AIF_OUT ||
+		    info->type == SND_SOC_TPLG_DAPM_DAI_IN ||
+		    info->type == SND_SOC_TPLG_DAPM_DAI_OUT)
+			continue;
+
+		if (count >= TB_NUM_WIDGETS_SUPPORTED) {
+			fprintf(stderr, "Error: Too many components for heap records, max %d.\n",
+				TB_NUM_WIDGETS_SUPPORTED);
+			break;
+		}
+
+		records[count].module_name = info->name;
+		records[count].heap_max = (int)dev->mod->priv.resources.heap_high_water_mark;
+		count++;
+	}
+
+	*count_out = count;
+	return 0;
+}
+#endif
+
+static void test_pipeline_stats(struct testbench_prm *tp, long long delta_t,
+				struct tb_heap_usage_record *heap_records,
+				int heap_records_count)
 {
 	long long file_cycles, pipeline_cycles;
 	float pipeline_mcps;
@@ -295,11 +336,15 @@ static void test_pipeline_stats(struct testbench_prm *tp, long long delta_t)
 			printf("Warning: Use -d 3 or smaller value to avoid traces to increase MCPS.\n");
 	}
 
-	if (delta_t)
-		printf("Total execution time: %lld us, %.2f x realtime\n",
-		       delta_t, (float)frames_out / tp->fs_out * 1000000 / delta_t);
+	if (heap_records_count > 0) {
+		for (i = 0; i < heap_records_count; i++)
+			printf("Heap usage for module %s: %d bytes\n",
+			       heap_records[i].module_name, heap_records[i].heap_max);
+	}
 
-	printf("\n");
+	if (delta_t)
+		printf("Total execution time: %lld us, %.2f x realtime\n\n", delta_t,
+		       (float)frames_out / tp->fs_out * 1000000 / delta_t);
 }
 
 /*
@@ -308,14 +353,16 @@ static void test_pipeline_stats(struct testbench_prm *tp, long long delta_t)
  */
 static int pipline_test(struct testbench_prm *tp)
 {
-	float samples_to_ns;
-	int dp_count = 0;
-	struct timespec td0, td1;
+	struct tb_heap_usage_record heap_usage_records[TB_NUM_WIDGETS_SUPPORTED];
 	struct file_state *out_stat;
+	struct timespec td0, td1;
 	long long delta_t;
 	int64_t next_control_ns;
 	int64_t time_ns;
+	float samples_to_ns;
 	int err;
+	int heap_usage_records_count = 0;
+	int dp_count = 0;
 
 	/* build, run and teardown pipelines */
 	while (dp_count < tp->dynamic_pipeline_iterations) {
@@ -396,6 +443,9 @@ static int pipline_test(struct testbench_prm *tp)
 		tb_gettime(&td1);
 
 out:
+#if CONFIG_IPC_MAJOR_4
+		tb_collect_heap_usage(tp, heap_usage_records, &heap_usage_records_count);
+#endif
 		err = tb_set_reset_state(tp);
 		if (err < 0) {
 			fprintf(stderr, "error: pipeline reset %d failed %d\n",
@@ -408,7 +458,7 @@ static int pipline_test(struct testbench_prm *tp)
 		 */
 		delta_t = (td1.tv_sec - td0.tv_sec) * 1000000;
 		delta_t += (td1.tv_nsec - td0.tv_nsec) / 1000;
-		test_pipeline_stats(tp, delta_t);
+		test_pipeline_stats(tp, delta_t, heap_usage_records, heap_usage_records_count);
 
 		err = tb_free_all_pipelines(tp);
 		if (err < 0) {