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

Timer Node is a node that lets you create animation curve(s), plays back and samples the value(s) along its time to output
values.
"""


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



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

    Class Members:
        node: Node being evaluated

    Attribute Value Properties:
        Inputs:
            inputs.duration
            inputs.endValue
            inputs.play
            inputs.startValue
        Outputs:
            outputs.finished
            outputs.updated
            outputs.value
    """

    # 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:duration', 'double', 0, 'Duration', 'Number of seconds to play interpolation', {ogn.MetadataKeys.DEFAULT: '1.0'}, True, 1.0, False, ''),
        ('inputs:endValue', 'double', 0, 'End Value', 'Value value of the end of the duration', {ogn.MetadataKeys.DEFAULT: '1.0'}, True, 1.0, False, ''),
        ('inputs:play', 'execution', 0, 'Play', 'Play the clip from current frame', {}, True, None, False, ''),
        ('inputs:startValue', 'double', 0, 'Start Value', 'Value value of the start of the duration', {ogn.MetadataKeys.DEFAULT: '0.0'}, True, 0.0, False, ''),
        ('outputs:finished', 'execution', 0, 'Finished', 'The Timer node has finished the playback', {}, True, None, False, ''),
        ('outputs:updated', 'execution', 0, 'Updated', 'The Timer node is ticked, and output value(s) resampled and updated', {}, True, None, False, ''),
        ('outputs:value', 'double', 0, 'Value', 'Value value of the Timer node between 0.0 and 1.0', {ogn.MetadataKeys.DEFAULT: '0.0'}, True, 0.0, False, ''),
    ])

    @classmethod
    def _populate_role_data(cls):
        """Populate a role structure with the non-default roles on this node type"""
        role_data = super()._populate_role_data()
        role_data.inputs.play = og.AttributeRole.EXECUTION
        role_data.outputs.finished = og.AttributeRole.EXECUTION
        role_data.outputs.updated = og.AttributeRole.EXECUTION
        return role_data

    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 duration(self):
            data_view = og.AttributeValueHelper(self._attributes.duration)
            return data_view.get()

        @duration.setter
        def duration(self, value):
            if self._setting_locked:
                raise og.ReadOnlyError(self._attributes.duration)
            data_view = og.AttributeValueHelper(self._attributes.duration)
            data_view.set(value)

        @property
        def endValue(self):
            data_view = og.AttributeValueHelper(self._attributes.endValue)
            return data_view.get()

        @endValue.setter
        def endValue(self, value):
            if self._setting_locked:
                raise og.ReadOnlyError(self._attributes.endValue)
            data_view = og.AttributeValueHelper(self._attributes.endValue)
            data_view.set(value)

        @property
        def play(self):
            data_view = og.AttributeValueHelper(self._attributes.play)
            return data_view.get()

        @play.setter
        def play(self, value):
            if self._setting_locked:
                raise og.ReadOnlyError(self._attributes.play)
            data_view = og.AttributeValueHelper(self._attributes.play)
            data_view.set(value)

        @property
        def startValue(self):
            data_view = og.AttributeValueHelper(self._attributes.startValue)
            return data_view.get()

        @startValue.setter
        def startValue(self, value):
            if self._setting_locked:
                raise og.ReadOnlyError(self._attributes.startValue)
            data_view = og.AttributeValueHelper(self._attributes.startValue)
            data_view.set(value)

        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 finished(self):
            data_view = og.AttributeValueHelper(self._attributes.finished)
            return data_view.get()

        @finished.setter
        def finished(self, value):
            data_view = og.AttributeValueHelper(self._attributes.finished)
            data_view.set(value)

        @property
        def updated(self):
            data_view = og.AttributeValueHelper(self._attributes.updated)
            return data_view.get()

        @updated.setter
        def updated(self, value):
            data_view = og.AttributeValueHelper(self._attributes.updated)
            data_view.set(value)

        @property
        def value(self):
            data_view = og.AttributeValueHelper(self._attributes.value)
            return data_view.get()

        @value.setter
        def value(self, value):
            data_view = og.AttributeValueHelper(self._attributes.value)
            data_view.set(value)

        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 = OgnTimerDatabase.ValuesForInputs(node, self.attributes.inputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_OUTPUT)
        self.outputs = OgnTimerDatabase.ValuesForOutputs(node, self.attributes.outputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_STATE)
        self.state = OgnTimerDatabase.ValuesForState(node, self.attributes.state, dynamic_attributes)
