glTF export: new option to export only visible/renderable/active collection

3 files changed, 45 insertions, 1 deletions

diff --git a/io_scene_gltf2/__init__.py b/io_scene_gltf2/__init__.py
index 9c2972be..55ba071e 100755
--- a/io_scene_gltf2/__init__.py
+++ b/io_scene_gltf2/__init__.py
@@ -15,7 +15,7 @@
 bl_info = {
     'name': 'glTF 2.0 format',
     'author': 'Julien Duroure, Scurest, Norbert Nopper, Urs Hanselmann, Moritz Becher, Benjamin Schmithüsen, Jim Eckerlein, and many external contributors',
-    "version": (1, 6, 15),
+    "version": (1, 6, 16),
     'blender': (2, 91, 0),
     'location': 'File > Import-Export',
     'description': 'Import-Export as glTF 2.0',
@@ -298,6 +298,24 @@ class ExportGLTF2_Base:
         default=False
     )
 
+    use_visible: BoolProperty(
+        name='Visible Objects',
+        description='Export visible objects only',
+        default=False
+    )
+
+    use_renderable: BoolProperty(
+        name='Renderable Objects',
+        description='Export renderable objects only',
+        default=False
+    )
+
+    use_active_collection: BoolProperty(
+        name='Active Collection',
+        description='Export objects in the active collection only',
+        default=False
+    )
+
     export_extras: BoolProperty(
         name='Custom Properties',
         description='Export custom properties as glTF extras',
@@ -464,6 +482,9 @@ class ExportGLTF2_Base:
         exceptional = [
             # options that don't start with 'export_'
             'use_selection',
+            'use_visible',
+            'use_renderable',
+            'use_active_collection',
             'use_mesh_edges',
             'use_mesh_vertices',
         ]
@@ -528,6 +549,10 @@ class ExportGLTF2_Base:
         else:
             export_settings['gltf_selected'] = self.use_selection
 
+        export_settings['gltf_visible'] = self.use_visible
+        export_settings['gltf_renderable'] = self.use_renderable
+        export_settings['gltf_active_collection'] = self.use_active_collection
+
         # export_settings['gltf_selected'] = self.use_selection This can be uncomment when removing compatibility of export_selected
         export_settings['gltf_layers'] = True  # self.export_layers
         export_settings['gltf_extras'] = self.export_extras
@@ -657,6 +682,9 @@ class GLTF_PT_export_include(bpy.types.Panel):
 
         col = layout.column(heading = "Limit to", align = True)
         col.prop(operator, 'use_selection')
+        col.prop(operator, 'use_visible')
+        col.prop(operator, 'use_renderable')
+        col.prop(operator, 'use_active_collection')
 
         col = layout.column(heading = "Data", align = True)
         col.prop(operator, 'export_extras')
diff --git a/io_scene_gltf2/blender/exp/gltf2_blender_export_keys.py b/io_scene_gltf2/blender/exp/gltf2_blender_export_keys.py
index a983c088..06fe6a47 100755
--- a/io_scene_gltf2/blender/exp/gltf2_blender_export_keys.py
+++ b/io_scene_gltf2/blender/exp/gltf2_blender_export_keys.py
@@ -26,6 +26,9 @@ FILTERED_CAMERAS = 'filtered_cameras'
 
 APPLY = 'gltf_apply'
 SELECTED = 'gltf_selected'
+VISIBLE = 'gltf_visible'
+RENDERABLE = 'gltf_renderable'
+ACTIVE_COLLECTION = 'gltf_active_collection'
 SKINS = 'gltf_skins'
 DISPLACEMENT = 'gltf_displacement'
 FORCE_SAMPLING = 'gltf_force_sampling'
diff --git a/io_scene_gltf2/blender/exp/gltf2_blender_gather_nodes.py b/io_scene_gltf2/blender/exp/gltf2_blender_gather_nodes.py
index 5332f4a9..7b58c0f9 100755
--- a/io_scene_gltf2/blender/exp/gltf2_blender_gather_nodes.py
+++ b/io_scene_gltf2/blender/exp/gltf2_blender_gather_nodes.py
@@ -127,6 +127,19 @@ def __filter_node(blender_object, blender_scene, export_settings):
     if export_settings[gltf2_blender_export_keys.SELECTED] and blender_object.select_get() is False:
         return False
 
+    if export_settings[gltf2_blender_export_keys.VISIBLE] and blender_object.visible_get() is False:
+        return False
+
+    # render_get() doesn't exist, so unfortunately this won't take into account the Collection settings
+    if export_settings[gltf2_blender_export_keys.RENDERABLE] and blender_object.hide_render is True:
+        return False
+
+    if export_settings[gltf2_blender_export_keys.ACTIVE_COLLECTION]:
+        found = any(x == blender_object for x in bpy.context.collection.all_objects)
+
+        if not found:
+            return False
+
     return True