From 39161cf246aa2b34c7c02ebb6c88da3776979670 Mon Sep 17 00:00:00 2001
From: kartofen <mladenovnasko0@gmail.com>
Date: Tue, 23 Aug 2022 20:45:56 +0300
Subject: Big Bang

---
 src/gen_chain.c |  32 ++++++++++++++
 src/main.c      |  24 +++++++++++
 src/markov.h    | 130 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 186 insertions(+)
 create mode 100644 src/gen_chain.c
 create mode 100644 src/main.c
 create mode 100644 src/markov.h

(limited to 'src')

diff --git a/src/gen_chain.c b/src/gen_chain.c
new file mode 100644
index 0000000..636d3ea
--- /dev/null
+++ b/src/gen_chain.c
@@ -0,0 +1,32 @@
+#include <stdio.h>
+
+#define PRINT_ITEM_FREQUENCY
+// #define PRINT_ON_WALK
+#define SAVE_TO_FILE_ON_WALK "files/chain.txt"
+#define WALK_LEN 100000
+
+#define ITEM_CAP 3
+int ITEMS = 3;
+
+double chain[ITEM_CAP][ITEM_CAP] = {
+    { 0.2, 0.6, 0.2 },
+    { 0.3, 0.0, 0.7 },
+    { 0.5, 0.0, 0.5 }
+};
+
+char item_names[ITEM_CAP][64] = {
+    "burger",
+    "pizza",
+    "hotdog"
+};
+
+#include "markov.h"
+
+int main(void)
+{
+    srand(time(NULL));
+
+    print_chain();
+    take_walk();
+    return 0;
+}
diff --git a/src/main.c b/src/main.c
new file mode 100644
index 0000000..e1a942e
--- /dev/null
+++ b/src/main.c
@@ -0,0 +1,24 @@
+#include <stdio.h>
+
+#define PRINT_ITEM_FREQUENCY
+// #define PRINT_ON_WALK
+// #define SAVE_TO_FILE_ON_WALK "walk.txt"
+#define WALK_LEN 1000000
+
+#define ITEM_CAP 30
+int ITEMS = 0;
+
+double chain[ITEM_CAP][ITEM_CAP] = {0};
+char item_names[ITEM_CAP][64] = {0};
+
+#include "markov.h"
+
+int main(void)
+{
+    srand(time(NULL));
+    generate_chain("files/chain.txt");
+
+    print_chain();
+    take_walk();
+    return 0;
+}
diff --git a/src/markov.h b/src/markov.h
new file mode 100644
index 0000000..052af53
--- /dev/null
+++ b/src/markov.h
@@ -0,0 +1,130 @@
+#ifndef MARKOV_H
+#define MARKOV_H
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <assert.h>
+#include <string.h>
+#include <time.h>
+
+extern int ITEMS;
+extern double chain[][ITEM_CAP];
+extern char item_names[][64];
+
+void generate_chain(char *file_path)
+{
+    FILE *fp = fopen(file_path, "r");
+    if(!fp) {
+        fprintf(stderr, "ERROR: Could not open file %s\n", file_path);
+        exit(EXIT_FAILURE);
+    }
+
+    int items_seen[ITEM_CAP][ITEM_CAP] = {0};
+    int cur_item = -1;
+
+    char line[256] = {0};
+    while((fgets(line, sizeof(line), fp)) != NULL)
+    {
+        if(line[strlen(line)-1] == '\n')
+            line[strlen(line)-1] = '\0';
+
+        int item = ITEMS;
+
+        for(int i = 0; i < ITEMS; i++)
+        {
+            if(strcmp(line, item_names[i]) == 0)
+                item = i;
+        }
+
+        if(item == ITEMS) {
+            ITEMS++;
+            strcpy(item_names[item], line);
+        }
+
+        if(cur_item != -1)
+            items_seen[cur_item][item]++;
+
+        cur_item = item;
+    }
+
+    fclose(fp);
+
+    for(int i = 0; i < ITEMS; i++)
+    {
+        int all = 0;
+        for(int j = 0; j < ITEMS; j++)
+            all += items_seen[i][j];
+
+        for(int j = 0; j < ITEMS; j++)
+            chain[i][j] = (double)items_seen[i][j]/all;
+    }
+}
+
+int walk_step(int item)
+{
+    int item_lim = 0;
+    int r = rand();
+
+    for(int i = 0; i < ITEMS; i++)
+    {
+        item_lim += RAND_MAX * chain[item][i];
+        if(r <= item_lim) return i;
+    }
+
+    return -1;
+}
+
+void take_walk()
+{
+#ifdef SAVE_TO_FILE_ON_WALK
+    FILE *fp = fopen(SAVE_TO_FILE_ON_WALK, "w");
+    if(!fp) {
+        fprintf(stderr, "ERROR: Could not open file %s\n", SAVE_TO_FILE_ON_WALK);
+        exit(EXIT_FAILURE);
+    }
+#endif
+
+    int items_seen[ITEM_CAP] = {0};
+    int cur_item = rand() % ITEMS;
+
+    for(size_t i = 0; i < WALK_LEN; i++)
+    {
+    #ifdef PRINT_ON_WALK
+        puts(item_names[cur_item]);
+    #endif
+    #ifdef SAVE_TO_FILE_ON_WALK
+        fprintf(fp, "%s\n", item_names[cur_item]);
+    #endif
+        items_seen[cur_item]++;
+        int ni = walk_step(cur_item);
+        assert(ni != -1);
+        cur_item = ni;
+    }
+
+#ifdef SAVE_TO_FILE_ON_WALK
+    fclose(fp);
+    printf("Saved file with %d items at %s\n", WALK_LEN, SAVE_TO_FILE_ON_WALK);
+#endif
+
+#ifdef PRINT_ITEM_FREQUENCY
+    puts("Item Frequency:");
+    for(int i = 0; i < ITEMS; i++)
+        printf("%f %s\n", (double)items_seen[i]/WALK_LEN, item_names[i]);
+    puts("");
+#endif
+}
+
+void print_chain()
+{
+    puts("Chain Matrix:");
+    for(int i = 0; i < ITEMS; i++)
+    {
+        printf("{");
+        for(int j = 0; j < ITEMS; j++)
+            printf(" %f,", chain[i][j]);
+        printf("}\n");
+    }
+    puts("");
+}
+
+#endif
-- 
cgit v1.2.3