#ifndef ARROW_PYTHON_PYTHON_TO_ARROW_H

#define ARROW_PYTHON_PYTHON_TO_ARROW_H

#include "arrow/python/platform.h"

#include <memory>

#include <vector>

#include "arrow/python/common.h"

#include "arrow/python/pyarrow.h"

#include "arrow/status.h"

#include "arrow/util/visibility.h"

namespace arrow {

class MemoryPool;

class RecordBatch;

class Tensor;

namespace io {

class OutputStream;

} // namespace io

namespace py {

struct ARROW_EXPORT SerializedPyObject {

std::shared_ptr<RecordBatch> batch;

std::vector<std::shared_ptr<Tensor>> tensors;

std::vector<std::shared_ptr<Buffer>> buffers;

/// \brief Write serialized Python object to OutputStream

/// \param[in,out] dst an OutputStream

/// \return Status

Status WriteTo(io::OutputStream* dst);

/// \brief Convert SerializedPyObject to a dict containing the message

/// components as Buffer instances with minimal memory allocation

///

/// {

/// 'num_tensors': N,

/// 'num_buffers': K,

/// 'data': [Buffer]

/// }

///

/// Each tensor is written as two buffers, one for the metadata and one for

/// the body. Therefore, the number of buffers in 'data' is 2 * N + K + 1,

/// with the first buffer containing the serialized record batch containing

/// the UnionArray that describes the whole object

Status GetComponents(MemoryPool* pool, PyObject** out);

};

Status SerializeObject(PyObject* context, PyObject* sequence, SerializedPyObject* out);

} // namespace py

} // namespace arrow

#endif // ARROW_PYTHON_PYTHON_TO_ARROW_H