"""Support for simplified access to data on nodes of type omni.syntheticdata.SdPostSemantic3dBoundingBoxCameraProjection

Synthetic Data node to project 3d bounding boxes data in camera space.
"""

import numpy

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



class OgnSdPostSemantic3dBoundingBoxCameraProjectionDatabase(og.Database):
    """Helper class providing simplified access to data on nodes of type omni.syntheticdata.SdPostSemantic3dBoundingBoxCameraProjection

    Class Members:
        node: Node being evaluated

    Attribute Value Properties:
        Inputs:
            inputs.cameraFisheyeParams
            inputs.cameraModel
            inputs.cameraNearFar
            inputs.exec
            inputs.gpu
            inputs.instanceMappingInfoSDPtr
            inputs.metersPerSceneUnit
            inputs.renderProductResolution
            inputs.rp
            inputs.sdSemBBoxExtentCudaPtr
            inputs.sdSemBBoxInfosCudaPtr
            inputs.semanticWorldTransformSDCudaPtr
        Outputs:
            outputs.exec
            outputs.sdSemBBox3dCamCornersCudaPtr
            outputs.sdSemBBox3dCamExtentCudaPtr

    Predefined Tokens:
        tokens.SemanticBoundingBox3DInfosSD
        tokens.SemanticBoundingBox3DCamCornersSD
        tokens.SemanticBoundingBox3DCamExtentSD
    """

    # 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:cameraFisheyeParams', 'float[]', 0, None, 'Camera fisheye projection parameters', {}, True, [], False, ''),
        ('inputs:cameraModel', 'int', 0, None, 'Camera model (pinhole or fisheye models)', {}, True, 0, False, ''),
        ('inputs:cameraNearFar', 'float2', 0, None, 'Camera near/far clipping range', {ogn.MetadataKeys.DEFAULT: '[1.0, 10000000.0]'}, True, [1.0, 10000000.0], False, ''),
        ('inputs:exec', 'execution', 0, None, 'Trigger', {}, True, None, False, ''),
        ('inputs:gpu', 'uint64', 0, 'gpuFoundations', 'Pointer to shared context containing gpu foundations', {}, True, 0, False, ''),
        ('inputs:instanceMappingInfoSDPtr', 'uint64', 0, None, 'uint buffer pointer containing the following information : [numInstances, minInstanceId, numSemantics, minSemanticId, numProtoSemantic]', {}, True, 0, False, ''),
        ('inputs:metersPerSceneUnit', 'float', 0, None, 'Scene units to meters scale', {ogn.MetadataKeys.DEFAULT: '0.01'}, True, 0.01, False, ''),
        ('inputs:renderProductResolution', 'int2', 0, None, 'RenderProduct resolution', {ogn.MetadataKeys.DEFAULT: '[65536, 65536]'}, True, [65536, 65536], False, ''),
        ('inputs:rp', 'uint64', 0, 'renderProduct', 'Pointer to render product for this view', {}, True, 0, False, ''),
        ('inputs:sdSemBBoxExtentCudaPtr', 'uint64', 0, None, 'Cuda buffer containing the extent of the bounding boxes as a float4=(u_min,v_min,u_max,v_max) for 2D or a float6=(xmin,ymin,zmin,xmax,ymax,zmax) in object space for 3D', {}, True, 0, False, ''),
        ('inputs:sdSemBBoxInfosCudaPtr', 'uint64', 0, None, 'Cuda buffer containing valid bounding boxes infos', {}, True, 0, False, ''),
        ('inputs:semanticWorldTransformSDCudaPtr', 'uint64', 0, None, 'cuda float44 buffer pointer of size numSemantics containing the world semantic transform', {}, True, 0, False, ''),
        ('outputs:exec', 'execution', 0, None, 'Trigger', {}, True, None, False, ''),
        ('outputs:sdSemBBox3dCamCornersCudaPtr', 'uint64', 0, None, 'Cuda buffer containing the projection of the 3d bounding boxes on the camera plane represented as a float4=(u,v,z,a) for each bounding box corners', {}, True, None, False, ''),
        ('outputs:sdSemBBox3dCamExtentCudaPtr', 'uint64', 0, None, 'Cuda buffer containing the 2d extent of the 3d bounding boxes on the camera plane represented as a float6=(u_min,u_max,v_min,v_max,z_min,z_max)', {}, True, None, False, ''),
    ])

    class tokens:
        SemanticBoundingBox3DInfosSD = "SemanticBoundingBox3DInfosSD"
        SemanticBoundingBox3DCamCornersSD = "SemanticBoundingBox3DCamCornersSD"
        SemanticBoundingBox3DCamExtentSD = "SemanticBoundingBox3DCamExtentSD"

    @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.exec = og.AttributeRole.EXECUTION
        role_data.outputs.exec = 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 cameraFisheyeParams(self):
            data_view = og.AttributeValueHelper(self._attributes.cameraFisheyeParams)
            return data_view.get()

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

        @semanticWorldTransformSDCudaPtr.setter
        def semanticWorldTransformSDCudaPtr(self, value):
            if self._setting_locked:
                raise og.ReadOnlyError(self._attributes.semanticWorldTransformSDCudaPtr)
            data_view = og.AttributeValueHelper(self._attributes.semanticWorldTransformSDCudaPtr)
            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 exec(self):
            data_view = og.AttributeValueHelper(self._attributes.exec)
            return data_view.get()

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

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

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

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

        @sdSemBBox3dCamExtentCudaPtr.setter
        def sdSemBBox3dCamExtentCudaPtr(self, value):
            data_view = og.AttributeValueHelper(self._attributes.sdSemBBox3dCamExtentCudaPtr)
            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 = OgnSdPostSemantic3dBoundingBoxCameraProjectionDatabase.ValuesForInputs(node, self.attributes.inputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_OUTPUT)
        self.outputs = OgnSdPostSemantic3dBoundingBoxCameraProjectionDatabase.ValuesForOutputs(node, self.attributes.outputs, dynamic_attributes)
        dynamic_attributes = self.dynamic_attribute_data(node, og.AttributePortType.ATTRIBUTE_PORT_TYPE_STATE)
        self.state = OgnSdPostSemantic3dBoundingBoxCameraProjectionDatabase.ValuesForState(node, self.attributes.state, dynamic_attributes)
