source: libcfa/src/gmp.hfa @ 65240bb

arm-ehjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-expr
Last change on this file since 65240bb was 65240bb, checked in by Peter A. Buhr <pabuhr@…>, 2 years ago

first attempt to add streams exit and abort, rename private iostream nl function to ends

  • Property mode set to 100644
File size: 16.9 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2016 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// gmp --
8//
9// Author           : Peter A. Buhr
10// Created On       : Tue Apr 19 08:43:43 2016
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Fri Jul 12 12:02:55 2019
13// Update Count     : 26
14//
15
16// https://gmplib.org/gmp-man-6.1.1.pdf
17
18#pragma once
19
20#include <gmp.h>                                                                                // GNU multi-precise integers
21#include <fstream.hfa>                                                                  // sout
22
23struct Int { mpz_t mpz; };                                                              // wrap GMP implementation
24
25static inline {
26        // constructor
27        void ?{}( Int & this ) { mpz_init( this.mpz ); }
28        void ?{}( Int & this, Int init ) { mpz_init_set( this.mpz, init.mpz ); }
29        void ?{}( Int & this, zero_t ) { mpz_init_set_si( this.mpz, 0 ); }
30        void ?{}( Int & this, one_t ) { mpz_init_set_si( this.mpz, 1 ); }
31        void ?{}( Int & this, signed long int init ) { mpz_init_set_si( this.mpz, init ); }
32        void ?{}( Int & this, unsigned long int init ) { mpz_init_set_ui( this.mpz, init ); }
33        void ?{}( Int & this, const char * val ) { if ( mpz_init_set_str( this.mpz, val, 0 ) ) abort(); }
34        void ^?{}( Int & this ) { mpz_clear( this.mpz ); }
35
36        // literal
37        Int ?`mp( signed long int init ) { return (Int){ init }; }
38        Int ?`mp( unsigned long int init ) { return (Int){ init }; }
39        Int ?`mp( const char * init ) { return (Int){ init }; }
40
41        // assignment
42        Int ?=?( Int & lhs, Int rhs ) { mpz_set( lhs.mpz, rhs.mpz ); return lhs; }
43        Int ?=?( Int & lhs, long int rhs ) { mpz_set_si( lhs.mpz, rhs ); return lhs; }
44        Int ?=?( Int & lhs, unsigned long int rhs ) { mpz_set_ui( lhs.mpz, rhs ); return lhs; }
45        Int ?=?( Int & lhs, const char * rhs ) { if ( mpz_set_str( lhs.mpz, rhs, 0 ) ) { printf( "invalid string conversion\n" ); abort(); } return lhs; }
46
47        char ?=?( char & lhs, Int rhs ) { char val = mpz_get_si( rhs.mpz ); lhs = val; return lhs; }
48        short int ?=?( short int & lhs, Int rhs ) { short int val = mpz_get_si( rhs.mpz ); lhs = val; return lhs; }
49        int ?=?( int & lhs, Int rhs ) { int val = mpz_get_si( rhs.mpz ); lhs = val; return lhs; }
50        long int ?=?( long int & lhs, Int rhs ) { long int val = mpz_get_si( rhs.mpz ); lhs = val; return lhs; }
51        unsigned char ?=?( unsigned char & lhs, Int rhs ) { unsigned char val = mpz_get_ui( rhs.mpz ); lhs = val; return lhs; }
52        unsigned short int ?=?( unsigned short int & lhs, Int rhs ) { unsigned short int val = mpz_get_ui( rhs.mpz ); lhs = val; return lhs; }
53        unsigned int ?=?( unsigned int & lhs, Int rhs ) { unsigned int val = mpz_get_ui( rhs.mpz ); lhs = val; return lhs; }
54        unsigned long int ?=?( unsigned long int & lhs, Int rhs ) { unsigned long int val = mpz_get_ui( rhs.mpz ); lhs = val; return lhs; }
55
56        // conversions
57        long int narrow( Int val ) { return mpz_get_si( val.mpz ); }
58        unsigned long int narrow( Int val ) { return mpz_get_ui( val.mpz ); }
59
60        // comparison
61        int ?==?( Int oper1, Int oper2 ) { return mpz_cmp( oper1.mpz, oper2.mpz ) == 0; }
62        int ?==?( Int oper1, long int oper2 ) { return mpz_cmp_si( oper1.mpz, oper2 ) == 0; }
63        int ?==?( long int oper2, Int oper1 ) { return mpz_cmp_si( oper1.mpz, oper2 ) == 0; }
64        int ?==?( Int oper1, unsigned long int oper2 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) == 0; }
65        int ?==?( unsigned long int oper2, Int oper1 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) == 0; }
66
67        int ?!=?( Int oper1, Int oper2 ) { return ! ( oper1 == oper2 ); }
68        int ?!=?( Int oper1, long int oper2 ) { return ! ( oper1 == oper2 ); }
69        int ?!=?( long int oper1, Int oper2 ) { return ! ( oper1 == oper2 ); }
70        int ?!=?( Int oper1, unsigned long int oper2 ) { return ! ( oper1 == oper2 ); }
71        int ?!=?( unsigned long int oper1, Int oper2 ) { return ! ( oper1 == oper2 ); }
72
73        int ?<?( Int oper1, Int oper2 ) { return mpz_cmp( oper1.mpz, oper2.mpz ) < 0; }
74        int ?<?( Int oper1, long int oper2 ) { return mpz_cmp_si( oper1.mpz, oper2 ) < 0; }
75        int ?<?( long int oper2, Int oper1 ) { return mpz_cmp_si( oper1.mpz, oper2 ) < 0; }
76        int ?<?( Int oper1, unsigned long int oper2 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) < 0; }
77        int ?<?( unsigned long int oper2, Int oper1 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) < 0; }
78
79        int ?<=?( Int oper1, Int oper2 ) { return mpz_cmp( oper1.mpz, oper2.mpz ) <= 0; }
80        int ?<=?( Int oper1, long int oper2 ) { return mpz_cmp_si( oper1.mpz, oper2 ) <= 0; }
81        int ?<=?( long int oper2, Int oper1 ) { return mpz_cmp_si( oper1.mpz, oper2 ) <= 0; }
82        int ?<=?( Int oper1, unsigned long int oper2 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) <= 0; }
83        int ?<=?( unsigned long int oper2, Int oper1 ) { return mpz_cmp_ui( oper1.mpz, oper2 ) <= 0; }
84
85        int ?>?( Int oper1, Int oper2 ) { return ! ( oper1 <= oper2 ); }
86        int ?>?( Int oper1, long int oper2 ) { return ! ( oper1 <= oper2 ); }
87        int ?>?( long int oper1, Int oper2 ) { return ! ( oper1 <= oper2 ); }
88        int ?>?( Int oper1, unsigned long int oper2 ) { return ! ( oper1 <= oper2 ); }
89        int ?>?( unsigned long int oper1, Int oper2 ) { return ! ( oper1 <= oper2 ); }
90
91        int ?>=?( Int oper1, Int oper2 ) { return ! ( oper1 < oper2 ); }
92        int ?>=?( Int oper1, long int oper2 ) { return ! ( oper1 < oper2 ); }
93        int ?>=?( long int oper1, Int oper2 ) { return ! ( oper1 < oper2 ); }
94        int ?>=?( Int oper1, unsigned long int oper2 ) { return ! ( oper1 < oper2 ); }
95        int ?>=?( unsigned long int oper1, Int oper2 ) { return ! ( oper1 < oper2 ); }
96
97        // arithmetic
98        Int +?( Int oper ) { Int pos; mpz_set( pos.mpz, oper.mpz ); return pos; }
99        Int -?( Int oper ) { Int neg; mpz_neg( neg.mpz, oper.mpz ); return neg; }
100        Int ~?( Int oper ) { Int comp; mpz_com( comp.mpz, oper.mpz ); return comp; }
101
102        Int ?&?( Int oper1, Int oper2 ) { Int conjunction; mpz_and( conjunction.mpz, oper1.mpz, oper2.mpz ); return conjunction; }
103        Int ?&?( Int oper1, long int oper2 ) { Int conjunction, temp; mpz_set_si( temp.mpz, oper2 ); mpz_and( conjunction.mpz, oper1.mpz, temp.mpz ); return conjunction; }
104        Int ?&?( long int oper1, Int oper2 ) { Int conjunction, temp; mpz_set_si( temp.mpz, oper1 ); mpz_and( conjunction.mpz, temp.mpz, oper2.mpz ); return conjunction; }
105        Int ?&?( Int oper1, unsigned long int oper2 ) { Int conjunction, temp; mpz_set_ui( temp.mpz, oper2 ); mpz_and( conjunction.mpz, oper1.mpz, temp.mpz ); return conjunction; }
106        Int ?&?( unsigned long int oper1, Int oper2 ) { Int conjunction, temp; mpz_set_ui( temp.mpz, oper1 ); mpz_and( conjunction.mpz, temp.mpz, oper2.mpz ); return conjunction; }
107        Int ?&=?( Int & lhs, Int rhs ) { return lhs = lhs & rhs; }
108
109        Int ?|?( Int oper1, Int oper2 ) { Int disjunction; mpz_ior( disjunction.mpz, oper1.mpz, oper2.mpz ); return disjunction; }
110        Int ?|?( Int oper1, long int oper2 ) { Int disjunction, temp; mpz_set_si( temp.mpz, oper2 ); mpz_ior( disjunction.mpz, oper1.mpz, temp.mpz ); return disjunction; }
111        Int ?|?( long int oper1, Int oper2 ) { Int disjunction, temp; mpz_set_si( temp.mpz, oper1 ); mpz_ior( disjunction.mpz, temp.mpz, oper2.mpz ); return disjunction; }
112        Int ?|?( Int oper1, unsigned long int oper2 ) { Int disjunction, temp; mpz_set_ui( temp.mpz, oper2 ); mpz_ior( disjunction.mpz, oper1.mpz, temp.mpz ); return disjunction; }
113        Int ?|?( unsigned long int oper1, Int oper2 ) { Int disjunction, temp; mpz_set_ui( temp.mpz, oper1 ); mpz_ior( disjunction.mpz, temp.mpz, oper2.mpz ); return disjunction; }
114        Int ?|=?( Int & lhs, Int rhs ) { return lhs = lhs | rhs; }
115
116        Int ?^?( Int oper1, Int oper2 ) { Int disjunction; mpz_xor( disjunction.mpz, oper1.mpz, oper2.mpz ); return disjunction; }
117        Int ?^?( Int oper1, long int oper2 ) { Int disjunction, temp; mpz_set_si( temp.mpz, oper2 ); mpz_ior( disjunction.mpz, oper1.mpz, temp.mpz ); return disjunction; }
118        Int ?^?( long int oper1, Int oper2 ) { Int disjunction, temp; mpz_set_si( temp.mpz, oper1 ); mpz_ior( disjunction.mpz, temp.mpz, oper2.mpz ); return disjunction; }
119        Int ?^?( Int oper1, unsigned long int oper2 ) { Int disjunction, temp; mpz_set_ui( temp.mpz, oper2 ); mpz_ior( disjunction.mpz, oper1.mpz, temp.mpz ); return disjunction; }
120        Int ?^?( unsigned long int oper1, Int oper2 ) { Int disjunction, temp; mpz_set_ui( temp.mpz, oper1 ); mpz_ior( disjunction.mpz, temp.mpz, oper2.mpz ); return disjunction; }
121        Int ?^=?( Int & lhs, Int rhs ) { return lhs = lhs ^ rhs; }
122
123        Int ?+?( Int addend1, Int addend2 ) { Int sum; mpz_add( sum.mpz, addend1.mpz, addend2.mpz ); return sum; }
124        Int ?+?( Int addend1, long int addend2 ) { Int sum; if ( addend2 >= 0 ) mpz_add_ui( sum.mpz, addend1.mpz, addend2 ); else mpz_sub_ui( sum.mpz, addend1.mpz, -addend2 ); return sum; }
125        Int ?+?( long int addend2, Int addend1 ) { Int sum; if ( addend2 >= 0 ) mpz_add_ui( sum.mpz, addend1.mpz, addend2 ); else mpz_sub_ui( sum.mpz, addend1.mpz, -addend2 ); return sum; }
126        Int ?+?( Int addend1, unsigned long int addend2 ) { Int sum; mpz_add_ui( sum.mpz, addend1.mpz, addend2 ); return sum; }
127        Int ?+?( unsigned long int addend2, Int addend1 ) { Int sum; mpz_add_ui( sum.mpz, addend1.mpz, addend2 ); return sum; }
128        Int ?+=?( Int & lhs, Int rhs ) { return lhs = lhs + rhs; }
129        Int ?+=?( Int & lhs, long int rhs ) { return lhs = lhs + rhs; }
130        Int ?+=?( Int & lhs, unsigned long int rhs ) { return lhs = lhs + rhs; }
131        Int ++?( Int & lhs ) { return lhs += 1; }
132        Int ?++( Int & lhs ) { Int ret = lhs; lhs += 1; return ret; }
133
134        Int ?-?( Int minuend, Int subtrahend ) { Int diff; mpz_sub( diff.mpz, minuend.mpz, subtrahend.mpz ); return diff; }
135        Int ?-?( Int minuend, long int subtrahend ) { Int diff; if ( subtrahend >= 0 ) mpz_sub_ui( diff.mpz, minuend.mpz, subtrahend ); else mpz_add_ui( diff.mpz, minuend.mpz, -subtrahend ); return diff; }
136        Int ?-?( long int minuend, Int subtrahend ) { Int diff; if ( subtrahend >= 0 ) mpz_ui_sub( diff.mpz, minuend, subtrahend.mpz ); else { mpz_add_ui( diff.mpz, subtrahend.mpz, -minuend ); mpz_neg( diff.mpz, diff.mpz ); } return diff; }
137        Int ?-?( Int minuend, unsigned long int subtrahend ) { Int diff; mpz_sub_ui( diff.mpz, minuend.mpz, subtrahend ); return diff; }
138        Int ?-?( unsigned long int minuend, Int subtrahend ) { Int diff; mpz_ui_sub( diff.mpz, minuend, subtrahend.mpz ); return diff; }
139        Int ?-=?( Int & lhs, Int rhs ) { return lhs = lhs - rhs; }
140        Int ?-=?( Int & lhs, long int rhs ) { return lhs = lhs - rhs; }
141        Int ?-=?( Int & lhs, unsigned long int rhs ) { return lhs = lhs - rhs; }
142        Int --?( Int & lhs ) { return lhs -= 1; }
143        Int ?--( Int & lhs ) { Int ret = lhs; lhs -= 1; return ret; }
144
145        Int ?*?( Int multiplicator, Int multiplicand ) { Int product; mpz_mul( product.mpz, multiplicator.mpz, multiplicand.mpz ); return product; }
146        Int ?*?( Int multiplicator, long int multiplicand ) { Int product; mpz_mul_si( product.mpz, multiplicator.mpz, multiplicand ); return product; }
147        Int ?*?( long int multiplicand, Int multiplicator ) { Int product; mpz_mul_si( product.mpz, multiplicator.mpz, multiplicand ); return product; }
148        Int ?*?( Int multiplicator, unsigned long int multiplicand ) { Int product; mpz_mul_ui( product.mpz, multiplicator.mpz, multiplicand ); return product; }
149        Int ?*?( unsigned long int multiplicand, Int multiplicator ) { Int product; mpz_mul_ui( product.mpz, multiplicator.mpz, multiplicand ); return product; }
150        Int ?*=?( Int & lhs, Int rhs ) { return lhs = lhs * rhs; }
151        Int ?*=?( Int & lhs, long int rhs ) { return lhs = lhs * rhs; }
152        Int ?*=?( Int & lhs, unsigned long int rhs ) { return lhs = lhs * rhs; }
153
154        // some code for operators "/" and "%" taken from g++ gmpxx.h
155        Int ?/?( Int dividend, Int divisor ) { Int quotient; mpz_tdiv_q( quotient.mpz, dividend.mpz, divisor.mpz ); return quotient; }
156        Int ?/?( Int dividend, unsigned long int divisor ) { Int quotient; mpz_tdiv_q_ui( quotient.mpz, dividend.mpz, divisor ); return quotient; }
157        Int ?/?( unsigned long int dividend, Int divisor ) {
158                Int quotient;
159                if ( mpz_sgn( divisor.mpz ) >= 0 ) {
160                        if ( mpz_fits_ulong_p( divisor.mpz ) )
161                                mpz_set_ui( quotient.mpz, dividend / mpz_get_ui( divisor.mpz ) );
162                        else
163                                mpz_set_ui( quotient.mpz, 0 );
164                } else {
165                        mpz_neg( quotient.mpz, divisor.mpz );
166                        if ( mpz_fits_ulong_p( quotient.mpz ) ) {
167                                mpz_set_ui( quotient.mpz, dividend / mpz_get_ui( quotient.mpz ) );
168                                mpz_neg( quotient.mpz, quotient.mpz );
169                        } else
170                                mpz_set_ui( quotient.mpz, 0 );
171                } // if
172                return quotient;
173        } // ?/?
174        Int ?/?( Int dividend, long int divisor ) {
175                Int quotient;
176                if ( divisor >= 0 )
177                        mpz_tdiv_q_ui( quotient.mpz, dividend.mpz, divisor );
178                else {
179                        mpz_tdiv_q_ui( quotient.mpz, dividend.mpz, -divisor );
180                        mpz_neg( quotient.mpz, quotient.mpz );
181                } // if
182                return quotient;
183        } // ?/?
184        Int ?/?( long int dividend, Int divisor ) {
185                Int quotient;
186                if ( mpz_fits_slong_p( divisor.mpz ) )
187                        mpz_set_si( quotient.mpz, dividend / mpz_get_si( divisor.mpz ) );
188                else {
189                        // if divisor is bigger than a long then the quotient must be zero, unless dividend==LONG_MIN and
190                        // dividend==-LONG_MIN in which case the quotient is -1
191                        mpz_set_si( quotient.mpz, mpz_cmpabs_ui( divisor.mpz, (dividend >= 0 ? dividend : -dividend)) == 0 ? -1 : 0 );
192                } // if
193                return quotient;
194        } // ?/?
195        Int ?/=?( Int & lhs, Int rhs ) { return lhs = lhs / rhs; }
196        Int ?/=?( Int & lhs, long int rhs ) { return lhs = lhs / rhs; }
197        Int ?/=?( Int & lhs, unsigned long int rhs ) { return lhs = lhs / rhs; }
198
199        [ Int, Int ] div( Int dividend, Int divisor ) { Int quotient, remainder; mpz_fdiv_qr( quotient.mpz, remainder.mpz, dividend.mpz, divisor.mpz ); return [ quotient, remainder ]; }
200        [ Int, Int ] div( Int dividend, unsigned long int divisor ) { Int quotient, remainder; mpz_fdiv_qr_ui( quotient.mpz, remainder.mpz, dividend.mpz, divisor ); return [ quotient, remainder ]; }
201
202        Int ?%?( Int dividend, Int divisor ) { Int remainder; mpz_tdiv_r( remainder.mpz, dividend.mpz, divisor.mpz ); return remainder; }
203        Int ?%?( Int dividend, unsigned long int divisor ) { Int remainder; mpz_tdiv_r_ui( remainder.mpz, dividend.mpz, divisor ); return remainder; }
204        Int ?%?( unsigned long int dividend, Int divisor ) {
205                Int remainder;
206                if ( mpz_sgn( divisor.mpz ) >= 0 ) {
207                        if ( mpz_fits_ulong_p( divisor.mpz ) )
208                                mpz_set_ui( remainder.mpz, dividend % mpz_get_ui( divisor.mpz ) );
209                        else
210                                mpz_set_ui( remainder.mpz, dividend );
211                } else {
212                        mpz_neg( remainder.mpz, divisor.mpz );
213                        if ( mpz_fits_ulong_p( remainder.mpz ) )
214                                mpz_set_ui( remainder.mpz, dividend % mpz_get_ui( remainder.mpz ) );
215                        else
216                                mpz_set_ui( remainder.mpz, dividend );
217                } // if
218                return remainder;
219        } // ?%?
220        Int ?%?( Int dividend, long int divisor ) {
221                Int remainder;
222                mpz_tdiv_r_ui( remainder.mpz, dividend.mpz, (divisor >= 0 ? divisor : -divisor));
223                return remainder;
224        } // ?%?
225        Int ?%?( long int dividend, Int divisor ) {
226                Int remainder;
227                if ( mpz_fits_slong_p( divisor.mpz ) )
228                        mpz_set_si( remainder.mpz, dividend % mpz_get_si( divisor.mpz ) );
229                else {
230                        // if divisor is bigger than a long then the remainder is dividend unchanged, unless dividend==LONG_MIN and
231                        // dividend==-LONG_MIN in which case the remainder is 0
232                        mpz_set_si( remainder.mpz, mpz_cmpabs_ui( divisor.mpz, (dividend >= 0 ? dividend : -dividend)) == 0 ? 0 : dividend);
233                } // if
234                return remainder;
235        } // ?%?
236        Int ?%=?( Int & lhs, Int rhs ) { return lhs = lhs % rhs; }
237        Int ?%=?( Int & lhs, long int rhs ) { return lhs = lhs % rhs; }
238        Int ?%=?( Int & lhs, unsigned long int rhs ) { return lhs = lhs % rhs; }
239
240        Int ?<<?( Int shiften, mp_bitcnt_t shift ) { Int shifted; mpz_mul_2exp( shifted.mpz, shiften.mpz, shift ); return shifted; }
241        Int ?<<=?( Int & lhs, mp_bitcnt_t shift ) { return lhs = lhs << shift; }
242        Int ?>>?( Int shiften, mp_bitcnt_t shift ) { Int shifted; mpz_fdiv_q_2exp( shifted.mpz, shiften.mpz, shift ); return shifted; }
243        Int ?>>=?( Int & lhs, mp_bitcnt_t shift ) { return lhs = lhs >> shift; }
244
245        // number functions
246        Int abs( Int oper ) { Int positive; mpz_abs( positive.mpz, oper.mpz ); return positive; }
247        Int fact( unsigned long int N ) { Int factorial; mpz_fac_ui( factorial.mpz, N ); return factorial; }
248        Int gcd( Int oper1, Int oper2 ) { Int gcdret; mpz_gcd( gcdret.mpz, oper1.mpz, oper2.mpz ); return gcdret; }
249        Int pow( Int base, unsigned long int exponent ) { Int power; mpz_pow_ui( power.mpz, base.mpz, exponent ); return power; }
250        Int pow( unsigned long int base, unsigned long int exponent ) { Int power; mpz_ui_pow_ui( power.mpz, base, exponent ); return power; }
251        void srandom( gmp_randstate_t state ) { gmp_randinit_default( state ); }
252        Int random( gmp_randstate_t state, mp_bitcnt_t n ) { Int rand; mpz_urandomb( rand.mpz, state, n ); return rand; }
253        Int random( gmp_randstate_t state, Int n ) { Int rand; mpz_urandomm( rand.mpz, state, n.mpz ); return rand; }
254        Int random( gmp_randstate_t state, mp_size_t max_size ) { Int rand; mpz_random( rand.mpz, max_size ); return rand; }
255        int sgn( Int oper ) { return mpz_sgn( oper.mpz ); }
256        Int sqrt( Int oper ) { Int root; mpz_sqrt( root.mpz, oper.mpz ); return root; }
257
258        // I/O
259        forall( dtype istype | istream( istype ) )
260                istype & ?|?( istype & is, Int & mp ) {
261                gmp_scanf( "%Zd", &mp );
262                return is;
263        } // ?|?
264
265        forall( dtype ostype | ostream( ostype ) ) {
266                ostype & ?|?( ostype & os, Int mp ) {
267                        if ( sepPrt( os ) ) fmt( os, "%s", sepGetCur( os ) );
268                        gmp_printf( "%Zd", mp.mpz );
269                        sepOn( os );
270                        return os;
271                } // ?|?
272
273                void ?|?( ostype & os, Int mp ) {
274                        (ostype)(os | mp); ends( os );
275                } // ?|?
276        } // distribution
277} // distribution
278
279// Local Variables: //
280// mode: c //
281// tab-width: 4 //
282// End: //
Note: See TracBrowser for help on using the repository browser.