Depsgraph: Cleanup, make it easier to create relations with flags

1 files changed, 8 insertions, 2 deletions

diff --git a/source/blender/depsgraph/intern/depsgraph.cc b/source/blender/depsgraph/intern/depsgraph.cc
index 7f47dce9f47..6dd7ae97073 100644
--- a/source/blender/depsgraph/intern/depsgraph.cc
+++ b/source/blender/depsgraph/intern/depsgraph.cc
@@ -393,34 +393,40 @@ void Depsgraph::clear_id_nodes()
 DepsRelation *Depsgraph::add_new_relation(OperationDepsNode *from,
                                           OperationDepsNode *to,
                                           const char *description,
-                                          bool check_unique)
+                                          bool check_unique,
+                                          int flags)
 {
 	DepsRelation *rel = NULL;
 	if (check_unique) {
 		rel = check_nodes_connected(from, to, description);
 	}
 	if (rel != NULL) {
+		rel->flag |= flags;
 		return rel;
 	}
 	/* Create new relation, and add it to the graph. */
 	rel = OBJECT_GUARDED_NEW(DepsRelation, from, to, description);
+	rel->flag |= flags;
 	return rel;
 }
 
 /* Add new relation between two nodes */
 DepsRelation *Depsgraph::add_new_relation(DepsNode *from, DepsNode *to,
                                           const char *description,
-                                          bool check_unique)
+                                          bool check_unique,
+                                          int flags)
 {
 	DepsRelation *rel = NULL;
 	if (check_unique) {
 		rel = check_nodes_connected(from, to, description);
 	}
 	if (rel != NULL) {
+		rel->flag |= flags;
 		return rel;
 	}
 	/* Create new relation, and add it to the graph. */
 	rel = OBJECT_GUARDED_NEW(DepsRelation, from, to, description);
+	rel->flag |= flags;
 	return rel;
 }