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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
|
#!/usr/bin/env python
#--*-python-*-
import optparse
import sys
import time
import numpy as np
from chroma.cache import Cache
from chroma.loader import load_geometry_from_string
from chroma.log import logger, logging
from chroma.bvh import make_recursive_grid_bvh
from chroma.gpu import create_cuda_context
from chroma.gpu.bvh import optimize_layer
def parse_bvh_id(id_str):
'''Parse a BVH identifier given on the command line into
a name of geometry and an optional BVH name, assuming the form:
"geo_name:bvh_name"
If no bvh_name is given, it is assumed to be "default."
'''
bvh_name = 'default'
parts = id_str.split(':')
if len(parts) > 1 and len(parts[1]) > 0:
bvh_name = parts[1]
geo_name = parts[0]
return geo_name, bvh_name
def node_swap(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
if args[1].endswith('-'):
redo_all_above = True
opt_layer_index = int(args[1][:-1])
else:
redo_all_above = False
opt_layer_index = int(args[1])
mesh_hash = cache.get_geometry_hash(geo_name)
bvh = cache.load_bvh(mesh_hash, bvh_name)
if opt_layer_index < 1 or opt_layer_index > bvh.layer_count() - 1:
print 'Can only optimize node order in this tree for layers 1 through %d' % (bvh.layer_count() - 1)
return
if redo_all_above:
opt_range = range(2, opt_layer_index+1)
opt_range.reverse()
else:
opt_range = [opt_layer_index]
context = create_cuda_context()
for opt_layer_index in opt_range:
original_parent_area = bvh.get_layer(opt_layer_index-1).area_fixed()
print 'Optimizing layer %d through node swaps' % opt_layer_index
opt_layer = bvh.get_layer(opt_layer_index)
new_layer_nodes = optimize_layer(bvh.get_layer(opt_layer_index).nodes)
bvh.nodes[bvh.layer_bounds[opt_layer_index]:bvh.layer_bounds[opt_layer_index+1]] = new_layer_nodes
print 'Rebuilding tree...'
new_nodes = rebuild_tree(bvh, opt_layer_index)
bvh.nodes = new_nodes
print 'Original parent area (layer %d): %e' % (opt_layer_index, original_parent_area)
print 'New parent area (layer %d): %e' % (opt_layer_index, bvh.get_layer(opt_layer_index-1).area_fixed())
print 'Saving new BVH...'
context.pop()
print_stat(geo_name, bvh_name, mesh_hash, bvh)
cache.save_bvh(bvh, mesh_hash, bvh_name)
def create(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
degree = int(args[1])
mesh_hash = cache.get_geometry_hash(geo_name)
print 'Loading geometry (MD5=%s): %s' % (mesh_hash, geo_name)
geometry = cache.load_geometry(geo_name)
print 'Creating degree %d BVH...' % degree
start = time.time()
context = create_cuda_context()
bvh = make_recursive_grid_bvh(geometry.mesh, target_degree=degree)
context.pop()
logger.info('BVH generated in %1.1f seconds.' % (time.time() - start))
cache.save_bvh(bvh, mesh_hash, bvh_name)
def list_cmd(cache, args):
geo_name = args[0]
mesh_hash = cache.get_geometry_hash(geo_name)
bvh_list = cache.list_bvh(mesh_hash)
print 'BVHs for %s (MD5=%s):' % (geo_name, mesh_hash)
print '\n'.join(bvh_list)
def copy(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
target_bvh_name = args[1]
mesh_hash = cache.get_geometry_hash(geo_name)
bvh = cache.load_bvh(mesh_hash, bvh_name)
cache.save_bvh(bvh, mesh_hash, target_bvh_name)
def remove(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
mesh_hash = cache.get_geometry_hash(geo_name)
cache.remove_bvh(mesh_hash, bvh_name)
def stat(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
mesh_hash = cache.get_geometry_hash(geo_name)
bvh = cache.load_bvh(mesh_hash, bvh_name)
print_stat(geo_name, bvh_name, mesh_hash, bvh)
def area_hist(cache, args):
geo_name, bvh_name = parse_bvh_id(args[0])
mesh_hash = cache.get_geometry_hash(geo_name)
bvh = cache.load_bvh(mesh_hash, bvh_name)
print 'BVH %s:%s' % (geo_name, bvh_name)
print
if (args[1] == 'all'):
layer_ids = range(bvh.layer_count())
else:
layer_ids = [int(args[1])]
for layer_id in layer_ids:
layer = bvh.get_layer(layer_id)
areas = layer.areas_fixed()
areas = areas[areas > 0] # Cull dummy nodes
min_area = areas.min()
max_area = areas.max()
mean = areas.mean()
stddev = areas.std()
print 'Layer %d (%d nodes, %1.2e area):' % (layer_id, len(areas), areas.sum())
print ' min|max|avg|stddev: %1.2e|%1.2e|%1.2e|%1.2e' \
% (min_area, max_area, mean, stddev)
print ' stddev/avg: %4.2f' % (stddev/mean)
from chroma.rootimport import ROOT
h = ROOT.TH1D('layer%d'%layer_id, ';Area', 100, min_area, max_area+1)
for a in areas:
h.Fill(a)
h.Draw()
ROOT.gPad.Update()
raw_input('Hit enter to continue.')
def print_stat(geo_name, bvh_name, mesh_hash, bvh):
num_layers = bvh.layer_count()
print '[BVH] %s:%s (MD5=%s)' % (geo_name, bvh_name, mesh_hash)
print '-' * 72
print 'World origin: (%f,%f,%f)' % tuple(bvh.world_coords.world_origin)
print 'World scale factor: %f' % bvh.world_coords.world_scale
print 'Nodes: %d' % len(bvh)
print 'Layers:'
areas = []
for i in xrange(num_layers):
layer = bvh.get_layer(i)
if len(layer) == 1:
node_str = 'node, '
else:
node_str = 'nodes,'
area = layer.area_fixed()
areas.append(area)
print ' % 3d) % 10s %s area = % 9e' % \
(i, len(layer), node_str, area)
print 'Log sum area: %f' % np.log(areas).sum()
commands = {
'stat' : stat,
'create': create,
'list' : list_cmd,
'copy' : copy,
'remove' : remove,
'node_swap' : node_swap,
'hist' : area_hist,
}
def main():
logger.setLevel(logging.INFO)
parser = optparse.OptionParser('%prog <cmd> <options>')
parser.add_option('-c', '--cache', dest='cache',
default=None, help='Chroma cache directory')
options, args = parser.parse_args()
if len(args) < 1:
sys.exit(parser.format_help())
if options.cache is None:
cache = Cache()
else:
cache = Cache(options.cache)
cmd = args[0]
cmd_args = args[1:]
if cmd in commands:
commands[cmd](cache, cmd_args)
else:
print 'error: unknown cmd %s' % cmd
sys.exit(1)
if __name__ == '__main__':
main()
|
