]>
Commit | Line | Data |
---|---|---|
1d09f67e TL |
1 | # Licensed to the Apache Software Foundation (ASF) under one |
2 | # or more contributor license agreements. See the NOTICE file | |
3 | # distributed with this work for additional information | |
4 | # regarding copyright ownership. The ASF licenses this file | |
5 | # to you under the Apache License, Version 2.0 (the | |
6 | # "License"); you may not use this file except in compliance | |
7 | # with the License. You may obtain a copy of the License at | |
8 | # | |
9 | # http://www.apache.org/licenses/LICENSE-2.0 | |
10 | # | |
11 | # Unless required by applicable law or agreed to in writing, | |
12 | # software distributed under the License is distributed on an | |
13 | # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
14 | # KIND, either express or implied. See the License for the | |
15 | # specific language governing permissions and limitations | |
16 | # under the License. | |
17 | ||
18 | from libcpp.memory cimport shared_ptr | |
19 | from pyarrow.includes.libarrow cimport (CArray, CBuffer, CDataType, | |
20 | CField, CRecordBatch, CSchema, | |
21 | CTable, CTensor, CSparseCOOTensor, | |
22 | CSparseCSRMatrix, CSparseCSCMatrix, | |
23 | CSparseCSFTensor) | |
24 | ||
25 | cdef extern from "arrow/python/pyarrow.h" namespace "arrow::py": | |
26 | cdef int import_pyarrow() except -1 | |
27 | cdef object wrap_buffer(const shared_ptr[CBuffer]& buffer) | |
28 | cdef object wrap_data_type(const shared_ptr[CDataType]& type) | |
29 | cdef object wrap_field(const shared_ptr[CField]& field) | |
30 | cdef object wrap_schema(const shared_ptr[CSchema]& schema) | |
31 | cdef object wrap_array(const shared_ptr[CArray]& sp_array) | |
32 | cdef object wrap_tensor(const shared_ptr[CTensor]& sp_tensor) | |
33 | cdef object wrap_sparse_tensor_coo( | |
34 | const shared_ptr[CSparseCOOTensor]& sp_sparse_tensor) | |
35 | cdef object wrap_sparse_tensor_csr( | |
36 | const shared_ptr[CSparseCSRMatrix]& sp_sparse_tensor) | |
37 | cdef object wrap_sparse_tensor_csc( | |
38 | const shared_ptr[CSparseCSCMatrix]& sp_sparse_tensor) | |
39 | cdef object wrap_sparse_tensor_csf( | |
40 | const shared_ptr[CSparseCSFTensor]& sp_sparse_tensor) | |
41 | cdef object wrap_table(const shared_ptr[CTable]& ctable) | |
42 | cdef object wrap_batch(const shared_ptr[CRecordBatch]& cbatch) |