###########################################################################
#
# Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
#
# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
#
# @file    utils.py
#
# @author  (last)        LTTS 
# @date    (last)      15-Jan-2022
#
############################################################################


import names
import test
import object
import squish
from configuration import config
from builtins import int as pyInt


    
def check_if_object_is_within_the_container(obj=None, container=None):
    """
    check if an object is inside a container
        @param obj - child UI object
        @param container - container UI object
        @return boolean
    """
    container = squish.findObject(container)
    containerPos = container.mapToGlobal(squish.QPoint(0, 0))
    container_x, container_y = pyInt(containerPos.x), pyInt(containerPos.y)
    container_width, container_height = pyInt(container.width), pyInt(container.height)
    
    obj = squish.findObject(obj)
    objPos = obj.mapToGlobal(squish.QPoint(0, 0))
    obj_x, obj_y = pyInt(objPos.x), pyInt(objPos.y)
    obj_width, obj_height = pyInt(obj.width), pyInt(obj.height)
    
    if obj_x >= container_x and obj_y >= container_y:
        if (obj_x + obj_width) <= (container_x + container_width) and (obj_y + obj_height) <= (container_y + container_height):
            return True
        
    return False
    
    
def scroll_to_zone(zone=None, screen_object=None, direction = None):
    """
    scroll to the UI, if object is hidden
        @param zone - object to be find out.
        @param screen_object - object of the screen.
        @return boolean
    """
    counter = 0
    while counter <= 100:
        try:
            counter += 1
            squish.findObject(zone)
            squish.snooze(0.5)
            if check_if_object_is_within_the_container(obj=zone, container=screen_object):
                return True
            else:
                raise RuntimeError
        except RuntimeError:
            ScreenObj = squish.findObject(screen_object)
            screenHeight = pyInt(ScreenObj.height)
            screenWidth = pyInt(ScreenObj.width)
            if direction is None:
                squish.mouseWheel(ScreenObj, (screenWidth-100),
                107, 0, -(screenHeight-460), squish.Qt.NoModifier)
            else:
                squish.mouseWheel(ScreenObj, (screenWidth-100),
                -(screenHeight-700), 0, 200, squish.Qt.NoModifier)
                
    raise LookupError("zone object is not in view to the user after " + \
             "trying 100 times")    
    
 
def page_step_indicator_verification(pre_treatment_step, pre_treatment_items_object):
    """
    Method to verify the Page Step indicators on top of the screen which indicates the steps passed, current, remained]
    @param pre_treatment_step :(int) indicates the Current pre-treatment step 
    @param pre_treatment_items_object :(dictionary) pre_treatment bullet object
    @return N/A
    """
    test.startSection("Method to verify the Page Step indicators")
    for page in range(1, len(config.PRE_TREATMENT_SCREENS)):
        pre_treatment_items_object["occurrence"] = page
        bullet_children = object.children(squish.waitForObject(pre_treatment_items_object))
        bullet_circle_color = bullet_children[0].color.name 
        bullet_border_color = bullet_children[0].border.color.name
        step_title =  bullet_children[2]
        #To verify the step indicators of the completed pre treatment screens
        if page < pre_treatment_step:
            test.verify(squish.waitForObjectExists(pre_treatment_items_object).complete, str(config.PRE_TREATMENT_SCREENS[page]+" complete attribute should be True"))
            test.verify(not squish.waitForObjectExists(pre_treatment_items_object).current, str(config.PRE_TREATMENT_SCREENS[page]+" current attribute should be false"))
            test.compare(bullet_circle_color, config.COMPLETE_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet color hex should be "+str(config.COMPLETE_COLOR)))
            test.compare(bullet_border_color, config.COMPLETE_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet border color hex should be "+str(config.COMPLETE_COLOR))) 
            test.compare(step_title.color.name, config.ENABLED_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" title color hex should be "+str(config.ENABLED_COLOR)))
        #To verify the step indicators of the current pre treatment screen
        elif page == pre_treatment_step:
            test.verify(squish.waitForObjectExists(pre_treatment_items_object).current, str(config.PRE_TREATMENT_SCREENS[page]+" complete attribute should be True"))
            test.verify(not squish.waitForObjectExists(pre_treatment_items_object).complete, str(config.PRE_TREATMENT_SCREENS[page]+" current attribute should be false"))
            test.compare(bullet_circle_color,config.CURRENT_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet color hex should be "+str(config.CURRENT_COLOR)))
            test.compare(bullet_border_color,config.COMPLETE_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet border color hex should be "+str(config.COMPLETE_COLOR)))
            test.compare(step_title.color.name,config.ENABLED_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" title color hex should be "+str(config.ENABLED_COLOR)))
            test.verify(step_title.font.bold, str(config.PRE_TREATMENT_SCREENS[page]+" title font bold should be "+str(config.ENABLED_COLOR)))  
        #To verify the step indicators of the remaining pre-treatment screens   
        else:
            test.verify(not squish.waitForObjectExists(pre_treatment_items_object).current, str(config.PRE_TREATMENT_SCREENS[page]+" complete attribute should be false"))
            test.verify(not squish.waitForObjectExists(pre_treatment_items_object).complete, str(config.PRE_TREATMENT_SCREENS[page]+" current attribute should be false"))
            test.compare(step_title.color.name,config.INCOMPLETE_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" title color hex should be "+str(config.INCOMPLETE_COLOR)))
            test.compare(bullet_circle_color,config.CURRENT_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet circle color hex should be "+str(config.CURRENT_COLOR))) 
            test.compare(bullet_border_color,config.INCOMPLETE_COLOR, str(config.PRE_TREATMENT_SCREENS[page]+" bullet border color hex should be "+str(config.INCOMPLETE_COLOR)))     

    test.endSection()


#Methods for create custom treatment
def set_venous_ranges_min_val(ven_low):
    """
    Method to set the Venous range maximum value to user expected value
    @param ven_low - (int) user expected value
    """
    test.startSection("Set Venous range minimum value to {}".format(ven_low))
    ven_min = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous)
    ven_min = pyInt(ven_min.minimum)
    ven_max = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous)
    ven_max = pyInt(ven_max.maximum)
    low_handler_parent = object.parent(squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous))
    low_handler_children = object.children(low_handler_parent)
    low_handler = low_handler_children[-2]
    width = pyInt(low_handler.width) - 15
    height = pyInt(low_handler.height) - 10
    if ven_min == ven_low:
        test.passes("Venous range minimum is already set to {}".format(ven_low))
    elif ven_min < ven_low:
        while ven_min != ven_low:
            squish.mouseDrag(low_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
            ven_min += 10
            if ven_min == ven_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
                squish.mouseDrag(low_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Venous range minimum value cannot be moved beyond {}".format(ven_min))
                break
            else:
                continue
    elif ven_min > ven_low:
        while ven_min != ven_low:
            squish.mouseDrag(low_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
            ven_min -= 10
            # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if ven_min == ven_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(low_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Venous range minimum value cannot be moved beyond {}".format(ven_min))
                break
            else:
                continue
    # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
    if ven_min == ven_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   
        test.compare(low_handler_parent.minValue, ven_min,
                     "Venous range minimum value cannot be moved beyond {}".format(ven_min))
    else:
        test.compare(ven_min, ven_low, 
                     "Actual Venous range minimum value: {} is equal to Expected value: {}".format(ven_min, ven_low))
    test.endSection()


def set_venous_ranges_max_val(ven_high):
    """
    Method to set the Venous range maximum value to user expected value
    @param ven_high - (int) user expected value
    """
    test.startSection("Set Venous range maximum value to {}".format(ven_high))
    ven_max = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous)
    ven_max = pyInt(ven_max.maximum)
    ven_min = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous)
    ven_min = pyInt(ven_min.minimum)
    high_handler_parent = object.parent(squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Venous))
    high_handler_children = object.children(high_handler_parent)
    high_handler = high_handler_children[-1]
    width = pyInt(high_handler.width) - 15
    height = pyInt(high_handler.height) - 10
    if ven_max == ven_high:
        test.passes("Venous range maximum is already set to {}".format(ven_high))
    elif ven_max < ven_high:
        while ven_max != ven_high:
            squish.mouseDrag(high_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
            ven_max += 10
            # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if ven_max == ven_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(high_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Venous range maximum value cannot be moved beyond {}".format(ven_max))
                break
            else:
                continue
    elif ven_max > ven_high:
        while ven_max != ven_high:
            squish.mouseDrag(high_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
            ven_max -= 10
            # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if ven_max == ven_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(high_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Venous range maximum value cannot be moved beyond {}".format(ven_max))
                break
            else:
                continue
    # venous blood pressure low limit should be lower than the high limit by at least 30mmHg 
    if ven_max == ven_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
        test.compare(high_handler_parent.maxValue, ven_max,
                "Venous range maximum value cannot be moved beyond {}".format(ven_max))
    else:
        test.compare(ven_max, ven_high, 
                     "Actual Venous range maximum value: {} is equal to Expected value: {}".format(ven_max, ven_high))
    test.endSection()


def set_arterial_ranges_min_val(art_low):
    """
    Method to set the Arterial range maximum value to user expected value
    @param art_low - (int) user expected value
    """
    test.startSection("Set Arterial range minimum value to {}".format(art_low))
    arterial_min = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery)
    arterial_min = pyInt(arterial_min.minimum)
    arterial_max = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery)
    arterial_max = pyInt(arterial_max.maximum)
    low_handler_parent = object.parent(squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery))
    low_handler_children = object.children(low_handler_parent)
    low_handler = low_handler_children[-2]
    width = pyInt(low_handler.width) - 8
    height = pyInt(low_handler.height)- 10
    if arterial_min == art_low:
        test.passes("Arterial range minimum is already set to {}".format(art_low))
    elif arterial_min < art_low:
        while arterial_min != art_low:
            squish.mouseDrag(low_handler, width, height, 1, 0, squish.Qt.NoModifier, squish.Qt.LeftButton)
            arterial_min += 10
            # arterial blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if arterial_min == arterial_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(low_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Arterial range minimum value cannot be moved beyond {}".format(arterial_min))
                break
            else:
                continue
    elif arterial_min > art_low:
        while arterial_min != art_low:
            squish.mouseDrag(low_handler, width, height, -1, 0, squish.Qt.NoModifier, squish.Qt.LeftButton)
            arterial_min -= 10
            # arterial blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if arterial_min == arterial_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(low_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Arterial range minimum value cannot be moved beyond {}".format(arterial_min))
                break
            else:
                continue
    # arterial blood pressure low limit should be lower than the high limit by atleast 30mmHg 
    if arterial_min == arterial_max - config.BUFFER_LOW_AND_HIGH_LIMITS:   
        test.compare(low_handler_parent.minValue, arterial_min,
                "Arterial range minimum value cannot be moved beyond {}".format(arterial_min))
    else:
        test.compare(arterial_min, art_low, 
            "Actual Arterial range minimum value: {} is equal to Expected value: {}".format(arterial_min, art_low))
    test.endSection()


def set_arterial_ranges_max_val(art_high):
    """
    Method to set the Arterial range maximum value to user expected value
    @param art_high - (int) user expected value
    """
    test.startSection("Set Arterial range maximum value to {}".format(art_high))
    arterial_max = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery)
    arterial_max = pyInt(arterial_max.maximum)
    arterial_min = squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery)
    arterial_min = pyInt(arterial_min.minimum)
    high_handler_parent = object.parent(squish.waitForObjectExists(names.o_PreTreatmentCreate_rangeRect_RangeRect_Artery))
    high_handler_children = object.children(high_handler_parent)
    high_handler = high_handler_children[-1]
    width = pyInt(high_handler.width) - 20
    height = pyInt(high_handler.height) - 25
    if arterial_max == art_high:
        test.passes("Arterial range maximum is already set to {}".format(art_high))
    elif arterial_max < art_high:
        while arterial_max != art_high:
            squish.mouseDrag(high_handler, -1, height, width, 0, squish.Qt.NoModifier, squish.Qt.LeftButton)
            arterial_max += 10
            # arterial blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if arterial_max == arterial_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(high_handler, width, height, 1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Arterial range maximum value cannot be moved beyond {}".format(arterial_max))
                break
            else:
                continue
    elif arterial_max > art_high:
        while arterial_max != art_high:
            squish.mouseDrag(high_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
            arterial_max -= 10
            # arterial blood pressure low limit should be lower than the high limit by at least 30mmHg 
            if arterial_max == arterial_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
                squish.mouseDrag(high_handler, width, height, -1, 0,
                          squish.Qt.NoModifier, squish.Qt.LeftButton)
                test.log("Arterial range maximum value cannot be moved beyond {}".format(arterial_max))
                break
            else:
                continue
    # arterial blood pressure low limit should be lower than the high limit by at least 30mmHg 
    if arterial_max == arterial_min + config.BUFFER_LOW_AND_HIGH_LIMITS:   
        test.compare(high_handler_parent.maxValue, arterial_max,
                     "Arterial range maximum value cannot be moved beyond {}".format(arterial_max))
    else:
        test.compare(arterial_max, art_high, 
                     "Actual Arterial range maximum value: {} is equal to Expected value: {}".format(arterial_max, art_high))
    test.endSection()