def write_tileset(in_folder, out_folder, octree_metadata, offset, scale, projection, rotation_matrix, include_rgb):
# compute tile transform matrix
if rotation_matrix is None:
transform = np.identity(4)
else:
transform = inverse_matrix(rotation_matrix)
transform = np.dot(transform, scale_matrix(1.0 / scale[0]))
transform = np.dot(translation_matrix(offset), transform)
# build fake points
if True:
root_node = Node('', octree_metadata.aabb, octree_metadata.spacing * 2)
root_node.children = []
inv_aabb_size = (1.0 / (octree_metadata.aabb[1] - octree_metadata.aabb[0])).astype(np.float32)
for child in ['0', '1', '2', '3', '4', '5', '6', '7']:
ondisk_tile = name_to_filename(out_folder, child.encode('ascii'), '.pnts')
if os.path.exists(ondisk_tile):
tile_content = TileContentReader.read_file(ondisk_tile)
fth = tile_content.body.feature_table.header
xyz = tile_content.body.feature_table.body.positions_arr.view(np.float32).reshape((fth.points_length, 3))
if include_rgb:
rgb = tile_content.body.feature_table.body.colors_arr.reshape((fth.points_length, 3))
else:
rgb = np.zeros(xyz.shape, dtype=np.uint8)
root_node.grid.insert(
octree_metadata.aabb[0].astype(np.float32),
inv_aabb_size,
xyz.copy(),
rgb)
pnts_writer.node_to_pnts(''.encode('ascii'), root_node, out_folder, include_rgb)
executor = concurrent.futures.ProcessPoolExecutor()
root_tileset = Node.to_tileset(executor, ''.encode('ascii'), octree_metadata.aabb, octree_metadata.spacing, out_folder, scale)
executor.shutdown()
root_tileset['transform'] = transform.T.reshape(16).tolist()
root_tileset['refine'] = 'REPLACE'
for child in root_tileset['children']:
child['refine'] = 'ADD'
tileset = {
'asset': {
'version': '1.0',
},
'geometricError': np.linalg.norm(
octree_metadata.aabb[1] - octree_metadata.aabb[0]) / scale[0],
'root': root_tileset,
}
with open('{}/tileset.json'.format(out_folder), 'w') as f:
f.write(json.dumps(tileset))
def main(args):
dest = '{}/tileset.json'.format(args.folder)
if os.path.exists(dest):
if args.overwrite:
remove_tileset(dest)
else:
print(
'Destination tileset {} already exists. Aborting'.format(dest))
sys.exit(1)
tilesets = []
for root, dirs, files in os.walk(args.folder):
t = ['{}/{}'.format(root, f) for f in files if f == 'tileset.json']
if t:
tilesets += t
if args.verbose >= 1:
print('Found {} tilesets to merge'.format(len(tilesets)))
if args.verbose >= 2:
print('Tilesets:', tilesets)
infos = init(tilesets)
aabb = infos['aabb']
base_transform = infos['transforms'][0]
inv_base_transform = inverse_matrix(base_transform)
print('------------------------')
# build hierarchical structure
result = build_tileset_quadtree(args.folder, aabb, infos['tilesets'],
base_transform, inv_base_transform, '')
result['transform'] = base_transform.T.reshape(16).tolist()
tileset = {
'asset': {
'version': '1.0',
'gltfUpAxis': 'Z',
},
'refine': 'REPLACE',
'geometricError': np.linalg.norm((aabb[1] - aabb[0])[0:3]),
'root': result
}
with open('{}/tileset.json'.format(args.folder), 'w') as f:
json.dump(tileset, f)
