Emmett Lalishandpca006132
954df798b2
* merging * mostly working * implemented, but crashes * runs serially * fix compile issues * fixed empty meshes * fix parallel error dumping * fixed parallelization * fix compiler * run black * improved triangulation * remove unreferenced verts * addressing feedback --------- Co-authored-by: pca006132 <john.lck40@gmail.com>
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
"""
|
|
Copyright 2022 The Manifold Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
https://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
import pathlib
|
|
import sys
|
|
import importlib
|
|
import trimesh
|
|
import numpy as np
|
|
from time import time
|
|
|
|
if __name__ == "__main__":
|
|
current_file = pathlib.Path(__file__)
|
|
current_dir = current_file.parent
|
|
files = [f.parts[-1][:-3] for f in current_dir.glob("*.py") if f != current_file]
|
|
|
|
export_models = len(sys.argv) == 2 and sys.argv[-1] == "-e"
|
|
|
|
for f in files:
|
|
module = importlib.import_module(f)
|
|
t0 = time()
|
|
model = module.run()
|
|
mesh = model.to_mesh()
|
|
if export_models:
|
|
if mesh.vert_properties.shape[1] > 3:
|
|
vertices = mesh.vert_properties[:, :3]
|
|
colors = (mesh.vert_properties[:, 3:] * 255).astype(np.uint8)
|
|
else:
|
|
vertices = mesh.vert_properties
|
|
colors = None
|
|
meshOut = trimesh.Trimesh(
|
|
vertices=vertices, faces=mesh.tri_verts, vertex_colors=colors
|
|
)
|
|
trimesh.exchange.export.export_mesh(meshOut, f"{f}.glb", "glb")
|
|
print(f"Exported model to {f}.glb")
|
|
t1 = time()
|
|
print(f"Took {(t1-t0)*1000:.1f}ms for {f}")
|