Renovating this code

This commit is contained in:
2025-11-05 19:58:51 -08:00
parent f7c9def9bf
commit 8964bb030b
4 changed files with 51 additions and 22 deletions

View File

@@ -27,16 +27,20 @@ class SpatialEnvelope:
exits_range: int
def sound_camera_spatial_envelope(scene: bpy.types.Scene, speaker_obj, considered_range: float) -> SpatialEnvelope:
def sound_camera_spatial_envelope(scene: bpy.types.Scene, speaker_obj,
considered_range: float) -> SpatialEnvelope:
min_dist = sys.float_info.max
min_dist_frame = scene.frame_start
enters_range_frame = None
exits_range_frame = None
assert scene.camera
in_range = False
for frame in range(scene.frame_start, scene.frame_end + 1):
scene.frame_set(frame)
rel = speaker_obj.matrix_world.to_translation() - scene.camera.matrix_world.to_translation()
rel = speaker_obj.matrix_world.to_translation() \
- scene.camera.matrix_world.to_translation()
dist = norm(rel)
if dist < considered_range and not in_range:
@@ -44,7 +48,7 @@ def sound_camera_spatial_envelope(scene: bpy.types.Scene, speaker_obj, considere
in_range = True
if dist < min_dist:
min_dist = dist
min_dist = float(dist)
min_dist_frame = frame
if dist > considered_range and in_range:
@@ -52,6 +56,9 @@ def sound_camera_spatial_envelope(scene: bpy.types.Scene, speaker_obj, considere
in_range = False
break
assert enters_range_frame
assert exits_range_frame
return SpatialEnvelope(considered_range=considered_range,
enters_range=enters_range_frame,
exits_range=exits_range_frame,
@@ -64,7 +71,8 @@ def closest_approach_to_camera(scene, speaker_object):
at_time = scene.frame_start
for frame in range(scene.frame_start, scene.frame_end + 1):
scene.frame_set(frame)
rel = speaker_object.matrix_world.to_translation() - scene.camera.matrix_world.to_translation()
rel = speaker_object.matrix_world.to_translation(
) - scene.camera.matrix_world.to_translation()
dist = norm(rel)
if dist < max_dist:
@@ -74,7 +82,7 @@ def closest_approach_to_camera(scene, speaker_object):
return (max_dist, at_time)
def track_speaker_to_camera(speaker, camera):
def track_speaker_to_camera(speaker):
camera_lock = speaker.constraints.new('TRACK_TO')
camera_lock.target = bpy.context.scene.camera
camera_lock.use_target_z = True
@@ -89,7 +97,8 @@ def spot_audio(context, speaker, trigger_mode, sync_peak, sound_peak, sound_leng
audio_scene_in = envelope.closest_range
elif trigger_mode == TriggerMode.RANDOM:
audio_scene_in = floor(uniform(context.scene.frame_start, context.scene.frame_end))
audio_scene_in = floor(
uniform(context.scene.frame_start, context.scene.frame_end))
elif trigger_mode == TriggerMode.RANDOM_GAUSSIAN:
mean = (context.scene.frame_end - context.scene.frame_start) / 2
audio_scene_in = floor(gauss(mean, gaussian_stddev))
@@ -127,11 +136,6 @@ def constrain_speaker_to_mesh(speaker_obj, mesh):
location_loc.target = mesh
location_loc.target = mesh
def apply_gain_envelope(speaker_obj, envelope):
pass
def add_speakers_to_meshes(meshes, context, sound=None,
sound_name_prefix=None,
sync_peak=False,
@@ -146,7 +150,8 @@ def add_speakers_to_meshes(meshes, context, sound=None,
print("object is not mesh")
continue
envelope = sound_camera_spatial_envelope(context.scene, mesh, considered_range=5.)
envelope = sound_camera_spatial_envelope(
context.scene, mesh, considered_range=5.)
speaker_obj = next((spk for spk in context.scene.objects
if spk.type == 'SPEAKER' and spk.constraints['Copy Location'].target == mesh), None)
@@ -156,7 +161,7 @@ def add_speakers_to_meshes(meshes, context, sound=None,
speaker_obj = context.selected_objects[0]
constrain_speaker_to_mesh(speaker_obj, mesh)
track_speaker_to_camera(speaker_obj, context.scene.camera)
track_speaker_to_camera(speaker_obj)
if sound_name_prefix is not None:
sound = sound_bank.random_sound()