1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
#include <arrow/api.h>
#include <arrow-glib/buffer.hpp>
#include <arrow-glib/error.hpp>
#include <arrow-glib/readable.hpp>
G_BEGIN_DECLS
/**
* SECTION: readable
* @title: GArrowReadable
* @short_description: Input interface
*
* #GArrowReadable is an interface for input. Input must be
* readable.
*/
G_DEFINE_INTERFACE(GArrowReadable,
garrow_readable,
G_TYPE_OBJECT)
static void
garrow_readable_default_init (GArrowReadableInterface *iface)
{
iface->buffer_new_raw = garrow_buffer_new_raw;
}
/**
* garrow_readable_read:
* @readable: A #GArrowReadable.
* @n_bytes: The number of bytes to be read.
* @error: (nullable): Return location for a #GError or %NULL.
*
* Returns: (transfer full) (nullable): #GArrowBuffer that has read
* data on success, %NULL if there was an error.
*/
GArrowBuffer *
garrow_readable_read(GArrowReadable *readable,
gint64 n_bytes,
GError **error)
{
const auto arrow_readable = garrow_readable_get_raw(readable);
auto arrow_buffer = arrow_readable->Read(n_bytes);
if (garrow::check(error, arrow_buffer, "[readable][read]")) {
auto *iface = GARROW_READABLE_GET_IFACE(readable);
return iface->buffer_new_raw(&(arrow_buffer.ValueOrDie()));
} else {
return NULL;
}
}
/**
* garrow_readable_read_bytes:
* @readable: A #GArrowReadable.
* @n_bytes: The number of bytes to be read.
* @error: (nullable): Return location for a #GError or %NULL.
*
* Returns: (transfer full) (nullable): #GBytes that has read data on
* success, %NULL if there was an error.
*
* Since: 0.17.0
*/
GBytes *
garrow_readable_read_bytes(GArrowReadable *readable,
gint64 n_bytes,
GError **error)
{
const auto arrow_readable = garrow_readable_get_raw(readable);
auto arrow_buffer_result = arrow_readable->Read(n_bytes);
if (!garrow::check(error, arrow_buffer_result, "[readable][read-bytes]")) {
return NULL;
}
auto arrow_cpu_buffer_result =
arrow::Buffer::ViewOrCopy(*arrow_buffer_result,
arrow::default_cpu_memory_manager());
if (!garrow::check(error,
arrow_cpu_buffer_result,
"[readable][read-bytes][view-or-copy]")) {
return NULL;
}
auto arrow_cpu_buffer = *arrow_cpu_buffer_result;
return g_bytes_new(arrow_cpu_buffer->data(),
arrow_cpu_buffer->size());
}
G_END_DECLS
std::shared_ptr<arrow::io::Readable>
garrow_readable_get_raw(GArrowReadable *readable)
{
auto *iface = GARROW_READABLE_GET_IFACE(readable);
return iface->get_raw(readable);
}
|
