/* nag_rand_geom (g05tcc) Example Program. * * Copyright 2008, Numerical Algorithms Group. * * Mark 9, 2009. */ /* Pre-processor includes */ #include #include #include #include #include int main(void) { /* Integer scalar and array declarations */ Integer exit_status = 0; Integer i, lr, lstate; Integer *state = 0, *x = 0; /* NAG structures */ NagError fail; /* Double scalar and array declarations */ double *r = 0; /* Set the distribution parameters */ double p = 0.0010e0; /* Set the mode we will be using. As p is small we will not use a reference vector */ Nag_ModeRNG mode = Nag_GenerateWithoutReference; /* Set the sample size */ Integer n = 10; /* Choose the base generator */ Nag_BaseRNG genid = Nag_Basic; Integer subid = 0; /* Set the seed */ Integer seed[] = { 1762543 }; Integer lseed = 1; /* Initialise the error structure */ INIT_FAIL(fail); printf("nag_rand_geom (g05tcc) Example Program Results\n\n"); /* Get the length of the state array */ lstate = -1; nag_rand_init_repeatable(genid, subid, seed, lseed, state, &lstate, &fail); if (fail.code != NE_NOERROR) { printf("Error from nag_rand_init_repeatable (g05kfc).\n%s\n", fail.message); exit_status = 1; goto END; } /* Calculate the size of the reference vector, if any */ lr = (mode != Nag_GenerateWithoutReference)?8+42/p:0; /* Allocate arrays */ if (!(r = NAG_ALLOC(lr, double)) || !(state = NAG_ALLOC(lstate, Integer)) || !(x = NAG_ALLOC(n, Integer))) { printf("Allocation failure\n"); exit_status = -1; goto END; } /* Initialise the generator to a repeatable sequence */ nag_rand_init_repeatable(genid, subid, seed, lseed, state, &lstate, &fail); if (fail.code != NE_NOERROR) { printf("Error from nag_rand_init_repeatable (g05kfc).\n%s\n", fail.message); exit_status = 1; goto END; } /* Generate the variates, dont use a reference vector as p is close to 0 */ nag_rand_geom(mode, n, p, r, lr, state, x, &fail); if (fail.code != NE_NOERROR) { printf("Error from nag_rand_geom (g05tcc).\n%s\n", fail.message); exit_status = 1; goto END; } /* Display the variates*/ for (i = 0; i < n; i++) printf("%12ld\n", x[i]); END: if (r) NAG_FREE(r); if (state) NAG_FREE(state); if (x) NAG_FREE(x); return exit_status; }