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)
