mirror of
https://github.com/blender/blender-addons.git
synced 2025-08-20 13:22:58 +00:00
e8da6131fd
Move copyright text to SPDX-FileCopyrightText or set to the Blender Foundation so "make check_licenses" now runs without warnings.
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
# SPDX-FileCopyrightText: 2011-2023 Blender Foundation
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
__all__ = (
|
|
"as_string",
|
|
)
|
|
|
|
import os
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Generic Utilities
|
|
|
|
def round_float_32(f):
|
|
from struct import pack, unpack
|
|
return unpack("f", pack("f", f))[0]
|
|
|
|
|
|
def repr_f32(f):
|
|
f_round = round_float_32(f)
|
|
f_str = repr(f)
|
|
f_str_frac = f_str.partition(".")[2]
|
|
if not f_str_frac:
|
|
return f_str
|
|
for i in range(1, len(f_str_frac)):
|
|
f_test = round(f, i)
|
|
f_test_round = round_float_32(f_test)
|
|
if f_test_round == f_round:
|
|
return "%.*f" % (i, f_test)
|
|
return f_str
|
|
|
|
|
|
def repr_pretty(v):
|
|
from pprint import pformat
|
|
# Float's are originally 32 bit, regular pretty print
|
|
# shows them with many decimal places (not so pretty).
|
|
if type(v) is float:
|
|
return repr_f32(v)
|
|
return pformat(v)
|
|
|
|
|
|
# -----------------------------------------------------------------------------
|
|
# Configuration Generation
|
|
|
|
def blend_list(path):
|
|
for dirpath, dirnames, filenames in os.walk(path):
|
|
# skip '.git'
|
|
dirnames[:] = [d for d in dirnames if not d.startswith(".")]
|
|
for filename in filenames:
|
|
if filename.lower().endswith(".blend"):
|
|
filepath = os.path.join(dirpath, filename)
|
|
yield filepath
|
|
|
|
|
|
def generate(dirpath, random_order, **kwargs):
|
|
files = list(blend_list(dirpath))
|
|
if random_order:
|
|
import random
|
|
random.shuffle(files)
|
|
else:
|
|
files.sort()
|
|
|
|
config = []
|
|
for f in files:
|
|
defaults = kwargs.copy()
|
|
defaults["file"] = f
|
|
config.append(defaults)
|
|
|
|
return config, dirpath
|
|
|
|
|
|
def as_string(dirpath, random_order, exit, **kwargs):
|
|
""" Config loader is in demo_mode.py
|
|
"""
|
|
cfg, dirpath = generate(dirpath, random_order, **kwargs)
|
|
|
|
# hint for reader, can be used if files are not found.
|
|
cfg_str = [
|
|
"# generated file\n",
|
|
"\n",
|
|
"# edit the search path so other systems may find the files below\n",
|
|
"# based on name only if the absolute paths cannot be found\n",
|
|
"# Use '//' for current blend file path.\n",
|
|
"\n",
|
|
"search_path = %r\n" % dirpath,
|
|
"\n",
|
|
"exit = %r\n" % exit,
|
|
"\n",
|
|
]
|
|
|
|
# Works, but custom formatting looks better.
|
|
# `cfg_str.append("config = %s" % pprint.pformat(cfg, indent=4, width=120))`.
|
|
|
|
def dict_as_kw(d):
|
|
return "dict(%s)" % ", ".join(("%s=%s" % (k, repr_pretty(v))) for k, v in sorted(d.items()))
|
|
|
|
ident = " "
|
|
cfg_str.append("config = [\n")
|
|
for cfg_item in cfg:
|
|
cfg_str.append("%s%s,\n" % (ident, dict_as_kw(cfg_item)))
|
|
cfg_str.append("%s]\n\n" % ident)
|
|
|
|
return "".join(cfg_str), len(cfg), dirpath
|