Cleanup: Main id looping: add FOREACH_MAIN_LISTBASE macro.

We don't want to use flow control like `break` statement into the basic
`FOREACH_MAIN_ID` macro, as this is a nested loop.

When refined behavior is needed (like breaking whole iteration, or just
skipping to next ID type), FOREACH_MAIN_LISTBASE and
FOREACH_MAIN_LISTBASE_ID macros should be used instead.

Based on D4382 by @campbellbarton
(Other potential solution, using flow control macros: D4384).

3 files changed, 26 insertions, 26 deletions

diff --git a/source/blender/blenkernel/BKE_main.h b/source/blender/blenkernel/BKE_main.h
index 03f9f9a21ea..6b9c35d90d3 100644
--- a/source/blender/blenkernel/BKE_main.h
+++ b/source/blender/blenkernel/BKE_main.h
@@ -152,31 +152,29 @@ struct GSet *BKE_main_gset_create(struct Main *bmain, struct GSet *gset);
 		}                                                                 \
 	} ((void)0)
 
-
-#define FOREACH_MAIN_ID_BEGIN(_bmain, _id)                                \
+#define FOREACH_MAIN_LISTBASE_BEGIN(_bmain, _lb)                          \
 	{                                                                     \
 		ListBase *_lbarray[MAX_LIBARRAY];                                 \
 		int _i = set_listbasepointers(_bmain, _lbarray);                  \
 		while (_i--) {                                                    \
-			FOREACH_MAIN_LISTBASE_ID_BEGIN(_lbarray[_i], _id)
+			_lb = _lbarray[_i];
 
-#define FOREACH_MAIN_ID_END                                               \
-			FOREACH_MAIN_LISTBASE_ID_END;                                 \
+#define FOREACH_MAIN_LISTBASE_END                                         \
 		}                                                                 \
 	} ((void)0)
 
-/** \param _do_break A boolean, to allow breaking iteration (only used to break by type,
- *                   you must also use an explicit `break;` operation if you want to
- *                   immediately break from inner by-ID loop).
- */
-#define FOREACH_MAIN_ID_BREAKABLE_BEGIN(_bmain, _id, _do_break)           \
+/* DO NOT use break statement with that macro, use FOREACH_MAIN_LISTBASE and FOREACH_MAIN_LISTBASE_ID instead
+ * if you need that kind of control flow. */
+#define FOREACH_MAIN_ID_BEGIN(_bmain, _id)                                \
 	{                                                                     \
-		ListBase *_lbarray[MAX_LIBARRAY];                                 \
-		int i = set_listbasepointers(_bmain, _lbarray);                   \
-		while (i-- && !_do_break) {                                       \
-			FOREACH_MAIN_LISTBASE_ID_BEGIN(_lbarray[i], _id)              \
+		ListBase *_lb;                                                    \
+		FOREACH_MAIN_LISTBASE_BEGIN(_bmain, _lb) {                        \
+			FOREACH_MAIN_LISTBASE_ID_BEGIN(_lbarray[_i], _id)
 
-#define FOREACH_MAIN_ID_BREAKABLE_END FOREACH_MAIN_ID_END
+#define FOREACH_MAIN_ID_END                                               \
+			FOREACH_MAIN_LISTBASE_ID_END;                                 \
+		} FOREACH_MAIN_LISTBASE_END;                                      \
+	} ((void)0)
 
 
 struct BlendThumbnail *BKE_main_thumbnail_from_imbuf(struct Main *bmain, struct ImBuf *img);
diff --git a/source/blender/blenkernel/intern/blendfile.c b/source/blender/blenkernel/intern/blendfile.c
index 455f73c65c0..cd17911d65d 100644
--- a/source/blender/blenkernel/intern/blendfile.c
+++ b/source/blender/blenkernel/intern/blendfile.c
@@ -460,16 +460,21 @@ bool BKE_blendfile_read_from_memfile(
 void BKE_blendfile_read_make_empty(bContext *C)
 {
 	Main *bmain = CTX_data_main(C);
+	ListBase *lb;
 	ID *id;
 
-	FOREACH_MAIN_ID_BEGIN(bmain, id)
+	FOREACH_MAIN_LISTBASE_BEGIN(bmain, lb)
 	{
-		if (ELEM(GS(id->name), ID_SCE, ID_SCR, ID_WM, ID_WS)) {
-			break;  /* Only breaks iter on that ID type, and continues with IDs of next type. */
+		FOREACH_MAIN_LISTBASE_ID_BEGIN(lb, id)
+		{
+			if (ELEM(GS(id->name), ID_SCE, ID_SCR, ID_WM, ID_WS)) {
+				break;
+			}
+			BKE_id_delete(bmain, id);
 		}
-		BKE_id_delete(bmain, id);
+		FOREACH_MAIN_LISTBASE_ID_END;
 	}
-	FOREACH_MAIN_ID_END;
+	FOREACH_MAIN_LISTBASE_END;
 }
 
 /* only read the userdef from a .blend */
diff --git a/source/blender/blenkernel/intern/library_query.c b/source/blender/blenkernel/intern/library_query.c
index 72e9d747206..9861b232dd9 100644
--- a/source/blender/blenkernel/intern/library_query.c
+++ b/source/blender/blenkernel/intern/library_query.c
@@ -1384,19 +1384,16 @@ void BKE_library_unused_linked_data_set_tag(Main *bmain, const bool do_init_tag)
 	}
 
 	for (bool do_loop = true; do_loop; ) {
-		bool do_break = false;
 		do_loop = false;
-		FOREACH_MAIN_ID_BREAKABLE_BEGIN(bmain, id, do_break)
+		FOREACH_MAIN_ID_BEGIN(bmain, id)
 		{
+			/* We only want to check that ID if it is currently known as used... */
 			if ((id->tag & LIB_TAG_DOIT) == 0) {
 				BKE_library_foreach_ID_link(
 				            bmain, id, foreach_libblock_used_linked_data_tag_clear_cb, &do_loop, IDWALK_READONLY);
 			}
-			/* Else it is an unused ID (so far), no need to check it further. */
-			do_break = true;
-			break;
 		}
-		FOREACH_MAIN_ID_BREAKABLE_END;
+		FOREACH_MAIN_ID_END;
 	}
 }