def encpng(self, filename):
nim = Nimage.load(filename)
if len(nim.data.shape) == 3:
ys = nim.data[:,:,3]
else:
ys = nim.data
assert ys.shape == self.shape
self.hshape = [(n+1)//2 for n in self.shape]
us = 128 * np.ones(self.hshape, np.uint8)
vs = 128 * np.ones(self.hshape, np.uint8)
self.encframe((ys,us,vs))
self.drain()
self.framei += 1
def drain(self):
for i,(info, (ys,us,vs)) in enumerate(self.iterframes()):
nim = Nimage(Grid(None, shape=ys.shape), data=ys)
nim.save_raster(os.path.join(self.outdir, '%04d.png'%self.framei))
self.framei += 1
