add cluster support to gen-data.rb, show centroids on generated png

1 files changed, 8 insertions, 0 deletions

diff --git a/km.h b/km.h
index ae1791d..82bd835 100644
--- a/km.h
+++ b/km.h
@@ -131,6 +131,12 @@ typedef _Bool (*km_find_fini_cb_t)(
   void *cb_data
 );
 
+typedef _Bool (*km_find_best_cb_t)(
+  const float score,
+  const km_set_t * const cluster_set,
+  void *cb_data
+);
+
 typedef void (*km_find_data_cb_t)(
   const km_find_data_t * const,
   void *
@@ -143,6 +149,7 @@ typedef struct {
   km_find_init_cb_t on_init;
   km_find_fini_cb_t on_fini;
   km_find_data_cb_t on_data;
+  km_find_best_cb_t on_best;
 } km_find_cbs_t;
 
 // repeatedly test different cluster sizes and report results
@@ -160,6 +167,7 @@ km_set_draw(
   uint8_t * const rgb,
   const size_t width,
   const size_t height,
+  const int dot_size,
   const uint32_t color
 );