"""Log a simple 3D asset."""

import sys

import rerun as rr

if len(sys.argv) < 2:
    print(f"Usage: {sys.argv[0]} <path_to_asset.[gltf|glb|obj|stl]>")
    sys.exit(1)

rr.init("rerun_example_asset3d", spawn=True)

rr.log("world", rr.ViewCoordinates.RIGHT_HAND_Z_UP, static=True)  # Set an up-axis
rr.log("world/asset", rr.Asset3D(path=sys.argv[1]))

"""Log a simple 3D asset with an out-of-tree transform which will not affect its children."""

import sys

import numpy as np
import rerun as rr
from rerun.components import OutOfTreeTransform3DBatch
from rerun.datatypes import TranslationRotationScale3D

if len(sys.argv) < 2:
    print(f"Usage: {sys.argv[0]} <path_to_asset.[gltf|glb]>")
    sys.exit(1)

rr.init("rerun_example_asset3d_out_of_tree", spawn=True)

rr.log("world", rr.ViewCoordinates.RIGHT_HAND_Z_UP, static=True)  # Set an up-axis

rr.set_time_sequence("frame", 0)
rr.log("world/asset", rr.Asset3D(path=sys.argv[1]))
# Those points will not be affected by their parent's out-of-tree transform!
rr.log(
    "world/asset/points",
    rr.Points3D(np.vstack([xyz.ravel() for xyz in np.mgrid[3 * [slice(-10, 10, 10j)]]]).T),
)

asset = rr.Asset3D(path=sys.argv[1])
for i in range(1, 20):
    rr.set_time_sequence("frame", i)

    translation = TranslationRotationScale3D(translation=[0, 0, i - 10.0])
    rr.log_components("world/asset", [OutOfTreeTransform3DBatch(translation)])