#filter
if args.index is not None:
print args.index
mask = np.zeros([len(vts_label)], dtype=np.bool )
for index in args.index:
if index == -1:
print "selected region :", index, "None"
else:
print "selected region :", index, label_name[index]
mask = np.logical_or(mask, (vts_label == index))
if args.inverse_mask:
mask = ~mask
vts_color[~mask] = [0,0,0]
if args.white:
vts_color[mask] = [255,255,255]
mesh.update_polydata()
mesh.set_colors(vts_color)
if args.v:
mesh.display()
if args.out_surface is not None:
mesh.save(args.o)
if args.out_vts_mask is not None:
np.save(args.out_vts_mask, mask)
