ile-de-test/addons/zylann.hterrain/tools/terrain_preview.gd
2023-10-05 20:02:23 +02:00

143 lines
4.4 KiB
GDScript

@tool
extends SubViewportContainer
const PREVIEW_MESH_LOD = 2
const HTerrainMesher = preload("../hterrain_mesher.gd")
const HT_Util = preload("../util/util.gd")
signal dragged(relative, button_mask)
@onready var _viewport : SubViewport = $Viewport
@onready var _mesh_instance : MeshInstance3D = $Viewport/MeshInstance
@onready var _camera : Camera3D = $Viewport/Camera
@onready var _light : DirectionalLight3D = $Viewport/DirectionalLight
# Use the simplest shader
var _shader : Shader = load("res://addons/zylann.hterrain/shaders/simple4_lite.gdshader")
var _yaw := 0.0
var _pitch := -PI / 6.0
var _distance := 0.0
var _default_distance := 0.0
var _sea_outline : MeshInstance3D = null
var _sea_plane : MeshInstance3D = null
var _mesh_resolution := 0
func _ready():
if _sea_outline == null:
var mesh := HT_Util.create_wirecube_mesh()
var mat2 := StandardMaterial3D.new()
mat2.shading_mode = BaseMaterial3D.SHADING_MODE_UNSHADED
mat2.albedo_color = Color(0, 0.5, 1)
mesh.surface_set_material(0, mat2)
_sea_outline = MeshInstance3D.new()
_sea_outline.mesh = mesh
_viewport.add_child(_sea_outline)
if _sea_plane == null:
var mesh := PlaneMesh.new()
mesh.size = Vector2(1, 1)
var mat2 := StandardMaterial3D.new()
mat2.shading_mode = BaseMaterial3D.SHADING_MODE_UNSHADED
mat2.albedo_color = Color(0, 0.5, 1, 0.5)
mat2.transparency = BaseMaterial3D.TRANSPARENCY_ALPHA
mesh.material = mat2
_sea_plane = MeshInstance3D.new()
_sea_plane.mesh = mesh
_sea_plane.hide()
_viewport.add_child(_sea_plane)
func setup(heights_texture: Texture2D, normals_texture: Texture2D):
var terrain_size := heights_texture.get_width()
var mesh_resolution := terrain_size / PREVIEW_MESH_LOD
if _mesh_resolution != mesh_resolution or not (_mesh_instance.mesh is ArrayMesh):
_mesh_resolution = mesh_resolution
var mesh := HTerrainMesher.make_flat_chunk(
_mesh_resolution, _mesh_resolution, PREVIEW_MESH_LOD, 0)
_mesh_instance.mesh = mesh
_default_distance = _mesh_instance.get_aabb().size.x
_distance = _default_distance
#_mesh_instance.translation -= 0.5 * Vector3(terrain_size, 0, terrain_size)
_update_camera()
var mat : ShaderMaterial = _mesh_instance.mesh.surface_get_material(0)
if mat == null:
mat = ShaderMaterial.new()
mat.shader = _shader
_mesh_instance.mesh.surface_set_material(0, mat)
mat.set_shader_parameter("u_terrain_heightmap", heights_texture)
mat.set_shader_parameter("u_terrain_normalmap", normals_texture)
mat.set_shader_parameter("u_terrain_inverse_transform", Transform3D())
mat.set_shader_parameter("u_terrain_normal_basis", Basis())
var aabb := _mesh_instance.get_aabb()
_sea_outline.scale = aabb.size
aabb = _mesh_instance.get_aabb()
_sea_plane.scale = Vector3(aabb.size.x, 1, aabb.size.z)
_sea_plane.position = Vector3(aabb.size.x, 0, aabb.size.z) / 2.0
func set_sea_visible(visible: bool):
_sea_plane.visible = visible
func set_shadows_enabled(enabled: bool):
_light.shadow_enabled = enabled
func _update_camera():
var aabb := _mesh_instance.get_aabb()
var target := aabb.position + 0.5 * aabb.size
var trans := Transform3D()
trans.basis = Basis(Quaternion(Vector3(0, 1, 0), _yaw) * Quaternion(Vector3(1, 0, 0), _pitch))
var back := trans.basis.z
trans.origin = target + back * _distance
_camera.transform = trans
func cleanup():
if _mesh_instance != null:
var mat : ShaderMaterial = _mesh_instance.mesh.surface_get_material(0)
assert(mat != null)
mat.set_shader_parameter("u_terrain_heightmap", null)
mat.set_shader_parameter("u_terrain_normalmap", null)
func _gui_input(event: InputEvent):
if HT_Util.is_in_edited_scene(self):
return
if event is InputEventMouseMotion:
if event.button_mask & MOUSE_BUTTON_MASK_MIDDLE:
var d : Vector2 = 0.01 * event.relative
_yaw -= d.x
_pitch -= d.y
_update_camera()
else:
var rel : Vector2 = 0.01 * event.relative
# Align dragging to view rotation
rel = -rel.rotated(-_yaw)
dragged.emit(rel, event.button_mask)
elif event is InputEventMouseButton:
if event.pressed:
var factor := 1.2
var max_factor := 10.0
var min_distance := _default_distance / max_factor
var max_distance := _default_distance
# Zoom in/out
if event.button_index == MOUSE_BUTTON_WHEEL_DOWN:
_distance = clampf(_distance * factor, min_distance, max_distance)
_update_camera()
elif event.button_index == MOUSE_BUTTON_WHEEL_UP:
_distance = clampf(_distance / factor, min_distance, max_distance)
_update_camera()