forked from Kitware/Danesfield
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ply2geon.py
50 lines (40 loc) · 1.56 KB
/
ply2geon.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/env python
###############################################################################
# Copyright Kitware Inc. and Contributors
# Distributed under the Apache License, 2.0 (apache.org/licenses/LICENSE-2.0)
# See accompanying Copyright.txt and LICENSE files for details
###############################################################################
import os
import sys
import time
import argparse
from danesfield.surface.geon import Geon
def main(args):
parser = argparse.ArgumentParser(
description='Generate OBJ file from PLY file.')
parser.add_argument('-p', '--ply_dir',
help='PLY file folder to read', required=True)
parser.add_argument('-d', '--dem',
help='DEM file name to read', required=True)
args = parser.parse_args(args)
if not os.path.isdir(args.ply_dir):
raise RuntimeError(
"Error: Failed to open PLY folder {}".format(args.ply_dir))
if not os.path.exists(args.dem):
raise RuntimeError("Error: Failed to open DEM {}".format(args.dem))
start_time = time.time()
m = Geon()
m.initialize(args.ply_dir, args.dem)
m.get_geons()
generate_model_time = time.time()
m.geons_to_json()
m.write_geonjson()
write_obj_time = time.time()
print(args.ply_dir + " completed!")
print("generate geons time: " + str(generate_model_time - start_time))
print("write geon json file time: " + str(write_obj_time - start_time))
if __name__ == "__main__":
try:
main(sys.argv[1:])
except Exception as e:
print(e)