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"
)
