-
-
Notifications
You must be signed in to change notification settings - Fork 1k
/
RInterface.h
180 lines (151 loc) · 5.69 KB
/
RInterface.h
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
#ifndef __RINTERFACE__H_
#define __RINTERFACE__H_
#include <shogun/lib/config.h>
#include <shogun/lib/common.h>
#include <shogun/ui/SGInterface.h>
extern "C" {
#include <R.h>
#include <Rdefines.h>
}
using namespace shogun;
class CRInterface : public CSGInterface
{
public:
CRInterface(SEXP prhs, bool skip=true);
~CRInterface();
/// reset to clean state
virtual void reset(SEXP prhs);
/** get functions - to pass data from the target interface to shogun */
/// get type of current argument (does not increment argument counter)
virtual IFType get_argument_type();
virtual int32_t get_int();
virtual float64_t get_real();
virtual bool get_bool();
virtual char* get_string(int32_t& len);
virtual void get_vector(uint8_t*& vec, int32_t& len);
virtual void get_vector(char*& vec, int32_t& len);
virtual void get_vector(int32_t*& vec, int32_t& len);
virtual void get_vector(float32_t*& vec, int32_t& len);
virtual void get_vector(float64_t*& vec, int32_t& len);
virtual void get_vector(int16_t*& vec, int32_t& len);
virtual void get_vector(uint16_t*& vec, int32_t& len);
virtual void get_matrix(
uint8_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
char*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
int32_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
float32_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
float64_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
int16_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_matrix(
uint16_t*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_ndarray(
uint8_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
char*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
int32_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
float32_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
float64_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
int16_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_ndarray(
uint16_t*& array, int32_t*& dims, int32_t& num_dims);
virtual void get_sparse_matrix(
SGSparseVector<float64_t>*& matrix, int32_t& num_feat, int32_t& num_vec);
virtual void get_string_list(
SGString<uint8_t>*& strings, int32_t& num_str,
int32_t& max_string_len);
virtual void get_string_list(
SGString<char>*& strings, int32_t& num_str,
int32_t& max_string_len);
virtual void get_string_list(
SGString<int32_t>*& strings, int32_t& num_str,
int32_t& max_string_len);
virtual void get_string_list(
SGString<int16_t>*& strings, int32_t& num_str,
int32_t& max_string_len);
virtual void get_string_list(
SGString<uint16_t>*& strings, int32_t& num_str,
int32_t& max_string_len);
virtual void get_attribute_struct(
const CDynamicArray<T_ATTRIBUTE>* &attrs);
/** set functions - to pass data from shogun to the target interface */
virtual void set_int(int32_t scalar);
virtual void set_real(float64_t scalar);
virtual void set_bool(bool scalar);
virtual bool create_return_values(int32_t num_val);
virtual void set_vector(const uint8_t* vec, int32_t len);
virtual void set_vector(const char* vec, int32_t len);
virtual void set_vector(const int32_t* vec, int32_t len);
virtual void set_vector(const float32_t* vec, int32_t len);
virtual void set_vector(const float64_t* vec, int32_t len);
virtual void set_vector(const int16_t* vec, int32_t len);
virtual void set_vector(const uint16_t* vec, int32_t len);
virtual void set_matrix(
const uint8_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const char* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const int32_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const float32_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const float64_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const int16_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_matrix(
const uint16_t* matrix, int32_t num_feat, int32_t num_vec);
virtual void set_sparse_matrix(
const SGSparseVector<float64_t>* matrix, int32_t num_feat,
int32_t num_vec, int64_t nnz);
virtual void set_string_list(
const SGString<uint8_t>* strings, int32_t num_str);
virtual void set_string_list(
const SGString<char>* strings, int32_t num_str);
virtual void set_string_list(
const SGString<int32_t>* strings, int32_t num_str);
virtual void set_string_list(
const SGString<int16_t>* strings, int32_t num_str);
virtual void set_string_list(
const SGString<uint16_t>* strings, int32_t num_str);
virtual void set_attribute_struct(
const CDynamicArray<T_ATTRIBUTE>* attrs);
SEXP get_return_values();
virtual bool cmd_run_python();
virtual bool cmd_run_octave();
static void run_r_init();
static void run_r_exit();
static bool run_r_helper(CSGInterface* from_if);
private:
const SEXP get_arg_increment()
{
ASSERT(m_rhs_counter>=0 && m_rhs_counter<m_nrhs+1); // +1 for action
SEXP retval=R_NilValue;
if (m_rhs)
retval=CAR(m_rhs);
if (m_rhs)
m_rhs=CDR(m_rhs);
m_rhs_counter++;
return retval;
}
void set_arg_increment(SEXP arg)
{
ASSERT(m_lhs_counter>=0 && m_lhs_counter<m_nlhs);
SET_VECTOR_ELT(m_lhs, m_lhs_counter, arg);
m_lhs_counter++;
}
/** @return object name */
inline virtual const char* get_name() const { return "RInterface"; }
private:
SEXP m_lhs;
SEXP m_rhs;
bool skip_value;
};
#endif // __RINTERFACE__H_