OpenPunk/scripts
1
0
Fork 0
mirror of https://github.com/OpenFusionProject/scripts.git synced 2024-09-28 22:48:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix vertex positioning and outfile name

Browse Source
This commit is contained in:
gsemaj 2023-07-20 14:04:00 -04:00
parent 3aabb35f33
commit ef8ae1fb6e
No known key found for this signature in database
GPG Key ID: 24B96BAA40497929
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
terrain_mesh_extractor/ExtractTerrainMeshes.py
View File

@ -16,9 +16,12 @@ def rip_terrain_mesh(f, outpath):
terrainData = dong.objects[k].read() terrainData = dong.objects[k].read()
terrain_width = terrainData['m_Heightmap']['m_Width'] - 1 terrain_width = terrainData['m_Heightmap']['m_Width'] - 1
terrain_height = terrainData['m_Heightmap']['m_Height'] - 1 terrain_height = terrainData['m_Heightmap']['m_Height'] - 1
scale_x = terrainData['m_Heightmap']['m_Scale']['x']
scale_z = terrainData['m_Heightmap']['m_Scale']['z']
scale_y = terrainData['m_Heightmap']['m_Scale']['y']
# create the terrain # create the terrain
bpy.ops.mesh.primitive_grid_add(x_subdivisions=terrain_width, y_subdivisions=terrain_height, size=40, enter_editmode=True, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1)) bpy.ops.mesh.primitive_grid_add(x_subdivisions=terrain_width, y_subdivisions=terrain_height, size=128, enter_editmode=True, align='WORLD', location=(0, 0, 0), scale=(1, 1, 1))
context = bpy.context context = bpy.context
grid = context.edit_object grid = context.edit_object
@ -32,8 +35,16 @@ def rip_terrain_mesh(f, outpath):
bm = bmesh.from_edit_mesh(context.edit_object.data) bm = bmesh.from_edit_mesh(context.edit_object.data)
bm.verts.ensure_lookup_table() bm.verts.ensure_lookup_table()
for index, height in enumerate(terrainData['m_Heightmap']['m_Heights']): for index, height in enumerate(terrainData['m_Heightmap']['m_Heights']):
height = height / terrainData['m_Heightmap']['m_Scale']['y'] # scale height
bm.verts[index].co.z = height height_norm = height / (2 ** 15 - 2)
bm.verts[index].co.z = height_norm * scale_y
# pivot and scale x
bm.verts[index].co.x += terrain_width / 2
bm.verts[index].co.x *= scale_x
# pivot and scale z
bm.verts[index].co.y += terrain_height / 2
bm.verts[index].co.y *= scale_z
#print(f"{bm.verts[index].co.x}, {bm.verts[index].co.y}, {bm.verts[index].co.z}")
indices = [] indices = []
shift_amt = abs(bm.verts[0].co.x - bm.verts[1].co.x) shift_amt = abs(bm.verts[0].co.x - bm.verts[1].co.x)
@ -52,11 +63,17 @@ def rip_terrain_mesh(f, outpath):
if flags & 0b0001: # -Y if flags & 0b0001: # -Y
v.co.y -= shift_amt v.co.y -= shift_amt
# flip to correct orientation # flip diagonally
for v in bm.verts:
tmp = v.co.x
v.co.x = v.co.y
v.co.y = tmp
# export
bpy.ops.object.mode_set(mode="OBJECT") bpy.ops.object.mode_set(mode="OBJECT")
bpy.ops.object.select_all(action='SELECT') bpy.ops.object.select_all(action='SELECT')
bpy.ops.transform.mirror(orient_type='GLOBAL', orient_matrix=((1, 0, 0), (0, 1, 0), (0, 0, 1)), orient_matrix_type='GLOBAL', constraint_axis=(False, True, False)) name = terrainData['m_Name']
outfile = f"{k}.fbx" outfile = f"{name}.fbx"
bpy.ops.export_scene.fbx(filepath=os.path.join(outpath, outfile)) bpy.ops.export_scene.fbx(filepath=os.path.join(outpath, outfile))
# select modified vertices # select modified vertices