squiggle.c

Self-contained Monte Carlo estimation in C99
Log | Files | Refs | README

example.c (1022B)

      1 #include "../../../squiggle.h"
      2 #include "../../../squiggle_more.h"
      3 #include <stdio.h>
      4 #include <stdlib.h>
      5 
      6 double sampler_result(uint64_t * seed)
      7 {
      8     double p_a = 0.8;
      9     double p_b = 0.5;
     10     double p_c = p_a * p_b;
     11 
     12     double sample_0(uint64_t * seed) { UNUSED(seed); return 0; }
     13     double sample_1(uint64_t * seed) { UNUSED(seed); return 1; }
     14     double sample_few(uint64_t * seed) { return sample_to(1, 3, seed); }
     15     double sample_many(uint64_t * seed) { return sample_to(2, 10, seed); }
     16 
     17     int n_dists = 4;
     18     double weights[] = { 1 - p_c, p_c / 2, p_c / 4, p_c / 4 };
     19     double (*samplers[])(uint64_t*) = { sample_0, sample_1, sample_few, sample_many };
     20     return sample_mixture(samplers, weights, n_dists, seed);
     21 }
     22 
     23 int main()
     24 {
     25 
     26     int n_samples = 1000 * 1000, n_threads = 16;
     27     double* results = malloc((size_t)n_samples * sizeof(double));
     28     sampler_parallel(sampler_result, results, n_threads, n_samples);
     29     printf("Avg: %f\n", array_sum(results, n_samples) / n_samples);
     30     free(results);
     31 }