# Copyright 2014 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.error
import its.target
import sys
import os
import os.path

# Change this to True, to have the test break at the first failure.
stop_at_first_failure = False

def main():
    """Test different combinations of output formats.
    """
    NAME = os.path.basename(__file__).split(".")[0]

    with its.device.ItsSession() as cam:

        props = cam.get_camera_properties()
        its.caps.skip_unless(its.caps.compute_target_exposure(props) and
                             its.caps.raw16(props))

        successes = []
        failures = []

        # Two different requests: auto, and manual.
        e, s = its.target.get_target_exposure_combos(cam)["midExposureTime"]
        req_aut = its.objects.auto_capture_request()
        req_man = its.objects.manual_capture_request(s, e)
        reqs = [req_aut, # R0
                req_man] # R1

        # 10 different combos of output formats; some are single surfaces, and
        # some are multiple surfaces.
        wyuv,hyuv = its.objects.get_available_output_sizes("yuv", props)[-1]
        wjpg,hjpg = its.objects.get_available_output_sizes("jpg", props)[-1]
        fmt_yuv_prev = {"format":"yuv", "width":wyuv, "height":hyuv}
        fmt_yuv_full = {"format":"yuv"}
        fmt_jpg_prev = {"format":"jpeg","width":wjpg, "height":hjpg}
        fmt_jpg_full = {"format":"jpeg"}
        fmt_raw_full = {"format":"raw"}
        fmt_combos =[
                [fmt_yuv_prev],                             # F0
                [fmt_yuv_full],                             # F1
                [fmt_jpg_prev],                             # F2
                [fmt_jpg_full],                             # F3
                [fmt_raw_full],                             # F4
                [fmt_yuv_prev, fmt_jpg_prev],               # F5
                [fmt_yuv_prev, fmt_jpg_full],               # F6
                [fmt_yuv_prev, fmt_raw_full],               # F7
                [fmt_yuv_prev, fmt_jpg_prev, fmt_raw_full], # F8
                [fmt_yuv_prev, fmt_jpg_full, fmt_raw_full]] # F9

        # Two different burst lengths: single frame, and 3 frames.
        burst_lens = [1, # B0
                      3] # B1

        # There are 2x10x2=40 different combinations. Run through them all.
        n = 0
        for r,req in enumerate(reqs):
            for f,fmt_combo in enumerate(fmt_combos):
                for b,burst_len in enumerate(burst_lens):
                    try:
                        caps = cam.do_capture([req]*burst_len, fmt_combo)
                        successes.append((n,r,f,b))
                        print "==> Success[%02d]: R%d F%d B%d" % (n,r,f,b)

                        # Dump the captures out to jpegs.
                        if not isinstance(caps, list):
                            caps = [caps]
                        elif isinstance(caps[0], list):
                            caps = sum(caps, [])
                        for c,cap in enumerate(caps):
                            img = its.image.convert_capture_to_rgb_image(cap,
                                    props=props)
                            its.image.write_image(img,
                                    "%s_n%02d_r%d_f%d_b%d_c%d.jpg"%(NAME,n,r,f,b,c))

                    except Exception as e:
                        print e
                        print "==> Failure[%02d]: R%d F%d B%d" % (n,r,f,b)
                        failures.append((n,r,f,b))
                        if stop_at_first_failure:
                            sys.exit(0)
                    n += 1

        num_fail = len(failures)
        num_success = len(successes)
        num_total = len(reqs)*len(fmt_combos)*len(burst_lens)
        num_not_run = num_total - num_success - num_fail

        print "\nFailures (%d / %d):" % (num_fail, num_total)
        for (n,r,f,b) in failures:
            print "  %02d: R%d F%d B%d" % (n,r,f,b)
        print "\nSuccesses (%d / %d):" % (num_success, num_total)
        for (n,r,f,b) in successes:
            print "  %02d: R%d F%d B%d" % (n,r,f,b)
        if num_not_run > 0:
            print "\nNumber of tests not run: %d / %d" % (num_not_run, num_total)
        print ""

        # The test passes if all the combinations successfully capture.
        assert(num_fail == 0)
        assert(num_success == num_total)

if __name__ == '__main__':
    main()