Skip to content

Commit a93062a

Browse files
committed
Refactor C
1 parent 99da87f commit a93062a

File tree

2 files changed

+30
-30
lines changed

2 files changed

+30
-30
lines changed

catboost/catboost_wrapper.c

Lines changed: 30 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,30 @@
11
#include "catboost_wrapper.h"
22

3+
typedef const char* (*TypeGetErrorString) (void);
4+
typedef ModelCalcerHandle* (*TypeModelCalcerCreate) (void);
5+
typedef bool (*TypeLoadFullModelFromBuffer) (ModelCalcerHandle* modelHandle, const void* binaryBuffer, size_t binaryBufferSize);
6+
typedef bool (*TypeCalcModelPredictionSingle) (
7+
ModelCalcerHandle* modelHandle,
8+
const float* floatFeatures, size_t floatFeaturesSize,
9+
const char** catFeatures, size_t catFeaturesSize,
10+
double* result, size_t resultSize);
11+
typedef bool (*TypeCalcModelPrediction) (
12+
ModelCalcerHandle* modelHandle,
13+
size_t docCount,
14+
const float** floatFeatures, size_t floatFeaturesSize,
15+
const char*** catFeatures, size_t catFeaturesSize,
16+
double* result, size_t resultSize);
17+
typedef size_t (*TypeGetFloatFeaturesCount) (ModelCalcerHandle* modelHandle);
18+
typedef size_t (*TypeGetCatFeaturesCount)(ModelCalcerHandle* modelHandle);
19+
typedef size_t (*TypeGetDimensionsCount) (ModelCalcerHandle* modelHandle);
20+
typedef bool (*TypeSetPredictionTypeString) (ModelCalcerHandle* modelHandle, const char* predictionTypeStr);
21+
typedef bool (*TypeGetModelUsedFeaturesNames) (ModelCalcerHandle* modelHandle, char*** featureNames, size_t* featureCount);
22+
typedef const char* (*TypeGetModelInfoValue) (ModelCalcerHandle* modelHandle, const char* keyPtr, size_t keySize);
23+
typedef bool (*TypeGetCatFeatureIndices) (ModelCalcerHandle* modelHandle, size_t** indices, size_t* count);
24+
typedef bool (*TypeGetFloatFeatureIndices) (ModelCalcerHandle* modelHandle, size_t** indices, size_t* count);
25+
typedef bool (*TypeGetSupportedEvaluatorTypes) (ModelCalcerHandle* modelHandle, size_t** formulaEvaluatorTypes, size_t* count);
26+
typedef bool (*TypeEnableGPUEvaluation) (ModelCalcerHandle* modelHandle, int deviceId);
27+
328
static TypeGetErrorString GetErrorStringFn = NULL;
429
static TypeModelCalcerCreate ModelCalcerCreateFn = NULL;
530
static TypeLoadFullModelFromBuffer LoadFullModelFromBufferFn = NULL;
@@ -155,11 +180,11 @@ char** makeCharArray1D(int size) {
155180
return calloc(sizeof(char*), size);
156181
}
157182

158-
void freeCharArray1D(char **array, int size) {
183+
void freeCharArray1D(char **a, int size) {
159184
int i;
160185
for (i = 0; i < size; i++)
161-
free(array[i]);
162-
free(array);
186+
free(a[i]);
187+
free(a);
163188
}
164189

165190
void freeCharArray2D(char ***a, int sizeX, int sizeY) {
@@ -169,8 +194,8 @@ void freeCharArray2D(char ***a, int sizeX, int sizeY) {
169194
free(a);
170195
}
171196

172-
void setCharArray1D(char **array, char *s, int n) {
173-
array[n] = s;
197+
void setCharArray1D(char **a, char *s, int n) {
198+
a[n] = s;
174199
}
175200

176201
void setCharArray2D(char ***a, char **s, int n) {

catboost/catboost_wrapper.h

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -2,31 +2,6 @@
22
#include <stdlib.h>
33
#include "c_api.h"
44

5-
typedef const char* (*TypeGetErrorString) (void);
6-
typedef ModelCalcerHandle* (*TypeModelCalcerCreate) (void);
7-
typedef bool (*TypeLoadFullModelFromBuffer) (ModelCalcerHandle* modelHandle, const void* binaryBuffer, size_t binaryBufferSize);
8-
typedef bool (*TypeCalcModelPredictionSingle) (
9-
ModelCalcerHandle* modelHandle,
10-
const float* floatFeatures, size_t floatFeaturesSize,
11-
const char** catFeatures, size_t catFeaturesSize,
12-
double* result, size_t resultSize);
13-
typedef bool (*TypeCalcModelPrediction) (
14-
ModelCalcerHandle* modelHandle,
15-
size_t docCount,
16-
const float** floatFeatures, size_t floatFeaturesSize,
17-
const char*** catFeatures, size_t catFeaturesSize,
18-
double* result, size_t resultSize);
19-
typedef size_t (*TypeGetFloatFeaturesCount) (ModelCalcerHandle* modelHandle);
20-
typedef size_t (*TypeGetCatFeaturesCount)(ModelCalcerHandle* modelHandle);
21-
typedef size_t (*TypeGetDimensionsCount) (ModelCalcerHandle* modelHandle);
22-
typedef bool (*TypeSetPredictionTypeString) (ModelCalcerHandle* modelHandle, const char* predictionTypeStr);
23-
typedef bool (*TypeGetModelUsedFeaturesNames) (ModelCalcerHandle* modelHandle, char*** featureNames, size_t* featureCount);
24-
typedef const char* (*TypeGetModelInfoValue) (ModelCalcerHandle* modelHandle, const char* keyPtr, size_t keySize);
25-
typedef bool (*TypeGetCatFeatureIndices) (ModelCalcerHandle* modelHandle, size_t** indices, size_t* count);
26-
typedef bool (*TypeGetFloatFeatureIndices) (ModelCalcerHandle* modelHandle, size_t** indices, size_t* count);
27-
typedef bool (*TypeGetSupportedEvaluatorTypes) (ModelCalcerHandle* modelHandle, size_t** formulaEvaluatorTypes, size_t* count);
28-
typedef bool (*TypeEnableGPUEvaluation) (ModelCalcerHandle* modelHandle, int deviceId);
29-
305
void SetGetErrorStringFn(void *fn);
316
void SetCalcModelPredictionSingleFn(void *fn);
327
void SetModelCalcerCreateFn(void *fn);

0 commit comments

Comments
 (0)