Extend libquery checks to test wether an ID is used locally and/or indirectly.

2 files changed, 55 insertions, 16 deletions

diff --git a/source/blender/blenkernel/BKE_library_query.h b/source/blender/blenkernel/BKE_library_query.h
index 3a2b49f911c..77a3b31f805 100644
--- a/source/blender/blenkernel/BKE_library_query.h
+++ b/source/blender/blenkernel/BKE_library_query.h
@@ -76,6 +76,8 @@ void BKE_library_update_ID_link_user(struct ID *id_dst, struct ID *id_src, const
 
 int BKE_library_ID_use_ID(struct ID *id_user, struct ID *id_used);
 
+bool BKE_library_ID_is_locally_used(struct Main *bmain, void *idv);
 bool BKE_library_ID_is_indirectly_used(struct Main *bmain, void *idv);
+void BKE_library_ID_test_usages(struct Main *bmain, void *idv, bool *is_used_local, bool *is_used_linked);
 
 #endif  /* __BKE_LIBRARY_QUERY_H__ */
diff --git a/source/blender/blenkernel/intern/library_query.c b/source/blender/blenkernel/intern/library_query.c
index 6e2e07eeeca..efaa5448fa3 100644
--- a/source/blender/blenkernel/intern/library_query.c
+++ b/source/blender/blenkernel/intern/library_query.c
@@ -867,7 +867,7 @@ int BKE_library_ID_use_ID(ID *id_user, ID *id_used)
 }
 
 
-static int foreach_libblock_check_indirect_usage_callback(
+static int foreach_libblock_check_usage_callback(
         void *user_data, ID *UNUSED(id_self), ID **id_p, int UNUSED(cb_flag))
 {
 	IDUsersIter *iter = user_data;
@@ -880,38 +880,75 @@ static int foreach_libblock_check_indirect_usage_callback(
 	return IDWALK_RET_NOP;
 }
 
-/**
- * Check weather given ID is used indirectly (i.e. by another linked ID).
- */
-bool BKE_library_ID_is_indirectly_used(Main *bmain, void *idv)
+static bool library_ID_is_used(Main *bmain, void *idv, const bool check_linked)
 {
 	IDUsersIter iter;
 	ListBase *lb_array[MAX_LIBARRAY];
 	int i = set_listbasepointers(bmain, lb_array);
+	bool is_defined = false;
 
 	iter.id = idv;
 	iter.count = 0;
-	while (i--) {
+	while (i-- && !is_defined) {
 		ID *id_curr = lb_array[i]->first;
 
-		for (; id_curr; id_curr = id_curr->next) {
-			if (!ID_IS_LINKED_DATABLOCK(id_curr)) {
+		for (; id_curr && !is_defined; id_curr = id_curr->next) {
+			if (check_linked != ID_IS_LINKED_DATABLOCK(id_curr)) {
 				continue;
 			}
 
 			iter.curr_id = id_curr;
 			BKE_library_foreach_ID_link(
-			            id_curr, foreach_libblock_check_indirect_usage_callback, &iter, IDWALK_NOP);
+			            id_curr, foreach_libblock_check_usage_callback, &iter, IDWALK_NOP);
 
-			if (iter.count) {
-				break;
-			}
-		}
-		if (iter.count) {
-			break;
+			is_defined = (iter.count != 0);
 		}
 	}
 
-	return (iter.count != 0);
+	return is_defined;
+}
+
+/**
+ * Check wether given ID is used locally (i.e. by another non-linked ID).
+ */
+bool BKE_library_ID_is_locally_used(Main *bmain, void *idv)
+{
+	return library_ID_is_used(bmain, idv, false);
 }
 
+/**
+ * Check wether given ID is used indirectly (i.e. by another linked ID).
+ */
+bool BKE_library_ID_is_indirectly_used(Main *bmain, void *idv)
+{
+	return library_ID_is_used(bmain, idv, true);
+}
+
+/**
+ * Combine \a BKE_library_ID_is_locally_used() and \a BKE_library_ID_is_indirectly_used() in a single call.
+ */
+void BKE_library_ID_test_usages(Main *bmain, void *idv, bool *is_used_local, bool *is_used_linked)
+{
+	IDUsersIter iter_local, iter_linked;
+	ListBase *lb_array[MAX_LIBARRAY];
+	int i = set_listbasepointers(bmain, lb_array);
+	bool is_defined = false;
+
+	iter_local.id = iter_linked.id = idv;
+	iter_local.count = iter_linked.count = 0;
+	while (i-- && !is_defined) {
+		ID *id_curr = lb_array[i]->first;
+
+		for (; id_curr && !is_defined; id_curr = id_curr->next) {
+			IDUsersIter *iter = (ID_IS_LINKED_DATABLOCK(id_curr)) ? &iter_linked : &iter_local;
+
+			iter->curr_id = id_curr;
+			BKE_library_foreach_ID_link(id_curr, foreach_libblock_check_usage_callback, iter, IDWALK_NOP);
+
+			is_defined = (iter_local.count != 0 && iter_linked.count != 0);
+		}
+	}
+
+	*is_used_local = (iter_local.count != 0);
+	*is_used_linked = (iter_linked.count != 0);
+}