admin/mlib
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve example by adding also mpfr

Browse Source
This commit is contained in:
Patrick Pelissier committed 2022-04-28 22:22:05 +02:00
1 parent bd5c9ce2a7
commit 94bfac3f75
1 file changed
+83 -17
Download Patch File Download Diff File Expand all files Collapse all files
example/ex-multi02.c
+83 -17
View File
@@ -1,16 +1,17 @@
#if HAVE_GMP
#if HAVE_GMP && HAVE_MPFR
#include <stdio.h>
#include <gmp.h>
#include <mpfr.h>
#include "m-variant.h"
#include "m-string.h"
#include "m-dict.h"
#include "m-array.h"
// GMP doesn't provide a fast hash function.
// GMP doesn't provide a fast hash function (It is needed for being able to use mpz_t as key in a hashmap)
static inline size_t
mpz_hash(const mpz_t z)
mpz_fast_hash(const mpz_t z)
{
mp_size_t n = z->_mp_size;
mp_limb_t *p = z->_mp_d;
@@ -34,7 +35,7 @@ mpz_hash(const mpz_t z)
(INIT(mpz_init), INIT_SET(mpz_init_set), SET(mpz_set), CLEAR(mpz_clear), \
EQUAL( API( mpz_cmp, EQ(0), ARG1, ARG2)), \
CMP(mpz_cmp), \
HASH(mpz_hash), \
HASH(mpz_fast_hash), \
OUT_STR( API( mpz_out_str, VOID, ARG1, 10, ARG2)), \
IN_STR( API( mpz_inp_str, GT(0), ARG1, ARG2, 10)), \
EMPLACE_TYPE( LIST( (_si, mpz_init_set_si, int), \
@@ -43,59 +44,124 @@ mpz_hash(const mpz_t z)
(_str, API(mpz_init_set_str, VOID, ARG1, ARG2, 10), const char *) )), \
TYPE(mpz_t) )
// MPFR doesn't provide a fast hash function (It is needed for being able to use mpz_t as key in a hashmap)
static size_t
mpfr_fast_hash (const mpfr_t f)
{
mp_limb_t *p = f->_mpfr_d;
mp_size_t n = (f->_mpfr_prec-1)/(sizeof(mp_limb_t)*CHAR_BIT)+1;
M_HASH_DECL (h);
M_HASH_UP (h, f->_mpfr_exp);
if (MPFR_SIGN (f) < 0)
M_HASH_UP (h, 1);
/* Special value: NAN, INF and ZERO */
if (mpfr_nan_p (f) || mpfr_inf_p (f) || mpfr_zero_p (f))
return M_HASH_FINAL (h);
/* Mantissa */
do {
M_HASH_UP (h, *p++);
} while (--n > 0);
return M_HASH_FINAL (h);
}
/* Define the OPLIST of a mpfr_t and register it globally */
#define M_OPL_mpfr_t() \
(INIT(mpfr_init), INIT_SET(API(mpfr_init_set, VOID, ARG1, ARG2, MPFR_RNDN)), SET(API(mpfr_set, VOID, ARG1, ARG2, MPFR_RNDN)), CLEAR(mpfr_clear), \
EQUAL( API( mpfr_cmp, EQ(0), ARG1, ARG2)), \
CMP(mpfr_cmp), \
HASH(mpfr_fast_hash), \
OUT_STR( API( mpfr_out_str, VOID, ARG1, 10, 0, ARG2, MPFR_RNDN)), \
IN_STR( API( mpfr_inp_str, GT(0), ARG1, ARG2, 10, MPFR_RNDN)), \
EMPLACE_TYPE( LIST( (_si, API(mpfr_init_set_si, VOID, ARG1, ARG2, MPFR_RNDN), int), \
(_ui, API(mpfr_init_set_ui, VOID, ARG1, ARG2, MPFR_RNDN), unsigned), \
(_d, API(mpfr_init_set_d, VOID, ARG1, ARG2, MPFR_RNDN), double), \
(_ld, API(mpfr_init_set_ld, VOID, ARG1, ARG2, MPFR_RNDN), long double), \
(_z, API(mpfr_init_set_z, VOID, ARG1, ARG2, MPFR_RNDN), mpz_t), \
(_str, API(mpfr_init_set_str, VOID, ARG1, ARG2, 10, MPFR_RNDN), const char *) )), \
TYPE(mpfr_t) )
/* Define a variant over two integers named integer_t
Notice than mpz_t has special init & clear functions
mpz_init & mpz_clear.
This is handle by the library.
*/
VARIANT_DEF2(my_integer,
VARIANT_DEF2(my_number,
(n, long long),
(z, mpz_t))
(z, mpz_t),
(fr, mpfr_t) )
/* Define the oplist associated to this variant and register it */
#define M_OPL_my_integer_t() VARIANT_OPLIST(my_integer, M_BASIC_OPLIST, M_OPL_mpz_t() )
#define M_OPL_my_number_t() VARIANT_OPLIST(my_number, M_BASIC_OPLIST, M_OPL_mpz_t() )
/* Define a dictionary of string --> integer named dict_my_t */
DICT_DEF2(my_dict, string_t, my_integer_t)
DICT_DEF2(my_dict, string_t, my_number_t)
/* Define the oplist associated to this dictionary and register it */
#define M_OPL_my_dict_t() DICT_OPLIST(my_dict, STRING_OPLIST, M_OPL_my_integer_t() )
#define M_OPL_my_dict_t() DICT_OPLIST(my_dict, STRING_OPLIST, M_OPL_my_number_t() )
/* Define an array of such dictionary named array_my_t */
ARRAY_DEF(my_array, my_dict_t)
/* Define the oplist associated to this array of dictionary of variant and register it */
#define M_OPL_my_array_t() ARRAY_OPLIST(my_array, M_OPL_my_dict_t())
// Show how to define an array of mpz_t & mpfr_t
// (Not needed for the sequel of this example)
ARRAY_DEF(array_mpz, mpz_t)
ARRAY_DEF(array_mpfr, mpfr_t)
int main(void)
{
M_LET(z, my_integer_t) /* Define & init & clear an instance of a variant of integer */
M_LET(z, my_number_t) /* Define & init & clear an instance of a variant of integer */
M_LET(dict, my_dict_t) /* Define & init & clear an instance of a dictionary of variant */
M_LET(array, my_array_t) /* Define & init & clear an intance of an array of dictionary of variant */
M_LET(gmp, mpz_t) /* M_LET allows even with non container type */
M_LET(mpfr, mpfr_t)
{
my_integer_set_n (z, 17LL); // Set z as the long long 17
my_number_set_n (z, 17LL); // Set z as the long long 17
my_dict_set_at(dict, STRING_CTE("n1"), z); // Push it in the dictionary
mpz_set_str(gmp, "25446846874687468746874687468746874686874", 10); // Set gmp to a very long integer.
my_integer_set_z (z, gmp); // Set z as the mpz_t variable
my_number_set_z (z, gmp); // Set z as the mpz_t variable
my_dict_set_at(dict, STRING_CTE("n2"), z); // Push it in the dictionary
my_array_push_back (array, dict); // Push the dictionary in the array
// Let's add a new element in the array
mpfr_set_str(mpfr, "42.17", 0, MPFR_RNDN); // Set mpfr to the real value given as a string
my_number_set_fr(z, mpfr); // Set z as this variable.
my_dict_set_at(dict, STRING_CTE("n3"), z); // Push it in the dictionary
my_array_push_back (array, dict); // Push the dictionary in the array
// Iterate over the array container
for M_EACH(item, array, my_array_t) {
// Iterate over the dictionary container
for M_EACH(p, *item, M_OPL_my_dict_t()) {
for M_EACH(p, *item, my_dict_t) {
printf("Key is '%s'. ", string_get_cstr(p->key));
// Dictionary iterator are pair (key,value)
if (my_integer_n_p(p->value)) {
if (my_number_n_p(p->value)) {
printf ("It is a long long, value = %lld\n",
*my_integer_cget_n(p->value));
*my_number_cget_n(p->value));
}
if (my_integer_z_p(p->value)) {
if (my_number_z_p(p->value)) {
printf ("It is a mpz_t, value =");
mpz_out_str(stdout, 10, *my_integer_cget_z(p->value));
mpz_out_str(stdout, 10, *my_number_cget_z(p->value));
printf("\n");
}
if (my_number_fr_p(p->value)) {
printf ("It is a mpfr_t, value =");
mpfr_out_str(stdout, 10, 0, *my_number_cget_fr(p->value), MPFR_RNDN);
printf("\n");
}
}
printf("Next element of array\n");
}
#if defined(__STDC_VERSION__) && __STDC_VERSION__ >= 201112L
// Print all the array in one step. Need to be done in C11 only
M_PRINT("The array is equal to ", (array, my_array_t), "\n");