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

        Result<std::shared_ptr<arrow::StructBuilder>>
            TranslationRotationScale3D::new_arrow_array_builder(arrow::MemoryPool* memory_pool) {
            if (memory_pool == nullptr) {
                return Error(ErrorCode::UnexpectedNullArgument, "Memory pool is null.");
            }

            return Result(std::make_shared<arrow::StructBuilder>(
                arrow_datatype(),
                memory_pool,
                std::vector<std::shared_ptr<arrow::ArrayBuilder>>({
                    rerun::datatypes::Vec3D::new_arrow_array_builder(memory_pool).value,
                    rerun::datatypes::Rotation3D::new_arrow_array_builder(memory_pool).value,
                    rerun::datatypes::Scale3D::new_arrow_array_builder(memory_pool).value,
                    std::make_shared<arrow::BooleanBuilder>(memory_pool),
                })
            ));
        }

        Error TranslationRotationScale3D::fill_arrow_array_builder(
            arrow::StructBuilder* builder, const TranslationRotationScale3D* elements,
            size_t num_elements
        ) {
            if (builder == nullptr) {
                return Error(ErrorCode::UnexpectedNullArgument, "Passed array builder is null.");
            }
            if (elements == nullptr) {
                return 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(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(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(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();
        }
    } // namespace datatypes
} // namespace rerun
