/* Test mpz_perfect_power_p.

   Contributed to the GNU project by Torbjorn Granlund and Martin Boij.

Copyright 2008-2010, 2014 Free Software Foundation, Inc.

This file is part of the GNU MP Library test suite.

The GNU MP Library test suite is free software; you can redistribute it

and/or modify it under the terms of the GNU General Public License as

published by the Free Software Foundation; either version 3 of the License,

or (at your option) any later version.

The GNU MP Library test suite is distributed in the hope that it will be

useful, but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General

Public License for more details.

You should have received a copy of the GNU General Public License along with

the GNU MP Library test suite.  If not, see https://www.gnu.org/licenses/.  */

#include <stdio.h>

#include <stdlib.h>

#include "gmp.h"

#include "gmp-impl.h"

#include "tests.h"

struct

{

  const char *num_as_str;

  char want;

} static tests[] =

  {

    { "0", 1},

    { "1", 1},

    {"-1", 1},

    { "2", 0},

    {"-2", 0},

    { "3", 0},

    {"-3", 0},

    { "4", 1},

    {"-4", 0},

    { "64", 1},

    {"-64", 1},

    { "128", 1},

    {"-128", 1},

    { "256", 1},

    {"-256", 0},

    { "512", 1},

    {"-512", 1},

    { "0x4000000", 1},

    {"-0x4000000", 1},

    { "0x3cab640", 1},

    {"-0x3cab640", 0},

    { "0x3e23840", 1},

    {"-0x3e23840", 0},

    { "0x3d3a7ed1", 1},

    {"-0x3d3a7ed1", 1},

    { "0x30a7a6000", 1},

    {"-0x30a7a6000", 1},

    { "0xf33e5a5a59", 1},

    {"-0xf33e5a5a59", 0},

    { "0xed1b1182118135d", 1},

    {"-0xed1b1182118135d", 1},

    { "0xe71f6eb7689cc276b2f1", 1},

    {"-0xe71f6eb7689cc276b2f1", 0},

    { "0x12644507fe78cf563a4b342c92e7da9fe5e99cb75a01", 1},

    {"-0x12644507fe78cf563a4b342c92e7da9fe5e99cb75a01", 0},

    { "0x1ff2e7c581bb0951df644885bd33f50e472b0b73a204e13cbe98fdb424d66561e4000000", 1},

    {"-0x1ff2e7c581bb0951df644885bd33f50e472b0b73a204e13cbe98fdb424d66561e4000000", 1},

    { "0x2b9b44db2d91a6f8165c8c7339ef73633228ea29e388592e80354e4380004aad84000000", 1},

    {"-0x2b9b44db2d91a6f8165c8c7339ef73633228ea29e388592e80354e4380004aad84000000", 1},

    { "0x28d5a2b8f330910a9d3cda06036ae0546442e5b1a83b26a436efea5b727bf1bcbe7e12b47d81", 1},

    {"-0x28d5a2b8f330910a9d3cda06036ae0546442e5b1a83b26a436efea5b727bf1bcbe7e12b47d81", 1},

    {NULL, 0}

  };

void

check_tests ()

{

  mpz_t x;

  int i;

  int got, want;

  mpz_init (x);

  for (i = 0; tests[i].num_as_str != NULL; i++)

    {

      mpz_set_str (x, tests[i].num_as_str, 0);

      got = mpz_perfect_power_p (x);

      want = tests[i].want;

      if (got != want)

	{

	  fprintf (stderr, "mpz_perfect_power_p returns %d when %d was expected\n", got, want);

	  fprintf (stderr, "fault operand: %s\n", tests[i].num_as_str);

	  abort ();

	}

    }

  mpz_clear (x);

}

#define NRP 15

void

check_random (int reps)

{

  mpz_t n, np, temp, primes[NRP];

  int i, j, k, unique, destroy, res;

  unsigned long int nrprimes, primebits;

  mp_limb_t g, exp[NRP], e;

  gmp_randstate_ptr rands;

  rands = RANDS;

  mpz_init (n);

  mpz_init (np);

  mpz_init (temp);

  for (i = 0; i < NRP; i++)

    mpz_init (primes[i]);

  for (i = 0; i < reps; i++)

    {

      mpz_urandomb (np, rands, 32);

      nrprimes = mpz_get_ui (np) % NRP + 1; /* 1-NRP unique primes */

      mpz_urandomb (np, rands, 32);

      g = mpz_get_ui (np) % 32 + 2; /* gcd 2-33 */

      for (j = 0; j < nrprimes;)

	{

	  mpz_urandomb (np, rands, 32);

	  primebits = mpz_get_ui (np) % 100 + 3; /* 3-102 bit primes */

	  mpz_urandomb (primes[j], rands, primebits);

	  mpz_nextprime (primes[j], primes[j]);

	  unique = 1;

	  for (k = 0; k < j; k++)

	    {

	      if (mpz_cmp (primes[j], primes[k]) == 0)

		{

		  unique = 0;

		  break;

		}

	    }

	  if (unique)

	    {

	      mpz_urandomb (np, rands, 32);

	      e = 371 / (10 * primebits) + mpz_get_ui (np) % 11 + 1; /* Magic constants */

	      exp[j++] = g * e;

	    }

	}

      if (nrprimes > 1)

	{

	  /* Destroy d exponents, d in [1, nrprimes - 1] */

	  if (nrprimes == 2)

	    {

	      destroy = 1;

	    }

	  else

	    {

	      mpz_urandomb (np, rands, 32);

	      destroy = mpz_get_ui (np) % (nrprimes - 2);

	    }

	  g = exp[destroy];

	  for (k = destroy + 1; k < nrprimes; k++)

	    g = mpn_gcd_1 (&g, 1, exp[k]);

	  for (j = 0; j < destroy; j++)

	    {

	      mpz_urandomb (np, rands, 32);

	      e = mpz_get_ui (np) % 50 + 1;

	      while (mpn_gcd_1 (&g, 1, e) > 1)

		e++;

	      exp[j] = e;

	    }

	}

      /* Compute n */

      mpz_pow_ui (n, primes[0], exp[0]);

      for (j = 1; j < nrprimes; j++)

	{

	  mpz_pow_ui (temp, primes[j], exp[j]);

	  mpz_mul (n, n, temp);

	}

      res = mpz_perfect_power_p (n);

      if (nrprimes == 1)

	{

	if (res == 0 && exp[0] > 1)

	  {

	    printf("n is a perfect power, perfpow_p disagrees\n");

	    gmp_printf("n = %Zu\nprimes[0] = %Zu\nexp[0] = %lu\n", n, primes[0], exp[0]);

	    abort ();

	  }

	else if (res == 1 && exp[0] == 1)

	  {

	    gmp_printf("n = %Zu\n", n);

	    printf("n is now a prime number, but perfpow_p still believes n is a perfect power\n");

	    abort ();

	  }

	}

      else

	{

	  if (res == 1 && destroy != 0)

	    {

	      gmp_printf("n = %Zu\nn was destroyed, but perfpow_p still believes n is a perfect power\n", n);

	      abort ();

	    }

	  else if (res == 0 && destroy == 0)

	    {

	      gmp_printf("n = %Zu\nn is a perfect power, perfpow_p disagrees\n", n);

	      abort ();

	    }

	}

    }

  mpz_clear (n);

  mpz_clear (np);

  mpz_clear (temp);

  for (i = 0; i < NRP; i++)

    mpz_clear (primes[i]);

}

int

main (int argc, char **argv)

{

  int n_tests;

  tests_start ();

  mp_trace_base = -16;

  check_tests ();

  n_tests = 500;

  if (argc == 2)

    n_tests = atoi (argv[1]);

  check_random (n_tests);

  tests_end ();

  exit (0);

}