-
Notifications
You must be signed in to change notification settings - Fork 2
/
operators.py
550 lines (478 loc) · 22 KB
/
operators.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
import bpy
from bpy.types import Operator
from bpy.app import translations
from multiprocessing import cpu_count
def message_box(title="Message Box", message="", icon="INFO"):
def draw(self, context):
self.layout.label(text=message)
bpy.context.window_manager.popup_menu(
draw, title=translations.pgettext(title), icon=icon
)
class TOGGLE_LANGUAGE_OT_toggle_language(Operator):
bl_idname = "toggle_language.toggle_language"
bl_label = "Toggle"
bl_description = "Click button to toggle language"
def execute(self, context):
userpref = context.preferences
addonpref = userpref.addons[__package__].preferences
lang = translations.locale
if userpref.version[0] >= 4:
if addonpref.first_lang != addonpref.second_lang:
if lang == addonpref.first_lang:
userpref.view.language = addonpref.second_lang
else:
userpref.view.language = addonpref.first_lang
else:
message_box(
title="Fail to Toggle Language",
message="Two languages are same! Please select two different languages for addon.",
icon="ERROR",
)
else:
if addonpref.first_lang_before_v4 != addonpref.second_lang_before_v4:
if lang == addonpref.first_lang_before_v4:
userpref.view.language = addonpref.second_lang_before_v4
else:
userpref.view.language = addonpref.first_lang_before_v4
else:
message_box(
title="Fail to Toggle Language",
message="Two languages are same! Please select two different languages for addon.",
icon="ERROR",
)
# 检测并修正 use_translate_new_dataname 选项值。
scene = context.scene
lang = translations.locale
if lang != "en_US":
if (
userpref.view.use_translate_new_dataname
!= scene.toggle_language_settings.translate_new_dataname
):
userpref.view.use_translate_new_dataname = (
scene.toggle_language_settings.translate_new_dataname
)
return {"FINISHED"}
class TOGGLE_LANGUAGE_OT_use_default_hint_scheme(Operator):
bl_idname = "toggle_language.use_default_hint_scheme"
bl_label = "Default Mode"
bl_description = "Show no extra information"
def execute(self, context):
userpref = context.preferences
userpref.view.show_developer_ui = False
userpref.view.show_tooltips_python = False
self.report({"INFO"}, "Switch to default mode!")
return {"FINISHED"}
class TOGGLE_LANGUAGE_OT_use_developer_hint_scheme(Operator):
bl_idname = "toggle_language.use_developer_hint_scheme"
bl_label = "Developer Mode"
bl_description = "Show tooltips and options for developers"
def execute(self, context):
userpref = context.preferences
userpref.view.show_developer_ui = True
userpref.view.show_tooltips_python = True
self.report({"INFO"}, "Switch to developer mode!")
return {"FINISHED"}
class TOGGLE_LANGUAGE_OT_load_my_settings(Operator):
bl_idname = "toggle_language.load_my_settings"
bl_label = "Load My Settings"
bl_description = "Load my customized settings of startup file and preferences"
def execute(self, context):
scene = context.scene
userpref = context.preferences
addonpref = userpref.addons[__package__].preferences
if userpref.version[1] >= 90:
blender_v290 = True
else:
blender_v290 = False
if userpref.version[0] >= 3:
blender_v3_plus = True
if (
userpref.version[0] == 3 and userpref.version[1] >= 4
) or userpref.version[0] >= 4:
blender_v34_plus = True
if (
userpref.version[0] == 4 and userpref.version[1] >= 1
) or userpref.version[0] >= 5:
blender_v41_plus = True
if (
userpref.version[0] == 4
and userpref.version[1] >= 2
or userpref.version[0] >= 5
):
blender_v42_plus = True
else:
blender_v42_plus = False
else:
blender_v41_plus = False
else:
blender_v34_plus = False
else:
blender_v3_plus = False
# v2.93 及之后版本的文件命名有所变化。
if blender_v42_plus:
dict_blender_theme_name = {
"blender_dark": "Blender_Dark.xml",
"blender_light": "Blender_Light.xml",
"deep_grey": "theme_deep_grey",
"maya": "theme_maya",
"minimal_dark": "theme_minimal_dark",
"modo": "theme_modo",
"print_friendly": "theme_print_friendly",
"white": "theme_white",
"xsi": "theme_xsi",
}
blender_keyconfig_name = "Blender"
elif blender_v3_plus or userpref.version[1] == 93:
dict_blender_theme_name = {
"blender_dark": "Blender_Dark.xml",
"blender_light": "Blender_Light.xml",
"deep_grey": "Deep_Grey.xml",
"maya": "Maya.xml",
"minimal_dark": "Minimal_Dark.xml",
"modo": "Modo.xml",
"print_friendly": "Print_Friendly.xml",
"white": "White.xml",
"xsi": "XSI.xml",
}
blender_keyconfig_name = "Blender"
else:
dict_blender_theme_name = {
"blender_dark": "blender_dark.xml",
"blender_light": "blender_light.xml",
"deep_grey": "deep_grey.xml",
"maya": "maya.xml",
"minimal_dark": "minimal_dark.xml",
"modo": "modo.xml",
"print_friendly": "print_friendly.xml",
"white": "white.xml",
"xsi": "xsi.xml",
}
blender_keyconfig_name = "blender"
userpref.view.ui_scale = 1.3
if blender_v290 or blender_v3_plus:
userpref.view.show_statusbar_stats = True
userpref.view.show_statusbar_memory = True
if userpref.view.is_property_readonly("show_statusbar_vram"):
pass # 无显卡时,该属性为只读,无法对其进行写操作。
else:
userpref.view.show_statusbar_vram = True
else:
# v2.83 及之前版本未引入系统原生的 API,用 SDL 替代
userpref.system.audio_device = "SDL"
blender_theme_path_variable = {
"blender_dark": "",
"blender_light": "",
"deep_grey": "addons/",
"maya": "addons/",
"minimal_dark": "addons/",
"modo": "addons/",
"print_friendly": "addons/",
"white": "addons/",
"xsi": "addons/",
}
if addonpref.disable_theme_setting == False:
if (
blender_v42_plus == True
and addonpref.preset_theme != "blender_dark"
and addonpref.preset_theme != "blender_light"
):
bpy.ops.extensions.userpref_allow_online()
bpy.ops.extensions.repo_sync_all()
bpy.ops.extensions.package_install(
repo_index=0, pkg_id=dict_blender_theme_name[addonpref.preset_theme]
)
else:
user_platform = bpy.app.build_platform
execution_path = bpy.app.binary_path
theme_path = (
"{First_Main_Number}.{Second_Main_Number}/scripts/"
+ blender_theme_path_variable[addonpref.preset_theme]
+ "presets/interface_theme/"
+ dict_blender_theme_name[addonpref.preset_theme]
)
theme_path = theme_path.format(
First_Main_Number=userpref.version[0],
Second_Main_Number=userpref.version[1],
)
# b"Windows" b"Darwin" b"Linux"
if user_platform == b"Windows":
theme_path = execution_path.replace("blender.exe", theme_path)
elif user_platform == b"Darwin":
theme_path = execution_path.replace(
"MacOS/Blender", "Resources/" + theme_path
)
else:
theme_path = execution_path[:-7] + theme_path
bpy.ops.script.execute_preset(
filepath=theme_path,
menu_idname="USERPREF_MT_interface_theme_presets",
)
else:
pass
bpy.ops.preferences.addon_enable(module="node_wrangler")
if blender_v42_plus:
bpy.ops.extensions.package_install(repo_index=0, pkg_id="cell_fracture")
bpy.ops.extensions.package_install(repo_index=0, pkg_id="icon_viewer")
else:
bpy.ops.preferences.addon_enable(module="object_fracture_cell")
bpy.ops.preferences.addon_enable(module="development_icon_get")
# v3.0 及之后版本用的是 cyclesX,其中的 Auto Tiles 功能是为了减少内存占用,
# 即渲染 tile 大小的图像数据并缓存进硬盘,渲染结束时再合并在一起。
# 与老版的 cycles 渲染调度逻辑不一样,因此 auto_tile_size 插件在 v3.0 中被移除。
if blender_v3_plus:
# TODO:根据当前可用内存自动设置合适大小(仍需查询资料或者查看源码确认显存是否会受影响)
# 设置平铺大小为4096px,避免渲染4k图像时导致分割
scene.cycles.tile_size = 4096
else:
bpy.ops.preferences.addon_enable(module="render_auto_tile_size")
userpref.inputs.use_rotate_around_active = True
userpref.inputs.use_zoom_to_mouse = True
kcpref = context.window_manager.keyconfigs[blender_keyconfig_name].preferences
kcpref.use_pie_click_drag = True
kcpref.use_v3d_shade_ex_pie = True
userpref.filepaths.use_file_compression = True
if addonpref.disable_paths_setting == False:
# userpref.filepaths.use_auto_save_temporary_files = False
userpref.filepaths.texture_directory = "H:/texture/"
# userpref.filepaths.temporary_directory = "E:/temp/"
# userpref.filepaths.render_cache_directory = "E:/temp/"
userpref.filepaths.render_output_directory = "D:/process/"
scene.render.filepath = "D:/process/"
else:
pass
userpref.edit.undo_steps = 256
# cycles渲染引擎设置
scene.render.engine = "CYCLES"
cpref = userpref.addons["cycles"].preferences
if blender_v3_plus:
cpref.refresh_devices() # 刷新设备。
else:
cpref.get_devices()
# 获取当前版本支持的设备类型,逐一设置以检测是否存在显卡。
for device_type in cpref.get_device_types(bpy.context):
try:
cpref.compute_device_type = device_type[0]
if cpref.has_active_device():
gpu_exist = True
# 当optix可用时优先选择。
if cpref.compute_device_type == "CUDA":
try:
cpref.compute_device_type = "OPTIX"
if cpref.has_active_device():
optix_exist = True
break
else:
optix_exist = False
cpref.compute_device_type = "CUDA"
except TypeError:
pass
break
else:
gpu_exist = False
cpref.compute_device_type = "NONE"
except TypeError:
pass
if gpu_exist:
if blender_v3_plus:
cpref.refresh_devices() # 刷新设备。
else:
cpref.get_devices()
for device in cpref.devices:
if device.type == "CPU":
device.use = addonpref.use_cpu_in_gpu_render_setting
else:
device.use = True
scene.cycles.device = "GPU"
else:
pass
if blender_v3_plus:
if not gpu_exist and blender_v34_plus:
scene.cycles.use_guiding = True
else:
scene.cycles.use_adaptive_sampling = True
scene.cycles.adaptive_threshold = 0.1
# render_auto_tile_size插件的tiles大小设置
scene.ats_settings.gpu_choice = "128"
# 渲染降噪设置
if blender_v290 or blender_v3_plus:
scene.cycles.use_denoising = True
scene.cycles.use_preview_denoising = True
scene.cycles.preview_denoising_input_passes = "RGB_ALBEDO_NORMAL"
# gpu_exist为false时,根据短路求值原理,不会执行后面的语句。因此即使optix_exist未赋值时直接引用,也不会报错“UnboundLocalError: local variable 'optix_exist' referenced before assignment”
if gpu_exist and optix_exist:
if blender_v41_plus:
scene.cycles.denoiser = "OPENIMAGEDENOISE"
scene.cycles.denoising_use_gpu = True
scene.cycles.preview_denoiser = "OPENIMAGEDENOISE"
scene.cycles.preview_denoising_use_gpu = True
scene.cycles.preview_denoising_prefilter = "ACCURATE"
else:
scene.cycles.denoiser = "OPTIX"
scene.cycles.preview_denoiser = "OPTIX"
elif gpu_exist and not optix_exist:
scene.cycles.denoiser = "OPENIMAGEDENOISE"
scene.cycles.preview_denoiser = "OPENIMAGEDENOISE"
scene.cycles.preview_denoising_prefilter = "ACCURATE"
if blender_v41_plus:
scene.cycles.denoising_use_gpu = True
scene.cycles.preview_denoising_use_gpu = True
elif blender_v290:
scene.cycles.denoiser = "NLM"
scene.cycles.preview_denoiser = "OPENIMAGEDENOISE"
else:
scene.cycles.denoiser = "OPENIMAGEDENOISE"
scene.cycles.preview_denoiser = "OPENIMAGEDENOISE"
scene.cycles.preview_denoising_prefilter = "ACCURATE"
else:
context.view_layer.cycles.use_denoising = True
if gpu_exist and optix_exist:
context.view_layer.cycles.use_optix_denoising = True
context.view_layer.cycles.denoising_optix_input_passes = (
"RGB_ALBEDO_NORMAL"
)
scene.cycles.preview_denoising = "OPTIX"
# cycles引擎采样设置
scene.cycles.samples = 250
scene.cycles.preview_samples = 1
# 其他优化渲染速度的设置
scene.render.use_persistent_data = True
scene.render.threads_mode = "FIXED"
scene.render.threads = max(1, cpu_count() - 2)
bpy.ops.file.autopack_toggle() # 自动打包资源,例如加载的外部纹理图片,避免路径改变后导致文件未找到
bpy.ops.wm.save_userpref()
if addonpref.disable_saving_startup_file == False:
bpy.ops.wm.save_homefile()
else:
pass
return {"FINISHED"}
def invoke(self, context, event):
return context.window_manager.invoke_confirm(self, event)
class TOGGLE_LANGUAGE_OT_load_factory_settings(Operator):
bl_idname = "toggle_language.load_factory_settings"
bl_label = "Load Factory Settings"
bl_description = "Load factory settings of startup file and preferences"
def execute(self, context):
bpy.ops.wm.read_factory_settings()
bpy.ops.wm.save_userpref()
bpy.ops.wm.save_homefile()
return {"FINISHED"}
def invoke(self, context, event):
return context.window_manager.invoke_confirm(self, event)
class TOGGLE_LANGUAGE_OT_delete_all_collections_and_objects(Operator):
bl_idname = "toggle_language.delete_all_collections_and_objects"
bl_label = "Delete All Collections and Objects"
bl_description = "Delete all collections and objects in current scene"
def execute(self, context):
scene = context.scene
# 删除 Scene Collection 子项的物体
for obj in scene.collection.objects:
# if obj.users != 0可以检测data数据引用次数,不为0就删除,避免出现删除错误:which still has 1 users (including 0 'extra' shallow users),实际测试无效
bpy.data.objects.remove(obj)
# 删除 Scene Collection 子项的集合,该操作也可直接删除子项集合中的物体
for collection in scene.collection.children:
bpy.data.collections.remove(collection)
# remove会一同删除项目本身及其子项,但object的data还存储在blend文件中,可通过垃圾回收orphans_purge清除
bpy.ops.outliner.orphans_purge(do_recursive=True)
self.report(
{"INFO"},
"Delete all collections and objects in current scene successfully!",
)
return {"FINISHED"}
def invoke(self, context, event):
return context.window_manager.invoke_confirm(self, event)
# TODO:可调参数,弹出窗口,待完善开发
# TODO:首次完成添加视频进度条后,弹出可调参数窗口
# TODO:旁边按钮添加一个调出控制窗口的按钮
# class TOGGLE_LANGUAGE_OT_adjust_video_progress_bar(Operator):
# bl_idname = "toggle_language.add_video_progress_bar"
# bl_label = ""
# def execute(self, context):
# return {"FINISHED"}
class TOGGLE_LANGUAGE_OT_add_video_progress_bar(Operator):
bl_idname = "toggle_language.add_video_progress_bar"
bl_label = "Add Video Progress Bar"
bl_description = "Add video progress bar depend on current scene settings"
def execute(self, context):
scene = context.scene
sequence_editor = scene.sequence_editor
sequences = sequence_editor.sequences
# 如果当前正在展开编辑meta片段就退出编辑,合闭meta片段元素,避免影响后续的选择创建meta片段
# meta_stack[-1]倒数第一个元素就是当前展开编辑的meta片段
if len(sequence_editor.meta_stack) > 0:
bpy.ops.sequencer.meta_toggle()
# 获取vse中没有片段最顶部的频道数字
top_empty_channel_number = 0
strips = sequence_editor.sequences_all
for strip in strips:
# 剔除meta子片段
if strip.parent_meta() == None:
if strip.channel > top_empty_channel_number:
top_empty_channel_number = strip.channel
# 获取当前场景的设置
start_frame = scene.frame_start
end_frame = scene.frame_end
if scene.toggle_language_settings.translate_new_dataname == False:
name_text_bottom = "video progress bar bottom mask"
name_text_roll = "video progress bar roll mask"
name_text_meta = "video progress bar mask"
else:
name_text_bottom = translations.pgettext("video progress bar bottom mask")
name_text_roll = translations.pgettext("video progress bar roll mask")
name_text_meta = translations.pgettext("video progress bar mask")
bottom_effect = sequences.new_effect(
name=name_text_bottom,
type="COLOR",
channel=top_empty_channel_number + 1,
frame_start=start_frame,
frame_end=end_frame,
)
roll_effect = sequences.new_effect(
name=name_text_roll,
type="COLOR",
channel=top_empty_channel_number + 2,
frame_start=start_frame,
frame_end=end_frame,
)
# 设置子进度条颜色
bottom_effect.color = (0.45, 0.45, 0.45)
roll_effect.color = (0.255, 0.255, 0.255)
# 设置滚动遮罩层的关键帧动画
date_path = "offset_x"
roll_effect.transform.keyframe_insert(date_path, frame=start_frame)
roll_effect.transform.offset_x = scene.render.resolution_x
roll_effect.transform.keyframe_insert(date_path, frame=end_frame)
bpy.ops.sequencer.select_all(action="DESELECT")
effect_list = [bottom_effect, roll_effect]
for effect in effect_list:
effect.select = True
bpy.ops.sequencer.meta_make()
active_strip = sequence_editor.active_strip
active_strip.name = name_text_meta
active_strip.channel = top_empty_channel_number + 1
# 裁切meta片段和设置透明度
crop_value = scene.render.resolution_y - 44
blend_alpha_value = 0.9
active_strip.crop.min_y = crop_value
active_strip.blend_alpha = blend_alpha_value
active_strip.select = False
# 强制刷新vse
bpy.ops.sequencer.refresh_all()
self.report({"INFO"}, "Add video progress bar successfully!")
return {"FINISHED"}
classes = (
TOGGLE_LANGUAGE_OT_toggle_language,
TOGGLE_LANGUAGE_OT_use_default_hint_scheme,
TOGGLE_LANGUAGE_OT_use_developer_hint_scheme,
TOGGLE_LANGUAGE_OT_load_my_settings,
TOGGLE_LANGUAGE_OT_load_factory_settings,
TOGGLE_LANGUAGE_OT_delete_all_collections_and_objects,
TOGGLE_LANGUAGE_OT_add_video_progress_bar,
)
def register():
from bpy.utils import register_class
for cls in classes:
register_class(cls)
def unregister():
from bpy.utils import unregister_class
for cls in classes:
unregister_class(cls)