Tests: various improvements to the bl_run_operators.py utility
- Use command line arguments to configure options such as random seed, running operators on existing blend files & how often the file is reset to the initial state. - Support for generating a script file to replay the actions, useful for creating a repeatable script for bug reports. - Add new context setup functions for edit-mode grease-pencil & hair.
This commit is contained in:
Campbell Barton
@@ -2,10 +2,20 @@
|
||||
#
|
||||
# SPDX-License-Identifier: GPL-2.0-or-later
|
||||
|
||||
# semi-useful script, runs all operators in a number of different
|
||||
# contexts, cheap way to find misc small bugs but is in no way a complete test.
|
||||
#
|
||||
# only error checked for here is a segfault.
|
||||
"""
|
||||
This script, runs all operators in a number of different contexts,
|
||||
it's a convenient way to find various bugs but is in no way a complete test.
|
||||
|
||||
It's good at catching errors where operators fail when run in unexpected contexts.
|
||||
|
||||
This can be run with the following command line arguments:
|
||||
|
||||
./blender.bin -P tests/utils/bl_run_operators.py -- --random --random-seed=123
|
||||
|
||||
Or from GDB:
|
||||
|
||||
gdb - ./blender.bin -ex=r --args ./blender.bin -P tests/utils/bl_run_operators.py -- --random --random-seed=123
|
||||
"""
|
||||
|
||||
__all__ = (
|
||||
"main",
|
||||
@@ -13,21 +23,237 @@ __all__ = (
|
||||
|
||||
import bpy
|
||||
import sys
|
||||
import random
|
||||
|
||||
|
||||
# This block is included in the script referenced by: `--generate-script`.
|
||||
# BEGIN UTILS TO EXPORT.
|
||||
|
||||
# TODO: support this via command line arguments.
|
||||
USE_ATTRSET = False
|
||||
USE_FILES = "" # "/mango/"
|
||||
USE_RANDOM = False
|
||||
USE_RANDOM_SCREEN = False
|
||||
RANDOM_SEED = [1] # so we can redo crashes
|
||||
RANDOM_RESET = 0.1 # 10% chance of resetting on each new operator
|
||||
RANDOM_MULTIPLY = 10
|
||||
|
||||
STATE = {
|
||||
"counter": 0,
|
||||
}
|
||||
|
||||
|
||||
op_blacklist = (
|
||||
def reset_blend(random_screen_index=-1):
|
||||
bpy.ops.wm.read_factory_settings()
|
||||
for scene in bpy.data.scenes:
|
||||
# Reduce range so any bake action doesn't take too long.
|
||||
scene.frame_end = scene.frame_start + 5
|
||||
|
||||
if random_screen_index != -1:
|
||||
for _ in range(random_screen_index % len(bpy.data.screens)):
|
||||
bpy.ops.screen.delete()
|
||||
|
||||
|
||||
def reset_file(filepath):
|
||||
bpy.ops.wm.open_mainfile(filepath=filepath)
|
||||
|
||||
|
||||
def temp_override_default_kwargs(
|
||||
context,
|
||||
area_type=None,
|
||||
region_type=None,
|
||||
):
|
||||
window = context.window_manager.windows[0]
|
||||
screen = window.screen
|
||||
|
||||
kwargs = {
|
||||
"window": window,
|
||||
"screen": screen,
|
||||
}
|
||||
|
||||
if (
|
||||
(area_type is not None) and
|
||||
(area := next(iter([area for area in screen.areas if area.type == area_type]), None))
|
||||
):
|
||||
kwargs["area"] = area
|
||||
if (
|
||||
(region_type is not None) and
|
||||
(region := next(iter([region for region in area.regions if region.type == region_type]), None))
|
||||
):
|
||||
kwargs["region"] = region
|
||||
|
||||
return kwargs
|
||||
|
||||
|
||||
def run_op(
|
||||
context,
|
||||
setup_fn,
|
||||
op,
|
||||
area_type=None,
|
||||
region_type=None,
|
||||
):
|
||||
op_id = op.idname_py()
|
||||
|
||||
print(" operator: {:04d}, {:s}".format(STATE["counter"], op_id))
|
||||
STATE["counter"] += 1
|
||||
sys.stdout.flush() # in case of crash
|
||||
|
||||
with context.temp_override(**temp_override_default_kwargs(context, area_type, region_type)):
|
||||
|
||||
# We can't be sure it will work (even if poll succeeds).
|
||||
try:
|
||||
setup_fn()
|
||||
except:
|
||||
pass
|
||||
|
||||
for mode in {
|
||||
'EXEC_DEFAULT',
|
||||
'INVOKE_DEFAULT',
|
||||
}:
|
||||
try:
|
||||
op(mode)
|
||||
except Exception:
|
||||
# import traceback
|
||||
# traceback.print_exc()
|
||||
pass
|
||||
|
||||
if USE_ATTRSET:
|
||||
attrset_data()
|
||||
|
||||
|
||||
# Contexts.
|
||||
|
||||
def ctx_nop():
|
||||
# This only exists so there is a callback name to reference that does nothing.
|
||||
pass
|
||||
|
||||
|
||||
def ctx_clear_scene(): # copied from batch_import.py
|
||||
bpy.ops.wm.read_factory_settings(use_empty=True)
|
||||
|
||||
|
||||
def ctx_editmode_mesh():
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mesh_extra():
|
||||
bpy.ops.object.vertex_group_add()
|
||||
bpy.ops.object.shape_key_add(from_mix=False)
|
||||
bpy.ops.object.shape_key_add(from_mix=True)
|
||||
bpy.ops.mesh.uv_texture_add()
|
||||
bpy.ops.object.material_slot_add()
|
||||
# editmode last!
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mesh_empty():
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.mesh.select_all(action='SELECT')
|
||||
bpy.ops.mesh.delete()
|
||||
|
||||
|
||||
def ctx_editmode_curves():
|
||||
bpy.ops.curve.primitive_nurbs_circle_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_curves_empty():
|
||||
bpy.ops.curve.primitive_nurbs_circle_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.curve.select_all(action='SELECT')
|
||||
bpy.ops.curve.delete(type='VERT')
|
||||
|
||||
|
||||
def ctx_editmode_surface():
|
||||
bpy.ops.surface.primitive_nurbs_surface_torus_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_hair():
|
||||
bpy.ops.mesh.primitive_plane_add()
|
||||
bpy.ops.object.curves_empty_hair_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.curves.add_circle()
|
||||
|
||||
|
||||
def ctx_editmode_hair_empty():
|
||||
bpy.ops.mesh.primitive_plane_add()
|
||||
bpy.ops.object.curves_empty_hair_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_grease_pencil():
|
||||
bpy.ops.object.grease_pencil_add(type='MONKEY')
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_grease_pencil_empty():
|
||||
bpy.ops.object.grease_pencil_add(type='EMPTY')
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mball():
|
||||
bpy.ops.object.metaball_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_text():
|
||||
bpy.ops.object.text_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_armature():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_armature_empty():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.armature.select_all(action='SELECT')
|
||||
bpy.ops.armature.delete()
|
||||
|
||||
|
||||
def ctx_editmode_lattice():
|
||||
bpy.ops.object.add(type='LATTICE')
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
# bpy.ops.object.vertex_group_add()
|
||||
|
||||
|
||||
def ctx_object_empty():
|
||||
bpy.ops.object.add(type='EMPTY')
|
||||
|
||||
|
||||
def ctx_object_pose():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='POSE')
|
||||
bpy.ops.pose.select_all(action='SELECT')
|
||||
|
||||
|
||||
def ctx_object_volume():
|
||||
bpy.ops.object.add(type='VOLUME')
|
||||
|
||||
|
||||
def ctx_object_empty_as_camera():
|
||||
bpy.ops.object.add(type='EMPTY')
|
||||
bpy.context.scene.camera = bpy.context.active_object
|
||||
bpy.context.active_object.data = bpy.data.images.new(name="Foo", width=1, height=1)
|
||||
|
||||
|
||||
def ctx_object_paint_weight():
|
||||
bpy.ops.object.mode_set(mode='WEIGHT_PAINT')
|
||||
|
||||
|
||||
def ctx_object_paint_vertex():
|
||||
bpy.ops.object.mode_set(mode='VERTEX_PAINT')
|
||||
|
||||
|
||||
def ctx_object_paint_sculpt():
|
||||
bpy.ops.object.mode_set(mode='SCULPT')
|
||||
|
||||
|
||||
def ctx_object_paint_texture():
|
||||
bpy.ops.object.mode_set(mode='TEXTURE_PAINT')
|
||||
|
||||
|
||||
# END UTILS TO EXPORT.
|
||||
|
||||
|
||||
operator_pattern_exclude = (
|
||||
"script.reload",
|
||||
"export*.*",
|
||||
"import*.*",
|
||||
@@ -125,49 +351,24 @@ def blend_list(mainpath):
|
||||
|
||||
def is_blend(filename):
|
||||
ext = splitext(filename)[1]
|
||||
return (ext in {".blend", })
|
||||
return (ext == ".blend")
|
||||
|
||||
return list(sorted(file_list(mainpath, is_blend)))
|
||||
|
||||
|
||||
if USE_FILES:
|
||||
USE_FILES_LS = blend_list(USE_FILES)
|
||||
# print(USE_FILES_LS)
|
||||
|
||||
|
||||
def filter_op_list(operators):
|
||||
from fnmatch import fnmatchcase
|
||||
|
||||
def is_op_ok(op):
|
||||
for op_match in op_blacklist:
|
||||
for op_match in operator_pattern_exclude:
|
||||
if fnmatchcase(op, op_match):
|
||||
print(" skipping: %s (%s)" % (op, op_match))
|
||||
print(" skipping: {:s} ({:s})".format(op, op_match))
|
||||
return False
|
||||
return True
|
||||
|
||||
operators[:] = [op for op in operators if is_op_ok(op[0])]
|
||||
|
||||
|
||||
def reset_blend():
|
||||
bpy.ops.wm.read_factory_settings()
|
||||
for scene in bpy.data.scenes:
|
||||
# reduce range so any bake action doesn't take too long
|
||||
scene.frame_start = 1
|
||||
scene.frame_end = 5
|
||||
|
||||
if USE_RANDOM_SCREEN:
|
||||
import random
|
||||
for _ in range(random.randint(0, len(bpy.data.screens))):
|
||||
bpy.ops.screen.delete()
|
||||
print("Scree IS", bpy.context.screen)
|
||||
|
||||
|
||||
def reset_file():
|
||||
import random
|
||||
f = USE_FILES_LS[random.randint(0, len(USE_FILES_LS) - 1)]
|
||||
bpy.ops.wm.open_mainfile(filepath=f)
|
||||
|
||||
|
||||
if USE_ATTRSET:
|
||||
def build_property_typemap(skip_classes):
|
||||
|
||||
@@ -189,11 +390,11 @@ if USE_ATTRSET:
|
||||
property_typemap[attr] = properties
|
||||
|
||||
return property_typemap
|
||||
CLS_BLACKLIST = (
|
||||
CLS_EXCLUDE = (
|
||||
bpy.types.BrushTextureSlot,
|
||||
bpy.types.Brush,
|
||||
)
|
||||
property_typemap = build_property_typemap(CLS_BLACKLIST)
|
||||
property_typemap = build_property_typemap(CLS_EXCLUDE)
|
||||
bpy_struct_type = bpy.types.Struct.__base__
|
||||
|
||||
def id_walk(value, parent):
|
||||
@@ -262,176 +463,116 @@ if USE_ATTRSET:
|
||||
pass
|
||||
|
||||
|
||||
def run_ops(operators, setup_func=None, reset=True):
|
||||
def run_ops(
|
||||
operators, # `list[str]`
|
||||
*,
|
||||
log_fn, # `BytesIO | None`
|
||||
setup_fn, # `Callable[[None], None]`
|
||||
use_random, # `bool`
|
||||
random_reset, # `float` (between 0 and 1).
|
||||
random_screen, # `bool`
|
||||
blend_files, # `list[str] | None`
|
||||
):
|
||||
from bpy import context
|
||||
print("\ncontext:", setup_func.__name__)
|
||||
|
||||
def temp_override_default_kwargs():
|
||||
return {
|
||||
"window": context.window_manager.windows[0],
|
||||
}
|
||||
print("\nContext:", setup_fn.__name__)
|
||||
|
||||
# first invoke
|
||||
for op_id, op in operators:
|
||||
with context.temp_override(window=context.window_manager.windows[0]):
|
||||
if not op.poll():
|
||||
continue
|
||||
if log_fn is not None:
|
||||
# Only when operators run else this just produces empty headings.
|
||||
if operators:
|
||||
log_fn("# Context: {:s}\n".format(setup_fn.__name__))
|
||||
|
||||
print(" operator: %4d, %s" % (STATE["counter"], op_id))
|
||||
STATE["counter"] += 1
|
||||
sys.stdout.flush() # in case of crash
|
||||
# This is more of a run-time check, ignore `log_fn`.
|
||||
if not operators:
|
||||
reset_blend()
|
||||
|
||||
with context.temp_override(**temp_override_default_kwargs(context, area_type=None, region_type=None)):
|
||||
# Empty operators is a signal the setup functions are being tested.
|
||||
if use_random and operators:
|
||||
# we can't be sure it will work
|
||||
try:
|
||||
setup_fn()
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
setup_fn()
|
||||
return
|
||||
|
||||
for filepath in ((None, ) if blend_files is None else blend_files):
|
||||
is_first = True
|
||||
random_int = 0
|
||||
|
||||
for op_id, op in operators:
|
||||
|
||||
# disable will get blender in a bad state and crash easy!
|
||||
if reset:
|
||||
reset_test = True
|
||||
if USE_RANDOM:
|
||||
import random
|
||||
if random.random() < (1.0 - RANDOM_RESET):
|
||||
if use_random:
|
||||
if random.random() < (1.0 - random_reset):
|
||||
reset_test = False
|
||||
|
||||
# Always reset on the first iteration.
|
||||
if is_first:
|
||||
is_first = False
|
||||
reset_test = True
|
||||
|
||||
if reset_test:
|
||||
if USE_FILES:
|
||||
reset_file()
|
||||
if use_random:
|
||||
random_int = random.randint(0, 0xffffffff)
|
||||
|
||||
if filepath is not None:
|
||||
if log_fn is not None:
|
||||
log_fn("reset_file({!r})\n".format(filepath))
|
||||
reset_file(filepath)
|
||||
else:
|
||||
reset_blend()
|
||||
if random_screen:
|
||||
random_screen_int = random.randint(0, 0xffffffff)
|
||||
else:
|
||||
random_screen_int = -1
|
||||
|
||||
if log_fn is not None:
|
||||
log_fn("reset_blend({:d})\n".format(random_screen_int))
|
||||
reset_blend(random_screen_int)
|
||||
|
||||
window = context.window_manager.windows[0]
|
||||
screen = window.screen
|
||||
|
||||
# Get the area & region, when random is used they may randomly be None.
|
||||
areas = list(screen.areas)
|
||||
if use_random:
|
||||
areas.append(None)
|
||||
area = areas[random_int % len(areas)]
|
||||
del areas
|
||||
|
||||
if area is not None:
|
||||
regions = list(area.regions)
|
||||
if use_random:
|
||||
regions.append(None)
|
||||
region = regions[random_int % len(regions)]
|
||||
del regions
|
||||
else:
|
||||
region = None
|
||||
|
||||
area_type = area.type if area else None
|
||||
region_type = region.type if (area and region) else None
|
||||
del reset_test
|
||||
del area, region
|
||||
|
||||
with context.temp_override(**temp_override_default_kwargs()):
|
||||
with context.temp_override(**temp_override_default_kwargs(
|
||||
context,
|
||||
area_type=area_type,
|
||||
region_type=region_type,
|
||||
)):
|
||||
if not op.poll():
|
||||
continue
|
||||
|
||||
if USE_RANDOM:
|
||||
# we can't be sure it will work
|
||||
try:
|
||||
setup_func()
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
setup_func()
|
||||
if log_fn is not None:
|
||||
log_fn("run_op(context, {:s}, bpy.ops.{:s}, {!r}, {!r})\n".format(
|
||||
setup_fn.__name__,
|
||||
op_id,
|
||||
area_type,
|
||||
region_type,
|
||||
))
|
||||
|
||||
for mode in {'EXEC_DEFAULT', 'INVOKE_DEFAULT'}:
|
||||
try:
|
||||
op(mode)
|
||||
except:
|
||||
# import traceback
|
||||
# traceback.print_exc()
|
||||
pass
|
||||
|
||||
if USE_ATTRSET:
|
||||
attrset_data()
|
||||
|
||||
if not operators:
|
||||
# run test
|
||||
if reset:
|
||||
reset_blend()
|
||||
|
||||
with context.temp_override(**temp_override_default_kwargs()):
|
||||
if USE_RANDOM:
|
||||
# we can't be sure it will work
|
||||
try:
|
||||
setup_func()
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
setup_func()
|
||||
|
||||
|
||||
# contexts
|
||||
def ctx_clear_scene(): # copied from batch_import.py
|
||||
bpy.ops.wm.read_factory_settings(use_empty=True)
|
||||
|
||||
|
||||
def ctx_editmode_mesh():
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mesh_extra():
|
||||
bpy.ops.object.vertex_group_add()
|
||||
bpy.ops.object.shape_key_add(from_mix=False)
|
||||
bpy.ops.object.shape_key_add(from_mix=True)
|
||||
bpy.ops.mesh.uv_texture_add()
|
||||
bpy.ops.object.material_slot_add()
|
||||
# editmode last!
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mesh_empty():
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.mesh.select_all(action='SELECT')
|
||||
bpy.ops.mesh.delete()
|
||||
|
||||
|
||||
def ctx_editmode_curves():
|
||||
bpy.ops.curve.primitive_nurbs_circle_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_curves_empty():
|
||||
bpy.ops.curve.primitive_nurbs_circle_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.curve.select_all(action='SELECT')
|
||||
bpy.ops.curve.delete(type='VERT')
|
||||
|
||||
|
||||
def ctx_editmode_surface():
|
||||
bpy.ops.surface.primitive_nurbs_surface_torus_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_mball():
|
||||
bpy.ops.object.metaball_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_text():
|
||||
bpy.ops.object.text_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_armature():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
|
||||
|
||||
def ctx_editmode_armature_empty():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
bpy.ops.armature.select_all(action='SELECT')
|
||||
bpy.ops.armature.delete()
|
||||
|
||||
|
||||
def ctx_editmode_lattice():
|
||||
bpy.ops.object.add(type='LATTICE')
|
||||
bpy.ops.object.mode_set(mode='EDIT')
|
||||
# bpy.ops.object.vertex_group_add()
|
||||
|
||||
|
||||
def ctx_object_empty():
|
||||
bpy.ops.object.add(type='EMPTY')
|
||||
|
||||
|
||||
def ctx_object_pose():
|
||||
bpy.ops.object.armature_add()
|
||||
bpy.ops.object.mode_set(mode='POSE')
|
||||
bpy.ops.pose.select_all(action='SELECT')
|
||||
|
||||
|
||||
def ctx_object_volume():
|
||||
bpy.ops.object.add(type='VOLUME')
|
||||
|
||||
|
||||
def ctx_object_paint_weight():
|
||||
bpy.ops.object.mode_set(mode='WEIGHT_PAINT')
|
||||
|
||||
|
||||
def ctx_object_paint_vertex():
|
||||
bpy.ops.object.mode_set(mode='VERTEX_PAINT')
|
||||
|
||||
|
||||
def ctx_object_paint_sculpt():
|
||||
bpy.ops.object.mode_set(mode='SCULPT')
|
||||
|
||||
|
||||
def ctx_object_paint_texture():
|
||||
bpy.ops.object.mode_set(mode='TEXTURE_PAINT')
|
||||
run_op(context, setup_fn, op, area_type, region_type)
|
||||
|
||||
|
||||
def bpy_check_type_duplicates():
|
||||
@@ -444,12 +585,44 @@ def bpy_check_type_duplicates():
|
||||
for t in sorted(bl_types_unique):
|
||||
tot = bl_types.count(t)
|
||||
if tot > 1:
|
||||
print(" '%s', %d" % (t, tot))
|
||||
print(" '{:s}', {:d}".format(t, tot))
|
||||
import sys
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def main():
|
||||
def extract_region_from_text_by_delimiters(text, mark_beg, mark_end):
|
||||
beg = text.find(mark_beg)
|
||||
end = text.find(mark_end)
|
||||
assert beg != -1 and end != -1
|
||||
return text[beg + len(mark_beg):end]
|
||||
|
||||
|
||||
def run_all(
|
||||
log_fn, # `File | None`
|
||||
*,
|
||||
use_random, # `bool`
|
||||
random_reset, # `float`
|
||||
random_multiply, # `int`
|
||||
random_screen, # `bool`
|
||||
blend_files, # `list[str] | None`
|
||||
):
|
||||
if log_fn is not None:
|
||||
log_fn(
|
||||
"import bpy\n"
|
||||
"import sys\n"
|
||||
"from bpy import context\n"
|
||||
"\n"
|
||||
)
|
||||
|
||||
# Extract utility functions form this file.
|
||||
with open(__file__, "r", encoding="utf-8") as fh:
|
||||
log_fn("# Utility functions.")
|
||||
log_fn(extract_region_from_text_by_delimiters(
|
||||
fh.read(),
|
||||
"# BEGIN UTILS TO EXPORT.",
|
||||
"# END UTILS TO EXPORT.",
|
||||
))
|
||||
log_fn("\n")
|
||||
|
||||
bpy_check_type_duplicates()
|
||||
|
||||
@@ -460,62 +633,218 @@ def main():
|
||||
mod = getattr(bpy.ops, mod_name)
|
||||
for submod_name in dir(mod):
|
||||
op = getattr(mod, submod_name)
|
||||
operators.append(("%s.%s" % (mod_name, submod_name), op))
|
||||
operators.append(("{:s}.{:s}".format(mod_name, submod_name), op))
|
||||
|
||||
operators.sort(key=lambda op: op[0])
|
||||
|
||||
filter_op_list(operators)
|
||||
|
||||
# for testing, mix the list up.
|
||||
# operators.reverse()
|
||||
if blend_files:
|
||||
setup_fn_list = [
|
||||
ctx_nop,
|
||||
]
|
||||
else:
|
||||
setup_fn_list = [
|
||||
ctx_clear_scene,
|
||||
# Object modes.
|
||||
ctx_object_empty,
|
||||
ctx_object_empty_as_camera,
|
||||
ctx_object_paint_sculpt,
|
||||
ctx_object_paint_texture,
|
||||
ctx_object_paint_vertex,
|
||||
ctx_object_paint_weight,
|
||||
ctx_object_pose,
|
||||
ctx_object_volume,
|
||||
|
||||
if USE_RANDOM:
|
||||
import random
|
||||
random.seed(RANDOM_SEED[0])
|
||||
operators = operators * RANDOM_MULTIPLY
|
||||
# Mesh.
|
||||
ctx_editmode_mesh,
|
||||
ctx_editmode_mesh_extra,
|
||||
ctx_editmode_mesh_empty,
|
||||
# Armature.
|
||||
ctx_editmode_armature,
|
||||
ctx_editmode_armature_empty,
|
||||
# Curves.
|
||||
ctx_editmode_curves,
|
||||
ctx_editmode_curves_empty,
|
||||
ctx_editmode_surface,
|
||||
# Hair.
|
||||
ctx_editmode_hair,
|
||||
ctx_editmode_hair_empty,
|
||||
|
||||
# Grease pencil.
|
||||
ctx_editmode_grease_pencil,
|
||||
ctx_editmode_grease_pencil_empty,
|
||||
|
||||
# Other.
|
||||
ctx_editmode_mball,
|
||||
ctx_editmode_text,
|
||||
ctx_editmode_lattice,
|
||||
]
|
||||
|
||||
if use_random:
|
||||
operators = operators * max(1, random_multiply)
|
||||
random.shuffle(operators)
|
||||
|
||||
# 2 passes, first just run setup_func to make sure they are ok
|
||||
for operators_test in ((), operators):
|
||||
# Run the operator tests in different contexts
|
||||
run_ops(operators_test, setup_func=lambda: None)
|
||||
# First just run `setup_fn` to make sure they work.
|
||||
for setup_fn in setup_fn_list:
|
||||
run_ops(
|
||||
(),
|
||||
log_fn=log_fn,
|
||||
setup_fn=setup_fn,
|
||||
use_random=False,
|
||||
random_reset=False,
|
||||
random_screen=False,
|
||||
blend_files=None,
|
||||
)
|
||||
print("All setup functions run fine!")
|
||||
|
||||
if USE_FILES:
|
||||
continue
|
||||
for setup_fn in setup_fn_list:
|
||||
run_ops(
|
||||
operators,
|
||||
log_fn=log_fn,
|
||||
setup_fn=setup_fn,
|
||||
use_random=use_random,
|
||||
random_reset=random_reset,
|
||||
random_screen=random_screen,
|
||||
blend_files=blend_files,
|
||||
)
|
||||
|
||||
run_ops(operators_test, setup_func=ctx_clear_scene)
|
||||
# object modes
|
||||
run_ops(operators_test, setup_func=ctx_object_empty)
|
||||
run_ops(operators_test, setup_func=ctx_object_pose)
|
||||
run_ops(operators_test, setup_func=ctx_object_paint_weight)
|
||||
run_ops(operators_test, setup_func=ctx_object_paint_vertex)
|
||||
run_ops(operators_test, setup_func=ctx_object_paint_sculpt)
|
||||
run_ops(operators_test, setup_func=ctx_object_paint_texture)
|
||||
# mesh
|
||||
run_ops(operators_test, setup_func=ctx_editmode_mesh)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_mesh_extra)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_mesh_empty)
|
||||
# armature
|
||||
run_ops(operators_test, setup_func=ctx_editmode_armature)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_armature_empty)
|
||||
# curves
|
||||
run_ops(operators_test, setup_func=ctx_editmode_curves)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_curves_empty)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_surface)
|
||||
# other
|
||||
run_ops(operators_test, setup_func=ctx_editmode_mball)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_text)
|
||||
run_ops(operators_test, setup_func=ctx_editmode_lattice)
|
||||
run_ops(operators_test, setup_func=ctx_object_volume)
|
||||
print("Finished {!r}".format(__file__))
|
||||
|
||||
if not operators_test:
|
||||
print("All setup functions run fine!")
|
||||
|
||||
print("Finished %r" % __file__)
|
||||
def parse_create():
|
||||
import argparse
|
||||
import os
|
||||
|
||||
parser = argparse.ArgumentParser(
|
||||
description=__doc__,
|
||||
formatter_class=argparse.RawTextHelpFormatter,
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--generate-script",
|
||||
dest="generate_script",
|
||||
metavar='FILEPATH',
|
||||
default="",
|
||||
type=str,
|
||||
help=(
|
||||
"When set, write a Python script to this destination.\n"
|
||||
"This can be used to replay events that crash more conveniently.\n"
|
||||
"\n"
|
||||
"To reply the script fun Blender, appending the arguments: --python FILEPATH"
|
||||
),
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--random",
|
||||
dest="random",
|
||||
default=False,
|
||||
action='store_true',
|
||||
help="When set, randomize the operator call order and when the blend-file resets.",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--random-screen",
|
||||
dest="random_screen",
|
||||
default=False,
|
||||
action='store_true',
|
||||
help="When set, randomize the screen used when resetting the blend-file.",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--random-seed",
|
||||
dest="random_seed",
|
||||
metavar='INT',
|
||||
default=0,
|
||||
type=int,
|
||||
help="The seed to use for randomization.\n",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--random-multiply",
|
||||
dest="random_multiply",
|
||||
metavar='INT',
|
||||
default=1,
|
||||
type=int,
|
||||
help="Expand the number of times operators are called.\n",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--random-reset",
|
||||
dest="random_reset",
|
||||
metavar='UNIT',
|
||||
default=0.1,
|
||||
type=float,
|
||||
help=(
|
||||
"The probability of resetting the blend-file between calling each operator.\n"
|
||||
"A value of 0.1 means there is a 10%% chance that the file will be reset.\n"
|
||||
"A value of 0.9 means there is a 90%% chance.\n"
|
||||
"\n"
|
||||
"Very low values such as 0.01 (a 1%% chance), means operators could perform actions\n"
|
||||
"that exhaust system resources by randomly generating heavy scenes, use with care!"
|
||||
),
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--blend-files",
|
||||
dest="blend_files",
|
||||
metavar='DIRECTORY',
|
||||
default="",
|
||||
type=str,
|
||||
help=(
|
||||
"Instead of using empty blend file, run operators.\n"
|
||||
"\n"
|
||||
"Multiple paths may be passed in separated by \"{:s}\"\n"
|
||||
"- Files ending with \".blend\" will be loaded.\n"
|
||||
"- Other paths will be recursively scanned for \".blend\" files.\n"
|
||||
"\n"
|
||||
"Note that all operators will run on each blend file."
|
||||
).format(os.pathsep),
|
||||
)
|
||||
|
||||
return parser
|
||||
|
||||
|
||||
def main():
|
||||
import os
|
||||
import contextlib
|
||||
|
||||
args = parse_create().parse_args(sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
|
||||
|
||||
blend_files = None
|
||||
if args.blend_files:
|
||||
blend_files = []
|
||||
for path in args.blend_files.split(os.pathsep):
|
||||
if path.endswith(".blend"):
|
||||
blend_files.append(path)
|
||||
else:
|
||||
blend_files.extend(blend_list(args.blend_files))
|
||||
if not blend_files:
|
||||
print("No blend files found in: {:s}".format(args.blend_files))
|
||||
return 1
|
||||
|
||||
if args.random:
|
||||
random.seed(args.random_seed)
|
||||
|
||||
# Disable buffering so the script is complete if Blender crashes.
|
||||
with (
|
||||
open(args.generate_script, "wb", buffering=0) if args.generate_script else
|
||||
contextlib.nullcontext()
|
||||
|
||||
) as fh:
|
||||
run_all(
|
||||
log_fn=(
|
||||
None if fh is None else
|
||||
(lambda text: fh.write(text.encode("utf-8")))
|
||||
),
|
||||
use_random=args.random,
|
||||
random_reset=args.random_reset,
|
||||
random_multiply=args.random_multiply,
|
||||
random_screen=args.random_screen,
|
||||
blend_files=blend_files,
|
||||
)
|
||||
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# for i in range(200):
|
||||
# RANDOM_SEED[0] += 1
|
||||
# main()
|
||||
main()
|
||||
sys.exit(main())
|
||||
|
||||
Reference in New Issue
Block a user