Home | History | Annotate | Download | only in scene0
      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 
     15 import os.path
     16 
     17 import its.caps
     18 import its.device
     19 import its.image
     20 import its.objects
     21 
     22 NAME = os.path.basename(__file__).split('.')[0]
     23 RTOL_EXP_GAIN = 0.97
     24 TEST_EXP_RANGE = [1E6, 1E9]  # ns [1ms, 1s]
     25 
     26 
     27 def 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 
    116 if __name__ == '__main__':
    117     main()
    118