Added PointCloud()

Author: dsblank

backend/kangas/__init__.py

@@ -26,6 +26,7 @@
     Tensor,
     Text,
     Video,
+    PointCloud,
 )
 from .integrations import export_to_comet, import_from_comet  # noqa
 from .server.queries import sqlite_query, sqlite_query_explain  # noqa

backend/kangas/datatypes/__init__.py

@@ -21,12 +21,14 @@
 from .tensor import Tensor
 from .text import Text
 from .video import Video
+from .pointcloud import PointCloud
 
 for name, cls, asset_type in [
     (None, Asset, "asset"),
     ("AUDIO-ASSET", Audio, "audio"),
     ("CURVE-ASSET", Curve, "curve"),
     ("IMAGE-ASSET", Image, "image"),
+    ("POINTCLOUD-ASSET", PointCloud, "pointcloud"),
     ("TEXT-ASSET", Text, "text"),
     ("VIDEO-ASSET", Video, "video"),
     ("EMBEDDING-ASSET", Embedding, "embedding"),

backend/kangas/datatypes/pointcloud.py

@@ -0,0 +1,52 @@
+# -*- coding: utf-8 -*-
+######################################################
+#     _____                  _____      _     _      #
+#    (____ \       _        |  ___)    (_)   | |     #
+#     _   \ \ ____| |_  ____| | ___ ___ _  _ | |     #
+#    | |  | )/ _  |  _)/ _  | |(_  / __) |/ || |     #
+#    | |__/ ( ( | | | ( ( | | |__| | | | ( (_| |     #
+#    |_____/ \_||_|___)\_||_|_____/|_| |_|\____|     #
+#                                                    #
+#    Copyright (c) 2023-2024 Kangas Development Team #
+#    All rights reserved                             #
+######################################################
+
+import logging
+import json
+
+from .base import Asset
+
+LOGGER = logging.getLogger(__name__)
+
+class PointCloud(Asset):
+    """
+    A PointCloud asset.
+    """
+
+    ASSET_TYPE = "PointCloud"
+
+    def __init__(
+        self,
+        scene_name=None,
+        points=None,
+        boxes=None,
+        step=None,
+        metadata=None,
+        source=None,
+        unserialize=False,
+    ):
+        super().__init__(source)
+        if unserialize:
+            # A function that takes the object to lookup
+            self._unserialize = unserialize
+            return
+
+        self.metadata["scene_name"] = scene_name
+        self.metadata["step"] = step
+
+        self.asset_data = json.dumps(
+            {
+                "points": points,
+                "boxes": boxes,
+            }
+        )

frontend/lib/consts/cellMap.js

@@ -63,6 +63,11 @@ const cellMap = {
         groupedWidth: 300,
         component: EmbeddingCell
     },
+    'POINTCLOUD-ASSET': {
+        width: 150,
+        groupedWidth: 300,
+        component: NACell
+    },
     ROW_ID: {
         width: 100,
         groupedWidth: 100,