From 02c18c23d7eb8e520e8584af183ca9b67937d45d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marc=20Andr=C3=A9=20Tanner?= <mat@brain-dump.org>
Date: Wed, 3 May 2017 17:21:46 +0200
Subject: vis: introduce count iterator to handle interrupted flag

---
 vis.c | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'vis.c')

diff --git a/vis.c b/vis.c
index be12efb..bdee2e7 100644
--- a/vis.c
+++ b/vis.c
@@ -1545,6 +1545,28 @@ void vis_count_set(Vis *vis, int count) {
 	vis->action.count = (count >= 0 ? count : VIS_COUNT_UNKNOWN);
 }
 
+VisCountIterator vis_count_iterator_get(Vis *vis, int def) {
+	return (VisCountIterator) {
+		.vis = vis,
+		.iteration = 0,
+		.count = vis_count_get_default(vis, def),
+	};
+}
+
+VisCountIterator vis_count_iterator_init(Vis *vis, int count) {
+	return (VisCountIterator) {
+		.vis = vis,
+		.iteration = 0,
+		.count = count,
+	};
+}
+
+bool vis_count_iterator_next(VisCountIterator *it) {
+	if (it->vis->interrupted)
+		return false;
+	return it->iteration++ < it->count;
+}
+
 void vis_exit(Vis *vis, int status) {
 	vis->running = false;
 	vis->exit_status = status;
-- 
cgit v1.2.3