testbench: file: Add support for stopping testbench on file size/copies

lrgirdwo · lgirdwood · commit 7ee22be7009b · 2022-01-31T16:16:27.000Z
Allow the testbench to stop on a file size or number of file(pipeline)
copies.

Signed-off-by: Liam Girdwood &lt;liam.r.girdwood@linux.intel.com&gt;
diff --git a/tools/testbench/file.c b/tools/testbench/file.c
@@ -443,7 +443,11 @@ static int file_s32(struct comp_dev *dev, struct audio_stream *sink,
 		return -EINVAL;
 	}
 
+	/* update sample counter and check if we have a sample limit */
 	cd->fs.n += n_samples;
+	if (cd->max_samples && cd->fs.n >= cd->max_samples)
+		cd->fs.reached_eof = 1;
+
 	return n_samples;
 }
 
@@ -473,7 +477,11 @@ static int file_s16(struct comp_dev *dev, struct audio_stream *sink,
 		return -EINVAL;
 	}
 
+	/* update sample counter and check if we have a sample limit */
 	cd->fs.n += n_samples;
+	if (cd->max_samples && cd->fs.n >= cd->max_samples)
+		cd->fs.reached_eof = 1;
+
 	return n_samples;
 }
 
@@ -503,7 +511,11 @@ static int file_s24(struct comp_dev *dev, struct audio_stream *sink,
 		return -EINVAL;
 	}
 
+	/* update sample counter and check if we have a sample limit */
 	cd->fs.n += n_samples;
+	if (cd->max_samples && cd->fs.n >= cd->max_samples)
+		cd->fs.reached_eof = 1;
+
 	return n_samples;
 }
 
@@ -603,6 +615,7 @@ static struct comp_dev *file_new(const struct comp_driver *drv,
 	cd->fs.reached_eof = false;
 	cd->fs.write_failed = false;
 	cd->fs.n = 0;
+	cd->fs.copy_count = 0;
 	dev->state = COMP_STATE_READY;
 	return dev;
 
@@ -772,6 +785,14 @@ static int file_copy(struct comp_dev *dev)
 		break;
 	}
 
+	cd->fs.copy_count++;
+	if (cd->fs.reached_eof || (cd->max_copies && cd->fs.copy_count >= cd->max_copies)) {
+		cd->fs.reached_eof = 1;
+		comp_info(dev, "file_copy(): copies %d max %d eof %d",
+			  cd->fs.copy_count, cd->max_copies,
+			  cd->fs.reached_eof);
+		schedule_task_cancel(dev->pipeline->pipe_task);
+	}
 	return ret;
 }
 
diff --git a/tools/testbench/include/testbench/file.h b/tools/testbench/include/testbench/file.h
@@ -36,6 +36,7 @@ struct file_state {
 	int n;
 	enum file_mode mode;
 	enum file_format f_format;
+	int copy_count;
 };
 
 /* file comp data */
@@ -48,6 +49,9 @@ struct file_comp_data {
 	int (*file_func)(struct comp_dev *dev, struct audio_stream *sink,
 			 struct audio_stream *source, uint32_t frames);
 
+	/* maximum limits */
+	int max_samples;
+	int max_copies;
 };
 
 #endif
