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_image_array_rgba_png(self): im_arr = ImageArray(dummy_image(10.0, 4)) im_arr.write_png('standard.png') im_arr.write_png('non-scaled.png', rescale=False) im_arr.write_png('black_bg.png', background='black') im_arr.write_png('white_bg.png', background='white') im_arr.write_png('green_bg.png', background=[0., 1., 0., 1.]) im_arr.write_png('transparent_bg.png', background=None)
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)
def test_image_array_rgb_png(self): im_arr = ImageArray(dummy_image(10.0, 3)) im_arr.write_png('standard.png')
def test_image_array_rgba_png(self): im_arr = ImageArray(dummy_image(10.0, 4)) im_arr.write_png("standard") im_arr.write_png("non-scaled.png", rescale=False) im_arr.write_png("black_bg.png", background="black") im_arr.write_png("white_bg.png", background="white") im_arr.write_png("green_bg.png", background=[0.0, 1.0, 0.0, 1.0]) im_arr.write_png("transparent_bg.png", background=None) with warnings.catch_warnings(record=True) as w: # Cause all warnings to always be triggered. warnings.simplefilter("always") # Trigger a warning. im_arr.write_png("clipped.png", clip_ratio=0.5) assert str(w[0].message) == ( "'clip_ratio' keyword is deprecated. Use 'sigma_clip' instead" )