void mpz_lucnum2_ui (mpz_ptr ln, mpz_ptr lnsub1, unsigned long n) { mp_ptr lp, l1p, f1p; mp_size_t size; mp_limb_t c; TMP_DECL; ASSERT (ln != lnsub1); /* handle small n quickly, and hide the special case for L[-1]=-1 */ if (n <= FIB_TABLE_LUCNUM_LIMIT) { mp_limb_t f = FIB_TABLE (n); mp_limb_t f1 = FIB_TABLE ((int) n - 1); /* L[n] = F[n] + 2F[n-1] */ PTR(ln)[0] = f + 2*f1; SIZ(ln) = 1; /* L[n-1] = 2F[n] - F[n-1], but allow for L[-1]=-1 */ PTR(lnsub1)[0] = (n == 0 ? 1 : 2*f - f1); SIZ(lnsub1) = (n == 0 ? -1 : 1); return; } TMP_MARK; size = MPN_FIB2_SIZE (n); f1p = TMP_ALLOC_LIMBS (size); MPZ_REALLOC (ln, size+1); MPZ_REALLOC (lnsub1, size+1); lp = PTR(ln); l1p = PTR(lnsub1); size = mpn_fib2_ui (l1p, f1p, n); /* L[n] = F[n] + 2F[n-1] */ #if HAVE_NATIVE_mpn_addlsh1_n c = mpn_addlsh1_n (lp, l1p, f1p, size); #else c = mpn_lshift1 (lp, f1p, size); c += mpn_add_n (lp, lp, l1p, size); #endif lp[size] = c; SIZ(ln) = size + (c != 0); /* L[n-1] = 2F[n] - F[n-1] */ c = mpn_double (l1p, size); c -= mpn_sub_n (l1p, l1p, f1p, size); ASSERT ((mp_limb_signed_t) c >= 0); l1p[size] = c; SIZ(lnsub1) = size + (c != 0); TMP_FREE; }
void mpz_fib2_ui (mpz_ptr fn, mpz_ptr fnsub1, mpir_ui n) { mp_ptr fp, f1p; mp_size_t size; size = MPN_FIB2_SIZE (n); MPZ_REALLOC (fn, size); MPZ_REALLOC (fnsub1, size); fp = PTR (fn); f1p = PTR (fnsub1); size = mpn_fib2_ui (fp, f1p, n); SIZ(fn) = size - (n == 0); SIZ(fnsub1) = size - (f1p[size-1] == 0); }
void mpz_fib2_ui (mpz_ptr fn, mpz_ptr fnsub1, unsigned long n) { mp_ptr fp, f1p; mp_size_t size; if (n <= FIB_TABLE_LIMIT) { MPZ_NEWALLOC (fn, 1)[0] = FIB_TABLE (n); SIZ(fn) = (n != 0); /* F[0]==0, others are !=0 */ MPZ_NEWALLOC (fnsub1, 1)[0] = FIB_TABLE ((int) n - 1); SIZ(fnsub1) = (n != 1); /* F[1-1]==0, others are !=0 */ return; } size = MPN_FIB2_SIZE (n); fp = MPZ_NEWALLOC (fn, size); f1p = MPZ_NEWALLOC (fnsub1, size); size = mpn_fib2_ui (fp, f1p, n); SIZ(fn) = size; SIZ(fnsub1) = size - (f1p[size-1] == 0); }
int main (int argc, char *argv[]) { unsigned long n; unsigned long limit = 100 * BITS_PER_MP_LIMB; mpz_t want_fn, want_fn1, got_fn, got_fn1; tests_start (); mp_trace_base = -16; if (argc > 1 && argv[1][0] == 'x') limit = ULONG_MAX; else if (argc > 1) limit = atoi (argv[1]); check_fib_table (); /* start at n==0 */ mpz_init_set_ui (want_fn1, 1); /* F[-1] */ mpz_init_set_ui (want_fn, 0); /* F[0] */ mpz_init (got_fn); mpz_init (got_fn1); for (n = 0; n < limit; n++) { /* check our float formula seems right */ if (MPZ_FIB_SIZE_FLOAT (n) < SIZ(want_fn)) { printf ("MPZ_FIB_SIZE_FLOAT wrong at n=%lu\n", n); printf (" MPZ_FIB_SIZE_FLOAT %ld\n", MPZ_FIB_SIZE_FLOAT (n)); printf (" SIZ(want_fn) %d\n", SIZ(want_fn)); abort (); } /* check MPN_FIB2_SIZE seems right, compared to actual size and compared to our float formula */ if (MPN_FIB2_SIZE (n) < MPZ_FIB_SIZE_FLOAT (n)) { printf ("MPN_FIB2_SIZE wrong at n=%lu\n", n); printf (" MPN_FIB2_SIZE %ld\n", MPN_FIB2_SIZE (n)); printf (" MPZ_FIB_SIZE_FLOAT %ld\n", MPZ_FIB_SIZE_FLOAT (n)); abort (); } if (MPN_FIB2_SIZE (n) < SIZ(want_fn)) { printf ("MPN_FIB2_SIZE wrong at n=%lu\n", n); printf (" MPN_FIB2_SIZE %ld\n", MPN_FIB2_SIZE (n)); printf (" SIZ(want_fn) %d\n", SIZ(want_fn)); abort (); } mpz_fib2_ui (got_fn, got_fn1, n); MPZ_CHECK_FORMAT (got_fn); MPZ_CHECK_FORMAT (got_fn1); if (mpz_cmp (got_fn, want_fn) != 0 || mpz_cmp (got_fn1, want_fn1) != 0) { printf ("mpz_fib2_ui(%lu) wrong\n", n); mpz_trace ("want fn ", want_fn); mpz_trace ("got fn ", got_fn); mpz_trace ("want fn1", want_fn1); mpz_trace ("got fn1", got_fn1); abort (); } mpz_fib_ui (got_fn, n); MPZ_CHECK_FORMAT (got_fn); if (mpz_cmp (got_fn, want_fn) != 0) { printf ("mpz_fib_ui(%lu) wrong\n", n); mpz_trace ("want fn", want_fn); mpz_trace ("got fn", got_fn); abort (); } mpz_add (want_fn1, want_fn1, want_fn); /* F[n+1] = F[n] + F[n-1] */ mpz_swap (want_fn1, want_fn); } mpz_clear (want_fn); mpz_clear (want_fn1); mpz_clear (got_fn); mpz_clear (got_fn1); tests_end (); exit (0); }