OverviewFilesCommitsIssuesDocsReleasesPeers

godot-witchy_world_demo / addons / scatter / src / modifiers / randomize_rotation.gd

71 lines · gd ref:
tool
extends "base_modifier.gd"


export var override_global_seed := false
export var custom_seed := 0
export var local_space := false
export var rotation := Vector3(360.0, 360.0, 360.0)
export var snap_angle := Vector3.ZERO

var _rng: RandomNumberGenerator


func _init() -> void:
	display_name = "Randomize Rotation"
	category = "Edit"


func _process_transforms(transforms, global_seed) -> void:
	_rng = RandomNumberGenerator.new()

	if override_global_seed:
		_rng.set_seed(custom_seed)
	else:
		_rng.set_seed(global_seed)

	var t: Transform
	var b: Basis

	var gt: Transform = transforms.path.get_global_transform()
	var gb: Basis = gt.basis
	var global_x: Vector3 = gb.xform_inv(Vector3.RIGHT).normalized()
	var global_y: Vector3 = gb.xform_inv(Vector3.UP).normalized()
	var global_z: Vector3 = gb.xform_inv(Vector3.DOWN).normalized()

	for i in transforms.list.size():
		t = transforms.list[i]
		b = t.basis

		if local_space:
			b = b.rotated(t.basis.x.normalized(), _random_angle(rotation.x, snap_angle.x))
			b = b.rotated(t.basis.y.normalized(), _random_angle(rotation.y, snap_angle.y))
			b = b.rotated(t.basis.z.normalized(), _random_angle(rotation.z, snap_angle.z))

		else:
			b = b.rotated(global_x, _random_angle(rotation.x, snap_angle.x))
			b = b.rotated(global_y, _random_angle(rotation.y, snap_angle.y))
			b = b.rotated(global_z, _random_angle(rotation.z, snap_angle.z))

		t.basis = b
		transforms.list[i] = t


func _random_vec3() -> Vector3:
	var vec3 = Vector3.ZERO
	vec3.x = _rng.randf_range(-1.0, 1.0)
	vec3.y = _rng.randf_range(-1.0, 1.0)
	vec3.z = _rng.randf_range(-1.0, 1.0)
	return vec3


func _random_angle(rot: float, snap: float) -> float:
	return deg2rad(stepify(_rng.randf_range(-1.0, 1.0) * rot, snap))


func _clamp_vector(vec3, vmin, vmax) -> Vector3:
	vec3.x = clamp(vec3.x, vmin.x, vmax.x)
	vec3.y = clamp(vec3.y, vmin.y, vmax.y)
	vec3.z = clamp(vec3.z, vmin.z, vmax.z)
	return vec3