aboutsummaryrefslogtreecommitdiff
path: root/apps/CameraITS/tests/scene1/test_linearity.py
diff options
context:
space:
mode:
Diffstat (limited to 'apps/CameraITS/tests/scene1/test_linearity.py')
-rw-r--r--apps/CameraITS/tests/scene1/test_linearity.py99
1 files changed, 0 insertions, 99 deletions
diff --git a/apps/CameraITS/tests/scene1/test_linearity.py b/apps/CameraITS/tests/scene1/test_linearity.py
deleted file mode 100644
index fed0324..0000000
--- a/apps/CameraITS/tests/scene1/test_linearity.py
+++ /dev/null
@@ -1,99 +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.caps
-import its.device
-import its.objects
-import its.target
-import numpy
-import math
-import pylab
-import os.path
-import matplotlib
-import matplotlib.pyplot
-
-def main():
- """Test that device processing can be inverted to linear pixels.
-
- Captures a sequence of shots with the device pointed at a uniform
- target. Attempts to invert all the ISP processing to get back to
- linear R,G,B pixel data.
- """
- NAME = os.path.basename(__file__).split(".")[0]
-
- RESIDUAL_THRESHOLD = 0.00005
-
- # The HAL3.2 spec requires that curves up to 64 control points in length
- # must be supported.
- L = 64
- LM1 = float(L-1)
-
- gamma_lut = numpy.array(
- sum([[i/LM1, math.pow(i/LM1, 1/2.2)] for i in xrange(L)], []))
- inv_gamma_lut = numpy.array(
- sum([[i/LM1, math.pow(i/LM1, 2.2)] for i in xrange(L)], []))
-
- with its.device.ItsSession() as cam:
- props = cam.get_camera_properties()
- if not its.caps.compute_target_exposure(props):
- print "Test skipped"
- return
-
- e,s = its.target.get_target_exposure_combos(cam)["midSensitivity"]
- s /= 2
- sens_range = props['android.sensor.info.sensitivityRange']
- sensitivities = [s*1.0/3.0, s*2.0/3.0, s, s*4.0/3.0, s*5.0/3.0]
- sensitivities = [s for s in sensitivities
- if s > sens_range[0] and s < sens_range[1]]
-
- req = its.objects.manual_capture_request(0, e)
- req["android.blackLevel.lock"] = True
- req["android.tonemap.mode"] = 0
- req["android.tonemap.curveRed"] = gamma_lut.tolist()
- req["android.tonemap.curveGreen"] = gamma_lut.tolist()
- req["android.tonemap.curveBlue"] = gamma_lut.tolist()
-
- r_means = []
- g_means = []
- b_means = []
-
- for sens in sensitivities:
- req["android.sensor.sensitivity"] = sens
- cap = cam.do_capture(req)
- img = its.image.convert_capture_to_rgb_image(cap)
- its.image.write_image(
- img, "%s_sens=%04d.jpg" % (NAME, sens))
- img = its.image.apply_lut_to_image(img, inv_gamma_lut[1::2] * LM1)
- tile = its.image.get_image_patch(img, 0.45, 0.45, 0.1, 0.1)
- rgb_means = its.image.compute_image_means(tile)
- r_means.append(rgb_means[0])
- g_means.append(rgb_means[1])
- b_means.append(rgb_means[2])
-
- pylab.plot(sensitivities, r_means, 'r')
- pylab.plot(sensitivities, g_means, 'g')
- pylab.plot(sensitivities, b_means, 'b')
- pylab.ylim([0,1])
- matplotlib.pyplot.savefig("%s_plot_means.png" % (NAME))
-
- # Check that each plot is actually linear.
- for means in [r_means, g_means, b_means]:
- line,residuals,_,_,_ = numpy.polyfit(range(5),means,1,full=True)
- print "Line: m=%f, b=%f, resid=%f"%(line[0], line[1], residuals[0])
- assert(residuals[0] < RESIDUAL_THRESHOLD)
-
-if __name__ == '__main__':
- main()
-