"""Support for simplified access to data on nodes of type omni.graph.nodes.SelectIf

Selects an output from the given inputs based on a boolean condition. If the condition is an array, and the inputs are arrays
of equal length, and values will be selected from ifTrue, ifFalse depending on the bool at the same index. If one input is
an array and the other is a scaler of the same base type, the scaler will be extended to the length of the other input.
"""

from typing import Any
import carb

import omni.graph.core as og
import omni.graph.core._omni_graph_core as _og
import omni.graph.tools.ogn as ogn



class OgnSelectIfDatabase(og.Database):
    """Helper class providing simplified access to data on nodes of type omni.graph.nodes.SelectIf

    Class Members:
        node: Node being evaluated

    Attribute Value Properties:
        Inputs:
            inputs.condition
            inputs.ifFalse
            inputs.ifTrue
        Outputs:
            outputs.result
    """

    # Imprint the generator and target ABI versions in the file for JIT generation
    GENERATOR_VERSION = (1, 41, 3)
    TARGET_VERSION = (2, 139, 12)

    # This is an internal object that provides per-class storage of a per-node data dictionary
    PER_NODE_DATA = {}

    # This is an internal object that describes unchanging attributes in a generic way
    # The values in this list are in no particular order, as a per-attribute tuple
    #     Name, Type, ExtendedTypeIndex, UiName, Description, Metadata,
    #     Is_Required, DefaultValue, Is_Deprecated, DeprecationMsg
    # You should not need to access any of this data directly, use the defined database interfaces
    INTERFACE = og.Database._get_interface([
        ('inputs:condition', 'bool,bool[]', 1, None, 'The selection variable', {}, True, None, False, ''),
        ('inputs:ifFalse', 'any', 2, 'If False', 'Value if condition is False', {}, True, None, False, ''),
        ('inputs:ifTrue', 'any', 2, 'If True', 'Value if condition is True', {}, True, None, False, ''),
        ('outputs:result', 'any', 2, 'Result', 'The selected value from ifTrue and ifFalse', {}, True, None, False, ''),
    ])

    class ValuesForInputs(og.DynamicAttributeAccess):
        LOCAL_PROPERTY_NAMES = { }
        """Helper class that creates natural hierarchical access to input attributes"""
        def __init__(self, node: og.Node, attributes, dynamic_attributes: og.DynamicAttributeInterface):
            """Initialize simplified access for the attribute data"""
            context = node.get_graph().get_default_graph_context()
            super().__init__(context, node, attributes, dynamic_attributes)
            self._batchedReadAttributes = []
            self._batchedReadValues = []

        @property
        def condition(self) -> og.RuntimeAttribute:
            """Get the runtime wrapper class for the attribute inputs.condition"""
            return og.RuntimeAttribute(self._attributes.condition.get_attribute_data(), self._context, True)

        @condition.setter
        def condition(self, value_to_set: Any):
            """Assign another attribute's value to outputs.condition"""
            if isinstance(value_to_set, og.RuntimeAttribute):
                self.condition.value = value_to_set.value
            else:
                self.condition.value = value_to_set

        @property
        def ifFalse(self) -> og.RuntimeAttribute:
            """Get the runtime wrapper class for the attribute inputs.ifFalse"""
            return og.RuntimeAttribute(self._attributes.ifFalse.get_attribute_data(), self._context, True)

        @ifFalse.setter
        def ifFalse(self, value_to_set: Any):
            """Assign another attribute's value to outputs.ifFalse"""
            if isinstance(value_to_set, og.RuntimeAttribute):
                self.ifFalse.value = value_to_set.value
            else:
                self.ifFalse.value = value_to_set

        @property
        def ifTrue(self) -> og.RuntimeAttribute:
            """Get the runtime wrapper class for the attribute inputs.ifTrue"""
            return og.RuntimeAttribute(self._attributes.ifTrue.get_attribute_data(), self._context, True)

        @ifTrue.setter
        def ifTrue(self, value_to_set: Any):
            """Assign another attribute's value to outputs.ifTrue"""
            if isinstance(value_to_set, og.RuntimeAttribute):
                self.ifTrue.value = value_to_set.value
            else:
                self.ifTrue.value = value_to_set

        def _prefetch(self):
            readAttributes = self._batchedReadAttributes
            newValues = _og._prefetch_input_attributes_data(readAttributes)
            if len(readAttributes) == len(newValues):
                self._batchedReadValues = newValues

    class ValuesForOutputs(og.DynamicAttributeAccess):
        LOCAL_PROPERTY_NAMES = { }
        """Helper class that creates natural hierarchical access to output attributes"""
        def __init__(self, node: og.Node, attributes, dynamic_attributes: og.DynamicAttributeInterface):
            """Initialize simplified access for the attribute data"""
            context = node.get_graph().get_default_graph_context()
            super().__init__(context, node, attributes, dynamic_attributes)
            self._batchedWriteValues = { }

        @property
        def result(self) -> og.RuntimeAttribute:
            """Get the runtime wrapper class for the attribute outputs.result"""
            return og.RuntimeAttribute(self._attributes.result.get_attribute_data(), self._context, False)

        @result.setter
        def result(self, value_to_set: Any):
            """Assign another attribute's value to outputs.result"""
            if isinstance(value_to_set, og.RuntimeAttribute):
                self.result.value = value_to_set.value
            else:
                self.result.value = value_to_set

        def _commit(self):
            _og._commit_output_attributes_data(self._batchedWriteValues)
            self._batchedWriteValues = { }

    class ValuesForState(og.DynamicAttributeAccess):
        """Helper class that creates natural hierarchical access to state attributes"""
        def __init__(self, node: og.Node, attributes, dynamic_attributes: og.DynamicAttributeInterface):
            """Initialize simplified access for the attribute data"""
            context = node.get_graph().get_default_graph_context()
            super().__init__(context, node, attributes, dynamic_attributes)

    def __init__(self, node):
        super().__init__(node)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_INPUT)
        self.inputs = OgnSelectIfDatabase.ValuesForInputs(node, self.attributes.inputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_OUTPUT)
        self.outputs = OgnSelectIfDatabase.ValuesForOutputs(node, self.attributes.outputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_STATE)
        self.state = OgnSelectIfDatabase.ValuesForState(node, self.attributes.state, dynamic_attributes)
