aboutsummaryrefslogtreecommitdiff
path: root/sem1/algo/workshop2/dfs/graph.c
diff options
context:
space:
mode:
Diffstat (limited to 'sem1/algo/workshop2/dfs/graph.c')
-rw-r--r--sem1/algo/workshop2/dfs/graph.c123
1 files changed, 123 insertions, 0 deletions
diff --git a/sem1/algo/workshop2/dfs/graph.c b/sem1/algo/workshop2/dfs/graph.c
new file mode 100644
index 0000000..502b1a9
--- /dev/null
+++ b/sem1/algo/workshop2/dfs/graph.c
@@ -0,0 +1,123 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include "graph.h"
+
+static vertex_t *create_vertex(char ref) {
+ // Get some space TODO check for null
+ vertex_t *v = malloc(sizeof(vertex_t));
+
+ // Set values
+ v->ref = ref;
+ v->color = COLOR_WHITE;
+ v->p = NULL;
+ v->adj = NULL;
+ return v;
+}
+
+static edge_t *create_edge(vertex_t *from, vertex_t *to, int weight) {
+ edge_t *old = from->adj;
+
+ // Create new list node
+ edge_t *e = malloc(sizeof(edge_t));
+ e->weight = weight;
+ e->from = from;
+ e->to = to;
+
+ // Do new link
+ e->next = old;
+ if( old ) {
+ e->prev = old->prev;
+ old->prev = e;
+ } else { e->prev = NULL; }
+
+ if( e->prev ) {
+ e->prev->next = e;
+ }
+
+ from->adj = e;
+
+ return e;
+}
+
+// For iterating edges
+edge_t *edge_next(graph_t *g, edge_t *e) {
+ if(e == NULL || e->next == NULL ) {
+ // Find next index
+
+ // Chose starting index 0 if NULL e
+ unsigned char i = e ? e->from->ref+1 : 0;
+ for( ; i < 128; i++ ) {
+ if( g->vertexes[i] && g->vertexes[i]->adj ) {
+ return g->vertexes[i]->adj;
+ }
+ }
+ // No next vertex
+ return NULL;
+ }
+
+ // Not finished with this adj list
+ return e->next;
+}
+
+vertex_t *vertex_next(graph_t *g, vertex_t *v) {
+ unsigned char ref = v ? v->ref+1 : 0;
+ for( ; ref < 128; ref++ ) {
+ if( g->vertexes[ref] ) {
+ return g->vertexes[ref];
+ }
+ }
+
+ return NULL;
+}
+
+void graph_edge(graph_t *g, char from, char to, int weight) {
+ // Does a exists
+ if( g->vertexes[from] == NULL ) {
+ g->vertexes[from] = create_vertex(from);
+ }
+ // What about b
+ if( g->vertexes[to] == NULL ) {
+ g->vertexes[to] = create_vertex(to);
+ }
+
+ // Add edge
+ create_edge(g->vertexes[from], g->vertexes[to], weight);
+}
+
+int graph_print_adj(graph_t *g, char ref) {
+ if( g->vertexes[ref] == NULL ) {
+ return 1;
+ }
+
+ edge_t *e = g->vertexes[ref]->adj;
+ printf("[ ");
+ while(e && e->from->ref == ref) {
+ printf("%c[%d] ", e->to->ref, e->weight);
+ e = e->next;
+ }
+ printf("]\n");
+
+ return 0;
+}
+
+int graph_to_dot(FILE *f, graph_t *g) {
+ // print pre stuff
+ fprintf(f, "digraph coolgraph {\n");
+
+ // Label all nodes
+ vertex_t *v = vertex_next(g, NULL);
+ while( v ) {
+ fprintf(f, "%c [label=\"%c\"];\n", v->ref, v->ref);
+ v = vertex_next(g, v);
+ }
+ // Print all the edges
+ edge_t *e = edge_next(g, NULL);
+ while( e ) {
+ fprintf(f, "%c -> %c [label = %d%s];\n", e->from->ref, e->to->ref, e->weight, e->to->p == e->from ? " color=blue" : "");
+ e = edge_next(g, e);
+ }
+
+ // done
+ fprintf(f, "}\n");
+}
+