Layers/Collections: Fix crash when deleting collection

If the collection to be deleted has a nested collection that is directly linked
to a view layer, we were getting a crash.
This commit is contained in:
Dalai Felinto 2018-01-02 18:19:25 -02:00
parent 8300b5e539
commit 07de8eff77
3 changed files with 64 additions and 0 deletions

View File

@ -188,6 +188,15 @@ bool BKE_collection_remove(ID *owner_id, SceneCollection *sc)
return false;
}
/* We need to do bottom up removal, otherwise we get a crash when we remove a collection that
* has one of its nested collections linked to a view layer. */
SceneCollection *scene_collection_nested = sc->scene_collections.first;
while (scene_collection_nested != NULL) {
SceneCollection *scene_collection_next = scene_collection_nested->next;
BKE_collection_remove(owner_id, scene_collection_nested);
scene_collection_nested = scene_collection_next;
}
/* Unlink from the respective collection tree. */
if (!collection_remlink(sc_master, sc)) {
BLI_assert(false);

View File

@ -170,6 +170,7 @@ VIEW_LAYER_TEST(move_into_layer_collection_i)
VIEW_LAYER_TEST(move_into_layer_collection_j)
VIEW_LAYER_TEST(layer_linking)
VIEW_LAYER_TEST(layer_syncing)
VIEW_LAYER_TEST(scene_collection_delete)
VIEW_LAYER_TEST(scene_copy_a)
VIEW_LAYER_TEST(scene_copy_b)
VIEW_LAYER_TEST(scene_copy_c)

View File

@ -0,0 +1,54 @@
# ############################################################
# Importing - Same For All Render Layer Tests
# ############################################################
import unittest
import os
import sys
from view_layer_common import *
# ############################################################
# Testing
# ############################################################
class UnitTesting(ViewLayerTesting):
def test_scene_collection_delete(self):
"""
See if a scene collection can be properly deleted even
when linked
"""
import bpy
# delete all initial objects
while bpy.data.objects:
bpy.data.objects.remove(bpy.data.objects[0])
# delete all initial collections
scene = bpy.context.scene
master_collection = scene.master_collection
while master_collection.collections:
master_collection.collections.remove(master_collection.collections[0])
collection_parent = master_collection.collections.new('parent')
collection_nested = collection_parent.collections.new('child linked')
bpy.context.view_layer.collections.link(collection_nested)
master_collection.collections.remove(collection_parent)
# Update depsgraph.
scene.update()
# ############################################################
# Main - Same For All Render Layer Tests
# ############################################################
if __name__ == '__main__':
import sys
extra_arguments = sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else []
sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 2:] if "--" in sys.argv else [])
UnitTesting._extra_arguments = extra_arguments
unittest.main()