# Copyright 2013 The Android Open Source Project # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import its.image import its.device import its.objects import sys import numpy import Image import pprint import math import pylab import os.path import matplotlib import matplotlib.pyplot def main(): """Test that the device will produce full black+white images. """ NAME = os.path.basename(__file__).split(".")[0] r_means = [] g_means = [] b_means = [] with its.device.ItsSession() as cam: props = cam.get_camera_properties() expt_range = props['android.sensor.info.exposureTimeRange'] sens_range = props['android.sensor.info.sensitivityRange'] # Take a shot with very low ISO and exposure time. Expect it to # be black. print "Black shot: sens = %d, exp time = %.4fms" % ( sens_range[0], expt_range[0]/1000000.0) req = its.objects.manual_capture_request(sens_range[0], expt_range[0]) cap = cam.do_capture(req) img = its.image.convert_capture_to_rgb_image(cap) its.image.write_image(img, "%s_black.jpg" % (NAME)) tile = its.image.get_image_patch(img, 0.45, 0.45, 0.1, 0.1) black_means = its.image.compute_image_means(tile) r_means.append(black_means[0]) g_means.append(black_means[1]) b_means.append(black_means[2]) print "Dark pixel means:", black_means # Take a shot with very high ISO and exposure time. Expect it to # be white. print "White shot: sens = %d, exp time = %.2fms" % ( sens_range[1], expt_range[1]/1000000.0) req = its.objects.manual_capture_request(sens_range[1], expt_range[1]) cap = cam.do_capture(req) img = its.image.convert_capture_to_rgb_image(cap) its.image.write_image(img, "%s_white.jpg" % (NAME)) tile = its.image.get_image_patch(img, 0.45, 0.45, 0.1, 0.1) white_means = its.image.compute_image_means(tile) r_means.append(white_means[0]) g_means.append(white_means[1]) b_means.append(white_means[2]) print "Bright pixel means:", white_means # Draw a plot. pylab.plot([0,1], r_means, 'r') pylab.plot([0,1], g_means, 'g') pylab.plot([0,1], b_means, 'b') pylab.ylim([0,1]) matplotlib.pyplot.savefig("%s_plot_means.png" % (NAME)) for val in black_means: assert(val < 0.025) for val in white_means: assert(val > 0.975) if __name__ == '__main__': main()