Code coverage tests

This page documents the degree to which the PARI/GP source code is tested by our public test suite, distributed with the source distribution in directory src/test/. This is measured by the gcov utility; we then process gcov output using the lcov frond-end.

We test a few variants depending on Configure flags on the pari.math.u-bordeaux.fr machine (x86_64 architecture), and agregate them in the final report:

The target is to exceed 90% coverage for all mathematical modules (given that branches depending on DEBUGLEVEL or DEBUGMEM are not covered). This script is run to produce the results below.

LCOV - code coverage report
Current view: top level - basemath - trans2.c (source / functions) Hit Total Coverage
Test: PARI/GP v2.12.0 lcov report (development 23036-b751c0af5) Lines: 1053 1096 96.1 %
Date: 2018-09-26 05:46:06 Functions: 81 81 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /* Copyright (C) 2000  The PARI group.
       2             : 
       3             : This file is part of the PARI/GP package.
       4             : 
       5             : PARI/GP is free software; you can redistribute it and/or modify it under the
       6             : terms of the GNU General Public License as published by the Free Software
       7             : Foundation. It is distributed in the hope that it will be useful, but WITHOUT
       8             : ANY WARRANTY WHATSOEVER.
       9             : 
      10             : Check the License for details. You should have received a copy of it, along
      11             : with the package; see the file 'COPYING'. If not, write to the Free Software
      12             : Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. */
      13             : 
      14             : /********************************************************************/
      15             : /**                                                                **/
      16             : /**                   TRANSCENDENTAL FUNCTIONS                     **/
      17             : /**                          (part 2)                              **/
      18             : /**                                                                **/
      19             : /********************************************************************/
      20             : #include "pari.h"
      21             : #include "paripriv.h"
      22             : 
      23             : GEN
      24      262129 : trans_fix_arg(long *prec, GEN *s0, GEN *sig, GEN *tau, pari_sp *av, GEN *res)
      25             : {
      26      262129 :   GEN p1, s = *s0 = cxtoreal(*s0);
      27             :   long l;
      28      262129 :   l = precision(s); if (!l) l = *prec;
      29      262129 :   if (l < LOWDEFAULTPREC) l = LOWDEFAULTPREC;
      30      262129 :   *res = cgetc(l); *av = avma;
      31      262129 :   if (typ(s) == t_COMPLEX)
      32             :   { /* s = sig + i t */
      33      207424 :     s = cxtofp(s, l+EXTRAPRECWORD);
      34      207424 :     *sig = gel(s,1);
      35      207424 :     *tau = gel(s,2);
      36             :   }
      37             :   else /* real number */
      38             :   {
      39       54705 :     *sig = s = gtofp(s, l+EXTRAPRECWORD);
      40       54705 :     *tau = gen_0;
      41       54705 :     p1 = trunc2nr(s, 0);
      42       54705 :     if (!signe(subri(s,p1))) *s0 = p1;
      43             :   }
      44      262129 :   *prec = l; return s;
      45             : }
      46             : 
      47             : /********************************************************************/
      48             : /**                                                                **/
      49             : /**                          ARCTANGENT                            **/
      50             : /**                                                                **/
      51             : /********************************************************************/
      52             : /* atan(b/a), real a and b, suitable for gerepileupto */
      53             : static GEN
      54         196 : atan2_agm(GEN a, GEN b, long prec)
      55         196 : { return gel(logagmcx(mkcomplex(a, b), prec), 2); }
      56             : static GEN
      57      553895 : mpatan(GEN x)
      58             : {
      59      553895 :   long l, l1, l2, n, m, i, lp, e, s, sx = signe(x);
      60             :   pari_sp av0, av;
      61             :   double alpha, beta, delta;
      62             :   GEN y, p1, p2, p3, p4, p5, unr;
      63             :   int inv;
      64             : 
      65      553895 :   if (!sx) return real_0_bit(expo(x));
      66      553832 :   l = lp = realprec(x);
      67      553832 :   if (absrnz_equal1(x)) { /* |x| = 1 */
      68        3688 :     y = Pi2n(-2, l+EXTRAPRECWORD); if (sx < 0) setsigne(y,-1);
      69        3688 :     return y;
      70             :   }
      71      550144 :   if (l > AGM_ATAN_LIMIT)
      72         175 :   { av = avma; return gerepileuptoleaf(av, atan2_agm(gen_1, x, l)); }
      73             : 
      74      549969 :   e = expo(x); inv = (e >= 0); /* = (|x| > 1 ) */
      75      549969 :   if (e > 0) lp += nbits2extraprec(e);
      76             : 
      77      549969 :   y = cgetr(lp); av0 = avma;
      78      549969 :   p1 = rtor(x, l+EXTRAPRECWORD); setabssign(p1); /* p1 = |x| */
      79      549969 :   if (inv) p1 = invr(p1);
      80      549969 :   e = expo(p1);
      81      549969 :   if (e < -100)
      82        4923 :     alpha = 1.65149612947 - e; /* log_2(Pi) - e */
      83             :   else
      84      545046 :     alpha = log2(M_PI / atan(rtodbl(p1)));
      85      549969 :   beta = (double)(prec2nbits(l)>>1);
      86      549969 :   delta = 1 + beta - alpha/2;
      87      549969 :   if (delta <= 0) { n = 1; m = 0; }
      88             :   else
      89             :   {
      90      546349 :     double fi = alpha-2;
      91      546349 :     if (delta >= fi*fi)
      92             :     {
      93      527362 :       double t = 1 + sqrt(delta);
      94      527362 :       n = (long)t;
      95      527362 :       m = (long)(t - fi);
      96             :     }
      97             :     else
      98             :     {
      99       18987 :       n = (long)(1+beta/fi);
     100       18987 :       m = 0;
     101             :     }
     102             :   }
     103      549969 :   l2 = l + nbits2extraprec(m);
     104      549969 :   p2 = rtor(p1, l2); av = avma;
     105     5287990 :   for (i=1; i<=m; i++)
     106             :   {
     107     4738021 :     p5 = addsr(1, sqrr(p2)); setprec(p5,l2);
     108     4738021 :     p5 = addsr(1, sqrtr_abs(p5)); setprec(p5,l2);
     109     4738021 :     affrr(divrr(p2,p5), p2); set_avma(av);
     110             :   }
     111      549969 :   p3 = sqrr(p2); l1 = minss(LOWDEFAULTPREC+EXTRAPRECWORD, l2); /* l1 increases to l2 */;
     112      549969 :   unr = real_1(l2); setprec(unr,l1);
     113      549969 :   p4 = cgetr(l2); setprec(p4,l1);
     114      549969 :   affrr(divru(unr,2*n+1), p4);
     115      549969 :   s = 0; e = expo(p3); av = avma;
     116     5935121 :   for (i = n; i > 1; i--) /* n >= 1. i = 1 done outside for efficiency */
     117             :   {
     118     5385152 :     setprec(p3,l1); p5 = mulrr(p4,p3);
     119     5385152 :     l1 += dvmdsBIL(s - e, &s); if (l1 > l2) l1 = l2;
     120     5385152 :     setprec(unr,l1); p5 = subrr(divru(unr,2*i-1), p5);
     121     5385152 :     setprec(p4,l1); affrr(p5,p4); set_avma(av);
     122             :   }
     123      549969 :   setprec(p3, l2); p5 = mulrr(p4,p3); /* i = 1 */
     124      549969 :   setprec(unr,l2); p4 = subrr(unr, p5);
     125             : 
     126      549969 :   p4 = mulrr(p2,p4); shiftr_inplace(p4, m);
     127      549969 :   if (inv) p4 = subrr(Pi2n(-1, lp), p4);
     128      549969 :   if (sx < 0) togglesign(p4);
     129      549969 :   affrr_fixlg(p4,y); set_avma(av0); return y;
     130             : }
     131             : 
     132             : GEN
     133       19374 : gatan(GEN x, long prec)
     134             : {
     135             :   pari_sp av;
     136             :   GEN a, y;
     137             : 
     138       19374 :   switch(typ(x))
     139             :   {
     140       11114 :     case t_REAL: return mpatan(x);
     141             :     case t_COMPLEX: /* atan(x) = -i atanh(ix) */
     142        7672 :       if (ismpzero(gel(x,2))) return gatan(gel(x,1), prec);
     143        7441 :       av = avma; return gerepilecopy(av, mulcxmI(gatanh(mulcxI(x),prec)));
     144             :     default:
     145         588 :       av = avma; if (!(y = toser_i(x))) break;
     146          28 :       if (valp(y) < 0) pari_err_DOMAIN("atan","valuation", "<", gen_0, x);
     147          21 :       if (lg(y)==2) return gerepilecopy(av, y);
     148             :       /* lg(y) > 2 */
     149          14 :       a = integser(gdiv(derivser(y), gaddsg(1,gsqr(y))));
     150          14 :       if (!valp(y)) a = gadd(a, gatan(gel(y,2),prec));
     151          14 :       return gerepileupto(av, a);
     152             :   }
     153         560 :   return trans_eval("atan",gatan,x,prec);
     154             : }
     155             : /********************************************************************/
     156             : /**                                                                **/
     157             : /**                             ARCSINE                            **/
     158             : /**                                                                **/
     159             : /********************************************************************/
     160             : /* |x| < 1, x != 0 */
     161             : static GEN
     162          98 : mpasin(GEN x) {
     163          98 :   pari_sp av = avma;
     164          98 :   GEN z, a = sqrtr(subsr(1, sqrr(x)));
     165          98 :   if (realprec(x) > AGM_ATAN_LIMIT)
     166           7 :     z = atan2_agm(a, x, realprec(x));
     167             :   else
     168          91 :     z = mpatan(divrr(x, a));
     169          98 :   return gerepileuptoleaf(av, z);
     170             : }
     171             : 
     172             : static GEN mpacosh(GEN x);
     173             : GEN
     174        8183 : gasin(GEN x, long prec)
     175             : {
     176             :   long sx;
     177             :   pari_sp av;
     178             :   GEN a, y, p1;
     179             : 
     180        8183 :   switch(typ(x))
     181             :   {
     182         483 :     case t_REAL: sx = signe(x);
     183         483 :       if (!sx) return real_0_bit(expo(x));
     184         476 :       if (absrnz_equal1(x)) { /* |x| = 1 */
     185          28 :         if (sx > 0) return Pi2n(-1, realprec(x)); /* 1 */
     186          14 :         y = Pi2n(-1, realprec(x)); setsigne(y, -1); return y; /* -1 */
     187             :       }
     188         448 :       if (expo(x) < 0) return mpasin(x);
     189         350 :       y = cgetg(3,t_COMPLEX);
     190         350 :       gel(y,1) = Pi2n(-1, realprec(x));
     191         350 :       gel(y,2) = mpacosh(x);
     192         350 :       if (sx < 0) togglesign(gel(y,1)); else togglesign(gel(y,2));
     193         350 :       return y;
     194             : 
     195             :     case t_COMPLEX: /* asin(z) = -i asinh(iz) */
     196        7637 :       if (ismpzero(gel(x,2))) return gasin(gel(x,1), prec);
     197        7406 :       av = avma;
     198        7406 :       return gerepilecopy(av, mulcxmI(gasinh(mulcxI(x), prec)));
     199             :     default:
     200          63 :       av = avma; if (!(y = toser_i(x))) break;
     201          42 :       if (gequal0(y)) return gerepilecopy(av, y);
     202             :       /* lg(y) > 2*/
     203          35 :       if (valp(y) < 0) pari_err_DOMAIN("asin","valuation", "<", gen_0, x);
     204          28 :       p1 = gsubsg(1,gsqr(y));
     205          28 :       if (gequal0(p1))
     206             :       {
     207          21 :         GEN t = Pi2n(-1,prec);
     208          21 :         if (gsigne(gel(y,2)) < 0) setsigne(t, -1);
     209          21 :         return gerepileupto(av, scalarser(t, varn(y), valp(p1)>>1));
     210             :       }
     211           7 :       p1 = gdiv(derivser(y), gsqrt(p1,prec));
     212           7 :       a = integser(p1);
     213           7 :       if (!valp(y)) a = gadd(a, gasin(gel(y,2),prec));
     214           7 :       return gerepileupto(av, a);
     215             :   }
     216          21 :   return trans_eval("asin",gasin,x,prec);
     217             : }
     218             : /********************************************************************/
     219             : /**                                                                **/
     220             : /**                             ARCCOSINE                          **/
     221             : /**                                                                **/
     222             : /********************************************************************/
     223             : static GEN
     224          14 : acos0(long e) { return Pi2n(-1, nbits2prec(e<0? -e: 1)); }
     225             : 
     226             : /* |x| < 1, x != 0 */
     227             : static GEN
     228         105 : mpacos(GEN x)
     229             : {
     230         105 :   pari_sp av = avma;
     231         105 :   GEN z, a = sqrtr(subsr(1, sqrr(x)));
     232         105 :   if (realprec(x) > AGM_ATAN_LIMIT)
     233          14 :     z = atan2_agm(x, a, realprec(x));
     234             :   else
     235             :   {
     236          91 :     z = mpatan(divrr(a, x));
     237          91 :     if (signe(x) < 0) z = addrr(mppi(realprec(z)), z);
     238             :   }
     239         105 :   return gerepileuptoleaf(av, z);
     240             : }
     241             : 
     242             : GEN
     243        7938 : gacos(GEN x, long prec)
     244             : {
     245             :   long sx;
     246             :   pari_sp av;
     247             :   GEN a, y, p1;
     248             : 
     249        7938 :   switch(typ(x))
     250             :   {
     251         252 :     case t_REAL: sx = signe(x);
     252         252 :       if (!sx) return acos0(expo(x));
     253         245 :       if (absrnz_equal1(x)) /* |x| = 1 */
     254          14 :         return sx > 0? real_0_bit( -(bit_prec(x)>>1) ) : mppi(realprec(x));
     255         231 :       if (expo(x) < 0) return mpacos(x);
     256             : 
     257         175 :       y = cgetg(3,t_COMPLEX); p1 = mpacosh(x);
     258         175 :       if (sx < 0) { gel(y,1) = mppi(realprec(x)); togglesign(p1); }
     259          91 :       else gel(y,1) = gen_0;
     260         175 :       gel(y,2) = p1; return y;
     261             : 
     262             :     case t_COMPLEX:
     263        7637 :       if (ismpzero(gel(x,2))) return gacos(gel(x,1), prec);
     264        7406 :       av = avma;
     265        7406 :       p1 = gadd(x, mulcxI(gsqrt(gsubsg(1,gsqr(x)), prec)));
     266        7406 :       y = glog(p1,prec); /* log(x + I*sqrt(1-x^2)) */
     267        7406 :       return gerepilecopy(av, mulcxmI(y));
     268             :     default:
     269          49 :       av = avma; if (!(y = toser_i(x))) break;
     270          35 :       if (valp(y) < 0) pari_err_DOMAIN("acos","valuation", "<", gen_0, x);
     271          28 :       if (lg(y) > 2)
     272             :       {
     273          21 :         p1 = gsubsg(1,gsqr(y));
     274          21 :         if (gequal0(p1)) { set_avma(av); return zeroser(varn(y), valp(p1)>>1); }
     275           7 :         p1 = integser(gdiv(gneg(derivser(y)), gsqrt(p1,prec)));
     276             :         /*y(t) = 1+O(t)*/
     277           7 :         if (gequal1(gel(y,2)) && !valp(y)) return gerepileupto(av, p1);
     278             :       }
     279           7 :       else p1 = y;
     280          14 :       a = (lg(y)==2 || valp(y))? Pi2n(-1, prec): gacos(gel(y,2),prec);
     281          14 :       return gerepileupto(av, gadd(a,p1));
     282             :   }
     283          14 :   return trans_eval("acos",gacos,x,prec);
     284             : }
     285             : /********************************************************************/
     286             : /**                                                                **/
     287             : /**                            ARGUMENT                            **/
     288             : /**                                                                **/
     289             : /********************************************************************/
     290             : 
     291             : /* we know that x and y are not both 0 */
     292             : static GEN
     293      542725 : mparg(GEN x, GEN y)
     294             : {
     295      542725 :   long prec, sx = signe(x), sy = signe(y);
     296             :   GEN z;
     297             : 
     298      542725 :   if (!sy)
     299             :   {
     300         112 :     if (sx > 0) return real_0_bit(expo(y) - expo(x));
     301          49 :     return mppi(realprec(x));
     302             :   }
     303      542613 :   prec = realprec(y); if (prec < realprec(x)) prec = realprec(x);
     304      542613 :   if (!sx)
     305             :   {
     306          14 :     z = Pi2n(-1, prec); if (sy < 0) setsigne(z,-1);
     307          14 :     return z;
     308             :   }
     309             : 
     310      542599 :   if (expo(x)-expo(y) > -2)
     311             :   {
     312      487807 :     z = mpatan(divrr(y,x)); if (sx > 0) return z;
     313      103092 :     return addrr_sign(z, signe(z), mppi(prec), sy);
     314             :   }
     315       54792 :   z = mpatan(divrr(x,y));
     316       54792 :   return addrr_sign(z, -signe(z), Pi2n(-1, prec), sy);
     317             : }
     318             : 
     319             : static GEN
     320     1085450 : rfix(GEN x,long prec)
     321             : {
     322     1085450 :   switch(typ(x))
     323             :   {
     324        2831 :     case t_INT: return itor(x, prec);
     325        5607 :     case t_FRAC: return fractor(x, prec);
     326     1077012 :     case t_REAL: break;
     327           0 :     default: pari_err_TYPE("rfix (conversion to t_REAL)",x);
     328             :   }
     329     1077012 :   return x;
     330             : }
     331             : 
     332             : static GEN
     333      542725 : cxarg(GEN x, GEN y, long prec)
     334             : {
     335      542725 :   pari_sp av = avma;
     336      542725 :   x = rfix(x,prec);
     337      542725 :   y = rfix(y,prec); return gerepileuptoleaf(av, mparg(x,y));
     338             : }
     339             : 
     340             : GEN
     341      543866 : garg(GEN x, long prec)
     342             : {
     343             :   long l;
     344      543866 :   if (gequal0(x)) pari_err_DOMAIN("arg", "argument", "=", gen_0, x);
     345      543866 :   switch(typ(x))
     346             :   {
     347        1141 :     case t_REAL: prec = realprec(x); /* fall through */
     348        1141 :     case t_INT: case t_FRAC: return (gsigne(x)>0)? real_0(prec): mppi(prec);
     349             :     case t_COMPLEX:
     350      542725 :       l = precision(x); if (l) prec = l;
     351      542725 :       return cxarg(gel(x,1),gel(x,2),prec);
     352             :   }
     353           0 :   return trans_eval("arg",garg,x,prec);
     354             : }
     355             : 
     356             : /********************************************************************/
     357             : /**                                                                **/
     358             : /**                      HYPERBOLIC COSINE                         **/
     359             : /**                                                                **/
     360             : /********************************************************************/
     361             : 
     362             : static GEN
     363        3584 : mpcosh(GEN x)
     364             : {
     365             :   pari_sp av;
     366             :   GEN z;
     367             : 
     368        3584 :   if (!signe(x)) { /* 1 + x */
     369           7 :     long e = expo(x);
     370           7 :     return e >= 0? real_0_bit(e): real_1_bit(-e);
     371             :   }
     372        3577 :   av = avma;
     373        3577 :   z = mpexp(x); z = addrr(z, invr(z)); shiftr_inplace(z, -1);
     374        3577 :   return gerepileuptoleaf(av, z);
     375             : }
     376             : 
     377             : GEN
     378        3668 : gcosh(GEN x, long prec)
     379             : {
     380             :   pari_sp av;
     381             :   GEN y, p1;
     382             : 
     383        3668 :   switch(typ(x))
     384             :   {
     385        3584 :     case t_REAL: return mpcosh(x);
     386             :     case t_COMPLEX:
     387          21 :       if (isintzero(gel(x,1))) return gcos(gel(x,2),prec);
     388             :       /* fall through */
     389             :     case t_PADIC:
     390          21 :       av = avma; p1 = gexp(x,prec); p1 = gadd(p1, ginv(p1));
     391          21 :       return gerepileupto(av, gmul2n(p1,-1));
     392             :     default:
     393          49 :       av = avma; if (!(y = toser_i(x))) break;
     394          28 :       if (gequal0(y) && valp(y) == 0) return gerepilecopy(av, y);
     395          28 :       p1 = gexp(y,prec); p1 = gadd(p1, ginv(p1));
     396          28 :       return gerepileupto(av, gmul2n(p1,-1));
     397             :   }
     398          21 :   return trans_eval("cosh",gcosh,x,prec);
     399             : }
     400             : /********************************************************************/
     401             : /**                                                                **/
     402             : /**                       HYPERBOLIC SINE                          **/
     403             : /**                                                                **/
     404             : /********************************************************************/
     405             : 
     406             : static GEN
     407       15365 : mpsinh(GEN x)
     408             : {
     409             :   pari_sp av;
     410       15365 :   long ex = expo(x), lx;
     411             :   GEN z, res;
     412             : 
     413       15365 :   if (!signe(x)) return real_0_bit(ex);
     414       15365 :   lx = realprec(x); res = cgetr(lx); av = avma;
     415       15365 :   if (ex < 1 - BITS_IN_LONG)
     416             :   { /* y = e^x-1; e^x - e^(-x) = y(1 + 1/(y+1)) */
     417           7 :     GEN y = mpexpm1(x);
     418           7 :     z = addrs(y,1); if (lg(z) > lx+1) z = rtor(z,lx+1); /* e^x */
     419           7 :     z = mulrr(y, addsr(1,invr(z)));
     420             :   }
     421             :   else
     422             :   {
     423       15358 :     z = mpexp(x);
     424       15358 :     z = subrr(z, invr(z));
     425             :   }
     426       15365 :   shiftr_inplace(z, -1);
     427       15365 :   affrr(z, res); set_avma(av); return res;
     428             : }
     429             : 
     430             : GEN
     431       17311 : gsinh(GEN x, long prec)
     432             : {
     433             :   pari_sp av;
     434             :   GEN y, p1;
     435             : 
     436       17311 :   switch(typ(x))
     437             :   {
     438       15365 :     case t_REAL: return mpsinh(x);
     439             :     case t_COMPLEX:
     440          21 :       if (isintzero(gel(x,1))) retmkcomplex(gen_0, gsin(gel(x,2),prec));
     441             :       /* fall through */
     442             :     case t_PADIC:
     443          14 :       av = avma; p1 = gexp(x,prec); p1 = gsub(p1, ginv(p1));
     444          14 :       return gerepileupto(av, gmul2n(p1,-1));
     445             :     default:
     446        1918 :       av = avma; if (!(y = toser_i(x))) break;
     447        1890 :       if (gequal0(y) && valp(y) == 0) return gerepilecopy(av, y);
     448        1890 :       p1 = gexp(y, prec); p1 = gsub(p1, ginv(p1));
     449        1890 :       return gerepileupto(av, gmul2n(p1,-1));
     450             :   }
     451          28 :   return trans_eval("sinh",gsinh,x,prec);
     452             : }
     453             : /********************************************************************/
     454             : /**                                                                **/
     455             : /**                      HYPERBOLIC TANGENT                        **/
     456             : /**                                                                **/
     457             : /********************************************************************/
     458             : 
     459             : static GEN
     460       77056 : mptanh(GEN x)
     461             : {
     462       77056 :   long lx, s = signe(x);
     463             :   GEN y;
     464             : 
     465       77056 :   if (!s) return real_0_bit(expo(x));
     466       77056 :   lx = realprec(x);
     467       77056 :   if (abscmprr(x, utor(prec2nbits(lx), LOWDEFAULTPREC)) >= 0) {
     468       24840 :     y = real_1(lx);
     469             :   } else {
     470       52216 :     pari_sp av = avma;
     471       52216 :     long ex = expo(x);
     472             :     GEN t;
     473       52216 :     if (ex < 1 - BITS_IN_LONG) x = rtor(x, lx + nbits2extraprec(-ex)-1);
     474       52216 :     t = exp1r_abs(gmul2n(x,1)); /* exp(|2x|) - 1 */
     475       52216 :     y = gerepileuptoleaf(av, divrr(t, addsr(2,t)));
     476             :   }
     477       77056 :   if (s < 0) togglesign(y); /* tanh is odd */
     478       77056 :   return y;
     479             : }
     480             : 
     481             : GEN
     482       77161 : gtanh(GEN x, long prec)
     483             : {
     484             :   pari_sp av;
     485             :   GEN y, t;
     486             : 
     487       77161 :   switch(typ(x))
     488             :   {
     489       77056 :     case t_REAL: return mptanh(x);
     490             :     case t_COMPLEX:
     491          35 :       if (isintzero(gel(x,1))) retmkcomplex(gen_0, gtan(gel(x,2),prec));
     492             :       /* fall through */
     493             :     case t_PADIC:
     494          28 :       av = avma;
     495          28 :       t = gexp(gmul2n(x,1),prec);
     496          28 :       t = gdivsg(-2, gaddgs(t,1));
     497          28 :       return gerepileupto(av, gaddsg(1,t));
     498             :     default:
     499          63 :       av = avma; if (!(y = toser_i(x))) break;
     500          28 :       if (gequal0(y)) return gerepilecopy(av, y);
     501          14 :       t = gexp(gmul2n(y, 1),prec);
     502          14 :       t = gdivsg(-2, gaddgs(t,1));
     503          14 :       return gerepileupto(av, gaddsg(1,t));
     504             :   }
     505          35 :   return trans_eval("tanh",gtanh,x,prec);
     506             : }
     507             : 
     508             : static GEN
     509           7 : mpcotanh(GEN x)
     510             : {
     511           7 :   long lx, s = signe(x);
     512             :   GEN y;
     513             : 
     514           7 :   if (!s) pari_err_DOMAIN("cotan", "argument", "=", gen_0, x);
     515             : 
     516           7 :   lx = realprec(x);
     517           7 :   if (abscmprr(x, utor(prec2nbits(lx), LOWDEFAULTPREC)) >= 0) {
     518           0 :     y = real_1(lx);
     519             :   } else {
     520           7 :     pari_sp av = avma;
     521           7 :     long ex = expo(x);
     522             :     GEN t;
     523           7 :     if (ex < 1 - BITS_IN_LONG) x = rtor(x, lx + nbits2extraprec(-ex)-1);
     524           7 :     t = exp1r_abs(gmul2n(x,1)); /* exp(|2x|) - 1 */
     525           7 :     y = gerepileuptoleaf(av, divrr(addsr(2,t), t));
     526             :   }
     527           7 :   if (s < 0) togglesign(y); /* cotanh is odd */
     528           7 :   return y;
     529             : }
     530             : 
     531             : GEN
     532          63 : gcotanh(GEN x, long prec)
     533             : {
     534             :   pari_sp av;
     535             :   GEN y, t;
     536             : 
     537          63 :   switch(typ(x))
     538             :   {
     539           7 :     case t_REAL: return mpcotanh(x);
     540             :     case t_COMPLEX:
     541          14 :       if (isintzero(gel(x,1))) retmkcomplex(gen_0, gcotan(gel(x,2),prec));
     542             :       /* fall through */
     543             :     case t_PADIC:
     544          14 :       av = avma;
     545          14 :       t = gexpm1(gmul2n(x,1),prec);
     546          14 :       return gerepileupto(av, gaddsg(1, gdivsg(2,t)));
     547             :     default:
     548          35 :       av = avma; if (!(y = toser_i(x))) break;
     549          28 :       if (gequal0(y)) return gerepilecopy(av, y);
     550          14 :       t = gexpm1(gmul2n(y,1),prec);
     551          14 :       return gerepileupto(av, gaddsg(1, gdivsg(2,t)));
     552             :   }
     553           7 :   return trans_eval("cotanh",gcotanh,x,prec);
     554             : }
     555             : 
     556             : /********************************************************************/
     557             : /**                                                                **/
     558             : /**                     AREA HYPERBOLIC SINE                       **/
     559             : /**                                                                **/
     560             : /********************************************************************/
     561             : 
     562             : /* x != 0 */
     563             : static GEN
     564         483 : mpasinh(GEN x)
     565             : {
     566             :   GEN z, res;
     567             :   pari_sp av;
     568         483 :   long lx = realprec(x), ex = expo(x);
     569             : 
     570         483 :   res = cgetr(lx); av = avma;
     571         483 :   if (ex < 1 - BITS_IN_LONG) x = rtor(x, lx + nbits2extraprec(-ex)-1);
     572         483 :   z = logr_abs( addrr_sign(x,1, sqrtr_abs( addrs(sqrr(x), 1) ), 1) );
     573         483 :   if (signe(x) < 0) togglesign(z);
     574         483 :   affrr(z, res); set_avma(av); return res;
     575             : }
     576             : 
     577             : GEN
     578       15715 : gasinh(GEN x, long prec)
     579             : {
     580             :   pari_sp av;
     581             :   GEN a, y, p1;
     582             : 
     583       15715 :   switch(typ(x))
     584             :   {
     585             :     case t_REAL:
     586         490 :       if (!signe(x)) return rcopy(x);
     587         483 :       return mpasinh(x);
     588             : 
     589             :     case t_COMPLEX:
     590       15036 :       if (ismpzero(gel(x,2))) return gasinh(gel(x,1), prec);
     591       14581 :       av = avma;
     592       14581 :       if (ismpzero(gel(x,1))) /* avoid cancellation */
     593         231 :         return gerepilecopy(av, mulcxI(gasin(gel(x,2), prec)));
     594       14350 :       p1 = gadd(x, gsqrt(gaddsg(1,gsqr(x)), prec));
     595       14350 :       y = glog(p1,prec); /* log (x + sqrt(1+x^2)) */
     596       14350 :       return gerepileupto(av, y);
     597             :     default:
     598         189 :       av = avma; if (!(y = toser_i(x))) break;
     599         161 :       if (gequal0(y)) return gerepilecopy(av, y);
     600         154 :       if (valp(y) < 0) pari_err_DOMAIN("asinh","valuation", "<", gen_0, x);
     601         147 :       p1 = gaddsg(1,gsqr(y));
     602         147 :       if (gequal0(p1))
     603             :       {
     604          14 :         GEN t = PiI2n(-1,prec);
     605          14 :         if ( gsigne(imag_i(gel(y,2))) < 0 ) setsigne(gel(t,2), -1);
     606          14 :         return gerepileupto(av, scalarser(t, varn(y), valp(p1)>>1));
     607             :       }
     608         133 :       p1 = gdiv(derivser(y), gsqrt(p1,prec));
     609         133 :       a = integser(p1);
     610         133 :       if (!valp(y)) a = gadd(a, gasinh(gel(y,2),prec));
     611         133 :       return gerepileupto(av, a);
     612             :   }
     613          28 :   return trans_eval("asinh",gasinh,x,prec);
     614             : }
     615             : /********************************************************************/
     616             : /**                                                                **/
     617             : /**                     AREA HYPERBOLIC COSINE                     **/
     618             : /**                                                                **/
     619             : /********************************************************************/
     620             : 
     621             : /* |x| >= 1, return ach(|x|) */
     622             : static GEN
     623         728 : mpacosh(GEN x)
     624             : {
     625         728 :   pari_sp av = avma;
     626             :   GEN z;
     627         728 :   if (absrnz_equal1(x)) return real_0_bit(- bit_prec(x) >> 1);
     628         721 :   z = logr_abs( addrr_sign(x, 1, sqrtr( subrs(sqrr(x), 1) ), 1) );
     629         721 :   return gerepileuptoleaf(av, z);
     630             : }
     631             : 
     632             : GEN
     633        7980 : gacosh(GEN x, long prec)
     634             : {
     635             :   pari_sp av;
     636             :   GEN y, p1;
     637             : 
     638        7980 :   switch(typ(x))
     639             :   {
     640             :     case t_REAL: {
     641         266 :       long s = signe(x), e = expo(x);
     642             :       GEN a, b;
     643         266 :       if (s > 0 && e >= 0) return mpacosh(x);
     644             :       /* x < 1 */
     645         147 :       y = cgetg(3,t_COMPLEX); a = gen_0;
     646         147 :       if (s == 0) b = acos0(e);
     647         140 :       else if (e < 0) b = mpacos(x); /* -1 < x < 1 */
     648             :       else {
     649          91 :         if (!absrnz_equal1(x)) a = mpacosh(x);
     650          91 :         b = mppi(realprec(x));
     651             :       }
     652         147 :       gel(y,1) = a;
     653         147 :       gel(y,2) = b; return y;
     654             :     }
     655             :     case t_COMPLEX:
     656        7637 :       if (ismpzero(gel(x,2))) return gacosh(gel(x,1), prec);
     657        7406 :       av = avma;
     658        7406 :       p1 = gadd(x, gsqrt(gaddsg(-1,gsqr(x)), prec));
     659        7406 :       y = glog(p1,prec); /* log(x + sqrt(x^2-1)) */
     660        7406 :       if (signe(real_i(y)) < 0) y = gneg(y);
     661        7406 :       return gerepileupto(av, y);
     662             :     default: {
     663             :       GEN a;
     664             :       long v;
     665          77 :       av = avma; if (!(y = toser_i(x))) break;
     666          49 :       v = valp(y);
     667          49 :       if (v < 0) pari_err_DOMAIN("acosh","valuation", "<", gen_0, x);
     668          42 :       if (gequal0(y))
     669             :       {
     670           7 :         if (!v) return gerepilecopy(av, y);
     671           7 :         return gerepileupto(av, gadd(y, PiI2n(-1, prec)));
     672             :       }
     673          35 :       p1 = gsubgs(gsqr(y),1);
     674          35 :       if (gequal0(p1)) { set_avma(av); return zeroser(varn(y), valp(p1)>>1); }
     675          21 :       p1 = gdiv(derivser(y), gsqrt(p1,prec));
     676          21 :       a = integser(p1);
     677          21 :       if (v)
     678           7 :         p1 = PiI2n(-1, prec); /* I Pi/2 */
     679             :       else
     680             :       {
     681          14 :         p1 = gel(y,2); if (gequal1(p1)) return gerepileupto(av,a);
     682           7 :         p1 = gacosh(p1, prec);
     683             :       }
     684          14 :       return gerepileupto(av, gadd(p1,a));
     685             :     }
     686             :   }
     687          28 :   return trans_eval("acosh",gacosh,x,prec);
     688             : }
     689             : /********************************************************************/
     690             : /**                                                                **/
     691             : /**                     AREA HYPERBOLIC TANGENT                    **/
     692             : /**                                                                **/
     693             : /********************************************************************/
     694             : 
     695             : /* |x| < 1, x != 0 */
     696             : static GEN
     697          98 : mpatanh(GEN x)
     698             : {
     699          98 :   pari_sp av = avma;
     700          98 :   long ex = expo(x);
     701             :   GEN z;
     702          98 :   if (ex < 1 - BITS_IN_LONG) x = rtor(x, realprec(x) + nbits2extraprec(-ex)-1);
     703          98 :   z = invr( subsr(1,x) ); shiftr_inplace(z, 1); /* 2/(1-x)*/
     704          98 :   z = logr_abs( addrs(z,-1) );
     705          98 :   shiftr_inplace(z, -1); return gerepileuptoleaf(av, z);
     706             : }
     707             : 
     708             : GEN
     709       15603 : gatanh(GEN x, long prec)
     710             : {
     711             :   long sx;
     712             :   pari_sp av;
     713             :   GEN a, y, z;
     714             : 
     715       15603 :   switch(typ(x))
     716             :   {
     717             :     case t_REAL:
     718         483 :       sx = signe(x);
     719         483 :       if (!sx) return real_0_bit(expo(x));
     720         476 :       if (expo(x) < 0) return mpatanh(x);
     721             : 
     722         378 :       y = cgetg(3,t_COMPLEX);
     723         378 :       av = avma;
     724         378 :       z = subrs(x,1);
     725         378 :       if (!signe(z)) pari_err_DOMAIN("atanh", "argument", "=", gen_1, x);
     726         364 :       z = invr(z); shiftr_inplace(z, 1); /* 2/(x-1)*/
     727         364 :       z = addrs(z,1);
     728         364 :       if (!signe(z)) pari_err_DOMAIN("atanh", "argument", "=", gen_m1, x);
     729         350 :       z = logr_abs(z);
     730         350 :       shiftr_inplace(z, -1); /* (1/2)log((1+x)/(x-1)) */
     731         350 :       gel(y,1) = gerepileuptoleaf(av, z);
     732         350 :       gel(y,2) = Pi2n(-1, realprec(x));
     733         350 :       if (sx > 0) togglesign(gel(y,2));
     734         350 :       return y;
     735             : 
     736             :     case t_COMPLEX: /* 2/(1-z) - 1 = (1+z) / (1-z) */
     737       15071 :       if (ismpzero(gel(x,2))) return gatanh(gel(x,1), prec);
     738       14616 :       av = avma; z = glog( gaddgs(gdivsg(2,gsubsg(1,x)),-1), prec );
     739       14616 :       return gerepileupto(av, gmul2n(z,-1));
     740             : 
     741             :     default:
     742          49 :       av = avma; if (!(y = toser_i(x))) break;
     743          28 :       if (valp(y) < 0) pari_err_DOMAIN("atanh","valuation", "<", gen_0, x);
     744          21 :       z = gdiv(derivser(y), gsubsg(1,gsqr(y)));
     745          14 :       a = integser(z);
     746          14 :       if (!valp(y)) a = gadd(a, gatanh(gel(y,2),prec));
     747          14 :       return gerepileupto(av, a);
     748             :   }
     749          21 :   return trans_eval("atanh",gatanh,x,prec);
     750             : }
     751             : /********************************************************************/
     752             : /**                                                                **/
     753             : /**                         EULER'S GAMMA                          **/
     754             : /**                                                                **/
     755             : /********************************************************************/
     756             : /* 0 < a < b */
     757             : static GEN
     758       22714 : mulu_interval_step_i(ulong a, ulong b, ulong step)
     759             : {
     760             :   ulong k, l, N, n;
     761             :   long lx;
     762             :   GEN x;
     763             : 
     764       22714 :   n = 1 + (b-a) / step;
     765       22714 :   b -= (b-a) % step;
     766             :   /* step | b-a */
     767       22714 :   lx = 1; x = cgetg(2 + n/2, t_VEC);
     768       22714 :   N = b + a;
     769      124701 :   for (k = a;; k += step)
     770             :   {
     771      226688 :     l = N - k; if (l <= k) break;
     772      101987 :     gel(x,lx++) = muluu(k,l);
     773             :   }
     774       22714 :   if (l == k) gel(x,lx++) = utoipos(k);
     775       22714 :   setlg(x, lx); return x;
     776             : }
     777             : static GEN
     778       99262 : _mul(void *data, GEN x, GEN y)
     779             : {
     780       99262 :   long prec = (long)data;
     781             :   /* switch to t_REAL ? */
     782       99262 :   if (typ(x) == t_INT && lgefint(x) > prec) x = itor(x, prec);
     783       99262 :   if (typ(y) == t_INT && lgefint(y) > prec) y = itor(y, prec);
     784       99262 :   return mpmul(x, y);
     785             : }
     786             : static GEN
     787       22714 : mulu_interval_step_prec(long l, long m, long s, long prec)
     788             : {
     789       22714 :   GEN v = mulu_interval_step_i(l, m, s);
     790       22714 :   return gen_product(v, (void*)prec, &_mul);
     791             : }
     792             : 
     793             : /* x * (i*(i+1)) */
     794             : static GEN
     795     8433452 : muliunu(GEN x, ulong i)
     796             : {
     797     8433452 :   if (i & HIGHMASK) /* i(i+1) >= 2^BITS_IN_LONG*/
     798           0 :     return muliu(muliu(x, i), i+1);
     799             :   else
     800     8433452 :     return muliu(x, i*(i+1));
     801             : }
     802             : /* x / (i*(i+1)) */
     803             : GEN
     804    22146279 : divrunu(GEN x, ulong i)
     805             : {
     806    22146279 :   if (i & HIGHMASK) /* i(i+1) >= 2^BITS_IN_LONG*/
     807           0 :     return divru(divru(x, i), i+1);
     808             :   else
     809    22146279 :     return divru(x, i*(i+1));
     810             : }
     811             : /* x / (i*(i+1)) */
     812             : GEN
     813      665997 : divgunu(GEN x, ulong i)
     814             : {
     815             : #ifdef LONG_IS_64BIT
     816      571248 :   if (i < 3037000500L) /* i(i+1) < 2^63 */
     817             : #else
     818       94749 :   if (i < 46341L) /* i(i+1) < 2^31 */
     819             : #endif
     820      665997 :     return gdivgs(x, i*(i+1));
     821             :   else
     822           0 :     return gdivgs(gdivgs(x, i), i+1);
     823             : }
     824             : 
     825             : /* arg(s+it) */
     826             : double
     827      255136 : darg(double s, double t)
     828             : {
     829             :   double x;
     830      255136 :   if (!t) return (s>0)? 0.: M_PI;
     831      203532 :   if (!s) return (t>0)? M_PI/2: -M_PI/2;
     832      203525 :   x = atan(t/s);
     833             :   return (s>0)? x
     834      203525 :               : ((t>0)? x+M_PI : x-M_PI);
     835             : }
     836             : 
     837             : void
     838      255136 : dcxlog(double s, double t, double *a, double *b)
     839             : {
     840      255136 :   *a = log(s*s + t*t) / 2; /* log |s| = Re(log(s)) */
     841      255136 :   *b = darg(s,t);          /* Im(log(s)) */
     842      255136 : }
     843             : 
     844             : double
     845       12838 : dabs(double s, double t) { return sqrt( s*s + t*t ); }
     846             : double
     847       25235 : dnorm(double s, double t) { return s*s + t*t; }
     848             : 
     849             : #if 0
     850             : /* x, z t_REAL. Compute unique x in ]-z,z] congruent to x mod 2z */
     851             : static GEN
     852             : red_mod_2z(GEN x, GEN z)
     853             : {
     854             :   GEN Z = gmul2n(z, 1), d = subrr(z, x);
     855             :   /* require little accuracy */
     856             :   if (!signe(d)) return x;
     857             :   setprec(d, nbits2prec(expo(d) - expo(Z)));
     858             :   return addrr(mulir(floorr(divrr(d, Z)), Z), x);
     859             : }
     860             : #endif
     861             : 
     862             : /* lngamma(1+z) = -Euler*z + sum_{i > 1} zeta(i)/i (-z)^i
     863             :  * at relative precision prec, |z| < 1 is small */
     864             : static GEN
     865        2279 : lngamma1(GEN z, long prec)
     866             : { /* sum_{i > l} |z|^(i-1) = |z|^l / (1-|z|) < 2^-B
     867             :    * for l > (B+1) / |log2(|z|)| */
     868        2279 :   long i, l = ceil((bit_accuracy(prec) + 1) / - dbllog2(z));
     869        2279 :   GEN zet, me = mpeuler(prec), s = gen_0;
     870        2279 :   setsigne(me, -1); /* -Euler */
     871        2279 :   if (l <= 1) return gmul(me, z);
     872        2104 :   zet = veczeta(gen_1, gen_2, l-1, prec); /* z[i] = zeta(i+1) */
     873       15942 :   for (i = l; i > 1; i--)
     874             :   {
     875       13838 :     GEN c = divru(gel(zet,i-1), i);
     876       13838 :     if (odd(i)) setsigne(c, -1);
     877       13838 :     s = gadd(gmul(s,z), c);
     878             :   }
     879        2104 :   return gmul(z, gadd(gmul(s,z), me));
     880             : }
     881             : /* B_i / (i(i-1)), i even. Sometimes NOT reduced (but gadd/gmul won't care)!*/
     882             : static GEN
     883     8433452 : bern_unu(long i)
     884     8433452 : { GEN B = bernfrac(i); return mkfrac(gel(B,1), muliunu(gel(B,2), i-1)); }
     885             : /* B_i / i, i even. Sometimes NOT reduced (but gadd/gmul won't care)!*/
     886             : static GEN
     887     1269334 : bern_u(long i)
     888     1269334 : { GEN B = bernfrac(i); return mkfrac(gel(B,1), muliu(gel(B,2), i)); }
     889             : /* sum_{i > 0} B_{2i}/(2i(2i-1)) * a^(i-1) */
     890             : static GEN
     891      227346 : lngamma_sum(GEN a, long N)
     892             : {
     893      227346 :   pari_sp av = avma;
     894      227346 :   GEN S = bern_unu(2*N);
     895             :   long i;
     896     8433452 :   for (i = 2*N-2; i > 0; i -= 2)
     897             :   {
     898     8206106 :     S = gadd(bern_unu(i), gmul(a,S));
     899     8206106 :     if (gc_needed(av,3))
     900             :     {
     901           0 :       if(DEBUGMEM>1) pari_warn(warnmem,"gamma: i = %ld", i);
     902           0 :       S = gerepileupto(av, S);
     903             :     }
     904             :   }
     905      227346 :   return S;
     906             : }
     907             : /* sum_{i > 0} B_{2i}/(2i) * a^i */
     908             : static GEN
     909       25242 : psi_sum(GEN a, long N)
     910             : {
     911       25242 :   pari_sp av = avma;
     912       25242 :   GEN S = bern_u(2*N);
     913             :   long i;
     914     1269334 :   for (i = 2*N-2; i > 0; i -= 2)
     915             :   {
     916     1244092 :     S = gadd(bern_u(i), gmul(a,S));
     917     1244092 :     if (gc_needed(av,3))
     918             :     {
     919           0 :       if(DEBUGMEM>1) pari_warn(warnmem,"psi: i = %ld", i);
     920           0 :       S = gerepileupto(av, S);
     921             :     }
     922             :   }
     923       25242 :   return gmul(a,S);
     924             : }
     925             : static GEN
     926      230104 : cxgamma(GEN s0, int dolog, long prec)
     927             : {
     928             :   GEN s, a, y, res, sig, tau, p1, nnx, pi, pi2, sqrtpi2;
     929             :   long i, lim, N, esig, et;
     930             :   pari_sp av, av2;
     931      230104 :   int funeq = 0;
     932             :   pari_timer T;
     933             : 
     934      230104 :   if (DEBUGLEVEL>5) timer_start(&T);
     935      230104 :   s = trans_fix_arg(&prec,&s0,&sig,&tau,&av,&res);
     936             : 
     937      230104 :   esig = expo(sig);
     938      230104 :   et = signe(tau)? expo(tau): 0;
     939      230104 :   if ((signe(sig) <= 0 || esig < -1) && et <= 16)
     940             :   { /* s <--> 1-s */
     941       30772 :     funeq = 1; s = gsubsg(1, s); sig = real_i(s);
     942             :   }
     943             : 
     944             :   /* find "optimal" parameters [lim, N] */
     945      230104 :   if (esig > 300 || et > 300)
     946          35 :   { /* |s| is HUGE ! Play safe and avoid inf / NaN */
     947             :     GEN S, iS, l2, la, u;
     948             :     double logla, l;
     949             : 
     950          35 :     S = gprec_w(s,LOWDEFAULTPREC);
     951             :     /* l2 ~ |lngamma(s))|^2 */
     952          35 :     l2 = gnorm(gmul(S, glog(S, LOWDEFAULTPREC)));
     953          35 :     l = (prec2nbits_mul(prec, M_LN2) - rtodbl(glog(l2,LOWDEFAULTPREC))/2) / 2.;
     954          35 :     if (l < 0) l = 0.;
     955             : 
     956          35 :     iS = imag_i(S);
     957          35 :     if (et > 0 && l > 0)
     958          21 :     {
     959          21 :       GEN t = gmul(iS, dbltor(M_PI / l)), logt = glog(t,LOWDEFAULTPREC);
     960          21 :       la = gmul(t, logt);
     961          21 :       if      (gcmpgs(la, 3) < 0)   { logla = log(3.); la = stoi(3); }
     962          14 :       else if (gcmpgs(la, 150) > 0) { logla = rtodbl(logt); la = t; }
     963           7 :       else logla = rtodbl(mplog(la));
     964             :     }
     965             :     else
     966             :     {
     967          14 :       logla = log(3.); la = stoi(3);
     968             :     }
     969          35 :     lim = (long)ceil(l / (1.+ logla));
     970          35 :     if (lim == 0) lim = 1;
     971             : 
     972          35 :     u = gmul(la, dbltor((lim-0.5)/M_PI));
     973          35 :     l2 = gsub(gsqr(u), gsqr(iS));
     974          35 :     if (signe(l2) > 0)
     975             :     {
     976          14 :       l2 = gsub(gsqrt(l2,3), sig);
     977          14 :       if (signe(l2) > 0) N = itos( gceil(l2) ); else N = 1;
     978             :     }
     979             :     else
     980          21 :       N = 1;
     981             :   }
     982             :   else
     983             :   { /* |s| is moderate. Use floats  */
     984      230069 :     double ssig = rtodbl(sig);
     985      230069 :     double st = typ(s) == t_REAL? 0.0: rtodbl(imag_i(s));
     986             :     double la, l,l2,u,v, rlogs, ilogs;
     987             : 
     988      230069 :     if (fabs(ssig-1) + fabs(st) < 1e-16)
     989             :     { /* s ~ 1: loggamma(1+u) ~ - Euler * u, cancellation */
     990        2758 :       if (funeq) /* s0 ~ 0: use lngamma(s0)+log(s0) = lngamma(s0+1) */
     991             :       {
     992          28 :         if (dolog)
     993           0 :           y = gsub(lngamma1(s0,prec), glog(s0,prec));
     994             :         else
     995          28 :           y = gdiv(gexp(lngamma1(s0,prec), prec), s0);
     996             :       }
     997             :       else
     998             :       {
     999        5488 :         if (isint1(s0)) { set_avma(av); return dolog? real_0(prec): real_1(prec); }
    1000        1026 :         y = lngamma1(gsubgs(s0,1),prec);
    1001        1026 :         if (!dolog) y = gexp(y,prec);
    1002             :       }
    1003        1054 :       set_avma(av); return affc_fixlg(y, res);
    1004             :     }
    1005      227311 :     dcxlog(ssig,st, &rlogs,&ilogs);
    1006             :     /* Re (s - 1/2) log(s) */
    1007      227311 :     u = (ssig - 0.5)*rlogs - st * ilogs;
    1008             :     /* Im (s - 1/2) log(s) */
    1009      227311 :     v = (ssig - 0.5)*ilogs + st * rlogs;
    1010             :     /* l2 = | (s - 1/2) log(s) - s + log(2Pi)/2 |^2 ~ |lngamma(s))|^2 */
    1011      227311 :     u = u - ssig + log(2.*M_PI)/2;
    1012      227311 :     v = v - st;
    1013      227311 :     l2 = u*u + v*v;
    1014      227311 :     if (l2 < 0.000001) l2 = 0.000001;
    1015      227311 :     l = (prec2nbits_mul(prec, M_LN2) - log(l2)/2) / 2.;
    1016      227311 :     if (l < 0) l = 0.;
    1017             : 
    1018      227311 :     if (st > 1 && l > 0)
    1019       63840 :     {
    1020       63840 :       double t = st * M_PI / l;
    1021       63840 :       la = t * log(t);
    1022       63840 :       if (la < 4.) la = 4.;
    1023       63840 :       if (la > 150) la = t;
    1024             :     }
    1025             :     else
    1026      163471 :       la = 4.; /* heuristic */
    1027      227311 :     lim = (long)ceil(l / (1.+ log(la)));
    1028      227311 :     if (lim == 0) lim = 1;
    1029             : 
    1030      227311 :     u = (lim-0.5) * la / M_PI;
    1031      227311 :     l2 = u*u - st*st;
    1032      227311 :     if (l2 > 0)
    1033             :     {
    1034      217741 :       double t = ceil(sqrt(l2) - ssig);
    1035      217741 :       N = (t < 1)? 1: (long)t;
    1036      217741 :       if (N < 1) N = 1;
    1037             :     }
    1038             :     else
    1039        9570 :       N = 1;
    1040             :   }
    1041      227346 :   if (DEBUGLEVEL>5) err_printf("lim, N: [%ld, %ld]\n",lim,N);
    1042      227346 :   incrprec(prec);
    1043             : 
    1044      227346 :   av2 = avma;
    1045      227346 :   y = s;
    1046      227346 :   if (typ(s0) == t_INT)
    1047             :   {
    1048        2143 :     ulong ss = itou_or_0(s0);
    1049        2143 :     if (signe(s0) <= 0)
    1050           0 :       pari_err_DOMAIN("gamma","argument", "=",
    1051             :                        strtoGENstr("non-positive integer"), s0);
    1052        2150 :     if (!ss || ss + (ulong)N < ss) {
    1053           7 :       for (i=1; i < N; i++)
    1054             :       {
    1055           0 :         y = mulri(y, addiu(s0, i));
    1056           0 :         if (gc_needed(av2,3))
    1057             :         {
    1058           0 :           if(DEBUGMEM>1) pari_warn(warnmem,"gamma");
    1059           0 :           y = gerepileuptoleaf(av2, y);
    1060             :         }
    1061             :       }
    1062             :     } else {
    1063       28932 :       for (i=1; i < N; i++)
    1064             :       {
    1065       26796 :         y = mulru(y, ss + i);
    1066       26796 :         if (gc_needed(av2,3))
    1067             :         {
    1068           0 :           if(DEBUGMEM>1) pari_warn(warnmem,"gamma");
    1069           0 :           y = gerepileuptoleaf(av2, y);
    1070             :         }
    1071             :       }
    1072             :     }
    1073        2143 :     if (dolog) y = logr_abs(y);
    1074             :   }
    1075             :   else
    1076             :   { /* Compute lngamma mod 2 I Pi */
    1077      225203 :     GEN sq = gsqr(s);
    1078      225203 :     pari_sp av3 = avma;
    1079     4728371 :     for (i = 1; i < N - 1; i += 2)
    1080             :     {
    1081     4503168 :       y = gmul(y, gaddsg(i*(i + 1), gadd(gmulsg(2*i + 1, s), sq)));
    1082     4503168 :       if (gc_needed(av2,3))
    1083             :       {
    1084           0 :         if(DEBUGMEM>1) pari_warn(warnmem,"gamma");
    1085           0 :         y = gerepileupto(av3, y);
    1086             :       }
    1087             :     }
    1088      225203 :     if (!odd(N)) y = gmul(y, gaddsg(N - 1, s));
    1089      225203 :     if (dolog)
    1090             :     {
    1091       25780 :       if (typ(s) == t_REAL) y = logr_abs(y);
    1092             :       else
    1093             :       { /* fix imaginary part */
    1094         273 :         long prec0 = LOWDEFAULTPREC;
    1095         273 :         GEN s0 = gprec_w(s, prec0), y0 = s0, k;
    1096         273 :         y0 = garg(y0, prec0); /* Im log(s) at low accuracy */
    1097         273 :         for (i=1; i < N; i++) y0 = gadd(y0, garg(gaddgs(s0,i), prec0));
    1098         273 :         y = glog(y, prec);
    1099         273 :         k = ground( gdiv(gsub(y0, imag_i(y)), Pi2n(1,prec0)) );
    1100         273 :         if (signe(k)) y = gadd(y, mulcxI(mulir(k, Pi2n(1, prec))));
    1101             :       }
    1102             :     }
    1103             :   }
    1104      227346 :   if (DEBUGLEVEL>5) timer_printf(&T,"product from 0 to N-1");
    1105      227346 :   constbern(lim);
    1106      227346 :   nnx = gaddgs(s, N); a = ginv(nnx);
    1107      227346 :   p1 = gsub(gmul(gsub(nnx, ghalf), glog(nnx,prec)), nnx);
    1108      227346 :   p1 = gadd(p1, gmul(a, lngamma_sum(gsqr(a), lim)));
    1109      227346 :   if (DEBUGLEVEL>5) timer_printf(&T,"Bernoulli sum");
    1110             : 
    1111      227346 :   pi = mppi(prec); pi2 = shiftr(pi, 1); sqrtpi2 = sqrtr(pi2);
    1112             : 
    1113      227346 :   if (dolog)
    1114             :   {
    1115       26026 :     if (funeq)
    1116             :     { /* recall that s = 1 - s0 */
    1117          14 :       GEN T = shiftr(sqrtpi2,-1); /* sqrt(2Pi)/2 */
    1118          14 :       if (typ(s) != t_REAL)
    1119             :       {
    1120             :         /* We compute log(sin(Pi s0)) so that it has branch cuts along
    1121             :         * (-oo, 0] and [1, oo).  To do this in a numerically stable way
    1122             :         * we must compute the log first then mangle its imaginary part.
    1123             :         * The rounding operation below is stable because we're rounding
    1124             :         * a number which is already within 1/4 of an integer. */
    1125             : 
    1126             :         /* z = log( sin(Pi s0) / (sqrt(2Pi)/2) ) */
    1127           7 :         GEN z = glog(gdiv(gsin(gmul(pi,s0),prec), T), prec);
    1128             :         /* b = (2 Re(s) - 1) / 4 */
    1129           7 :         GEN b = shiftr(subrs(shiftr(sig, 1), 1), -2);
    1130           7 :         y = gsub(y, z);
    1131           7 :         if (gsigne(imag_i(s)) > 0) togglesign(b);
    1132             :         /* z = 2Pi round( Im(z)/2Pi - b ) */
    1133           7 :         z = gmul(roundr(gsub(gdiv(imag_i(z), pi2), b)), pi2);
    1134           7 :         if (signe(z)) { /* y += I*z, z a t_REAL */
    1135           0 :           if (typ(y) == t_COMPLEX)
    1136           0 :             gel(y,2) = gadd(gel(y,2), z);
    1137             :           else
    1138           0 :             y = mkcomplex(y, z);
    1139             :         }
    1140             :       }
    1141             :       else
    1142             :       { /* s0 < 0, formula simplifies: imag(lngamma(s0)) = - Pi * floor(s0) */
    1143           7 :         GEN z = logr_abs(divrr(mpsin(gmul(pi,s0)), T));
    1144           7 :         y = gsub(y, z);
    1145           7 :         y = mkcomplex(y, mulri(pi, gfloor(s0)));
    1146             :       }
    1147          14 :       p1 = gneg(p1);
    1148             :     }
    1149             :     else /* y --> sqrt(2Pi) / y */
    1150       26012 :       y = gsub(logr_abs(sqrtpi2), y);
    1151       26026 :     y = gadd(p1, y);
    1152             :   }
    1153             :   else
    1154             :   {
    1155      201320 :     if (funeq)
    1156             :     { /* y --> y Pi/(sin(Pi s) * sqrt(2Pi)) = y sqrt(Pi/2)/sin(Pi s) */
    1157       30730 :       y = gdiv(gmul(shiftr(sqrtpi2,-1),y), gsin(gmul(pi,s0), prec));
    1158             :       /* don't use s above: sin(pi s0) = sin(pi s) and the former is
    1159             :        * more accurate, esp. if s0 ~ 0 */
    1160       30730 :       p1 = gneg(p1);
    1161             :     }
    1162             :     else /* y --> sqrt(2Pi) / y */
    1163      170590 :       y = gdiv(sqrtpi2, y);
    1164      201320 :     y = gmul(gexp(p1, prec), y);
    1165             :   }
    1166      227346 :   set_avma(av); return affc_fixlg(y, res);
    1167             : }
    1168             : 
    1169             : /* Theory says n > C * b^1.5 / log(b). Timings:
    1170             :  * b = 64*[1, 2, 3, 4, 5, 6, 7, 10, 20, 30, 50, 100, 200, 500];
    1171             :  * n = [1450, 1930, 2750, 3400, 4070, 5000, 6000, 8800, 26000, 50000, 130000,
    1172             :  *      380000, 1300000, 6000000]; */
    1173             : static long
    1174       32025 : gamma2_n(long prec)
    1175             : {
    1176       32025 :   long b = bit_accuracy(prec);
    1177       32025 :   if (b <=  64) return 1450;
    1178       31458 :   if (b <= 128) return 1930;
    1179       29645 :   if (b <= 192) return 2750;
    1180       16977 :   if (b <= 256) return 3400;
    1181        6806 :   if (b <= 320) return 4070;
    1182        6444 :   if (b <= 384) return 5000;
    1183        3986 :   if (b <= 448) return 6000;
    1184        3786 :   return 10.0 * b * sqrt(b) / log(b);
    1185             : }
    1186             : 
    1187             : /* m even, Gamma((m+1) / 2) */
    1188             : static GEN
    1189       32025 : gammahs(long m, long prec)
    1190             : {
    1191       32025 :   GEN y = cgetr(prec), z;
    1192       32025 :   pari_sp av = avma;
    1193       32025 :   long ma = labs(m);
    1194             : 
    1195       32025 :   if (ma > gamma2_n(prec))
    1196             :   {
    1197           0 :     z = stor(m + 1, prec); shiftr_inplace(z, -1);
    1198           0 :     affrr(cxgamma(z,0,prec), y);
    1199           0 :     set_avma(av); return y;
    1200             :   }
    1201       32025 :   z = sqrtr( mppi(prec) );
    1202       32025 :   if (m)
    1203             :   {
    1204       18452 :     GEN t = mulu_interval_step_prec(1, ma-1, 2, prec + EXTRAPRECWORD);
    1205       18452 :     if (m >= 0) z = mpmul(z,t);
    1206             :     else
    1207             :     {
    1208          77 :       z = mpdiv(z,t);
    1209          77 :       if ((m&3) == 2) setsigne(z,-1);
    1210             :     }
    1211       18452 :     shiftr_inplace(z, -m/2);
    1212             :   }
    1213       32025 :   affrr(z, y); set_avma(av); return y;
    1214             : }
    1215             : GEN
    1216          28 : ggammah(GEN x, long prec)
    1217             : {
    1218          28 :   switch(typ(x))
    1219             :   {
    1220             :     case t_INT:
    1221             :     {
    1222          21 :       long k = itos_or_0(x);
    1223          21 :       if (!k && signe(x)) pari_err_OVERFLOW("gamma");
    1224          21 :       return gammahs(k * 2, prec);
    1225             :     }
    1226             :     case t_REAL: case t_COMPLEX: case t_PADIC: case t_SER: {
    1227           7 :       pari_sp av = avma;
    1228           7 :       return gerepileupto(av, ggamma(gadd(x,ghalf), prec));
    1229             :     }
    1230             :   }
    1231           0 :   return trans_eval("gammah",ggammah,x,prec);
    1232             : }
    1233             : 
    1234             : /* find n such that n+v_p(n!)>=k p^2/(p-1)^2 */
    1235             : static long
    1236          28 : nboft(long k, long p)
    1237             : {
    1238          28 :   pari_sp av = avma;
    1239             :   long s, n;
    1240             : 
    1241          28 :   if (k <= 0) return 0;
    1242          28 :   k = itou( gceil(gdiv(mului(k, sqru(p)), sqru(p-1))) );
    1243          28 :   set_avma(av);
    1244          28 :   for (s=0, n=0; n+s < k; n++, s += u_lval(n, p));
    1245          28 :   return n;
    1246             : }
    1247             : 
    1248             : /* Using Dwork's expansion, compute \Gamma(px+1)=-\Gamma(px) with x a unit.
    1249             :  * See p-Adic Gamma Functions and Dwork Cohomology, Maurizio Boyarsky
    1250             :  * Transactions of the AMS, Vol. 257, No. 2. (Feb., 1980), pp. 359-369.
    1251             :  * Inspired by a GP script by Fernando Rodriguez-Villegas */
    1252             : static GEN
    1253          28 : gadw(GEN x, long p)
    1254             : {
    1255          28 :   pari_sp ltop = avma;
    1256          28 :   GEN s, t, u = cgetg(p+1, t_VEC);
    1257          28 :   long j, k, kp, n = nboft(precp(x)+valp(x)+1, p);
    1258             : 
    1259          28 :   t = s = gaddsg(1, zeropadic(gel(x,2), n));
    1260          28 :   gel(u, 1) = s;
    1261          28 :   gel(u, 2) = s;
    1262          42 :   for (j = 2; j < p; ++j)
    1263          14 :     gel(u, j+1) = gdivgs(gel(u, j), j);
    1264         378 :   for (k = 1, kp = p; k < n; ++k, kp += p) /* kp = k*p */
    1265             :   {
    1266             :     GEN c;
    1267         350 :     gel(u, 1) = gdivgs(gadd(gel(u, 1), gel(u, p)), kp);
    1268         812 :     for (j = 1; j < p; ++j)
    1269         462 :       gel(u, j+1) = gdivgs(gadd(gel(u, j), gel(u, j+1)), kp + j);
    1270             : 
    1271         350 :     t = gmul(t, gaddgs(x, k-1));
    1272         350 :     c = leafcopy(gel(u,1)); setvalp(c, valp(c) + k); /* c = u[1] * p^k */
    1273         350 :     s = gadd(s, gmul(c, t));
    1274         350 :     if ((k&0xFL)==0) gerepileall(ltop, 3, &u,&s,&t);
    1275             :   }
    1276          28 :   return gneg(s);
    1277             : }
    1278             : 
    1279             : /*Use Dwork expansion*/
    1280             : /*This is a O(p*e*log(pe)) algorithm, should be used when p small
    1281             :  * If p==2 this is a O(pe) algorithm. */
    1282             : static GEN
    1283          28 : Qp_gamma_Dwork(GEN x, long p)
    1284             : {
    1285          28 :   pari_sp ltop = avma;
    1286          28 :   long k = padic_to_Fl(x, p);
    1287             :   GEN p1;
    1288             :   long j;
    1289          28 :   long px = precp(x);
    1290          28 :   if (p==2 && px)
    1291             :   {
    1292          14 :     x = shallowcopy(x);
    1293          14 :     setprecp(x, px+1);
    1294          14 :     gel(x,3) = shifti(gel(x,3),1);
    1295             :   }
    1296          28 :   if (k)
    1297             :   {
    1298          21 :     GEN x_k = gsubgs(x,k);
    1299          21 :     x = gdivgs(x_k, p);
    1300          21 :     p1 = gadw(x, p); if (!odd(k)) p1 = gneg(p1);
    1301          21 :     for (j = 1; j < k; ++j) p1 = gmul(p1, gaddgs(x_k, j));
    1302             :   }
    1303             :   else
    1304           7 :     p1 = gneg(gadw(gdivgs(x, p), p));
    1305          28 :   return gerepileupto(ltop, p1);
    1306             : }
    1307             : 
    1308             : /* Compute Qp_gamma using the definition. This is a O(x*M(log(pe))) algorithm.
    1309             :  * This should be used if x is very small. */
    1310             : static GEN
    1311          49 : Qp_gamma_Morita(long n, GEN p, long e)
    1312             : {
    1313          49 :   pari_sp ltop=avma;
    1314          49 :   GEN p2 = gaddsg((n&1)?-1:1, zeropadic(p, e));
    1315             :   long i;
    1316          49 :   long pp=is_bigint(p)? 0: itos(p);
    1317         154 :   for (i = 2; i < n; i++)
    1318         105 :     if (!pp || i%pp)
    1319             :     {
    1320          63 :       p2 = gmulgs(p2, i);
    1321          63 :       if ((i&0xFL) == 0xFL)
    1322           0 :         p2 = gerepileupto(ltop, p2);
    1323             :     }
    1324          49 :   return gerepileupto(ltop, p2);
    1325             : }
    1326             : 
    1327             : /* x\in\N: Gamma(-x)=(-1)^(1+x+x\p)*Gamma(1+x) */
    1328             : static GEN
    1329          28 : Qp_gamma_neg_Morita(long n, GEN p, long e)
    1330             : {
    1331          28 :   GEN g = ginv(Qp_gamma_Morita(n+1, p, e));
    1332          28 :   return ((n^sdivsi(n,p)) & 1)? g: gneg(g);
    1333             : }
    1334             : 
    1335             : /* p-adic Gamma function for x a p-adic integer */
    1336             : /* If n < p*e : use Morita's definition.
    1337             :  * Else : use Dwork's expansion.
    1338             :  * If both n and p are big : itos(p) will fail.
    1339             :  * TODO: handle p=2 better (Qp_gamma_Dwork is slow for p=2). */
    1340             : GEN
    1341          77 : Qp_gamma(GEN x)
    1342             : {
    1343          77 :   GEN n, m, N, p = gel(x,2);
    1344          77 :   long s, e = precp(x);
    1345          77 :   if (absequaliu(p, 2) && e == 2) e = 1;
    1346          77 :   if (valp(x) < 0) pari_err_DOMAIN("gamma","v_p(x)", "<", gen_0, x);
    1347          77 :   n = gtrunc(x);
    1348          77 :   m = gtrunc(gneg(x));
    1349          77 :   N = cmpii(n,m)<=0?n:m;
    1350          77 :   s = itos_or_0(N);
    1351          77 :   if (s && cmpsi(s, muliu(p,e)) < 0) /* s < p*e */
    1352          49 :     return (N == n) ? Qp_gamma_Morita(s,p,e): Qp_gamma_neg_Morita(s,p,e);
    1353          28 :   return Qp_gamma_Dwork(x, itos(p));
    1354             : }
    1355             : 
    1356             : /* gamma(1+x) - 1, |x| < 1 is "small" */
    1357             : GEN
    1358        1211 : ggamma1m1(GEN x, long prec) { return gexpm1(lngamma1(x, prec), prec); }
    1359             : 
    1360             : /* lngamma(y) with 0 constant term, using (lngamma y)' = y' psi(y) */
    1361             : static GEN
    1362       17661 : serlngamma0(GEN y, long prec)
    1363             : {
    1364             :   GEN t;
    1365       17661 :   if (valp(y)) pari_err_DOMAIN("lngamma","valuation", "!=", gen_0, y);
    1366       17654 :   t = derivser(y);
    1367             :   /* don't compute psi if y'=0 */
    1368       17654 :   if (signe(t)) t = gmul(t, gpsi(y,prec));
    1369       17654 :   return integser(t);
    1370             : }
    1371             : 
    1372             : static GEN
    1373       17626 : serlngamma(GEN y, long prec)
    1374             : {
    1375             :   GEN z, y0, Y;
    1376       17626 :   if (lg(y) == 2) pari_err_DOMAIN("gamma", "argument", "=", gen_0,y);
    1377             :   /* exp(lngamma) */
    1378       17619 :   if (valp(y) > 0) return gdiv(gexp(glngamma(gaddgs(y,1),prec),prec),y);
    1379       17402 :   y0 = simplify_shallow(gel(y,2));
    1380       17402 :   z = NULL; Y = y;
    1381       17402 :   if (isint(y0, &y0))
    1382             :   { /* fun eq. avoids log singularity of lngamma at negative ints */
    1383        7245 :     long s = signe(y0);
    1384             :     /* possible if y[2] is an inexact 0 */
    1385        7245 :     if (!s) return gdiv(gexp(glngamma(gaddgs(y,1),prec),prec),y);
    1386        7238 :     if (signe(y0) < 0) { Y = gsubsg(1, y); y0 = subsi(1, y0); }
    1387        7238 :     if (abscmpiu(y0, 50) < 0) z = mpfact(itos(y0)-1); /* more precise */
    1388             :   }
    1389       17395 :   if (!z) z = ggamma(y0,prec);
    1390       17395 :   z = gmul(z, gexp(serlngamma0(Y,prec),prec));
    1391       17395 :   if (Y != y)
    1392             :   {
    1393          77 :     GEN pi = mppi(prec);
    1394          77 :     z = gdiv(mpodd(y0)? pi: negr(pi),
    1395             :              gmul(z, gsin(gmul(pi,serchop0(y)), prec)));
    1396             :   }
    1397       17395 :   return z;
    1398             : }
    1399             : 
    1400             : static GEN
    1401        1225 : sqrtu(ulong a, long prec) { return sqrtr_abs(utor(a, prec)); }
    1402             : static GEN
    1403          14 : cbrtu(ulong a, long prec) { return sqrtnr_abs(utor(a, prec), 3); }
    1404             : /* N | 6 */
    1405             : static GEN
    1406        1092 : ellkprime(long N, GEN s2, GEN s3)
    1407             : {
    1408             :   GEN z;
    1409        1092 :   switch(N)
    1410             :   {
    1411         623 :     case 1: return shiftr(s2, -1);
    1412          28 :     case 2: return sqrtr_abs(shiftr(subrs(s2,1), 1));
    1413         357 :     case 3: return shiftr(mulrr(s2, addrs(s3, 1)), -2);
    1414             :     default: /* 6 */
    1415          84 :       z = mulrr(subrr(s3,s2), subsr(2,s3));
    1416          84 :       return mulrr(addsr(2,s2), sqrtr_abs(z));
    1417             :   }
    1418             : }
    1419             : 
    1420             : static GEN
    1421        1092 : ellKk(long N, GEN s2, GEN s3, long prec)
    1422        1092 : { return gdiv(Pi2n(-1,prec), agm(ellkprime(N,s2,s3), gen_1, prec)); }
    1423             : 
    1424             : /* Gamma(1/3) */
    1425             : static GEN
    1426         273 : G3(GEN s2, GEN s3, long prec)
    1427             : {
    1428         273 :   GEN A = ellKk(3, s2,s3, prec), pi = mppi(prec);
    1429         273 :   A = shiftr(divrs(powrs(mulrr(pi, A), 12), 27), 28);
    1430         273 :   return sqrtnr_abs(A, 36);
    1431             : }
    1432             : /* Gamma(1/4) */
    1433             : static GEN
    1434         511 : G4(GEN s2, long prec)
    1435             : {
    1436         511 :   GEN A = ellKk(1, s2,NULL, prec), pi = mppi(prec);
    1437         511 :   return shiftr(sqrtr_abs(mulrr(sqrtr_abs(pi), A)), 1);
    1438             : }
    1439             : 
    1440             : /* Gamma(n / 24), n = 1,5,7,11 */
    1441             : static GEN
    1442          84 : Gn24(long n, GEN s2, GEN s3, long prec)
    1443             : {
    1444          84 :   GEN A, B, C, t, t1, t2, t3, t4, pi = mppi(prec);
    1445          84 :   A = ellKk(1, s2,s3, prec);
    1446          84 :   B = ellKk(3, s2,s3, prec);
    1447          84 :   C = ellKk(6, s2,s3, prec);
    1448          84 :   t1 = sqrtr_abs(mulur(3, addsr(2, s3)));
    1449          84 :   t2 = sqrtr_abs(divrr(s3, pi));
    1450          84 :   t2 = mulrr(t2, shiftr(mulrr(addrr(s2,s3), A), 2));
    1451          84 :   t3 = mulrr(divur(3,pi), sqrr(B));
    1452          84 :   t3 = mulrr(addsr(2,s2), sqrtnr_abs(shiftr(powrs(t3, 3), 8), 9));
    1453          84 :   t4 = mulrr(mulrr(addsr(1, s2), subrr(s3, s2)), subsr(2, s3));
    1454          84 :   t4 = mulrr(mulrr(mulur(384, t4), pi), sqrr(C));
    1455          84 :   switch (n)
    1456             :   {
    1457          42 :     case 1: t = mulrr(mulrr(t1, t2), mulrr(t3, t4)); break;
    1458          14 :     case 5: t = divrr(mulrr(t2, t4), mulrr(t1, t3)); break;
    1459          14 :     case 7: t = divrr(mulrr(t3, t4), mulrr(t1, t2)); break;
    1460          14 :     default:t = divrr(mulrr(t1, t4), mulrr(t2, t3)); break;
    1461             :   }
    1462          84 :   return sqrtnr_abs(t, 4);
    1463             : }
    1464             : /* sin(x/2) = sqrt((1-c) / 2) > 0 given c = cos(x) */
    1465             : static GEN
    1466          28 : sinx2(GEN c)
    1467          28 : { c = subsr(1, c); shiftr_inplace(c,-1); return sqrtr_abs(c); }
    1468             : /* sin(Pi/12), given sqrt(3) */
    1469             : static GEN
    1470          21 : sin12(GEN s3)
    1471          21 : { GEN t = subsr(2, s3); shiftr_inplace(t, -2); return sqrtr_abs(t); }
    1472             : /* cos(Pi/12) = sin(5Pi/12), given sqrt(3) */
    1473             : static GEN
    1474          21 : cos12(GEN s3)
    1475          21 : { GEN t = addsr(2, s3); shiftr_inplace(t, -2); return sqrtr_abs(t); }
    1476             : /* 0 < n < d, (n, d) = 1, 2 < d | 24 */
    1477             : static GEN
    1478         868 : gammafrac24_s(long n, long d, long prec)
    1479             : {
    1480         868 :   GEN A, B, s2, s3, pi = mppi(prec);
    1481         868 :   s2 = sqrtu(2, prec);
    1482         868 :   s3 = d % 3? NULL: sqrtu(3, prec);
    1483         868 :   switch(d)
    1484             :   {
    1485             :     case 3:
    1486         231 :       A = G3(s2,s3,prec);
    1487         231 :       if (n == 1) return A;
    1488         133 :       return divrr(Pi2n(1, prec), mulrr(s3, A));
    1489             :     case 4:
    1490         483 :       A = G4(s2,prec);
    1491         483 :       if (n == 1) return A;
    1492         189 :       return divrr(mulrr(pi, s2), A);
    1493             :     case 6:
    1494          14 :       A = sqrr(G3(s2,s3,prec));
    1495          14 :       A = mulrr(A, sqrtr_abs(divsr(3, pi)));
    1496          14 :       A = divrr(A, cbrtu(2, prec));
    1497          14 :       if (n == 1) return A;
    1498           7 :       return divrr(Pi2n(1, prec), A);
    1499             :     case 8:
    1500          28 :       A = ellKk(1, s2,s3, prec);
    1501          28 :       B = ellKk(2, s2,s3, prec);
    1502          28 :       A = shiftr(sqrtr_abs(divrr(mulrr(addsr(1, s2), A), sqrtr_abs(pi))), 1);
    1503          28 :       B = shiftr(mulrr(sqrtr_abs(gmul(subrs(s2, 1), mulrr(s2, pi))), B), 3);
    1504          28 :       switch (n)
    1505             :       {
    1506             :         GEN t;
    1507           7 :         case 1: return sqrtr_abs(mulrr(A, B));
    1508           7 :         case 3: return sqrtr_abs(divrr(B, A));
    1509           7 :         case 5: A = sqrtr_abs(divrr(B, A));
    1510           7 :           t = sqrtr_abs(shiftr(addsr(1, shiftr(s2, -1)), -1)); /*sin(3Pi/8)*/
    1511           7 :           return divrr(pi, mulrr(t, A));
    1512           7 :         default: A = sqrtr_abs(mulrr(A, B));
    1513           7 :           t = sqrtr_abs(shiftr(subsr(1, shiftr(s2, -1)), -1)); /*sin(Pi/8)*/
    1514           7 :           return divrr(pi, mulrr(t, A));
    1515             :       }
    1516             :     case 12:
    1517          28 :       A = G3(s2,s3,prec);
    1518          28 :       B = G4(s2,prec);
    1519          28 :       switch (n)
    1520             :       {
    1521             :         GEN t2;
    1522             :         case 1: case 11:
    1523          14 :           t2 = shiftr(mulur(27, powrs(divrr(addsr(1,s3), pi), 4)), -2);
    1524          14 :           t2 = mulrr(sqrtnr_abs(t2, 8), mulrr(A, B));
    1525          14 :           if (n == 1) return t2;
    1526           7 :           return divrr(pi, mulrr(sin12(s3), t2));
    1527             :         case 5: case 7:
    1528          14 :           t2 = shiftr(divrs(powrs(mulrr(subrs(s3,1), pi), 4), 3), 2);
    1529          14 :           t2 = mulrr(sqrtnr_abs(t2, 8), divrr(B, A));
    1530          14 :           if (n == 5) return t2;
    1531           7 :           return divrr(pi, mulrr(cos12(s3), t2));
    1532             :       }
    1533             :     default: /* n = 24 */
    1534          84 :       if (n > 12)
    1535             :       {
    1536             :         GEN t;
    1537          28 :         n = 24 - n;
    1538          28 :         A = Gn24(n, s2,s3, prec);
    1539          28 :         switch(n)
    1540             :         { /* t = sin(n*Pi/24) */
    1541           7 :           case 1: t = cos12(s3); t = sinx2(t); break;
    1542           7 :           case 5: t = sin12(s3); t = sinx2(t); break;
    1543           7 :           case 7: t = sin12(s3); togglesign(t); t = sinx2(t); break;
    1544           7 :           default:t = cos12(s3); togglesign(t); t = sinx2(t); break; /* n=11 */
    1545             :         }
    1546          28 :         return divrr(pi, mulrr(A, t));
    1547             :       }
    1548          56 :       return Gn24(n, s2,s3, prec);
    1549             :   }
    1550             : }
    1551             : 
    1552             : /* (a,b) = 1. If 0 < x < b, m >= 0
    1553             : gamma(x/b + m) = gamma(x/b) * mulu_interval_step(x, x+(m-1)*b, b) / b^m
    1554             : gamma(x/b - m) = gamma(x/b) / mulu_interval_step(b-x, b*m-x, b) * (-b)^m */
    1555             : static GEN
    1556       33432 : gammafrac24(GEN a, GEN b, long prec)
    1557             : {
    1558             :   pari_sp av;
    1559             :   long A, B, m, x, bit;
    1560             :   GEN z0, z, t;
    1561       33432 :   if (!(A = itos_or_0(a)) || !(B = itos_or_0(b)) || B > 24) return NULL;
    1562       33355 :   switch(B)
    1563             :   {
    1564       32004 :     case 2: return gammahs(A-1, prec);
    1565             :     case 3: case 4: case 6: case 8: case 12: case 24:
    1566         882 :       m = A / B;
    1567         882 :       x = A % B; /* = A - m*B */
    1568         882 :       if (x < 0) { x += B; m--; } /* now 0 < x < B, A/B = x/B + m */
    1569         882 :       bit = bit_accuracy(prec);
    1570             :       /* Depending on B and prec, we must experimentally replace the 0.5
    1571             :        * by 0.4 to 2.0 for optimal value. Play safe. */
    1572         882 :       if (labs(m) > 0.5 * bit * sqrt(bit) / log(bit)) return NULL;
    1573         868 :       z0 = cgetr(prec); av = avma;
    1574         868 :       prec += EXTRAPRECWORD;
    1575         868 :       z = gammafrac24_s(x, B, prec);
    1576         868 :       if (m)
    1577             :       {
    1578         245 :         if (m > 0)
    1579         231 :           t = mpdiv(mulu_interval_step(x, (m-1)*B + x, B), rpowuu(B,m,prec));
    1580             :         else
    1581             :         {
    1582          14 :           m = -m;
    1583          14 :           t = mpdiv(rpowuu(B,m,prec), mulu_interval_step(B-x, m*B - x, B));
    1584          14 :           if (odd(m)) togglesign(t);
    1585             :         }
    1586         245 :         z = mpmul(z,t);
    1587             :       }
    1588         868 :       affrr(z, z0); set_avma(av); return z0;
    1589             :   }
    1590         469 :   return NULL;
    1591             : }
    1592             : GEN
    1593      314209 : ggamma(GEN x, long prec)
    1594             : {
    1595             :   pari_sp av;
    1596             :   GEN y;
    1597             : 
    1598      314209 :   switch(typ(x))
    1599             :   {
    1600             :     case t_INT:
    1601       60690 :       if (signe(x) <= 0)
    1602           0 :         pari_err_DOMAIN("gamma","argument", "=",
    1603             :                          strtoGENstr("non-positive integer"), x);
    1604       60690 :       return mpfactr(itos(x) - 1, prec);
    1605             : 
    1606             :     case t_REAL: case t_COMPLEX:
    1607      202391 :       return cxgamma(x, 0, prec);
    1608             : 
    1609             :     case t_FRAC:
    1610             :     {
    1611       33432 :       GEN a = gel(x,1), b = gel(x,2), c = gammafrac24(a, b, prec);
    1612       33432 :       if (c) return c;
    1613         560 :       av = avma; c = subii(a,b);
    1614         560 :       if (expi(c) - expi(b) < -50)
    1615             :       { /* x = 1 + c/b is close to 1 */
    1616           7 :         x = mkfrac(c,b);
    1617           7 :         if (lgefint(b) >= prec) x = fractor(x,prec);
    1618           7 :         y = mpexp(lngamma1(x, prec));
    1619             :       }
    1620         553 :       else if (signe(a) < 0 || cmpii(shifti(a,1), b) < 0)
    1621         203 :       { /* gamma will use functional equation x -> z = 1-x = -c/b >= 1/2.
    1622             :          * Gamma(x) = Pi / (sin(Pi z) * Gamma(z)) */
    1623         203 :         GEN z = mkfrac(negi(c), b), q = ground(z), r = gsub(z,q);
    1624         203 :         GEN pi = mppi(prec); /* |r| <= 1/2 */
    1625         203 :         z = fractor(z, prec+EXTRAPRECWORD);
    1626         203 :         y = divrr(pi, mulrr(mpsin(gmul(pi, r)), cxgamma(z, 0, prec)));
    1627         203 :         if (mpodd(q)) togglesign(y);
    1628             :       }
    1629             :       else
    1630             :       {
    1631         350 :         x = fractor(x, prec);
    1632         350 :         y = cxgamma(x, 0, prec);
    1633             :       }
    1634         560 :       return gerepileupto(av, y);
    1635             :     }
    1636             : 
    1637          70 :     case t_PADIC: return Qp_gamma(x);
    1638             :     default:
    1639       17626 :       av = avma; if (!(y = toser_i(x))) break;
    1640       17626 :       return gerepileupto(av, serlngamma(y, prec));
    1641             :   }
    1642           0 :   return trans_eval("gamma",ggamma,x,prec);
    1643             : }
    1644             : 
    1645             : static GEN
    1646         510 : mpfactr_basecase(long n, long prec)
    1647             : {
    1648         510 :   GEN v = cgetg(expu(n) + 2, t_VEC);
    1649         510 :   long k, prec2 = prec + EXTRAPRECWORD;
    1650             :   GEN a;
    1651        4772 :   for (k = 1;; k++)
    1652        4262 :   {
    1653        4772 :     long m = n >> (k-1), l;
    1654        4772 :     if (m <= 2) break;
    1655        4262 :     l = (1 + (n >> k)) | 1;
    1656             :     /* product of odd numbers in ]n / 2^k, 2 / 2^(k-1)] */
    1657        4262 :     a = mulu_interval_step_prec(l, m, 2, prec2);
    1658        4262 :     gel(v,k) = k == 1? a: gpowgs(a, k);
    1659             :   }
    1660         510 :   a = gel(v,--k); while (--k) a = mpmul(a, gel(v,k));
    1661         510 :   if (typ(a) == t_INT) a = itor(a, prec); else a = gprec_wtrunc(a, prec);
    1662         510 :   shiftr_inplace(a, factorial_lval(n, 2));
    1663         510 :   return a;
    1664             : }
    1665             : /* Theory says n > C * b^1.5 / log(b). Timings:
    1666             :  * b = [64, 128, 192, 256, 512, 1024, 2048, 4096, 8192, 16384]
    1667             :  * n = [1930, 2650, 3300, 4270, 9000, 23000, 75000, 210000, 750000, 2400000] */
    1668             : static long
    1669         622 : mpfactr_n(long prec)
    1670             : {
    1671         622 :   long b = bit_accuracy(prec);
    1672         622 :   if (b <=  64) return 1930;
    1673          76 :   if (b <= 128) return 2650;
    1674          62 :   if (b <= 192) return 3300;
    1675          62 :   return b * sqrt(b);
    1676             : }
    1677             : static GEN
    1678        7714 : mpfactr_small(long n, long prec)
    1679             : {
    1680        7714 :   GEN f = cgetr(prec);
    1681        7714 :   pari_sp av = avma;
    1682        7714 :   if (n < 410)
    1683        7714 :     affir(mpfact(n), f);
    1684             :   else
    1685           0 :     affrr(mpfactr_basecase(n, prec), f);
    1686        7714 :   set_avma(av); return f;
    1687             : }
    1688             : GEN
    1689      104859 : mpfactr(long n, long prec)
    1690             : {
    1691      104859 :   GEN f = cgetr(prec);
    1692      104859 :   pari_sp av = avma;
    1693             : 
    1694      104859 :   if (n < 410)
    1695      104237 :     affir(mpfact(n), f);
    1696             :   else
    1697             :   {
    1698         622 :     long N = mpfactr_n(prec);
    1699         622 :     GEN z = n <= N? mpfactr_basecase(n, prec)
    1700         622 :                   : cxgamma(utor(n+1, prec), 0, prec);
    1701         622 :     affrr(z, f);
    1702             :   }
    1703      104859 :   set_avma(av); return f;
    1704             : }
    1705             : 
    1706             : /* First a little worse than mpfactr_n because of the extra logarithm.
    1707             :  * Asymptotically same. */
    1708             : static ulong
    1709        7714 : lngamma_n(long prec)
    1710             : {
    1711        7714 :   long b = bit_accuracy(prec);
    1712             :   double N;
    1713        7714 :   if (b <=  64) return 1450UL;
    1714        7714 :   if (b <= 128) return 2010UL;
    1715         133 :   if (b <= 192) return 2870UL;
    1716           0 :   N = b * sqrt(b);
    1717           0 :   if (b <= 256) return N/1.25;
    1718           0 :   if (b <= 512) return N/1.2;
    1719           0 :   if (b <= 2048) return N/1.1;
    1720           0 :   return N;
    1721             : }
    1722             : 
    1723             : GEN
    1724       35049 : glngamma(GEN x, long prec)
    1725             : {
    1726       35049 :   pari_sp av = avma;
    1727             :   GEN y, y0, t;
    1728             : 
    1729       35049 :   switch(typ(x))
    1730             :   {
    1731             :     case t_INT:
    1732             :     {
    1733             :       ulong n;
    1734        7721 :       if (signe(x) <= 0)
    1735           0 :         pari_err_DOMAIN("lngamma","argument", "=",
    1736             :                          strtoGENstr("non-positive integer"), x);
    1737        7721 :       n = itou_or_0(x);
    1738        7721 :       if (!n || n > lngamma_n(prec)) return cxgamma(x, 1, prec);
    1739        7714 :       return gerepileuptoleaf(av, logr_abs( mpfactr_small(n-1, prec) ));
    1740             :     }
    1741             :     case t_FRAC:
    1742             :     {
    1743        8253 :       GEN a = gel(x,1), b = gel(x,2), c = subii(a,b);
    1744        8253 :       long e = expi(b) - expi(c);
    1745        8253 :       if (e > 50)
    1746             :       {
    1747           7 :         x = mkfrac(c,b);
    1748           7 :         if (lgefint(b) >= prec) x = fractor(x,prec + nbits2nlong(e));
    1749           7 :         y = lngamma1(x, prec);
    1750             :       }
    1751        8246 :       else if (signe(a) < 0 || cmpii(shifti(a,1), b) < 0)
    1752          21 :       { /* gamma will use functional equation x -> z = 1-x = -c/b >= 1/2.
    1753             :          * lngamma(x) = log |Pi / (sin(Pi z) * Gamma(z))| + I*Pi * floor(x) */
    1754          21 :         GEN z = mkfrac(negi(c), b), q = ground(z), r = gsub(z,q);
    1755          21 :         GEN pi = mppi(prec); /* |r| <= 1/2 */
    1756          21 :         z = fractor(z, prec+EXTRAPRECWORD);
    1757          21 :         y = subrr(logr_abs(divrr(pi, mpsin(gmul(pi, r)))), cxgamma(z, 1, prec));
    1758          21 :         if (signe(a) < 0) y = gadd(y, mkcomplex(gen_0, mulri(pi, gfloor(x))));
    1759             :       }
    1760             :       else
    1761             :       {
    1762        8225 :         x = fractor(x, e > 1? prec+EXTRAPRECWORD: prec);
    1763        8225 :         y = cxgamma(x, 1, prec);
    1764             :       }
    1765        8253 :       return gerepileupto(av, y);
    1766             :     }
    1767             : 
    1768             :     case t_REAL: case t_COMPLEX:
    1769       18795 :       return cxgamma(x, 1, prec);
    1770             : 
    1771             :     default:
    1772         273 :       if (!(y = toser_i(x))) break;
    1773         273 :       if (lg(y) == 2) pari_err_DOMAIN("lngamma", "argument", "=", gen_0,y);
    1774         266 :       t = serlngamma0(y,prec);
    1775         252 :       y0 = simplify_shallow(gel(y,2));
    1776             :       /* no constant term if y0 = 1 or 2 */
    1777         252 :       if (!isint(y0,&y0) || signe(y0) <= 0 || abscmpiu(y0,2) > 2)
    1778           7 :         t = gadd(t, glngamma(y0,prec));
    1779         252 :       return gerepileupto(av, t);
    1780             : 
    1781           7 :     case t_PADIC: return gerepileupto(av, Qp_log(Qp_gamma(x)));
    1782             :   }
    1783           0 :   return trans_eval("lngamma",glngamma,x,prec);
    1784             : }
    1785             : /********************************************************************/
    1786             : /**                                                                **/
    1787             : /**                  PSI(x) = GAMMA'(x)/GAMMA(x)                   **/
    1788             : /**                                                                **/
    1789             : /********************************************************************/
    1790             : static GEN
    1791       25242 : cxpsi(GEN s0, long prec)
    1792             : {
    1793             :   pari_sp av, av2;
    1794             :   GEN sum, z, a, res, sig, tau, s, unr, s2, sq;
    1795             :   long lim, nn, k;
    1796       25242 :   const long la = 3;
    1797       25242 :   int funeq = 0;
    1798             :   pari_timer T;
    1799             : 
    1800       25242 :   if (DEBUGLEVEL>2) timer_start(&T);
    1801       25242 :   s = trans_fix_arg(&prec,&s0,&sig,&tau,&av,&res);
    1802       25242 :   if (signe(sig) <= 0) { funeq = 1; s = gsub(gen_1, s); sig = real_i(s); }
    1803       25242 :   if (typ(s0) == t_INT && signe(s0) <= 0)
    1804           0 :     pari_err_DOMAIN("psi","argument", "=",
    1805             :                     strtoGENstr("non-positive integer"), s0);
    1806             : 
    1807       25242 :   if (expo(sig) > 300 || (typ(s) == t_COMPLEX && gexpo(gel(s,2)) > 300))
    1808           7 :   { /* |s| is HUGE. Play safe */
    1809           7 :     GEN L, S = gprec_w(s,LOWDEFAULTPREC), rS = real_i(S), iS = imag_i(S);
    1810             :     double l;
    1811             : 
    1812           7 :     l = rtodbl( gnorm(glog(S, 3)) );
    1813           7 :     l = log(l) / 2.;
    1814           7 :     lim = 2 + (long)ceil((prec2nbits_mul(prec, M_LN2) - l) / (2*(1+log((double)la))));
    1815           7 :     if (lim < 2) lim = 2;
    1816             : 
    1817           7 :     l = (2*lim-1)*la / (2.*M_PI);
    1818           7 :     L = gsub(dbltor(l*l), gsqr(iS));
    1819           7 :     if (signe(L) < 0) L = gen_0;
    1820             : 
    1821           7 :     L = gsub(gsqrt(L, 3), rS);
    1822           7 :     if (signe(L) > 0) nn = (long)ceil(rtodbl(L)); else nn = 1;
    1823           7 :     if (DEBUGLEVEL>2) err_printf("lim, nn: [%ld, %ld]\n",lim,nn);
    1824             :   }
    1825             :   else
    1826             :   {
    1827       25235 :     double ssig = rtodbl(sig);
    1828       25235 :     double st = typ(s) == t_REAL? 0.0: rtodbl(imag_i(s));
    1829             :     double l;
    1830             :     {
    1831             :       double rlog, ilog; /* log (s - Euler) */
    1832       25235 :       dcxlog(ssig - 0.57721566, st, &rlog,&ilog);
    1833       25235 :       l = dnorm(rlog,ilog);
    1834             :     }
    1835       25235 :     if (l < 0.000001) l = 0.000001;
    1836       25235 :     l = log(l) / 2.;
    1837       25235 :     lim = 2 + (long)ceil((prec2nbits_mul(prec, M_LN2) - l) / (2*(1+log((double)la))));
    1838       25235 :     if (lim < 2) lim = 2;
    1839             : 
    1840       25235 :     l = (2*lim-1)*la / (2.*M_PI);
    1841       25235 :     l = l*l - st*st;
    1842       25235 :     if (l < 0.) l = 0.;
    1843       25235 :     nn = (long)ceil( sqrt(l) - ssig );
    1844       25235 :     if (nn < 1) nn = 1;
    1845       25235 :     if (DEBUGLEVEL>2) err_printf("lim, nn: [%ld, %ld]\n",lim,nn);
    1846             :   }
    1847       25242 :   incrprec(prec); unr = real_1(prec); /* one extra word of precision */
    1848       25242 :   s2 = gmul2n(s, 1); sq = gsqr(s);
    1849       25242 :   a = gdiv(unr, gaddgs(s, nn)); /* 1 / (s+n) */
    1850       25242 :   av2 = avma; sum = gmul2n(a, -1);
    1851      611772 :   for (k = 0; k < nn - 1; k += 2)
    1852             :   {
    1853      586530 :     GEN tmp = gaddsg(k*(k + 1), gadd(gmulsg(2*k + 1, s), sq));
    1854      586530 :     sum = gadd(sum, gdiv(gaddsg(2*k + 1, s2), tmp));
    1855      586530 :     if ((k & 1023) == 0) sum = gerepileupto(av2, sum);
    1856             :   }
    1857       25242 :   if (odd(nn)) sum = gadd(sum, gdiv(unr, gaddsg(nn - 1, s)));
    1858       25242 :   z = gsub(glog(gaddgs(s, nn), prec), sum);
    1859       25242 :   if (DEBUGLEVEL>2) timer_printf(&T,"sum from 0 to N - 1");
    1860       25242 :   constbern(lim);
    1861       25242 :   z = gsub(z, psi_sum(gsqr(a), lim));
    1862       25242 :   if (DEBUGLEVEL>2) timer_printf(&T,"Bernoulli sum");
    1863       25242 :   if (funeq)
    1864             :   {
    1865        3976 :     GEN pi = mppi(prec);
    1866        3976 :     z = gadd(z, gmul(pi, gcotan(gmul(pi,s), prec)));
    1867             :   }
    1868       25242 :   set_avma(av); return affc_fixlg(z, res);
    1869             : }
    1870             : 
    1871             : /* n > 0; return psi(1+x) + O(x^n), x = pol_x(v) */
    1872             : static GEN
    1873        6944 : serpsi1(long n, long v, long prec)
    1874             : {
    1875        6944 :   long i, l = n+3;
    1876        6944 :   GEN z, g, s = cgetg(l, t_SER);
    1877        6944 :   s[1] = evalsigne(1)|evalvalp(0)|evalvarn(v);
    1878        6944 :   g = mpeuler(prec); setsigne(g, -1);
    1879        6944 :   z = veczeta(gen_1, gen_2, n, prec); /* zeta(2..n) */
    1880        6944 :   gel(s,2) = g;
    1881       29428 :   for (i = 2; i < l-1; i++)
    1882             :   {
    1883       22484 :     GEN c = gel(z,i-1); /* zeta(i) */
    1884       22484 :     if (odd(i)) setsigne(c, -1);
    1885       22484 :     gel(s,i+1) = c;
    1886             :   }
    1887        6944 :   return s;
    1888             : }
    1889             : /* T an RgX, return T(X + z0) + O(X^L) */
    1890             : static GEN
    1891      956687 : tr(GEN T, GEN z0, long L)
    1892             : {
    1893      956687 :   GEN s = RgX_to_ser(RgX_translate(T, z0), L+3);
    1894      956687 :   setvarn(s, 0); return s;
    1895             : }
    1896             : /* z0 a complex number with Re(z0) > 1/2; return psi(z0+x) + O(x^L)
    1897             :  * using Luke's rational approximation for psi(x) */
    1898             : static GEN
    1899        4424 : serpsiz0(GEN z0, long L, long v, long prec)
    1900             : {
    1901             :   pari_sp av;
    1902             :   GEN A,A1,A2, B,B1,B2, Q;
    1903             :   long n;
    1904        4424 :   n = gprecision(z0); if (n) prec = n;
    1905        4424 :   z0 = gtofp(z0, prec + EXTRAPRECWORD);
    1906             :   /* Start from n = 3; in Luke's notation, A2 := A_{n-2}, A1 := A_{n-1},
    1907             :    * A := A_n. Same for B */
    1908        4424 :   av = avma;
    1909        4424 :   A2= gdivgs(mkpoln(2, gen_1, utoipos(6)), 2);
    1910        4424 :   B2 = scalarpol_shallow(utoipos(4), 0);
    1911        4424 :   A1= gdivgs(mkpoln(3, gen_1, utoipos(82), utoipos(96)), 6);
    1912        4424 :   B1 = mkpoln(2, utoipos(8), utoipos(28));
    1913        4424 :   A = gdivgs(mkpoln(4, gen_1, utoipos(387), utoipos(2906), utoipos(1920)), 12);
    1914        4424 :   B = mkpoln(3, utoipos(14), utoipos(204), utoipos(310));
    1915        4424 :   A2= tr(A2,z0, L);
    1916        4424 :   B2= tr(B2,z0, L);
    1917        4424 :   A1= tr(A1,z0, L);
    1918        4424 :   B1= tr(B1,z0, L);
    1919        4424 :   A = tr(A, z0, L);
    1920        4424 :   B = tr(B, z0, L); Q = gdiv(A, B);
    1921             :   /* work with z0+x as a variable */
    1922      308573 :   for (n = 4;; n++)
    1923      304149 :   {
    1924      308573 :     GEN Q0 = Q, a, b, r, c3,c2,c1,c0 = muluu(2*n-3, n+1);
    1925      308573 :     GEN u = subiu(muluu(n, 7*n-9), 6);
    1926      308573 :     GEN t = addiu(muluu(n, 7*n-19), 4);
    1927             :     /* c1=(2*n-1)*(3*(n-1)*z+7*n^2-9*n-6);
    1928             :      * c2=(2*n-3)*(z-n-1)*(-3*(n-1)*z+7*n^2-19*n+4);
    1929             :      * c3=(2*n-1)*(n-3)*(z-n)*(z-(n+1))*(z+(n-4)); */
    1930      308573 :     c1 = deg1pol_shallow(muluu(3*(n-1),2*n-1), muliu(u,2*n-1), 0);
    1931      308573 :     c2 = ZX_mul(deg1pol_shallow(utoipos(2*n-3), negi(muluu(2*n-3,n+1)), 0),
    1932      308573 :                 deg1pol_shallow(utoineg(3*(n-1)), t, 0));
    1933      308573 :     r = mkvec3(utoipos(n), utoipos(n+1), stoi(4-n));
    1934      308573 :     c3 = ZX_Z_mul(roots_to_pol(r,0), muluu(2*n-1,n-3));
    1935      308573 :     c1 = tr(c1, z0, L+3);
    1936      308573 :     c2 = tr(c2, z0, L+3);
    1937      308573 :     c3 = tr(c3, z0, L+3);
    1938             : 
    1939             :     /* A_{n+1}, B_{n+1} */
    1940      308573 :     a = gdiv(gadd(gadd(gmul(c1,A),gmul(c2,A1)),gmul(c3,A2)), c0);
    1941      308573 :     b = gdiv(gadd(gadd(gmul(c1,B),gmul(c2,B1)),gmul(c3,B2)), c0);
    1942      308573 :     Q = gdiv(a,b);
    1943      308573 :     if (gexpo(gsub(Q,Q0)) < -prec2nbits(prec)) break;
    1944      304149 :     A2 = A1; A1 = A; A = a;
    1945      304149 :     B2 = B1; B1 = B; B = b;
    1946      304149 :     if (gc_needed(av,1))
    1947             :     {
    1948           0 :       if(DEBUGMEM>1) pari_warn(warnmem,"serpsiz0, n = %ld", n);
    1949           0 :       gerepileall(av, 7, &A,&A1,&A2, &B,&B1,&B2, &Q);
    1950             :     }
    1951             :   }
    1952        4424 :   Q = gmul(Q, gmul2n(gsubsg(1, ginv(tr(pol_x(v),z0, L))), 1));
    1953        4424 :   setvarn(Q, v);
    1954        4424 :   return gadd(negr(mpeuler(prec)), Q);
    1955             : }
    1956             : /* sum (-1)^k*H(m,k)x^k + O(x^L); L > 0;
    1957             :  * H(m,k) = (-1)^{k * \delta_{m > 0}} sum_{1<=i<m} 1/i^(k+1) */
    1958             : static GEN
    1959         756 : Hseries(long m, long L, long v, long prec)
    1960             : {
    1961         756 :   long i, k, bit, l = L+3, M = m < 0? 1-m: m;
    1962         756 :   pari_sp av = avma;
    1963         756 :   GEN H = cgetg(l, t_SER);
    1964         756 :   H[1] = evalsigne(1)|evalvarn(v)|evalvalp(0);
    1965         756 :   prec++;
    1966         756 :   bit = -prec2nbits(prec);
    1967         756 :   for(k = 2; k < l; k++) gel(H,k) = gen_1; /* i=1 */
    1968         784 :   for (i = 2; i < M; i++)
    1969             :   {
    1970          28 :     GEN ik = invr(utor(i, prec));
    1971         203 :     for (k = 2; k < l; k++)
    1972             :     {
    1973         175 :       if (k > 2) { ik = divru(ik, i); if (expo(ik) < bit) break; }
    1974         175 :       gel(H,k) = gadd(gel(H,k), ik);
    1975             :     }
    1976          28 :     if (gc_needed(av,3))
    1977             :     {
    1978           0 :       if(DEBUGMEM>1) pari_warn(warnmem,"Hseries, i = %ld/%ld", i,M);
    1979           0 :       H = gerepilecopy(av, H);
    1980             :     }
    1981             :   }
    1982         756 :   if (m > 0)
    1983         742 :     for (k = 3; k < l; k+=2) togglesign_safe(&gel(H,k));
    1984         756 :   return H;
    1985             : }
    1986             : 
    1987             : static GEN
    1988       11382 : serpsi(GEN y, long prec)
    1989             : {
    1990       11382 :   GEN Q = NULL, z0, Y = y, Y2;
    1991       11382 :   long L = lg(y)-2, v  = varn(y), vy = valp(y);
    1992             : 
    1993       11382 :   if (!L) pari_err_DOMAIN("psi", "argument", "=", gen_0,y);
    1994       11375 :   if (vy < 0) pari_err_DOMAIN("psi", "series valuation", "<", gen_0,y);
    1995       11375 :   if (vy)
    1996          14 :     z0 = gen_0;
    1997             :   else
    1998             :   {
    1999       11361 :     z0 = simplify_shallow(gel(y,2));
    2000       11361 :     (void)isint(z0, &z0);
    2001             :   }
    2002       11375 :   if (typ(z0) == t_INT && !is_bigint(z0))
    2003             :   {
    2004        6951 :     long m = itos(z0);
    2005        6951 :     if (abscmpiu(muluu(prec2nbits(prec),L), labs(m)) > 0)
    2006             :     { /* psi(m+x) = psi(1+x) + sum_{1 <= i < m} 1/(i+x) for m > 0
    2007             :                     psi(1+x) - sum_{0 <= i < -m} 1/(i+x) for m <= 0 */
    2008        6951 :       GEN H = NULL;
    2009        6951 :       if (m <= 0) L--; /* lose series accuracy due to 1/x term */
    2010        6951 :       if (L)
    2011             :       {
    2012        6944 :         Q = serpsi1(L, v, prec);
    2013        6944 :         if (m && m != 1) { H = Hseries(m, L, v, prec); Q = gadd(Q, H); }
    2014        6944 :         if (m <= 0) Q = gsub(Q, ginv(pol_x(v)));
    2015             :       }
    2016             :       else
    2017             :       {
    2018           7 :         Q = scalarser(gen_m1, v, 1);
    2019           7 :         setvalp(Q,-1);
    2020             :       }
    2021             :     }
    2022             :   }
    2023       11375 :   if (!Q)
    2024             :   { /* use psi(1-y)=psi(y)+Pi*cotan(Pi*y) ? */
    2025        4424 :     if (gcmp(real_i(z0),ghalf) < 0) { z0 = gsubsg(1,z0); Y = gsubsg(1,y); }
    2026        4424 :     Q = serpsiz0(z0, L, v, prec);
    2027             :   }
    2028       11375 :   Y2 = serchop0(Y); if (signe(Y2)) Q = gsubst(Q, v, Y2);
    2029             :   /* psi(z0 + Y2) = psi(Y) */
    2030       11375 :   if (Y != y)
    2031             :   { /* psi(y) = psi(Y) + Pi cotan(Pi Y) */
    2032          49 :     GEN pi = mppi(prec);
    2033          49 :     if (typ(z0) == t_INT) Y = Y2; /* in this case cotan(Pi*Y2) = cotan(Pi*Y) */
    2034          49 :     Q = gadd(Q, gmul(pi, gcotan(gmul(pi,Y), prec)));
    2035             :   }
    2036       11375 :   return Q;
    2037             : }
    2038             : 
    2039             : GEN
    2040       57876 : gpsi(GEN x, long prec)
    2041             : {
    2042             :   pari_sp av;
    2043             :   GEN y;
    2044       57876 :   switch(typ(x))
    2045             :   {
    2046       25242 :     case t_REAL: case t_COMPLEX: return cxpsi(x,prec);
    2047             :     default:
    2048       32634 :       av = avma; if (!(y = toser_i(x))) break;
    2049       11382 :       return gerepileupto(av, serpsi(y,prec));
    2050             :   }
    2051       21252 :   return trans_eval("psi",gpsi,x,prec);
    2052             : }

Generated by: LCOV version 1.13