# import matplotlib.pyplot as plt
import numpy as np
import scipy.misc
import screen
def image_to_screen(imfile, brightness):
f = scipy.misc.imread(imfile, False, "RGB")
f = scipy.misc.imresize(f, (screen.Screen.W, screen.Screen.H))
# plt.imshow(f)
# plt.show()
s = screen.Screen(f.tolist(), brightness)
return s
def screen_to_image(screen, outf):
arr = np.array(screen.colordata, dtype=np.float64)
arr /= 255.
arr = scipy.misc.imresize(arr, (128, 128), "nearest")
scipy.misc.imsave(outf, arr, "png")