61 lines
1.5 KiB
C
61 lines
1.5 KiB
C
#include <ngi.h>
|
|
#include <assert.h>
|
|
#include <float.h>
|
|
#include <math.h>
|
|
#include <dlfcn.h>
|
|
|
|
int server_reduce(FILE * file, ngi_op_t op, ngi_type_t type, void * out_buff, size_t off, size_t size){
|
|
fseek(file, off, SEEK_SET);
|
|
|
|
assert(type == NGI_TYPE_FLOAT);
|
|
assert(op == NGI_OP_MAX);
|
|
|
|
float * data = malloc(size);
|
|
fread(data, size, 1, file);
|
|
|
|
float mx = -INFINITY;
|
|
for(size_t i=0; i < size/sizeof(float); i++){
|
|
mx = data[i] > mx ? data[i]: mx;
|
|
}
|
|
*(float*) out_buff = mx;
|
|
|
|
free(data);
|
|
return 0;
|
|
}
|
|
|
|
int server_reduce_any_func(FILE * file, void (*func)(int8_t * data, void * out_buff, size_t off, size_t size), void * out_buff, size_t max_out_data_size, size_t off, size_t size){
|
|
fseek(file, off, SEEK_SET);
|
|
|
|
int8_t * data = malloc(size);
|
|
fread(data, size, 1, file);
|
|
func(data, out_buff, off, size);
|
|
free(data);
|
|
return 0;
|
|
}
|
|
|
|
int server_reduce_any_func_lib(FILE * file, char const * libname, char const * funcname, void * out_buff, size_t max_out_data_size, size_t off, size_t size){
|
|
void *handle;
|
|
handle = dlopen(libname, RTLD_LAZY);
|
|
if(! handle){
|
|
printf("Error %s\n", dlerror());
|
|
return 1;
|
|
}
|
|
|
|
fseek(file, off, SEEK_SET);
|
|
|
|
void (*func)(int8_t * data, void * out_buff, size_t off, size_t size) = dlsym(handle, funcname);
|
|
if(! func){
|
|
printf("Error %s\n", dlerror());
|
|
dlclose(handle);
|
|
return 1;
|
|
}
|
|
|
|
int8_t * data = malloc(size);
|
|
fread(data, size, 1, file);
|
|
func(data, out_buff, off, size);
|
|
free(data);
|
|
|
|
dlclose(handle);
|
|
return 0;
|
|
}
|