1# Copyright 2018 The Android Open Source Project 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14 15import os.path 16 17import its.caps 18import its.device 19import its.image 20import its.objects 21 22NAME = os.path.basename(__file__).split('.')[0] 23RTOL_EXP_GAIN = 0.97 24TEST_EXP_RANGE = [1E6, 1E9] # ns [1ms, 1s] 25 26 27def main(): 28 """Test that the device will write/read correct exp/gain values.""" 29 30 with its.device.ItsSession() as cam: 31 props = cam.get_camera_properties() 32 its.caps.skip_unless(its.caps.manual_sensor(props) and 33 its.caps.per_frame_control(props)) 34 35 # determine capture format 36 debug = its.caps.debug_mode() 37 largest_yuv = its.objects.get_largest_yuv_format(props) 38 if debug: 39 fmt = largest_yuv 40 else: 41 match_ar = (largest_yuv['width'], largest_yuv['height']) 42 fmt = its.objects.get_smallest_yuv_format(props, match_ar=match_ar) 43 44 # grab exp/gain ranges from camera 45 sensor_exp_range = props['android.sensor.info.exposureTimeRange'] 46 sens_range = props['android.sensor.info.sensitivityRange'] 47 print 'sensor e range:', sensor_exp_range 48 print 'sensor s range:', sens_range 49 50 # determine if exposure test range is within sensor reported range 51 exp_range = [] 52 if sensor_exp_range[0] < TEST_EXP_RANGE[0]: 53 exp_range.append(TEST_EXP_RANGE[0]) 54 else: 55 exp_range.append(sensor_exp_range[0]) 56 if sensor_exp_range[1] > TEST_EXP_RANGE[1]: 57 exp_range.append(TEST_EXP_RANGE[1]) 58 else: 59 exp_range.append(sensor_exp_range[1]) 60 61 # build requests 62 reqs = [] 63 index_list = [] 64 for exp in exp_range: 65 for sens in sens_range: 66 reqs.append(its.objects.manual_capture_request(sens, exp)) 67 index_list.append((exp, sens)) 68 69 # take shots 70 caps = cam.do_capture(reqs, fmt) 71 72 # extract exp/sensitivity data 73 data = {} 74 for i, cap in enumerate(caps): 75 e_read = cap['metadata']['android.sensor.exposureTime'] 76 s_read = cap['metadata']['android.sensor.sensitivity'] 77 data[index_list[i]] = (e_read, s_read) 78 79 # check read/write match across all shots 80 e_failed = [] 81 s_failed = [] 82 for e_write in exp_range: 83 for s_write in sens_range: 84 (e_read, s_read) = data[(e_write, s_write)] 85 if e_write < e_read or e_read/float(e_write) <= RTOL_EXP_GAIN: 86 e_failed.append({'e_write': e_write, 87 'e_read': e_read, 88 's_write': s_write, 89 's_read': s_read}) 90 if s_write < s_read or s_read/float(s_write) <= RTOL_EXP_GAIN: 91 s_failed.append({'e_write': e_write, 92 'e_read': e_read, 93 's_write': s_write, 94 's_read': s_read}) 95 96 # print results 97 if e_failed: 98 print '\nFAILs for exposure time' 99 for fail in e_failed: 100 print ' e_write: %d, e_read: %d, RTOL: %.2f, ' % ( 101 fail['e_write'], fail['e_read'], RTOL_EXP_GAIN), 102 print 's_write: %d, s_read: %d, RTOL: %.2f' % ( 103 fail['s_write'], fail['s_read'], RTOL_EXP_GAIN) 104 if s_failed: 105 print 'FAILs for sensitivity(ISO)' 106 for fail in s_failed: 107 print 's_write: %d, s_read: %d, RTOL: %.2f, ' % ( 108 fail['s_write'], fail['s_read'], RTOL_EXP_GAIN), 109 print ' e_write: %d, e_read: %d, RTOL: %.2f' % ( 110 fail['e_write'], fail['e_read'], RTOL_EXP_GAIN) 111 112 # assert PASS/FAIL 113 assert not e_failed+s_failed 114 115 116if __name__ == '__main__': 117 main() 118