| 1043 | } |
| 1044 | |
| 1045 | static Status MakeRecordBatch(FBB& fbb, int64_t length, int64_t body_length, |
| 1046 | const std::vector<FieldMetadata>& nodes, |
| 1047 | const std::vector<BufferMetadata>& buffers, |
| 1048 | const std::vector<int64_t>& variadic_buffer_counts, |
| 1049 | const IpcWriteOptions& options, RecordBatchOffset* offset) { |
| 1050 | FieldNodeVector fb_nodes; |
| 1051 | RETURN_NOT_OK(WriteFieldNodes(fbb, nodes, &fb_nodes)); |
| 1052 | |
| 1053 | BufferVector fb_buffers; |
| 1054 | RETURN_NOT_OK(WriteBuffers(fbb, buffers, &fb_buffers)); |
| 1055 | |
| 1056 | BodyCompressionOffset fb_compression; |
| 1057 | RETURN_NOT_OK(GetBodyCompression(fbb, options, &fb_compression)); |
| 1058 | |
| 1059 | flatbuffers::Offset<flatbuffers::Vector<int64_t>> fb_variadic_buffer_counts{}; |
| 1060 | if (!variadic_buffer_counts.empty()) { |
| 1061 | fb_variadic_buffer_counts = fbb.CreateVector(variadic_buffer_counts); |
| 1062 | } |
| 1063 | |
| 1064 | *offset = flatbuf::CreateRecordBatch(fbb, length, fb_nodes, fb_buffers, fb_compression, |
| 1065 | fb_variadic_buffer_counts); |
| 1066 | return Status::OK(); |
| 1067 | } |
| 1068 | |
| 1069 | Status MakeSparseTensorIndexCOO(FBB& fbb, const SparseCOOIndex& sparse_index, |
| 1070 | const std::vector<BufferMetadata>& buffers, |
no test coverage detected