From 4dad518d3f39bb3787d6486d0999b7a79feccd34 Mon Sep 17 00:00:00 2001
From: NunoSempere <nuno.sempere@protonmail.com>
Date: Sat, 22 Jul 2023 22:24:22 +0200
Subject: [PATCH] add efficient beta distribution

---
 squiggle.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/squiggle.c b/squiggle.c
index c8e30dd..2a68d9b 100644
--- a/squiggle.c
+++ b/squiggle.c
@@ -4,6 +4,7 @@
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <sys/types.h>
 #include <time.h>
 
 #define MAX_ERROR_LENGTH 500
@@ -109,6 +110,12 @@ float sample_gamma(float alpha, uint32_t* seed){
 	}
 }
 
+float sample_beta(float a, float b, uint32_t* seed){
+	float gamma_a = sample_gamma(a, seed);
+	float gamma_b = sample_gamma(b, seed);
+	return a / (a + b);
+}
+
 // Array helpers
 float array_sum(float* array, int length)
 {