add km_set_print(), refactor main.c

1 files changed, 11 insertions, 1 deletions

diff --git a/src/km-set.c b/src/km-set.c
index 69ae6c0..fcf1054 100644
--- a/src/km-set.c
+++ b/src/km-set.c
@@ -263,5 +263,15 @@ km_set_get_row(
   const size_t i
 ) {
   const size_t num_floats = set->shape.num_floats;
-  return set->floats + i * num_floats;
+  return (num_floats) ? (set->floats + i * num_floats) : NULL;
+}
+
+// get row from data set
+int *
+km_set_get_row_ints(
+  const km_set_t * const set,
+  const size_t i
+) {
+  const size_t num_ints = set->shape.num_ints;
+  return (num_ints) ? (set->ints + i * num_ints) : NULL;
 }