#include <stdio.h>
#include "gen.h"

void my_gen_fn (gen_int_t gen)
{
	gen_int_yield(gen, 1);
	gen_int_yield(gen, 2);
	gen_int_yield(gen, 3);
}

void example_while (void)
{
	gen_int_t gen;
	int v;
	gen = gen_int_create(my_gen_fn, SIGSTKSZ);

	while (gen_int_next(gen, &v))
	{
		printf("Got: %d\n", v);
	}
}

void example_for (void)
{
	gen_int_t gen;
	int v;

	for (gen = gen_int_create(my_gen_fn, SIGSTKSZ); gen_int_next(gen, &v); )
	{
		printf("Got: %d\n", v);
	}
}

int main (int argc, char **argv)
{
	example_while();
	example_for();

	return 0;
}