X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=ccan%2Fagar%2Ftest%2Fparallel.c;fp=ccan%2Fagar%2Ftest%2Fparallel.c;h=741ff3a37033a3775aebc7ecf781afe3d283abc6;hb=c2966d1879c825cfaf0e7d6848a5da052ee4a038;hp=0000000000000000000000000000000000000000;hpb=06162212353c882249d7e207756ea81ea645fc30;p=ccan

diff --git a/ccan/agar/test/parallel.c b/ccan/agar/test/parallel.c
new file mode 100644
index 00000000..741ff3a3
--- /dev/null
+++ b/ccan/agar/test/parallel.c
@@ -0,0 +1,62 @@
+#include "config.h"
+
+#include <assert.h>
+
+#include <ccan/agar/agar.h>
+#include <ccan/container_of/container_of.h>
+#include <ccan/ptrint/ptrint.h>
+
+#include "simple-graphr.h"
+
+static const void *parallel_first_edge_r(const struct agar_graph *gr,
+					 const void *nr)
+{
+	const struct parallel_graphr *pgr
+		= container_of(gr, struct parallel_graphr, gr);
+
+	if (ptr2int(nr) != 1) {
+		assert(ptr2int(nr) == 2);
+		return NULL;
+	}
+
+	if (pgr->nlinks)
+		return int2ptr(1);
+	else
+		return NULL;
+}
+
+static const void *parallel_next_edge_r(const struct agar_graph *gr,
+					const void *nr, const void *edge)
+{
+	const struct parallel_graphr *pgr
+		= container_of(gr, struct parallel_graphr, gr);
+	int index = ptr2int(edge);
+
+	if (ptr2int(nr) != 1) {
+		assert(ptr2int(nr) == 2);
+		return NULL;
+	}
+
+	if (index < pgr->nlinks)
+		return int2ptr(index + 1);
+	else
+		return NULL;
+}
+
+static int parallel_edge_info_r(const struct agar_graph *gr,
+				const void *nr, const void *edge,
+				struct agar_edge_info *eir)
+{
+	assert(ptr2int(nr) == 1);
+
+	eir->to = int2ptr(2);
+	return 0;
+}
+
+void parallel_graphr_init(struct parallel_graphr *pgr, int nlinks)
+{
+	pgr->nlinks = nlinks;
+
+	agar_init_graph(&pgr->gr, parallel_first_edge_r, parallel_next_edge_r,
+			parallel_edge_info_r);
+}