#include <rerun.hpp>

struct CustomPosition3D {
    rerun::Position3D position;
};

template <>
struct rerun::Loggable<CustomPosition3D> {
    static constexpr const ComponentDescriptor Descriptor = ComponentDescriptor(
        "user.CustomArchetype", "user.CustomArchetype:custom_positions",
        "user.CustomPosition3D"
    );

    static const std::shared_ptr<arrow::DataType>& arrow_datatype() {
        return rerun::Loggable<rerun::Position3D>::arrow_datatype();
    }

    // TODO(#4257) should take a rerun::Collection instead of pointer and size.
    static rerun::Result<std::shared_ptr<arrow::Array>> to_arrow(
        const CustomPosition3D* instances, size_t num_instances
    ) {
        return rerun::Loggable<rerun::Position3D>::to_arrow(
            reinterpret_cast<const rerun::Position3D*>(instances),
            num_instances
        );
    }
};

int main(int argc, char* argv[]) {
    const auto rec =
        rerun::RecordingStream("rerun_example_descriptors_custom_component");
    rec.spawn().exit_on_failure();

    rec.log_static(
        "data",
        rerun::ComponentBatch::from_loggable<rerun::Position3D>(
            {1.0f, 2.0f, 3.0f},
            rerun::Loggable<CustomPosition3D>::Descriptor
        )
    );

    // The tags are indirectly checked by the Rust version (have a look over there for more info).
}