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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
import sys
import os, os.path
import traceback
import time
import argparse
try:
import collada
except:
sys.exit("Could not find pycollada library.")
def main():
parser = argparse.ArgumentParser(
description='Recursively scans a directory, loading any .dae file found.')
parser.add_argument('directory', help='Directory to scan')
parser.add_argument('--show-time', '-t', default=False, action='store_true',
help='Show how much time (in seconds) it took to load file')
parser.add_argument('--show-warnings', '-w', default=False, action='store_true',
help='If warnings present, print warning type')
parser.add_argument('--show-errors', '-e', default=False, action='store_true',
help='If errors present, print error and traceback')
parser.add_argument('--show-summary', '-s', default=False, action='store_true',
help='Print a summary at the end of how many files had warnings and errors')
parser.add_argument('--zip', '-z', default=False, action='store_true',
help='Include .zip files when searching for files to load')
args = parser.parse_args()
if not os.path.isdir(args.directory):
sys.exit("Given path '%s' is not a directory." % args.directory)
directories = [args.directory]
collada_files = []
while len(directories) > 0:
directory = directories.pop()
for name in os.listdir(directory):
fullpath = os.path.join(directory,name)
(root, ext) = os.path.splitext(fullpath)
if os.path.isfile(fullpath) and ext.lower() == ".dae":
collada_files.append(fullpath)
elif os.path.isfile(fullpath) and ext.lower() == ".zip":
collada_files.append(fullpath)
elif os.path.isdir(fullpath):
directories.append(fullpath)
collada_files.sort()
file_success_count = 0
file_warning_count = 0
file_error_count = 0
for c in collada_files:
(root, leaf) = os.path.split(c)
print "'%s'..." % leaf,
sys.stdout.flush()
start_time = time.time()
try:
col = collada.Collada(c, \
ignore=[collada.DaeUnsupportedError, collada.DaeBrokenRefError])
if len(col.errors) > 0:
print "WARNINGS:", len(col.errors)
file_warning_count += 1
err_names = [type(e).__name__ for e in col.errors]
unique = set(err_names)
type_cts = [(e, err_names.count(e)) for e in unique]
if args.show_warnings:
for e, ct in type_cts:
for err in col.errors:
if type(err).__name__ == e:
print " %s" % str(err)
break
if ct > 1:
print " %s: %d additional warnings of this type" % (e, ct-1)
else:
print "SUCCESS"
file_success_count += 1
#do some sanity checks looping through result
if not col.scene is None:
for geom in col.scene.objects('geometry'):
for prim in geom.primitives():
assert(len(prim) >= 0)
for cam in col.scene.objects('camera'):
assert(cam.original.id)
except (KeyboardInterrupt, SystemExit):
print
sys.exit("Keyboard interrupt. Exiting.")
except:
print "ERROR"
file_error_count += 1
if args.show_errors:
print
traceback.print_exc()
print
end_time = time.time()
if args.show_time:
print " Loaded in %.3f seconds" % (end_time-start_time)
if args.show_summary:
print
print
print "Summary"
print "======="
print "Files loaded successfully: %d" % file_success_count
print "Files with warnings: %d" % file_warning_count
print "Files with errors: %d" % file_error_count
if __name__ == "__main__":
main()
|
