def test_image_array_background(self): im_arr = ImageArray(dummy_image(10.0, 4)) im_arr.rescale() new_im = im_arr.add_background_color([1., 0., 0., 1.], inline=False) new_im.write_png('red_bg.png') im_arr.add_background_color('black') im_arr.write_png('black_bg2.png')
def test_rgba_rescale(): im_arr = ImageArray(dummy_image(10.0, 4)) new_im = im_arr.rescale(inline=False) yield assert_equal, im_arr[:, :, :3].max(), 2 * 10. yield assert_equal, im_arr[:, :, 3].max(), 3 * 10. yield assert_equal, new_im[:, :, :3].sum(axis=2).max(), 1.0 yield assert_equal, new_im[:, :, 3].max(), 1.0 im_arr.rescale() yield assert_equal, im_arr[:, :, :3].sum(axis=2).max(), 1.0 yield assert_equal, im_arr[:, :, 3].max(), 1.0
def test_rgba_rescale(): im_arr = ImageArray(dummy_image(10.0, 4)) new_im = im_arr.rescale(inline=False) assert_equal(im_arr[:, :, :3].max(), 2 * 10.0) assert_equal(im_arr[:, :, 3].max(), 3 * 10.0) assert_equal(new_im[:, :, :3].sum(axis=2).max(), 1.0) assert_equal(new_im[:, :, 3].max(), 1.0) im_arr.rescale() assert_equal(im_arr[:, :, :3].sum(axis=2).max(), 1.0) assert_equal(im_arr[:, :, 3].max(), 1.0) im_arr.rescale(cmax=0.0, amax=0.0) assert_equal(im_arr[:, :, :3].sum(axis=2).max(), 1.0) assert_equal(im_arr[:, :, 3].max(), 1.0)
transfer_function=transfer_function.tf, ) print "ROTATION SET 3",mol images3 = ytcubes[mol].quick_render_movie('ChemMovie/', size=size, nframes=nframes, start_index=nframes*2, camera_angle=(0,0,1), rot_vector=(0.5,0.5,0), image_prefix=mol, transfer_function=transfer_function.tf, ) all_images[mol] = images1+images2+images3 molecules = ('HNC','HCOp','HCN','HC3N') from matplotlib import image from yt.data_objects.image_array import ImageArray combined = [np.sum([image.imread("ChemMovie/{mol}{ii:04d}.png".format(mol=mol, ii=ii))[:,:,:3] for mol in molecules], axis=0) for ii in range(nframes*3)] cmax = max(np.percentile(img[:, :, :3].sum(axis=2), 99.5) for img in combined) for ii, img in enumerate(combined): img = ImageArray(img) img = img.rescale(cmax=cmax).swapaxes(0,1) img.write_png("ChemMovie/combined_%04i.png" % (ii), rescale=False) import spectral_cube.ytcube spectral_cube.ytcube._make_movie('ChemMovie', prefix='combined_', filename='combined_%i.mp4' % size)