example.c (935B)
1 #include "../../../squiggle.h" 2 #include <stdio.h> 3 #include <stdlib.h> 4 5 // Estimate functions 6 7 double sample_0(uint64_t* seed) { UNUSED(seed); return 0; } 8 double sample_1(uint64_t* seed) { UNUSED(seed); return 1; } 9 double sample_few(uint64_t* seed) { return sample_to(1, 3, seed); } 10 double sample_many(uint64_t* seed) { return sample_to(2, 10, seed); } 11 12 double sample_model(uint64_t* seed){ 13 14 double p_a = 0.8; 15 double p_b = 0.5; 16 double p_c = p_a * p_b; 17 18 int n_dists = 4; 19 double weights[] = { 1 - p_c, p_c / 2, p_c / 4, p_c / 4 }; 20 double (*samplers[])(uint64_t*) = { sample_0, sample_1, sample_few, sample_many }; 21 double result = sample_mixture(samplers, weights, n_dists, seed); 22 23 return result; 24 } 25 26 int main() 27 { 28 // set randomness seed 29 uint64_t* seed = malloc(sizeof(uint64_t)); 30 *seed = 1000; // xorshift can't start with 0 31 32 printf("result_one: %f\n", sample_model(seed)); 33 free(seed); 34 }