spacepaste

Paste details

reply | raw

caml_pam.c

  1.  
  2. #include <string.h>
  3. #include <caml/bigarray.h>
  4. #include <caml/mlvalues.h>
  5. #include <caml/memory.h>
  6. #include <caml/alloc.h>
  7. #include <caml/custom.h>
  8. #include <gsl/gsl_matrix.h>
  10. #include <stdio.h>
  12. size_t *pam(gsl_matrix *, size_t, /*OUT*/ float *);
  14. /* rows for leaves; columns for masses */
  15. CAMLprim value caml_pam(value k_value, value dist_value)
  16. {
  17. CAMLparam2(k_value, dist_value);
  18. double *dist = Data_bigarray_val(dist_value);
  19. gsl_matrix_view m;
  20. float work;
  21. size_t *medoids, *medoids_ptr;
  22. size_t nrow, ncol;
  23. value res_bigarr, *res_ptr;
  24. int i, k;
  25. nrow = Bigarray_val(dist_value)->dim[0];
  26. ncol = Bigarray_val(dist_value)->dim[1];
  27. m = gsl_matrix_view_array(dist, nrow, ncol);
  28. k = Int_val(k_value);
  29. medoids = pam(&m.matrix, k, &work);
  31. res_bigarr = alloc_bigarray_dims(BIGARRAY_CAML_INT | BIGARRAY_C_LAYOUT, 1, NULL, k);
  32. res_ptr = Data_bigarray_val(res_bigarr);
  33. medoids_ptr = medoids;
  34. for (i = 0; i < k; ++i) {
  35. *res_ptr++ = Val_long(*medoids_ptr++);
  36. }
  37. free(medoids);
  38. CAMLreturn(res_bigarr);
  39. }
  40.  

pam_solver.ml

  1.  
  2. module BA = Bigarray
  3. module BA1 = Bigarray.Array1
  5. type int_vector = (int, BA.int_elt, BA.c_layout) BA1.t
  6. external c_pam: int -> Matrix.matrix -> int_vector = "caml_pam"
  7.  
Powered by spacepaste. Paste removal.