forked from bartvdbraak/blender
449fbde7d1
Requested by Aaron Carlisle (@blendify) over IRC.
259 lines
8.4 KiB
Python
259 lines
8.4 KiB
Python
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
|
|
import bpy
|
|
from bpy.types import Operator
|
|
from bpy.props import (
|
|
StringProperty,
|
|
BoolProperty,
|
|
CollectionProperty,
|
|
)
|
|
|
|
# ########## Datablock previews... ##########
|
|
|
|
|
|
class WM_OT_previews_batch_generate(Operator):
|
|
"""Generate selected .blend file's previews"""
|
|
bl_idname = "wm.previews_batch_generate"
|
|
bl_label = "Batch-Generate Previews"
|
|
bl_options = {'REGISTER'}
|
|
|
|
# -----------
|
|
# File props.
|
|
files = CollectionProperty(
|
|
type=bpy.types.OperatorFileListElement,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
directory = StringProperty(
|
|
maxlen=1024,
|
|
subtype='FILE_PATH',
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
# Show only images/videos, and directories!
|
|
filter_blender = BoolProperty(
|
|
default=True,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
filter_folder = BoolProperty(
|
|
default=True,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
# -----------
|
|
# Own props.
|
|
use_scenes = BoolProperty(
|
|
default=True,
|
|
name="Scenes",
|
|
description="Generate scenes' previews",
|
|
)
|
|
use_groups = BoolProperty(
|
|
default=True,
|
|
name="Groups",
|
|
description="Generate groups' previews",
|
|
)
|
|
use_objects = BoolProperty(
|
|
default=True,
|
|
name="Objects",
|
|
description="Generate objects' previews",
|
|
)
|
|
use_intern_data = BoolProperty(
|
|
default=True,
|
|
name="Mat/Tex/...",
|
|
description="Generate 'internal' previews (materials, textures, images, etc.)",
|
|
)
|
|
|
|
use_trusted = BoolProperty(
|
|
default=False,
|
|
name="Trusted Blend Files",
|
|
description="Enable python evaluation for selected files",
|
|
)
|
|
use_backups = BoolProperty(
|
|
default=True,
|
|
name="Save Backups",
|
|
description="Keep a backup (.blend1) version of the files when saving with generated previews",
|
|
)
|
|
|
|
def invoke(self, context, event):
|
|
context.window_manager.fileselect_add(self)
|
|
return {'RUNNING_MODAL'}
|
|
|
|
def execute(self, context):
|
|
if "subprocess" in locals():
|
|
import imp
|
|
imp.reload(preview_render)
|
|
else:
|
|
import os
|
|
import subprocess
|
|
from bl_previews_utils import bl_previews_render as preview_render
|
|
|
|
context.window_manager.progress_begin(0, len(self.files))
|
|
context.window_manager.progress_update(0)
|
|
for i, fn in enumerate(self.files):
|
|
blen_path = os.path.join(self.directory, fn.name)
|
|
cmd = [
|
|
bpy.app.binary_path,
|
|
"--background",
|
|
"--factory-startup",
|
|
"-noaudio",
|
|
]
|
|
if self.use_trusted:
|
|
cmd.append("--enable-autoexec")
|
|
cmd.extend([
|
|
blen_path,
|
|
"--python",
|
|
os.path.join(os.path.dirname(preview_render.__file__), "bl_previews_render.py"),
|
|
"--",
|
|
])
|
|
if not self.use_scenes:
|
|
cmd.append('--no_scenes')
|
|
if not self.use_groups:
|
|
cmd.append('--no_groups')
|
|
if not self.use_objects:
|
|
cmd.append('--no_objects')
|
|
if not self.use_intern_data:
|
|
cmd.append('--no_data_intern')
|
|
if not self.use_backups:
|
|
cmd.append("--no_backups")
|
|
if subprocess.call(cmd):
|
|
self.report({'ERROR'}, "Previews generation process failed for file '%s'!" % blen_path)
|
|
context.window_manager.progress_end()
|
|
return {'CANCELLED'}
|
|
context.window_manager.progress_update(i + 1)
|
|
context.window_manager.progress_end()
|
|
|
|
return {'FINISHED'}
|
|
|
|
|
|
class WM_OT_previews_batch_clear(Operator):
|
|
"""Clear selected .blend file's previews"""
|
|
bl_idname = "wm.previews_batch_clear"
|
|
bl_label = "Batch-Clear Previews"
|
|
bl_options = {'REGISTER'}
|
|
|
|
# -----------
|
|
# File props.
|
|
files = CollectionProperty(
|
|
type=bpy.types.OperatorFileListElement,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
directory = StringProperty(
|
|
maxlen=1024,
|
|
subtype='FILE_PATH',
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
# Show only images/videos, and directories!
|
|
filter_blender = BoolProperty(
|
|
default=True,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
filter_folder = BoolProperty(
|
|
default=True,
|
|
options={'HIDDEN', 'SKIP_SAVE'},
|
|
)
|
|
|
|
# -----------
|
|
# Own props.
|
|
use_scenes = BoolProperty(
|
|
default=True,
|
|
name="Scenes",
|
|
description="Clear scenes' previews",
|
|
)
|
|
use_groups = BoolProperty(default=True,
|
|
name="Groups",
|
|
description="Clear groups' previews",
|
|
)
|
|
use_objects = BoolProperty(
|
|
default=True,
|
|
name="Objects",
|
|
description="Clear objects' previews",
|
|
)
|
|
use_intern_data = BoolProperty(
|
|
default=True,
|
|
name="Mat/Tex/...",
|
|
description="Clear 'internal' previews (materials, textures, images, etc.)",
|
|
)
|
|
|
|
use_trusted = BoolProperty(
|
|
default=False,
|
|
name="Trusted Blend Files",
|
|
description="Enable python evaluation for selected files",
|
|
)
|
|
use_backups = BoolProperty(
|
|
default=True,
|
|
name="Save Backups",
|
|
description="Keep a backup (.blend1) version of the files when saving with cleared previews",
|
|
)
|
|
|
|
def invoke(self, context, event):
|
|
context.window_manager.fileselect_add(self)
|
|
return {'RUNNING_MODAL'}
|
|
|
|
def execute(self, context):
|
|
if "subprocess" in locals():
|
|
import imp
|
|
imp.reload(preview_render)
|
|
else:
|
|
import os
|
|
import subprocess
|
|
from bl_previews_utils import bl_previews_render as preview_render
|
|
|
|
context.window_manager.progress_begin(0, len(self.files))
|
|
context.window_manager.progress_update(0)
|
|
for i, fn in enumerate(self.files):
|
|
blen_path = os.path.join(self.directory, fn.name)
|
|
cmd = [
|
|
bpy.app.binary_path,
|
|
"--background",
|
|
"--factory-startup",
|
|
"-noaudio",
|
|
]
|
|
if self.use_trusted:
|
|
cmd.append("--enable-autoexec")
|
|
cmd.extend([
|
|
blen_path,
|
|
"--python",
|
|
os.path.join(os.path.dirname(preview_render.__file__), "bl_previews_render.py"),
|
|
"--",
|
|
"--clear",
|
|
])
|
|
if not self.use_scenes:
|
|
cmd.append('--no_scenes')
|
|
if not self.use_groups:
|
|
cmd.append('--no_groups')
|
|
if not self.use_objects:
|
|
cmd.append('--no_objects')
|
|
if not self.use_intern_data:
|
|
cmd.append('--no_data_intern')
|
|
if not self.use_backups:
|
|
cmd.append("--no_backups")
|
|
if subprocess.call(cmd):
|
|
self.report({'ERROR'}, "Previews clear process failed for file '%s'!" % blen_path)
|
|
context.window_manager.progress_end()
|
|
return {'CANCELLED'}
|
|
context.window_manager.progress_update(i + 1)
|
|
context.window_manager.progress_end()
|
|
|
|
return {'FINISHED'}
|
|
|