Vectors2D

SUMMARY

A batch of vectors in 2D space.

from telekinesis import datatypes
vectors2d = datatypes.Vectors2D([[1.0, 2.0], [3.0, 4.0]])

API Reference

Complete API documentation for Vectors2D, including parameters, attributes, and methods.

View Reference →

Parameters

Parameter	Type	Default	Description
data	np.ndarray | list | tuple	Required	Array-like data of shape (N, 2).

Raises

Exception	Condition
TypeError	data can't be converted to a float32 array (e.g. non-numeric elements)
ValueError	data's rank isn't 2, its last axis isn't length 2, or it contains a non-finite value (NaN/Inf)

Attributes

Attribute	Type	Description
shape_spec	ClassVar[tuple[int | None, ...]]	Class-level shape spec, (None, 2) — None means the batch size is variable.
data	np.ndarray	The wrapped batch, shape (N, 2). Reading it returns a copy, so mutating the result doesn't affect the Vectors2D; assigning a new value re-validates it the same way as construction.
shape	tuple[int, ...]	(N, 2).
ndim	int	Always 2.
dtype	np.dtype	Always float32.
size	int	2 * N.

Methods

Method	Type	Description
Vectors2D.coerce(value)	Vectors2D	Converts array-like data into a Vectors2D. Accepts a np.ndarray, list, or tuple of shape (N, 2). If value is already a Vectors2D, it is returned unchanged.
to_numpy(copy=True)	np.ndarray	Returns the batch as a plain array. With the default copy=True you get an independent copy; pass copy=False to get a direct reference to the internal array instead, so mutating it also mutates the Vectors2D.
copy()	Vectors2D	Returns a new, independent Vectors2D with the same data.

Operators

Operation	Behavior
v1 == v2	True only if other is exactly a Vectors2D (not a subclass, not another datatype) with element-equal data. False for anything else.
len(v)	Batch size N. 0 for an empty batch (shape (0, 2)) — that's a valid, constructible batch.
np.asarray(v)	Returns a copy of data as an np.ndarray; NumPy functions accept a Vectors2D directly.

Visualization

import rerun as rr

# Your code block
# ....

rr.init("vectors2d_example", spawn=True)
datatypes.visualize(vectors2d, entity_path="/vectors2d", label="Vectors2D")

Example

"""Demonstrates the Telekinesis Vectors2D datatype."""

import time

import numpy as np
import rerun as rr
from loguru import logger

from telekinesis import datatypes

def vectors2d_example():
    """Demonstrate creation, inspection, operations, visualization, and serialization."""

    # ======================= Create ============================================
    vectors = [[1.0, 2.0], [3.0, 4.0]]
    vectors2d = datatypes.Vectors2D(vectors)
    logger.info(f"Created Vectors2D: {vectors2d}")

    empty_vectors2d = datatypes.Vectors2D(np.empty((0, 2), dtype=np.float32))
    logger.info(f"Created empty Vectors2D batch: {empty_vectors2d}")

    # ======================= Inspect ===========================================
    logger.info(f"data={vectors2d.data}")
    logger.info(f"shape={vectors2d.shape}")
    logger.info(f"ndim={vectors2d.ndim}")
    logger.info(f"dtype={vectors2d.dtype}")
    logger.info(f"size={vectors2d.size}")

    # ======================= Operations =========================================
    vectors2d.data = [[5.0, 6.0], [7.0, 8.0]]
    logger.info(f"Updated Vectors2D: {vectors2d}")

    vectors2d_copy = vectors2d.copy()
    logger.info(f"Copied Vectors2D: {vectors2d_copy}")

    vectors2d_numpy = vectors2d.to_numpy(copy=True)
    logger.info(f"NumPy Vectors2D:\n{vectors2d_numpy}")

    numpy_array = np.asarray(vectors2d)
    column_sums = np.sum(vectors2d, axis=0)
    logger.info(f"NumPy array:\n{numpy_array}")
    logger.info(f"Column sums: {column_sums}")

    # ======================= Visualize =========================================
    rr.init("vectors2d_example", spawn=True)
    datatypes.visualize(vectors2d, entity_path="/vectors2d", label=["Vector 1", "Vector 2"])

    # ======================= Serialize / Deserialize ===========================
    start = time.perf_counter()
    serialized = datatypes.serialize(vectors2d)
    serialization_ms = (time.perf_counter() - start) * 1000

    start = time.perf_counter()
    deserialized = datatypes.deserialize(serialized)["param_0"]
    deserialization_ms = (time.perf_counter() - start) * 1000

    logger.info(f"Deserialized Vectors2D: {deserialized}")
    logger.info(f"Round-trip successful: {vectors2d == deserialized}")
    logger.info(f"Serialization time: {serialization_ms:.3f} ms")
    logger.info(f"Deserialization time: {deserialization_ms:.3f} ms")


if __name__ == "__main__":
    vectors2d_example()

See also Vector2D for a single 2D vector.