#
#   Focusing application skeleton
#


import OBSUTILS as obsutil
import sys
import numpy as np
import pathlib as pl


def focussing_app(parser, init_seq_func, set_focus_func, get_image_func):
    """
    Focussing sequence standard application skeleton

        parser - argparse.ArgumentParser
        set_focus_func: a function of signature:
                        set_focus_func(foc_val)
        get_image_func: a function of signature:
                          get_image_func(image_filename, exp_time)
                        exp_time in seconds

        Returns:
            0 - OK
            non-zero - ERROR
    """

    parser.add_argument("focus_start", help="focus start value", type=float)
    parser.add_argument("focus_stop", help="focus stop value", type=float)
    parser.add_argument(
        "focus_step", help="focus step value", type=float, nargs="?", default=np.nan
    )

    parser.add_argument(
        "--guess",
        help="Use of two-steps focussing strategy. Input focus start and stop values are interpretated"
        + "as a guess range while step value is one for precise measurements.",
        action="store_true",
        default=False,
    )

    parser.add_argument(
        "-N",
        "--num",
        help="Number of points for precise stage measurement. Default is 7. Ignored if no '--guess'.",
        type=np.uint64,
        default=7,
    )

    parser.add_argument(
        "-v", "--verbose", help="Verbose output", action="store_true", default=False
    )

    parser = obsutil.getFocussingSequenceCmdlinePars(parser)

    args = parser.parse_args()

    if (args.num < 3) and (args.guess):
        if args.verbose:
            print("Number of focus values must be greater than 2! Abort!")
        return -1

    focus_start = float(args.focus_start)
    focus_stop = float(args.focus_stop)
    focus_step = float(args.focus_step)

    if np.isfinite(focus_step):
        if np.isclose(focus_step, 0.0):
            if args.verbose:
                print("Invalid (zero) focus step value! Abort!")
            return 1

        if np.isclose(focus_start, focus_stop):
            if args.verbose:
                print("Invalid focus range parameters (an empty range)! Abort!")
            return 1

        if (focus_stop - focus_start) * focus_step < 0:
            if args.verbose:
                print("Invalid focus range parameters! Abort!")
            return 1

        # if (focus_start > focus_stop) and (focus_step > 0):
        #     if args.verbose:
        #         print("Invalid focus range parameters! Abort!")
        #     return 1

        # if (focus_start < focus_stop) and (focus_step < 0):
        #     if args.verbose:
        #         print("Invalid focus range parameters! Abort!")
        #     return 1
    else:
        if args.guess:
            if args.verbose:
                print("If '--guess' is set then step value must be given! Abort!")
            return 1

    log = []

    seq_kwds = obsutil.parsFocussingSequenceCmdlinePars(args, log)

    if args.verbose:
        if len(log):
            for log_str in log:
                print(log_str)

    if args.verbose:
        print("START FOCUSSING SEQUENCE ...")

    if args.guess:  # rough focus estimation
        if args.verbose:
            print("\trough focus estimation ...")

        # add '_rough' suffixes
        seq_rough_kwds = seq_kwds.copy()
        if seq_rough_kwds["focus_curve_filename"] is not None:
            pt = pl.Path(seq_rough_kwds["focus_curve_filename"])
            seq_rough_kwds["focus_curve_filename"] = str(
                pt.with_stem(pt.stem + "_rough")
            )

        if seq_rough_kwds["result_data_filename"] is not None:
            pt = pl.Path(seq_rough_kwds["result_data_filename"])
            seq_rough_kwds["result_data_filename"] = str(
                pt.with_stem(pt.stem + "_rough")
            )

        # rough estimation: be sure that only a single
        # acquisition per focus value
        seq_rough_kwds["acq_per_value"] = 1

        result = obsutil.focussingSequence(
            [focus_start, focus_stop],
            init_seq_func,
            set_focus_func,
            get_image_func,
            None,
            sys.stdout,
            **seq_rough_kwds
        )

        if result["ret_code"] != 0:
            print("\n\tERROR: Cannot perform rought focus estimation! Abort!")
            print("\nFAIL!")
            return result["ret_code"]

        # compute precise range start and stop values
        r = (args.num - 1) * focus_step / 2.0
        focus_start = result["focus_value"] - r
        focus_stop = result["focus_value"] + r

        if args.verbose:
            print("\n\t\t rough focus position: {:g}".format(result["focus_value"]))
            print(
                "\t\testimate guess focus position in range of [{:g},{:g}]".format(
                    focus_start, focus_stop
                )
            )

    if np.isfinite(args.focus_step):
        result = obsutil.focussingSequence(
            [focus_start, focus_stop, focus_step],
            init_seq_func,
            set_focus_func,
            get_image_func,
            None,
            sys.stdout,
            **seq_kwds
        )
    else:
        result = obsutil.focussingSequence(
            [focus_start, focus_stop],
            init_seq_func,
            set_focus_func,
            get_image_func,
            None,
            sys.stdout,
            **seq_kwds
        )

    if result["ret_code"]:
        if args.verbose:
            print(
                "\tFocussing sequence returned error code: {:g}".format(
                    result["ret_code"]
                )
            )
            print("\tCannot set best focus value!")

        return result["ret_code"]

    if args.verbose:
        print("\n\tThe best focus value: {:g}".format(result["focus_value"]))

    ret_code = 0

    if not args.do_not_set:
        if args.verbose:
            print("\n\tSet focus to the best value ...", end="")

        ret_code = set_focus_func(result["focus_value"])

        if ret_code:
            if args.verbose:
                print("\tFAILED")
        else:
            if args.verbose:
                print("\tOK")

    if args.verbose:
        print("DONE.")

    return ret_code