diff options
Diffstat (limited to 'apps/CameraITS/tests/inprog/test_blc_lsc.py')
-rw-r--r-- | apps/CameraITS/tests/inprog/test_blc_lsc.py | 106 |
1 files changed, 0 insertions, 106 deletions
diff --git a/apps/CameraITS/tests/inprog/test_blc_lsc.py b/apps/CameraITS/tests/inprog/test_blc_lsc.py deleted file mode 100644 index ce120a2..0000000 --- a/apps/CameraITS/tests/inprog/test_blc_lsc.py +++ /dev/null @@ -1,106 +0,0 @@ -# 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 pylab -import os.path -import matplotlib -import matplotlib.pyplot - -def main(): - """Test that BLC and LSC look reasonable. - """ - NAME = os.path.basename(__file__).split(".")[0] - - r_means_center = [] - g_means_center = [] - b_means_center = [] - r_means_corner = [] - g_means_corner = [] - b_means_corner = [] - - with its.device.ItsSession() as cam: - props = cam.get_camera_properties() - expt_range = props['android.sensor.info.exposureTimeRange'] - - # Get AE+AWB lock first, so the auto values in the capture result are - # populated properly. - r = [[0,0,1,1,1]] - ae_sen,ae_exp,awb_gains,awb_transform,_ \ - = cam.do_3a(r,r,r,do_af=False,get_results=True) - print "AE:", ae_sen, ae_exp / 1000000.0 - print "AWB:", awb_gains, awb_transform - - # Set analog gain (sensitivity) to 800 - ae_exp = ae_exp * ae_sen / 800 - ae_sen = 800 - - # Capture range of exposures from 1/100x to 4x of AE estimate. - exposures = [ae_exp*x/100.0 for x in [1]+range(10,401,40)] - exposures = [e for e in exposures - if e >= expt_range[0] and e <= expt_range[1]] - - # Convert the transform back to rational. - awb_transform_rat = its.objects.float_to_rational(awb_transform) - - # Linear tonemap - tmap = sum([[i/63.0,i/63.0] for i in range(64)], []) - - reqs = [] - for e in exposures: - req = its.objects.manual_capture_request(ae_sen,e) - req["android.tonemap.mode"] = 0 - req["android.tonemap.curveRed"] = tmap - req["android.tonemap.curveGreen"] = tmap - req["android.tonemap.curveBlue"] = tmap - req["android.colorCorrection.transform"] = awb_transform_rat - req["android.colorCorrection.gains"] = awb_gains - reqs.append(req) - - caps = cam.do_capture(reqs) - for i,cap in enumerate(caps): - img = its.image.convert_capture_to_rgb_image(cap) - its.image.write_image(img, "%s_i=%d.jpg"%(NAME, i)) - - tile_center = its.image.get_image_patch(img, 0.45, 0.45, 0.1, 0.1) - rgb_means = its.image.compute_image_means(tile_center) - r_means_center.append(rgb_means[0]) - g_means_center.append(rgb_means[1]) - b_means_center.append(rgb_means[2]) - - tile_corner = its.image.get_image_patch(img, 0.0, 0.0, 0.1, 0.1) - rgb_means = its.image.compute_image_means(tile_corner) - r_means_corner.append(rgb_means[0]) - g_means_corner.append(rgb_means[1]) - b_means_corner.append(rgb_means[2]) - - fig = matplotlib.pyplot.figure() - pylab.plot(exposures, r_means_center, 'r') - pylab.plot(exposures, g_means_center, 'g') - pylab.plot(exposures, b_means_center, 'b') - pylab.ylim([0,1]) - matplotlib.pyplot.savefig("%s_plot_means_center.png" % (NAME)) - - fig = matplotlib.pyplot.figure() - pylab.plot(exposures, r_means_corner, 'r') - pylab.plot(exposures, g_means_corner, 'g') - pylab.plot(exposures, b_means_corner, 'b') - pylab.ylim([0,1]) - matplotlib.pyplot.savefig("%s_plot_means_corner.png" % (NAME)) - -if __name__ == '__main__': - main() - |