X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fsn_network.c;fp=src%2Fsn_network.c;h=8462020e38bc3d909541e609f79fcbd1dfc7b287;hb=927c3f062bda3494fbbd5803c5d08c3552138b89;hp=d8403c0527e57630d5bb06e7c3d2da422307b647;hpb=8398b792d7a73a06353855baa8fa941645ca2544;p=sort-networks.git

diff --git a/src/sn_network.c b/src/sn_network.c
index d8403c0..8462020 100644
--- a/src/sn_network.c
+++ b/src/sn_network.c
@@ -219,6 +219,47 @@ sn_network_t *sn_network_clone (const sn_network_t *n) /* {{{ */
   return (n_copy);
 } /* }}} sn_network_t *sn_network_clone */
 
+int sn_network_comparator_add (sn_network_t *n, /* {{{ */
+    const sn_comparator_t *c)
+{
+  sn_stage_t *s;
+
+  if ((n == NULL) || (c == NULL))
+    return (-1);
+
+  if (n->stages_num > 0)
+  {
+    s = n->stages[n->stages_num - 1];
+    
+    if (sn_stage_comparator_check_conflict (s, c) == 0)
+    {
+      sn_stage_comparator_add (s, c);
+      return (0);
+    }
+  }
+
+  s = sn_stage_create (n->stages_num);
+  sn_stage_comparator_add (s, c);
+  sn_network_stage_add (n, s);
+
+  return (0);
+} /* }}} int sn_network_comparator_add */
+
+int sn_network_get_comparator_num (const sn_network_t *n) /* {{{ */
+{
+  int num;
+  int i;
+
+  if (n == NULL)
+    return (-1);
+
+  num = 0;
+  for (i = 0; i < n->stages_num; i++)
+    num += n->stages[i]->comparators_num;
+
+  return (num);
+} /* }}} int sn_network_get_comparator_num */
+
 int sn_network_show (sn_network_t *n) /* {{{ */
 {
   int i;