Treelite
c_api_common.h
Go to the documentation of this file.
1 
9 #ifndef TREELITE_C_API_COMMON_H_
10 #define TREELITE_C_API_COMMON_H_
11 
12 #ifdef __cplusplus
13 #define TREELITE_EXTERN_C extern "C"
14 #include <cstdio>
15 #include <cstdint>
16 #else
17 #define TREELITE_EXTERN_C
18 #include <stdio.h>
19 #include <stdint.h>
20 #endif
21 
22 /* special symbols for DLL library on Windows */
23 #if defined(_MSC_VER) || defined(_WIN32)
24 #define TREELITE_DLL TREELITE_EXTERN_C __declspec(dllexport)
25 #else
26 #define TREELITE_DLL TREELITE_EXTERN_C
27 #endif
28 
30 typedef void* DMatrixHandle;
31 
37 TREELITE_DLL const char* TreeliteGetLastError(void);
38 
46 TREELITE_DLL int TreeliteRegisterLogCallback(void (*callback)(const char*));
47 
64 TREELITE_DLL int TreeliteDMatrixCreateFromCSR(
65  const void* data, const char* data_type, const uint32_t* col_ind, const size_t* row_ptr,
66  size_t num_row, size_t num_col, DMatrixHandle* out);
77 TREELITE_DLL int TreeliteDMatrixCreateFromMat(
78  const void* data, const char* data_type, size_t num_row, size_t num_col,
79  const void* missing_value, DMatrixHandle* out);
88 TREELITE_DLL int TreeliteDMatrixGetDimension(DMatrixHandle handle,
89  size_t* out_num_row,
90  size_t* out_num_col,
91  size_t* out_nelem);
92 
98 TREELITE_DLL int TreeliteDMatrixFree(DMatrixHandle handle);
101 #endif // TREELITE_C_API_COMMON_H_
int TreeliteRegisterLogCallback(void(*callback)(const char *))
register callback function for LOG(INFO) messages – helpful messages that are not errors...
Definition: c_api_common.cc:25
int TreeliteDMatrixFree(DMatrixHandle handle)
delete DMatrix from memory
Definition: c_api_common.cc:66
int TreeliteDMatrixCreateFromCSR(const void *data, const char *data_type, const uint32_t *col_ind, const size_t *row_ptr, size_t num_row, size_t num_col, DMatrixHandle *out)
create DMatrix from a (in-memory) CSR matrix
Definition: c_api_common.cc:32
void * DMatrixHandle
handle to a data matrix
Definition: c_api_common.h:30
int TreeliteDMatrixGetDimension(DMatrixHandle handle, size_t *out_num_row, size_t *out_num_col, size_t *out_nelem)
get dimensions of a DMatrix
Definition: c_api_common.cc:54
const char * TreeliteGetLastError(void)
display last error; can be called by multiple threads Note. Each thread will get the last error occur...
Definition: c_api_error.cc:17
int TreeliteDMatrixCreateFromMat(const void *data, const char *data_type, size_t num_row, size_t num_col, const void *missing_value, DMatrixHandle *out)
create DMatrix from a (in-memory) dense matrix
Definition: c_api_common.cc:43