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
|
/* gsl-ocaml - OCaml interface to GSL */
/* Copyright (©) 2002-2012 - Olivier Andrieu */
/* Distributed under the terms of the GPL version 3 */
#include <gsl/gsl_vector.h>
#include <caml/bigarray.h>
#include "wrappers.h"
#ifndef TYPE
#error pb with include files
#endif
static inline void TYPE(mlgsl_vec_of_bigarray)(TYPE(gsl_vector) *cvec,
value vvec){
struct caml_bigarray *bigarr = Bigarray_val(vvec);
cvec->block = NULL;
cvec->owner = 0;
cvec->size = bigarr->dim[0];
cvec->stride = 1;
cvec->data = bigarr->data;
}
#ifdef CONV_FLAT
static inline void TYPE(mlgsl_vec_of_floatarray)(TYPE(gsl_vector) *cvec,
value vvec){
cvec->block = NULL;
cvec->owner = 0;
cvec->size = Int_val(Field(vvec, 2));
cvec->stride = Int_val(Field(vvec, 3));
cvec->data = (double *)Field(vvec, 0) + Int_val(Field(vvec, 1));
}
#endif
static inline void TYPE(mlgsl_vec_of_value)(TYPE(gsl_vector) *cvec,
value vvec){
if(Tag_val(vvec) == 0 && Wosize_val(vvec) == 2)
/* value is a polymorphic variant */
vvec = Field(vvec, 1);
if(Tag_val(vvec) == Custom_tag)
/* value is a bigarray */
TYPE(mlgsl_vec_of_bigarray)(cvec, vvec);
#ifdef CONV_FLAT
else
/* value is a record wrapping a float array */
TYPE(mlgsl_vec_of_floatarray)(cvec, vvec);
#endif
}
#define _DECLARE_VECTOR(a) TYPE(gsl_vector) v_##a
#define _DECLARE_VECTOR2(a,b) _DECLARE_VECTOR(a); _DECLARE_VECTOR(b)
#define _DECLARE_VECTOR3(a,b,c) _DECLARE_VECTOR2(a,b); _DECLARE_VECTOR(c)
#define _DECLARE_VECTOR4(a,b,c,d) _DECLARE_VECTOR2(a,b); _DECLARE_VECTOR2(c,d)
#define _DECLARE_VECTOR5(a,b,c,d,e) _DECLARE_VECTOR4(a,b,c,d); _DECLARE_VECTOR(e)
#define _CONVERT_VECTOR(a) TYPE(mlgsl_vec_of_value)(&v_##a, a)
#define _CONVERT_VECTOR2(a,b) _CONVERT_VECTOR(a); _CONVERT_VECTOR(b)
#define _CONVERT_VECTOR3(a,b,c) _CONVERT_VECTOR2(a,b); _CONVERT_VECTOR(c)
#define _CONVERT_VECTOR4(a,b,c,d) _CONVERT_VECTOR2(a,b); _CONVERT_VECTOR2(c,d)
#define _CONVERT_VECTOR5(a,b,c,d,e) _CONVERT_VECTOR4(a,b,c,d); _CONVERT_VECTOR(e)
|