// DO NOT EDIT! This file was auto-generated by crates/re_types_builder/src/codegen/cpp/mod.rs
// Based on "crates/re_types/definitions/rerun/datatypes/translation_rotation_scale3d.fbs".

#include "translation_rotation_scale3d.hpp"

#include "rotation3d.hpp"
#include "scale3d.hpp"
#include "vec3d.hpp"

#include <arrow/builder.h>
#include <arrow/type_fwd.h>

namespace rerun::datatypes {}

namespace rerun {
    const std::shared_ptr<arrow::DataType>&
        Loggable<datatypes::TranslationRotationScale3D>::arrow_datatype() {
        static const auto datatype = arrow::struct_({
            arrow::field("translation", Loggable<rerun::datatypes::Vec3D>::arrow_datatype(), true),
            arrow::field(
                "rotation",
                Loggable<rerun::datatypes::Rotation3D>::arrow_datatype(),
                true
            ),
            arrow::field("scale", Loggable<rerun::datatypes::Scale3D>::arrow_datatype(), true),
            arrow::field("from_parent", arrow::boolean(), false),
        });
        return datatype;
    }

    rerun::Error Loggable<datatypes::TranslationRotationScale3D>::fill_arrow_array_builder(
        arrow::StructBuilder* builder, const datatypes::TranslationRotationScale3D* elements,
        size_t num_elements
    ) {
        if (builder == nullptr) {
            return rerun::Error(ErrorCode::UnexpectedNullArgument, "Passed array builder is null.");
        }
        if (elements == nullptr) {
            return rerun::Error(
                ErrorCode::UnexpectedNullArgument,
                "Cannot serialize null pointer to arrow array."
            );
        }

        {
            auto field_builder =
                static_cast<arrow::FixedSizeListBuilder*>(builder->field_builder(0));
            ARROW_RETURN_NOT_OK(field_builder->Reserve(static_cast<int64_t>(num_elements)));
            for (size_t elem_idx = 0; elem_idx < num_elements; elem_idx += 1) {
                const auto& element = elements[elem_idx];
                if (element.translation.has_value()) {
                    RR_RETURN_NOT_OK(Loggable<rerun::datatypes::Vec3D>::fill_arrow_array_builder(
                        field_builder,
                        &element.translation.value(),
                        1
                    ));
                } else {
                    ARROW_RETURN_NOT_OK(field_builder->AppendNull());
                }
            }
        }
        {
            auto field_builder = static_cast<arrow::DenseUnionBuilder*>(builder->field_builder(1));
            ARROW_RETURN_NOT_OK(field_builder->Reserve(static_cast<int64_t>(num_elements)));
            for (size_t elem_idx = 0; elem_idx < num_elements; elem_idx += 1) {
                const auto& element = elements[elem_idx];
                if (element.rotation.has_value()) {
                    RR_RETURN_NOT_OK(
                        Loggable<rerun::datatypes::Rotation3D>::fill_arrow_array_builder(
                            field_builder,
                            &element.rotation.value(),
                            1
                        )
                    );
                } else {
                    ARROW_RETURN_NOT_OK(field_builder->AppendNull());
                }
            }
        }
        {
            auto field_builder = static_cast<arrow::DenseUnionBuilder*>(builder->field_builder(2));
            ARROW_RETURN_NOT_OK(field_builder->Reserve(static_cast<int64_t>(num_elements)));
            for (size_t elem_idx = 0; elem_idx < num_elements; elem_idx += 1) {
                const auto& element = elements[elem_idx];
                if (element.scale.has_value()) {
                    RR_RETURN_NOT_OK(Loggable<rerun::datatypes::Scale3D>::fill_arrow_array_builder(
                        field_builder,
                        &element.scale.value(),
                        1
                    ));
                } else {
                    ARROW_RETURN_NOT_OK(field_builder->AppendNull());
                }
            }
        }
        {
            auto field_builder = static_cast<arrow::BooleanBuilder*>(builder->field_builder(3));
            ARROW_RETURN_NOT_OK(field_builder->Reserve(static_cast<int64_t>(num_elements)));
            for (size_t elem_idx = 0; elem_idx < num_elements; elem_idx += 1) {
                ARROW_RETURN_NOT_OK(field_builder->Append(elements[elem_idx].from_parent));
            }
        }
        ARROW_RETURN_NOT_OK(builder->AppendValues(static_cast<int64_t>(num_elements), nullptr));

        return Error::ok();
    }

    Result<std::shared_ptr<arrow::Array>> Loggable<datatypes::TranslationRotationScale3D>::to_arrow(
        const datatypes::TranslationRotationScale3D* instances, size_t num_instances
    ) {
        // TODO(andreas): Allow configuring the memory pool.
        arrow::MemoryPool* pool = arrow::default_memory_pool();
        auto datatype = arrow_datatype();

        ARROW_ASSIGN_OR_RAISE(auto builder, arrow::MakeBuilder(datatype, pool))
        if (instances && num_instances > 0) {
            RR_RETURN_NOT_OK(
                Loggable<datatypes::TranslationRotationScale3D>::fill_arrow_array_builder(
                    static_cast<arrow::StructBuilder*>(builder.get()),
                    instances,
                    num_instances
                )
            );
        }
        std::shared_ptr<arrow::Array> array;
        ARROW_RETURN_NOT_OK(builder->Finish(&array));
        return array;
    }
} // namespace rerun
