ASoC: SOF: ipc4-topology: reset pipeline mem_usage during unprepare

The mem_usage for a pipeline is updated when the widgets in the
pipeline are prepare. So, reset it when a widget in the pipeline is
unprepared. The ipc_unprepare for the pipeline widget might be called
multiple times when each widget associated with the pipeline is
unprepared and it is harmless.

Signed-off-by: Ranjani Sridharan <ranjani.sridharan@linux.intel.com>

Author: ranj063

sound/soc/sof/ipc4-topology.c

@@ -1501,6 +1501,14 @@ static int sof_ipc4_init_input_audio_fmt(struct snd_sof_dev *sdev,
 	return i;
 }
 
+static void sof_ipc4_unprepare_pipeline_module(struct snd_sof_widget *swidget)
+{
+	struct sof_ipc4_pipeline *pipeline = swidget->private;
+
+	/* reset pipeline memory usage */
+	pipeline->mem_usage = 0;
+}
+
 static void sof_ipc4_unprepare_copier_module(struct snd_sof_widget *swidget)
 {
 	struct sof_ipc4_copier *ipc4_copier = NULL;
@@ -3011,7 +3019,6 @@ static int sof_ipc4_widget_free(struct snd_sof_dev *sdev, struct snd_sof_widget
 			dev_err(sdev->dev, "failed to free pipeline widget %s\n",
 				swidget->widget->name);
 
-		pipeline->mem_usage = 0;
 		pipeline->state = SOF_IPC4_PIPE_UNINITIALIZED;
 		ida_free(&pipeline_ida, swidget->instance_id);
 		swidget->instance_id = -EINVAL;
@@ -3637,7 +3644,7 @@ static const struct sof_ipc_tplg_widget_ops tplg_ipc4_widget_ops[SND_SOC_DAPM_TY
 	[snd_soc_dapm_scheduler] = {sof_ipc4_widget_setup_comp_pipeline,
 				    sof_ipc4_widget_free_comp_pipeline,
 				    pipeline_token_list, ARRAY_SIZE(pipeline_token_list), NULL,
-				    NULL, NULL},
+				    NULL, sof_ipc4_unprepare_pipeline_module},
 	[snd_soc_dapm_pga] = {sof_ipc4_widget_setup_comp_pga, sof_ipc4_widget_free_comp_pga,
 			      pga_token_list, ARRAY_SIZE(pga_token_list), NULL,
 			      sof_ipc4_prepare_gain_module,

sound/soc/sof/sof-audio.c

@@ -447,10 +447,16 @@ sof_unprepare_widgets_in_path(struct snd_sof_dev *sdev, struct snd_soc_dapm_widg
 		goto sink_unprepare;
 
 	widget_ops = tplg_ops ? tplg_ops->widget : NULL;
-	if (widget_ops && widget_ops[widget->id].ipc_unprepare)
+	if (widget_ops && widget_ops[widget->id].ipc_unprepare) {
+		struct snd_sof_widget *pipe_widget = swidget->spipe->pipe_widget;
+
 		/* unprepare the source widget */
 		widget_ops[widget->id].ipc_unprepare(swidget);
 
+		/* unprepare the widget's pipeline widget */
+		widget_ops[pipe_widget->id].ipc_unprepare(pipe_widget);
+	}
+
 	swidget->prepared = false;
 
 sink_unprepare:
@@ -1085,10 +1091,16 @@ void snd_sof_unprepare_widgets_in_pipeline(struct snd_soc_dapm_widget *w,
 		goto unprepare;
 
 	widget_ops = tplg_ops ? tplg_ops->widget : NULL;
-	if (widget_ops && widget_ops[w->id].ipc_unprepare)
+	if (widget_ops && widget_ops[w->id].ipc_unprepare) {
+		struct snd_sof_widget *pipe_widget = swidget->spipe->pipe_widget;
+
 		/* unprepare the source widget */
 		widget_ops[w->id].ipc_unprepare(swidget);
 
+		/* unprepare the widget's pipeline widget */
+		widget_ops[pipe_widget->id].ipc_unprepare(pipe_widget);
+	}
+
 	swidget->prepared = false;
 
 unprepare: