You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.0 KiB
68 lines
2.0 KiB
2 years ago
|
import json
|
||
|
|
import argparse
|
||
|
|
import pathlib
|
||
|
|
|
||
|
|
# import igl
|
||
|
|
import pymesh
|
||
|
|
|
||
|
|
import fixture_utils
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser(description='Component splitter')
|
||
|
|
parser.add_argument('--input', type=pathlib.Path, help='input path')
|
||
|
|
parser.add_argument('--rho', '--density', type=float,
|
||
|
|
default=1000, help='density')
|
||
|
|
parser.add_argument('--output-path', type=pathlib.Path,
|
||
|
|
default='', help='output path')
|
||
|
|
parser.add_argument('--output-json', type=pathlib.Path,
|
||
|
|
default='out.json', help='output json file')
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
mesh_name = args.input.stem
|
||
|
|
|
||
|
|
# in_path = os.path.dirname(args.input)
|
||
|
|
# mesh = os.path.basename(args.input)
|
||
|
|
#
|
||
|
|
# V, F = igl.read_triangle_mesh(args.input)
|
||
|
|
# C = igl.face_components(F)
|
||
|
|
# c_idx = C.max()
|
||
|
|
#
|
||
|
|
# Fs = [F[C == c] for c in range(c_idx + 1)]
|
||
|
|
#
|
||
|
|
# FFs = []
|
||
|
|
# VVs = []
|
||
|
|
#
|
||
|
|
# for i in range(len(Fs)):
|
||
|
|
# VV, FF, _, _ = igl.remove_unreferenced(V, Fs[i])
|
||
|
|
# FFs.append(FF)
|
||
|
|
# VVs.append(VV)
|
||
|
|
#
|
||
|
|
# paths = []
|
||
|
|
# for i in range(len(Fs)):
|
||
|
|
# path = os.path.join(args.output_path, "{}_part_{}.obj".format(mesh, i))
|
||
|
|
# igl.write_triangle_mesh(path, VVs[i], FFs[i])
|
||
|
|
#
|
||
|
|
# paths.append(path)
|
||
|
|
|
||
|
|
mesh = pymesh.load_mesh(str(args.input))
|
||
|
|
|
||
|
|
paths = []
|
||
|
|
for i, component_mesh in enumerate(pymesh.separate_mesh(mesh)):
|
||
|
|
path = args.output_path / f"{mesh_name}-part{i:03d}{args.input.suffix}"
|
||
|
|
pymesh.save_mesh(str(path), component_mesh)
|
||
|
|
paths.append(path)
|
||
|
|
|
||
|
|
out = {
|
||
|
|
"scene_type": "distance_barrier_rb_problem",
|
||
|
|
"max_time": 10,
|
||
|
|
"timestep": 0.01,
|
||
|
|
"rigid_body_problem": {
|
||
|
|
"rigid_bodies": []
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
out["rigid_body_problem"]["rigid_bodies"] = [
|
||
|
|
{"mesh": str(path), "density": args.rho} for path in paths]
|
||
|
|
|
||
|
|
fixture_utils.save_fixture(out, args.output_json)
|