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

Test node to test out the effects of vectorization.
"""

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

    Class Members:
        node: Node being evaluated

    Attribute Value Properties:
        Inputs:
            inputs.base_name
            inputs.num_instances
        Outputs:
            outputs.gathered_paths
            outputs.rotations
    """
    # 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:base_name', 'token', 0, None, 'The base name of the pattern to match', {ogn.MetadataKeys.DEFAULT: '""'}, True, '', False, ''),
        ('inputs:num_instances', 'int', 0, None, 'How many instances are involved', {ogn.MetadataKeys.DEFAULT: '1'}, True, 1, False, ''),
        ('outputs:gathered_paths', 'token[]', 0, None, 'The paths of the gathered objects', {ogn.MetadataKeys.DEFAULT: '[]'}, True, [], False, ''),
        ('outputs:rotations', 'double3[]', 0, None, 'The rotations of the gathered points', {ogn.MetadataKeys.DEFAULT: '[]'}, True, [], False, ''),
    ])
    class ValuesForInputs(og.DynamicAttributeAccess):
        LOCAL_PROPERTY_NAMES = {"base_name", "num_instances", "_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.base_name, self._attributes.num_instances]
            self._batchedReadValues = ["", 1]

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

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

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

        @num_instances.setter
        def num_instances(self, value):
            self._batchedReadValues[1] = 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 = { }
        """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.gathered_paths_size = 0
            self.rotations_size = 0
            self._batchedWriteValues = { }

        @property
        def gathered_paths(self):
            data_view = og.AttributeValueHelper(self._attributes.gathered_paths)
            return data_view.get(reserved_element_count=self.gathered_paths_size)

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

        @property
        def rotations(self):
            data_view = og.AttributeValueHelper(self._attributes.rotations)
            return data_view.get(reserved_element_count=self.rotations_size)

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

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