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
|
/*
* 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.
*/
#pragma once
#include <arrow/api.h>
#ifdef _WIN32
# define gmtime_r gmtime_r_ruby_win32
# define localtime_r localtime_r_ruby_win32
# include <ruby.h>
# undef gmtime_r
# undef localtime_r
#endif
#include <arrow-glib/arrow-glib.hpp>
#include <rbgobject.h>
namespace red_arrow {
extern VALUE cDate;
extern VALUE cArrowTime;
extern VALUE ArrowTimeUnitSECOND;
extern VALUE ArrowTimeUnitMILLI;
extern VALUE ArrowTimeUnitMICRO;
extern VALUE ArrowTimeUnitNANO;
extern ID id_BigDecimal;
extern ID id_jd;
extern ID id_new;
extern ID id_to_datetime;
namespace symbols {
extern VALUE day;
extern VALUE millisecond;
extern VALUE month;
extern VALUE nanosecond;
}
VALUE array_values(VALUE obj);
VALUE chunked_array_values(VALUE obj);
VALUE record_batch_raw_records(VALUE obj);
VALUE table_raw_records(VALUE obj);
VALUE record_batch_each_raw_record(VALUE obj);
VALUE table_each_raw_record(VALUE obj);
inline VALUE time_unit_to_scale(const arrow::TimeUnit::type unit) {
switch (unit) {
case arrow::TimeUnit::SECOND:
return INT2FIX(1);
case arrow::TimeUnit::MILLI:
return INT2FIX(1000);
case arrow::TimeUnit::MICRO:
return INT2FIX(1000 * 1000);
case arrow::TimeUnit::NANO:
// NOTE: INT2FIX works for 1e+9 because: FIXNUM_MAX >= (1<<30) - 1 > 1e+9
return INT2FIX(1000 * 1000 * 1000);
default:
rb_raise(rb_eArgError, "invalid arrow::TimeUnit: %d", unit);
return Qnil;
}
}
inline VALUE time_unit_to_enum(const arrow::TimeUnit::type unit) {
switch (unit) {
case arrow::TimeUnit::SECOND:
return red_arrow::ArrowTimeUnitSECOND;
case arrow::TimeUnit::MILLI:
return red_arrow::ArrowTimeUnitMILLI;
case arrow::TimeUnit::MICRO:
return red_arrow::ArrowTimeUnitMICRO;
case arrow::TimeUnit::NANO:
return red_arrow::ArrowTimeUnitNANO;
default:
rb_raise(rb_eArgError, "invalid arrow::TimeUnit: %d", unit);
return Qnil;
}
}
inline void check_status(const arrow::Status&& status, const char* context) {
GError* error = nullptr;
if (!garrow_error_check(&error, status, context)) {
RG_RAISE_ERROR(error);
}
}
}
|
