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

Read the state of the last viewport drag event from the specified viewport. Note that viewport mouse events must be enabled
on the specified viewport using a SetViewportMode node.
"""

import omni.graph.core as og
import omni.graph.core._omni_graph_core as _og
import omni.graph.tools.ogn as ogn
import numpy
class OgnReadViewportDragStateDatabase(og.Database):
    """Helper class providing simplified access to data on nodes of type omni.graph.ui.ReadViewportDragState

    Class Members:
        node: Node being evaluated

    Attribute Value Properties:
        Inputs:
            inputs.gesture
            inputs.useNormalizedCoords
            inputs.viewport
        Outputs:
            outputs.currentPosition
            outputs.initialPosition
            outputs.isDragInProgress
            outputs.isValid
            outputs.velocity

    Predefined Tokens:
        tokens.LeftMouseDrag
        tokens.RightMouseDrag
        tokens.MiddleMouseDrag
    """
    # 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:gesture', 'token', 0, 'Gesture', 'The input gesture to trigger viewport drag events', {'displayGroup': 'parameters', ogn.MetadataKeys.ALLOWED_TOKENS: 'Left Mouse Drag,Right Mouse Drag,Middle Mouse Drag', ogn.MetadataKeys.ALLOWED_TOKENS_RAW: '{"LeftMouseDrag": "Left Mouse Drag", "RightMouseDrag": "Right Mouse Drag", "MiddleMouseDrag": "Middle Mouse Drag"}', ogn.MetadataKeys.DEFAULT: '"Left Mouse Drag"'}, True, 'Left Mouse Drag', False, ''),
        ('inputs:useNormalizedCoords', 'bool', 0, 'Use Normalized Coords', 'When true, the components of 2D position and velocity outputs are scaled to between 0 and 1,\nwhere 0 is top/left and 1 is bottom/right.\nWhen false, components are in viewport render resolution pixels.', {ogn.MetadataKeys.DEFAULT: 'false'}, True, False, False, ''),
        ('inputs:viewport', 'token', 0, 'Viewport', 'Name of the viewport window to watch for drag events', {ogn.MetadataKeys.DEFAULT: '"Viewport"'}, True, 'Viewport', False, ''),
        ('outputs:currentPosition', 'double2', 0, 'Current Position', 'The current mouse position if a drag is in progress,\nor the final mouse position of the most recent drag if a drag is not in progress', {}, True, None, False, ''),
        ('outputs:initialPosition', 'double2', 0, 'Initial Position', 'The mouse position at which the most recent drag began', {}, True, None, False, ''),
        ('outputs:isDragInProgress', 'bool', 0, 'Is Drag In Progress', 'True if a viewport drag event is currently in progress in the specified viewport', {}, True, None, False, ''),
        ('outputs:isValid', 'bool', 0, 'Is Valid', 'True if a valid event state was detected and the outputs of this node are valid, and false otherwise', {}, True, None, False, ''),
        ('outputs:velocity', 'double2', 0, 'Velocity', 'A vector representing the change in position of the mouse since the previous frame\nif a drag is in progress, otherwise (0,0)', {}, True, None, False, ''),
    ])
    class tokens:
        LeftMouseDrag = "Left Mouse Drag"
        RightMouseDrag = "Right Mouse Drag"
        MiddleMouseDrag = "Middle Mouse Drag"
    class ValuesForInputs(og.DynamicAttributeAccess):
        LOCAL_PROPERTY_NAMES = {"gesture", "useNormalizedCoords", "viewport", "_setting_locked", "_batchedReadAttributes", "_batchedReadValues"}
        """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._attributes.gesture, self._attributes.useNormalizedCoords, self._attributes.viewport]
            self._batchedReadValues = ["Left Mouse Drag", False, "Viewport"]

        @property
        def gesture(self):
            return self._batchedReadValues[0]

        @gesture.setter
        def gesture(self, value):
            self._batchedReadValues[0] = value

        @property
        def useNormalizedCoords(self):
            return self._batchedReadValues[1]

        @useNormalizedCoords.setter
        def useNormalizedCoords(self, value):
            self._batchedReadValues[1] = value

        @property
        def viewport(self):
            return self._batchedReadValues[2]

        @viewport.setter
        def viewport(self, value):
            self._batchedReadValues[2] = value

        def __getattr__(self, item: str):
            if item in self.LOCAL_PROPERTY_NAMES:
                return object.__getattribute__(self, item)
            else:
                return super().__getattr__(item)

        def __setattr__(self, item: str, new_value):
            if item in self.LOCAL_PROPERTY_NAMES:
                object.__setattr__(self, item, new_value)
            else:
                super().__setattr__(item, new_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 = {"currentPosition", "initialPosition", "isDragInProgress", "isValid", "velocity", "_batchedWriteValues"}
        """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 currentPosition(self):
            value = self._batchedWriteValues.get(self._attributes.currentPosition)
            if value:
                return value
            else:
                data_view = og.AttributeValueHelper(self._attributes.currentPosition)
                return data_view.get()

        @currentPosition.setter
        def currentPosition(self, value):
            self._batchedWriteValues[self._attributes.currentPosition] = value

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

        @initialPosition.setter
        def initialPosition(self, value):
            self._batchedWriteValues[self._attributes.initialPosition] = value

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

        @isDragInProgress.setter
        def isDragInProgress(self, value):
            self._batchedWriteValues[self._attributes.isDragInProgress] = value

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

        @isValid.setter
        def isValid(self, value):
            self._batchedWriteValues[self._attributes.isValid] = value

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

        @velocity.setter
        def velocity(self, value):
            self._batchedWriteValues[self._attributes.velocity] = value

        def __getattr__(self, item: str):
            if item in self.LOCAL_PROPERTY_NAMES:
                return object.__getattribute__(self, item)
            else:
                return super().__getattr__(item)

        def __setattr__(self, item: str, new_value):
            if item in self.LOCAL_PROPERTY_NAMES:
                object.__setattr__(self, item, new_value)
            else:
                super().__setattr__(item, new_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 = OgnReadViewportDragStateDatabase.ValuesForInputs(node, self.attributes.inputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_OUTPUT)
        self.outputs = OgnReadViewportDragStateDatabase.ValuesForOutputs(node, self.attributes.outputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_STATE)
        self.state = OgnReadViewportDragStateDatabase.ValuesForState(node, self.attributes.state, dynamic_attributes)
