From 4dc0a92ee9442c0c2bd8ab49949b98b70bcd7507 Mon Sep 17 00:00:00 2001
From: Alexandria Barghi <abarghi@nvidia.com>
Date: Mon, 11 Sep 2023 10:23:25 -0700
Subject: [PATCH] fix c api issues

---
 cpp/include/cugraph_c/sampling_algorithms.h | 27 +++++++++++++++++++++
 cpp/src/c_api/uniform_neighbor_sampling.cpp | 10 ++++++++
 2 files changed, 37 insertions(+)

diff --git a/cpp/include/cugraph_c/sampling_algorithms.h b/cpp/include/cugraph_c/sampling_algorithms.h
index 36f27abef83..6ef813d59f3 100644
--- a/cpp/include/cugraph_c/sampling_algorithms.h
+++ b/cpp/include/cugraph_c/sampling_algorithms.h
@@ -205,6 +205,17 @@ typedef enum cugraph_prior_sources_behavior_t {
                    but exclude any vertex that has already been used as a source */
 } cugraph_prior_sources_behavior_t;
 
+/**
+ * @brief     Enumeration for compression type
+ */
+typedef enum cugraph_compression_type_t {
+  COO = 0,
+  CSR,
+  CSC,
+  DCSR,
+  DCSC
+} cugraph_compression_type_t;
+
 /**
  * @brief   Create sampling options object
  *
@@ -225,6 +236,14 @@ cugraph_error_code_t cugraph_sampling_options_create(cugraph_sampling_options_t*
  */
 void cugraph_sampling_set_renumber_results(cugraph_sampling_options_t* options, bool_t value);
 
+/**
+ * @brief   Set whether to compress per-hop (True) or globally (False)
+ * 
+ * @param options - opaque pointer to the sampling options
+ * @param value - Boolean value to assign to the option
+ */
+void cugraph_sampling_set_compress_per_hop(cugraph_sampling_options_t* options, bool_t value);
+
 /**
  * @brief   Set flag to sample with_replacement
  *
@@ -241,6 +260,14 @@ void cugraph_sampling_set_with_replacement(cugraph_sampling_options_t* options,
  */
 void cugraph_sampling_set_return_hops(cugraph_sampling_options_t* options, bool_t value);
 
+/**
+ * @brief   Set compression type
+ * 
+ * @param options - opaque pointer to the sampling options
+ * @param value - Enum defining the compresion type
+ */
+void cugraph_sampling_set_compression_type(cugraph_sampling_options_t* options, cugraph_compression_type_t value);
+
 /**
  * @brief   Set prior sources behavior
  *
diff --git a/cpp/src/c_api/uniform_neighbor_sampling.cpp b/cpp/src/c_api/uniform_neighbor_sampling.cpp
index dcf278c6d09..17b66ec8aaa 100644
--- a/cpp/src/c_api/uniform_neighbor_sampling.cpp
+++ b/cpp/src/c_api/uniform_neighbor_sampling.cpp
@@ -382,6 +382,11 @@ extern "C" void cugraph_sampling_set_renumber_results(cugraph_sampling_options_t
   internal_pointer->renumber_results_ = value;
 }
 
+extern "C" void cugraph_sampling_set_compress_per_hop(cugraph_sampling_options_t* options, bool_t value) {
+  auto internal_pointer = reinterpret_cast<cugraph::c_api::cugraph_sampling_options_t*>(options);
+  internal_pointer->compress_per_hop_ = value;
+}
+
 extern "C" void cugraph_sampling_set_with_replacement(cugraph_sampling_options_t* options,
                                                       bool_t value)
 {
@@ -395,6 +400,11 @@ extern "C" void cugraph_sampling_set_return_hops(cugraph_sampling_options_t* opt
   internal_pointer->return_hops_ = value;
 }
 
+extern "C" void cugraph_sampling_set_compression_type(cugraph_sampling_options_t* options, cugraph_compression_type_t value) {
+  auto internal_pointer = reinterpret_cast<cugraph::c_api::cugraph_sampling_options_t*>(options);
+  internal_pointer->compression_type_ = value;
+}
+
 extern "C" void cugraph_sampling_set_prior_sources_behavior(cugraph_sampling_options_t* options,
                                                             cugraph_prior_sources_behavior_t value)
 {