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-bordeaux1.fr machine (x86_64 architecture), and agregate them in the final report:

The target is 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 - modsym.c (source / functions) Hit Total Coverage
Test: PARI/GP v2.8.0 lcov report (development 17236-ec2ebc9) Lines: 1428 1468 97.3 %
Date: 2014-12-19 Functions: 159 159 100.0 %
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed Branches: 555 659 84.2 %

           Branch data     Line data    Source code
       1                 :            : /* $Id$
       2                 :            : 
       3                 :            : Copyright (C) 2011  The PARI group.
       4                 :            : 
       5                 :            : This file is part of the PARI/GP package.
       6                 :            : 
       7                 :            : PARI/GP is free software; you can redistribute it and/or modify it under the
       8                 :            : terms of the GNU General Public License as published by the Free Software
       9                 :            : Foundation. It is distributed in the hope that it will be useful, but WITHOUT
      10                 :            : ANY WARRANTY WHATSOEVER.
      11                 :            : 
      12                 :            : Check the License for details. You should have received a copy of it, along
      13                 :            : with the package; see the file 'COPYING'. If not, write to the Free Software
      14                 :            : Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
      15                 :            : 
      16                 :            : #include "pari.h"
      17                 :            : #include "paripriv.h"
      18                 :            : 
      19                 :            : /* Adapted from shp_package/moments by Robert Pollack
      20                 :            :  * http://www.math.mcgill.ca/darmon/programs/shp/shp.html */
      21                 :            : static GEN mskinit(ulong N, long k, long sign);
      22                 :            : static GEN mshecke_i(GEN W, ulong p);
      23                 :            : static GEN msnew_trivial(GEN W);
      24                 :            : static GEN mscuspidal_trivial(GEN W0);
      25                 :            : static GEN ZGl2Q_star(GEN v);
      26                 :            : static GEN getMorphism_trivial(GEN WW1, GEN WW2, GEN v);
      27                 :            : static GEN getMorphism(GEN W1, GEN W2, GEN v);
      28                 :            : static GEN voo_act_Gl2Q(GEN g, long k);
      29                 :            : static GEN mscuspidal_i(GEN W);
      30                 :            : 
      31                 :            : /* Input: P^1(Z/NZ) (formed by create_p1mod)
      32                 :            :    Output: # P^1(Z/NZ) */
      33                 :            : static long
      34                 :        784 : p1_size(GEN p1N) { return lg(gel(p1N,1)) - 1; }
      35                 :            : static ulong
      36                 :   22777692 : p1N_get_N(GEN p1N) { return gel(p1N,3)[2]; }
      37                 :            : static GEN
      38                 :   20792632 : p1N_get_hash(GEN p1N) { return gel(p1N,2); }
      39                 :            : static GEN
      40                 :        259 : p1N_get_fa(GEN p1N) { return gel(p1N,4); }
      41                 :            : static GEN
      42                 :        196 : p1N_get_div(GEN p1N) { return gel(p1N,5); }
      43                 :            : /* ms-specific accessors */
      44                 :            : /* W = msinit or msfromell */
      45                 :            : static GEN
      46         [ +  + ]:    3891790 : get_ms(GEN W) { return lg(W) == 4? gel(W,1): W; }
      47                 :            : static GEN
      48                 :       1680 : ms_get_p1N(GEN W) { W = get_ms(W); return gel(W,1); }
      49                 :            : static long
      50                 :        441 : ms_get_N(GEN W) { return p1N_get_N(ms_get_p1N(W)); }
      51                 :            : static GEN
      52                 :        112 : ms_get_hashcusps(GEN W) { W = get_ms(W); return gel(W,16); }
      53                 :            : static GEN
      54                 :      33103 : ms_get_section(GEN W) { W = get_ms(W); return gel(W,12); }
      55                 :            : static GEN
      56                 :       1519 : ms_get_genindex(GEN W) { W = get_ms(W); return gel(W,5); }
      57                 :            : static long
      58                 :       1071 : ms_get_nbgen(GEN W) { return lg(ms_get_genindex(W))-1; }
      59                 :            : static long
      60                 :    1936865 : ms_get_nbE1(GEN W)
      61                 :            : {
      62                 :            :   GEN W11;
      63                 :    1936865 :   W = get_ms(W); W11 = gel(W,11);
      64                 :    1936865 :   return W11[4] - W11[3];
      65                 :            : }
      66                 :            : /* msk-specific accessors */
      67                 :            : static long
      68                 :          7 : msk_get_dim(GEN W) { return gmael(W,3,2)[2]; }
      69                 :            : static GEN
      70                 :       1071 : msk_get_basis(GEN W) { return gmael(W,3,1); }
      71                 :            : static long
      72                 :       1190 : msk_get_weight(GEN W) { return gmael(W,3,2)[1]; }
      73                 :            : static GEN
      74                 :        840 : msk_get_st(GEN W) { return gmael(W,3,3); }
      75                 :            : static GEN
      76                 :        840 : msk_get_link(GEN W) { return gmael(W,3,4); }
      77                 :            : static GEN
      78                 :        840 : msk_get_invphiblock(GEN W) { return gmael(W,3,5); }
      79                 :            : static long
      80                 :        217 : msk_get_sign(GEN W)
      81                 :            : {
      82                 :        217 :   GEN t = gel(W,2);
      83         [ +  - ]:        217 :   return typ(t)==t_INT? 0: itos(gel(t,1));
      84                 :            : }
      85                 :            : static GEN
      86                 :        112 : msk_get_star(GEN W) { return gmael(W,2,2); }
      87                 :            : static GEN
      88                 :         21 : msk_get_starproj(GEN W) { return gmael(W,2,3); }
      89                 :            : 
      90                 :            : void
      91                 :        343 : checkms(GEN W)
      92                 :            : {
      93 [ +  - ][ -  + ]:        343 :   if (typ(W) != t_VEC || lg(W) != 4)
      94                 :          0 :     pari_err_TYPE("checkms [please apply msinit]", W);
      95                 :        343 : }
      96                 :            : 
      97                 :            : /** MODULAR TO SYM **/
      98                 :            : 
      99                 :            : /* q a t_FRAC or t_INT */
     100                 :            : static GEN
     101                 :    1981371 : modulartosym_init(ulong N, GEN q)
     102                 :            : {
     103                 :            :   long l, n;
     104                 :            :   GEN Q;
     105                 :            : 
     106                 :    1981371 :   q = gboundcf(q, 0);
     107                 :    1981371 :   l = lg(q);
     108                 :    1981371 :   Q = cgetg(l, t_VECSMALL);
     109                 :    1981371 :   Q[1] = 1;
     110         [ +  + ]:   20645156 :   for (n=2; n <l; n++) Q[n] = umodiu(gel(q,n), N);
     111         [ +  + ]:   18665129 :   for (n=3; n < l; n++)
     112                 :   16683758 :     Q[n] = Fl_add(Fl_mul(Q[n], Q[n-1], N), Q[n-2], N);
     113                 :    1981371 :   return Q;
     114                 :            : }
     115                 :            : 
     116                 :            : /** INIT MODSYM STRUCTURE, WEIGHT 2 **/
     117                 :            : 
     118                 :            : /* num = [Gamma : Gamma_0(N)] = N * Prod_{p|N} (1+p^-1) */
     119                 :            : static ulong
     120                 :        196 : count_Manin_symbols(ulong N, GEN P)
     121                 :            : {
     122                 :        196 :   long i, l = lg(P);
     123                 :        196 :   ulong num = N;
     124         [ +  + ]:        448 :   for (i = 1; i < l; i++) { ulong p = P[i]; num *= p+1; num /= p; }
     125                 :        196 :   return num;
     126                 :            : }
     127                 :            : /* returns the list of "Manin symbols" (c,d) in (Z/NZ)^2, (c,d,N) = 1
     128                 :            :  * generating H^1(X_0(N), Z) */
     129                 :            : static GEN
     130                 :        196 : generatemsymbols(ulong N, ulong num, GEN divN)
     131                 :            : {
     132                 :        196 :   GEN ret = cgetg(num+1, t_VEC);
     133                 :        196 :   ulong c, d, curn = 0;
     134                 :            :   long i, l;
     135                 :            :   /* generate Manin-symbols in two lists: */
     136                 :            :   /* list 1: (c:1) for 0 <= c < N */
     137         [ +  + ]:      29827 :   for (c = 0; c < N; c++) gel(ret, ++curn) = mkvecsmall2(c, 1);
     138         [ -  + ]:        196 :   if (N == 1) return ret;
     139                 :            :   /* list 2: (c:d) with 1 <= c < N, c | N, 0 <= d < N, gcd(d,N) > 1, gcd(c,d)=1.
     140                 :            :    * Furthermore, d != d0 (mod N/c) with c,d0 already in the list */
     141                 :        196 :   l = lg(divN) - 1;
     142                 :            :   /* c = 1 first */
     143                 :        196 :   gel(ret, ++curn) = mkvecsmall2(1,0);
     144         [ +  + ]:      29435 :   for (d = 2; d < N; d++)
     145         [ +  + ]:      29239 :     if (ugcd(d,N) != 1UL)
     146                 :       8596 :       gel(ret, ++curn) = mkvecsmall2(1,d);
     147                 :            :   /* omit c = 1 (first) and c = N (last) */
     148         [ +  + ]:        553 :   for (i=2; i < l; i++)
     149                 :            :   {
     150                 :            :     ulong Novc, d0;
     151                 :        357 :     c = divN[i];
     152                 :        357 :     Novc = N / c;
     153         [ +  + ]:      17346 :     for (d0 = 2; d0 <= Novc; d0++)
     154                 :            :     {
     155                 :      16989 :       ulong k, d = d0;
     156         [ +  + ]:      16989 :       if (ugcd(d, Novc) == 1UL) continue;
     157         [ +  + ]:      62482 :       for (k = 0; k < c; k++, d += Novc)
     158         [ +  + ]:      56238 :         if (ugcd(c,d) == 1UL)
     159                 :            :         {
     160                 :       2506 :           gel(ret, ++curn) = mkvecsmall2(c,d);
     161                 :       2506 :           break;
     162                 :            :         }
     163                 :            :     }
     164                 :            :   }
     165         [ -  + ]:        196 :   if (curn != num) pari_err_BUG("generatemsymbols [wrong number of symbols]");
     166                 :        196 :   return ret;
     167                 :            : }
     168                 :            : 
     169                 :            : #if OLD_HASH
     170                 :            : static ulong
     171                 :            : hash2(GEN H, long N, long a, long b)
     172                 :            : { return ucoeff(H, smodss(a,N) + 1, smodss(b,N) + 1); }
     173                 :            : /* symbols from generatemsymbols(). Returns H such that
     174                 :            :  * H[ nc mod N, nd mod N ] = index of (c,d) in 'symbols', n < N, (n,N) = 1 */
     175                 :            : static GEN
     176                 :            : inithashmsymbols(ulong N, GEN symbols)
     177                 :            : {
     178                 :            :   GEN H = zero_Flm_copy(N, N);
     179                 :            :   long k, l = lg(symbols);
     180                 :            :   ulong n;
     181                 :            :   for (n = 1; n < N; n++)
     182                 :            :     if (ugcd(n,N) == 1)
     183                 :            :       for (k=1; k < l; k++)
     184                 :            :       {
     185                 :            :         GEN s = gel(symbols, k);
     186                 :            :         ucoeff(H, Fl_mul(s[1],n,N) + 1, Fl_mul(s[2],n,N) + 1) = k;
     187                 :            :       }
     188                 :            :   return H;
     189                 :            : }
     190                 :            : #else
     191                 :            : static GEN
     192                 :        196 : inithashmsymbols(ulong N, GEN symbols)
     193                 :            : {
     194                 :        196 :   GEN H = zerovec(N);
     195                 :        196 :   long k, l = lg(symbols);
     196                 :            :   /* skip the (c:1), 0 <= c < N and (1:0) */
     197         [ +  + ]:      11298 :   for (k=N+2; k < l; k++)
     198                 :            :   {
     199                 :      11102 :     GEN s = gel(symbols, k);
     200                 :      11102 :     ulong c = s[1], d = s[2], Novc = N/c;
     201         [ +  + ]:      11102 :     if (gel(H,c) == gen_0) gel(H,c) = const_vecsmall(Novc+1,0);
     202 [ +  + ][ +  + ]:      11102 :     if (c != 1) { d %= Novc; if (!d) d = Novc; }
     203                 :      11102 :     mael(H, c, d) = k;
     204                 :            :   }
     205                 :        196 :   return H;
     206                 :            : }
     207                 :            : #endif
     208                 :            : 
     209                 :            : /** Helper functions for Sl2(Z) / Gamma_0(N) **/
     210                 :            : /* M a 2x2 ZM in SL2(Z) */
     211                 :            : static GEN
     212                 :       9352 : SL2_inv(GEN M)
     213                 :            : {
     214                 :       9352 :   GEN a=gcoeff(M,1,1), b=gcoeff(M,1,2), c=gcoeff(M,2,1), d=gcoeff(M,2,2);
     215                 :       9352 :   return mkmat2(mkcol2(d, negi(c)), mkcol2(negi(b), a));
     216                 :            : }
     217                 :            : /* M a 2x2 zm in SL2(Z) */
     218                 :            : static GEN
     219                 :      31206 : sl2_inv(GEN M)
     220                 :            : {
     221                 :      31206 :   long a=coeff(M,1,1), b=coeff(M,1,2), c=coeff(M,2,1), d=coeff(M,2,2);
     222                 :      31206 :   return mkmat2(mkvecsmall2(d, -c), mkvecsmall2(-b, a));
     223                 :            : }
     224                 :            : /* Return the zm [a,b; c,d] */
     225                 :            : static GEN
     226                 :     276983 : mat2(long a, long b, long c, long d)
     227                 :     276983 : { return mkmat2(mkvecsmall2(a,c), mkvecsmall2(b,d)); }
     228                 :            : 
     229                 :            : /* Input: a = 2-vector = path = {r/s,x/y}
     230                 :            :  * Output: either [r,x;s,y] or [-r,x;-s,y], whichever has determinant > 0 */
     231                 :            : static GEN
     232                 :     158158 : path_to_zm(GEN a)
     233                 :            : {
     234                 :     158158 :   GEN v = gel(a,1), w = gel(a,2);
     235                 :     158158 :   long r = v[1], s = v[2], x = w[1], y = w[2];
     236         [ +  + ]:     158158 :   if (cmpii(mulss(r,y), mulss(x,s)) < 0) { r = -r; s = -s; }
     237                 :     158158 :   return mat2(r,x,s,y);
     238                 :            : }
     239                 :            : /* path from c1 to c2 */
     240                 :            : static GEN
     241                 :     116662 : mkpath(GEN c1, GEN c2) { return mat2(c1[1], c2[1], c1[2], c2[2]); }
     242                 :            : static long
     243                 :     116998 : cc(GEN M) { GEN v = gel(M,1); return v[2]; }
     244                 :            : static long
     245                 :     116998 : dd(GEN M) { GEN v = gel(M,2); return v[2]; }
     246                 :            : 
     247                 :            : /*Input: a,b = 2 paths, N = integer
     248                 :            :  *Output: 1 if the a,b are \Gamma_0(N)-equivalent; 0 otherwise */
     249                 :            : static int
     250                 :      13587 : gamma_equiv(GEN a, GEN b, ulong N)
     251                 :            : {
     252                 :      13587 :   pari_sp av = avma;
     253                 :      13587 :   GEN m = path_to_zm(a);
     254                 :      13587 :   GEN n = path_to_zm(b);
     255                 :      13587 :   GEN d = subii(mulss(cc(m),dd(n)), mulss(dd(m),cc(n)));
     256                 :      13587 :   ulong res = umodiu(d, N);
     257                 :      13587 :   avma = av; return res == 0;
     258                 :            : }
     259                 :            : /* Input: a,b = 2 paths that are \Gamma_0(N)-equivalent, N = integer
     260                 :            :  * Output: M in \Gamma_0(N) such that Mb=a */
     261                 :            : static GEN
     262                 :       7028 : gamma_equiv_matrix(GEN a, GEN b)
     263                 :            : {
     264                 :       7028 :   GEN m = zm_to_ZM( path_to_zm(a) );
     265                 :       7028 :   GEN n = zm_to_ZM( path_to_zm(b) );
     266                 :       7028 :   return ZM_mul(m, SL2_inv(n));
     267                 :            : }
     268                 :            : 
     269                 :            : /*************/
     270                 :            : /* P^1(Z/NZ) */
     271                 :            : /*************/
     272                 :            : 
     273                 :            : /* Input: N = integer
     274                 :            :  * Output: creates P^1(Z/NZ) = [symbols, H, N]
     275                 :            :  *   symbols: list of vectors [x,y] that give a set of representatives
     276                 :            :  *            of P^1(Z/NZ)
     277                 :            :  *   H: an M by M grid whose value at the r,c-th place is the index of the
     278                 :            :  *      "standard representative" equivalent to [r,c] occuring in the first
     279                 :            :  *      list. If gcd(r,c,N) > 1 the grid has value 0. */
     280                 :            : static GEN
     281                 :        196 : create_p1mod(ulong N)
     282                 :            : {
     283                 :        196 :   GEN fa = factoru(N), div = divisorsu(N);
     284                 :        196 :   ulong nsym = count_Manin_symbols(N, gel(fa,1));
     285                 :        196 :   GEN symbols = generatemsymbols(N, nsym, div);
     286                 :        196 :   GEN H = inithashmsymbols(N,symbols);
     287                 :        196 :   return mkvec5(symbols, H, utoipos(N), fa, div);
     288                 :            : }
     289                 :            : 
     290                 :            : /* result is known to be representable as an ulong */
     291                 :            : static ulong
     292                 :     411243 : lcmuu(ulong a, ulong b) { ulong d = ugcd(a,b); return (a/d) * b; }
     293                 :            : /* a != 0 in Z/NZ. Return u in (Z/NZ)^* such that au = gcd(a, N) (mod N)*/
     294                 :            : static ulong
     295                 :    5625697 : Fl_inverse(ulong a, ulong N)
     296                 :            : {
     297                 :            :   pari_sp av;
     298                 :    5625697 :   ulong d, d0, d1, e, u = Fl_invgen(a, N, &d);
     299         [ +  + ]:    5625697 :   if (d == 1) return u;
     300                 :     462196 :   e = N/d;
     301                 :     462196 :   d0 = ucoprime_part(d, e); /* d = d0 d1, d0 coprime to N/d, core(d1) | N/d */
     302         [ +  + ]:     462196 :   if (d0 == 1) return u;
     303                 :     411243 :   av = avma;
     304                 :     411243 :   d1 = d / d0;
     305                 :     411243 :   e = lcmuu(e, d1);
     306                 :     411243 :   u = itou(Z_chinese_coprime(utoipos(u), gen_1,
     307                 :            :                              utoipos(e), utoipos(d0), utoipos(e*d0)));
     308                 :    5625697 :   avma = av; return u;
     309                 :            : }
     310                 :            : /* Let (c : d) in P1(Z/NZ).
     311                 :            :  * If c = 0 return (0:1). If d = 0 return (1:0).
     312                 :            :  * Else replace by (cu : du), where u in (Z/NZ)^* such that C := cu = gcd(c,N).
     313                 :            :  * In create_p1mod(), (c : d) is represented by (C:D) where D = du (mod N/c)
     314                 :            :  * is smallest such that gcd(C,D) = 1. Return (C : du mod N/c), which need
     315                 :            :  * not belong to P1(Z/NZ) ! A second component du mod N/c = 0 is replaced by
     316                 :            :  * N/c in this case to avoid problems with array indices */
     317                 :            : static GEN
     318                 :   20792632 : p1_std_form(long c, long d, ulong N)
     319                 :            : {
     320                 :            :   ulong u;
     321                 :   20792632 :   c = smodss(c, N);
     322                 :   20792632 :   d = smodss(d, N);
     323         [ +  + ]:   20792632 :   if (!c) return mkvecsmall2(0, 1);
     324         [ +  + ]:   18523778 :   if (!d) return mkvecsmall2(1, 0);
     325                 :   18236141 :   u = Fl_invsafe(d, N);
     326         [ +  + ]:   18236141 :   if (u != 0) return mkvecsmall2(Fl_mul(c,u,N), 1); /* (d,N) = 1 */
     327                 :            : 
     328                 :    5620531 :   u = Fl_inverse(c, N);
     329         [ +  + ]:    5620531 :   if (u > 1) { c = Fl_mul(c,u,N); d = Fl_mul(d,u,N); }
     330                 :            :   /* c | N */
     331         [ +  + ]:    5620531 :   if (c != 1) d = d % (N/c);
     332         [ +  + ]:    5620531 :   if (!d) d = N/c;
     333                 :   20792632 :   return mkvecsmall2(c, d);
     334                 :            : }
     335                 :            : 
     336                 :            : /* Input: v = [x,y] = elt of P^1(Z/NZ) = class in Gamma_0(N) \ PSL2(Z)
     337                 :            :  * Output: returns the index of the standard rep equivalent to v */
     338                 :            : static long
     339                 :   20792632 : p1_index(long x, long y, GEN p1N)
     340                 :            : {
     341                 :   20792632 :   ulong N = p1N_get_N(p1N);
     342                 :   20792632 :   GEN H = p1N_get_hash(p1N), c;
     343                 :            : 
     344                 :            : #ifdef OLD_HASH
     345                 :            :   return hash2(p1N_get_hash(p1N), N, x, y);
     346                 :            : #else
     347                 :   20792632 :   c = p1_std_form(x, y, N);
     348                 :   20792632 :   x = c[1];
     349                 :   20792632 :   y = c[2];
     350         [ +  + ]:   20792632 :   if (y == 1) return x+1;
     351         [ +  + ]:    5908168 :   if (y == 0) return N+1;
     352         [ -  + ]:    5620531 :   if (mael(H,x,y) == 0) pari_err_BUG("p1_index");
     353                 :   20792632 :   return mael(H,x,y);
     354                 :            : #endif
     355                 :            : }
     356                 :            : 
     357                 :            : /* Cusps for \Gamma_0(N) */
     358                 :            : 
     359                 :            : /* \sum_{d | N} \phi(gcd(d, N/d)), using multiplicativity. fa = factor(N) */
     360                 :            : static ulong
     361                 :        196 : nbcusp(GEN fa)
     362                 :            : {
     363                 :        196 :   GEN P = gel(fa,1), E = gel(fa,2);
     364                 :        196 :   long i, l = lg(P);
     365                 :        196 :   ulong T = 1;
     366         [ +  + ]:        448 :   for (i = 1; i < l; i++)
     367                 :            :   {
     368                 :        252 :     long e = E[i] >> 1; /* floor(E[i] / 2) */
     369                 :        252 :     ulong p = P[i];
     370         [ +  + ]:        252 :     if (odd(E[i]))
     371                 :        224 :       T *= 2 * upowuu(p, e);
     372                 :            :     else
     373                 :         28 :       T *= (p+1) * upowuu(p, e - 1);
     374                 :            :   }
     375                 :        196 :   return T;
     376                 :            : }
     377                 :            : 
     378                 :            : /* to each cusp in \Gamma_0(N) P1(Q), represented by p/q, we associate a
     379                 :            :  * unique index. Canonical representative: (1:0) or (p:q) with q | N, q < N,
     380                 :            :  * p defined modulo d := gcd(N/q,q), (p,d) = 1.
     381                 :            :  * Return [[N, nbcusps], H, cusps]*/
     382                 :            : static GEN
     383                 :        196 : inithashcusps(GEN p1N)
     384                 :            : {
     385                 :        196 :   ulong N = p1N_get_N(p1N);
     386                 :        196 :   GEN div = p1N_get_div(p1N), H = zerovec(N+1);
     387                 :        196 :   long k, ind, l = lg(div), ncusp = nbcusp(p1N_get_fa(p1N));
     388                 :        196 :   GEN cusps = cgetg(ncusp+1, t_VEC);
     389                 :            : 
     390                 :        196 :   gel(H,1) = mkvecsmall2(0/*empty*/, 1/* first cusp: (1:0) */);
     391                 :        196 :   gel(cusps, 1) = mkvecsmall2(1,0);
     392                 :        196 :   ind = 2;
     393         [ +  + ]:        749 :   for (k=1; k < l-1; k++) /* l-1: remove q = N */
     394                 :            :   {
     395                 :        553 :     ulong p, q = div[k], d = ugcd(q, N/q);
     396                 :        553 :     GEN h = const_vecsmall(d+1,0);
     397                 :        553 :     gel(H,q+1) = h ;
     398         [ +  + ]:       2009 :     for (p = 0; p < d; p++)
     399         [ +  + ]:       1456 :       if (ugcd(p,d) == 1)
     400                 :            :       {
     401                 :        931 :         h[p+1] = ind;
     402                 :        931 :         gel(cusps, ind) = mkvecsmall2(p,q);
     403                 :        931 :         ind++;
     404                 :            :       }
     405                 :            :   }
     406                 :        196 :   return mkvec3(mkvecsmall2(N,ind-1), H, cusps);
     407                 :            : }
     408                 :            : /* c = [p,q], (p,q) = 1, return a canonical representative for
     409                 :            :  * \Gamma_0(N)(p/q) */
     410                 :            : static GEN
     411                 :       5278 : cusp_std_form(GEN c, GEN S)
     412                 :            : {
     413                 :       5278 :   long p, N = gel(S,1)[1], q = smodss(c[2], N);
     414                 :            :   ulong u, d;
     415         [ +  + ]:       5278 :   if (q == 0) return mkvecsmall2(1, 0);
     416                 :       5166 :   p = smodss(c[1], N);
     417                 :       5166 :   u = Fl_inverse(q, N);
     418                 :       5166 :   q = Fl_mul(q,u, N);
     419                 :       5166 :   d = ugcd(q, N/q);
     420                 :       5278 :   return mkvecsmall2(Fl_div(p % d,u % d, d), q);
     421                 :            : }
     422                 :            : /* c = [p,q], (p,q) = 1, return the index of the corresponding cusp.
     423                 :            :  * S from inithashcusps */
     424                 :            : static ulong
     425                 :       5278 : cusp_index(GEN c, GEN S)
     426                 :            : {
     427                 :            :   long p, q;
     428                 :       5278 :   GEN H = gel(S,2);
     429                 :       5278 :   c = cusp_std_form(c, S);
     430                 :       5278 :   p = c[1]; q = c[2];
     431         [ -  + ]:       5278 :   if (!mael(H,q+1,p+1)) pari_err_BUG("cusp_index");
     432                 :       5278 :   return mael(H,q+1,p+1);
     433                 :            : }
     434                 :            : 
     435                 :            : /* M a square invertible ZM, return a ZM iM such that iM M = M iM = d.Id */
     436                 :            : static GEN
     437                 :        497 : ZM_inv_denom(GEN M)
     438                 :            : {
     439                 :        497 :   GEN ciM = ZM_det(M);
     440                 :        497 :   GEN c, iM = Q_primitive_part(ZM_inv(M, ciM), &c);
     441         [ +  + ]:        497 :   if (c) ciM = diviiexact(ciM, c);
     442                 :        497 :   return mkvec2(iM, ciM);
     443                 :            : }
     444                 :            : /* return M^(-1) v, dinv = ZM_inv_denom(M) OR Qevproj_init(M) */
     445                 :            : static GEN
     446                 :       1687 : ZC_apply_dinv(GEN dinv, GEN v)
     447                 :            : {
     448                 :            :   GEN x, c, iM;
     449         [ +  + ]:       1687 :   if (lg(dinv) == 3)
     450                 :            :   {
     451                 :        721 :     iM = gel(dinv,1);
     452                 :        721 :     c = gel(dinv,2);
     453                 :            :   }
     454                 :            :   else
     455                 :            :   { /* Qevproj_init */
     456                 :        966 :     iM = gel(dinv,2);
     457                 :        966 :     c = gel(dinv,3);
     458                 :        966 :     v = typ(v) == t_MAT? rowpermute(v, gel(dinv,4))
     459         [ -  + ]:        966 :                        : vecpermute(v, gel(dinv,4));
     460                 :            :   }
     461                 :       1687 :   x = RgM_RgC_mul(iM, v);
     462         [ +  - ]:       1687 :   if (!isint1(c)) x = RgC_Rg_div(x, c);
     463                 :       1687 :   return x;
     464                 :            : }
     465                 :            : 
     466                 :            : /* M an n x d ZM of rank d (basis of a Q-subspace), n >= d.
     467                 :            :  * Initialize a projector on M */
     468                 :            : GEN
     469                 :        413 : Qevproj_init(GEN M)
     470                 :            : {
     471                 :            :   GEN v, perm, MM, dinv, iM, ciM;
     472                 :        413 :   v = ZM_indexrank(M); perm = gel(v,1);
     473                 :        413 :   MM = rowpermute(M, perm); /* square invertible */
     474                 :        413 :   dinv = ZM_inv_denom(MM);
     475                 :        413 :   iM = gel(dinv,1);
     476                 :        413 :   ciM= gel(dinv,2);
     477                 :        413 :   return mkvec4(M, iM, ciM, perm);
     478                 :            : }
     479                 :            : /* same with typechecks */
     480                 :            : static GEN
     481                 :         21 : Qevproj_init0(GEN M)
     482                 :            : {
     483      [ +  -  - ]:         21 :   switch(typ(M))
     484                 :            :   {
     485                 :            :     case t_VEC:
     486         [ +  - ]:         21 :       if (lg(M) == 5) return M;
     487                 :          0 :       break;
     488                 :            :     case t_MAT:
     489                 :          0 :       M = Q_primpart(M);
     490                 :          0 :       RgM_check_ZM(M,"Qevproj_init");
     491                 :          0 :       return Qevproj_init(M);
     492                 :            :   }
     493                 :          0 :   pari_err_TYPE("Qevproj_init",M);
     494                 :         21 :   return NULL;
     495                 :            : }
     496                 :            : 
     497                 :            : /* T an n x n QM, stabilizing d-dimensional Q-vector space spanned by the
     498                 :            :  * columns of M, pro = Qevproj_init(M). Return d x d matrix of T acting
     499                 :            :  * on M */
     500                 :            : GEN
     501                 :         98 : Qevproj_apply(GEN T, GEN pro)
     502                 :            : {
     503                 :         98 :   GEN M = gel(pro,1), iM = gel(pro,2), ciM = gel(pro,3), perm = gel(pro,4);
     504                 :         98 :   return RgM_Rg_div(RgM_mul(iM, RgM_mul(rowpermute(T,perm), M)), ciM);
     505                 :            : }
     506                 :            : /* Qevproj_apply(T,pro)[,k] */
     507                 :            : GEN
     508                 :         70 : Qevproj_apply_vecei(GEN T, GEN pro, long k)
     509                 :            : {
     510                 :         70 :   GEN M = gel(pro,1), iM = gel(pro,2), ciM = gel(pro,3), perm = gel(pro,4);
     511                 :         70 :   GEN v = RgM_RgC_mul(iM, RgM_RgC_mul(rowpermute(T,perm), gel(M,k)));
     512                 :         70 :   return RgC_Rg_div(v, ciM);
     513                 :            : }
     514                 :            : 
     515                 :            : /* normalize a Q-basis*/
     516                 :            : static GEN
     517                 :        441 : Q_primpart_basis(GEN M)
     518                 :            : {
     519                 :            :   long i, l;
     520                 :        441 :   GEN N = cgetg_copy(M, &l);
     521         [ +  + ]:       6951 :   for (i = 1; i < l; i++) gel(N,i) = Q_primpart(gel(M,i));
     522                 :        441 :   return N;
     523                 :            : }
     524                 :            : static GEN
     525                 :        364 : ZM_ker(GEN M) { return Q_primpart_basis(keri(M)); }
     526                 :            : static GEN
     527                 :        273 : QM_ker(GEN M) { return ZM_ker(Q_primpart(M)); }
     528                 :            : static GEN
     529                 :         70 : QM_image(GEN A)
     530                 :            : {
     531                 :         70 :   A = Q_primpart_basis(A);
     532                 :         70 :   return vecpermute(A, ZM_indeximage(A));
     533                 :            : }
     534                 :            : 
     535                 :            : /* Decompose the subspace H (Qevproj format) in simple subspaces.
     536                 :            :  * Eg for H = msnew */
     537                 :            : static GEN
     538                 :         14 : mssplit_i(GEN W, GEN H)
     539                 :            : {
     540                 :         14 :   ulong p, N = ms_get_N(W);
     541                 :            :   long first, dim;
     542                 :            :   forprime_t S;
     543                 :         14 :   GEN T1 = NULL, T2 = NULL, V;
     544                 :         14 :   dim = lg(gel(H,1))-1;
     545                 :         14 :   V = vectrunc_init(dim+1);
     546         [ -  + ]:         14 :   if (!dim) return V;
     547                 :         14 :   (void)u_forprime_init(&S, 2, ULONG_MAX);
     548                 :         14 :   vectrunc_append(V, H);
     549                 :         14 :   first = 1; /* V[1..first-1] contains simple subspaces */
     550         [ +  - ]:         21 :   while ((p = u_forprime_next(&S)))
     551                 :            :   {
     552                 :            :     GEN T;
     553                 :            :     long n, j, lV;
     554         [ -  + ]:         21 :     if (N % p == 0) continue;
     555 [ +  + ][ -  + ]:         21 :     if (T1 && T2)
     556                 :            :     {
     557                 :          0 :       T = RgM_add(T1,T2);
     558                 :          0 :       T2 = NULL;
     559                 :            :     }
     560                 :            :     else
     561                 :            :     {
     562                 :         21 :       T2 = T1;
     563                 :         21 :       T1 = T = mshecke_i(W, p);
     564                 :            :     }
     565                 :         21 :     lV = lg(V);
     566         [ +  + ]:         42 :     for (j = first; j < lV; j++)
     567                 :            :     {
     568                 :         21 :       pari_sp av = avma;
     569                 :         21 :       GEN Vj = gel(V,j), P = gel(Vj,1);
     570                 :         21 :       GEN TVj = Qevproj_apply(T, Vj); /* c T | V_j */
     571                 :         21 :       GEN ch = QM_charpoly_ZX(TVj), fa = ZX_factor(ch), F, E;
     572                 :            :       long k;
     573                 :         21 :       F = gel(fa, 1);
     574                 :         21 :       E = gel(fa, 2);
     575                 :         21 :       n = lg(F)-1;
     576         [ +  + ]:         21 :       if (n == 1)
     577                 :            :       {
     578         [ +  - ]:         14 :         if (isint1(gel(E,1)))
     579                 :            :         { /* simple subspace */
     580                 :         14 :           swap(gel(V,first), gel(V,j));
     581                 :         14 :           first++;
     582                 :            :         }
     583                 :            :         else
     584                 :          0 :           avma = av;
     585                 :            :       }
     586                 :            :       else
     587                 :            :       { /* can split Vj */
     588                 :            :         GEN pows;
     589                 :          7 :         long D = 1;
     590         [ +  + ]:         42 :         for (k = 1; k <= n; k++)
     591                 :            :         {
     592                 :         35 :           long d = degpol(gel(F,k));
     593         [ +  + ]:         35 :           if (d > D) D = d;
     594                 :            :         }
     595                 :            :         /* remove V[j] */
     596                 :          7 :         swap(gel(V,j), gel(V,lg(V)-1)); setlg(V, lg(V)-1);
     597                 :          7 :         pows = RgM_powers(TVj, minss((long)2*sqrt(D), D));
     598         [ +  + ]:         42 :         for (k = 1; k <= n; k++)
     599                 :            :         {
     600                 :         35 :           GEN f = gel(F,k);
     601                 :         35 :           GEN M = RgX_RgMV_eval(f, pows); /* f(TVj) */
     602                 :         35 :           GEN K = QM_ker(M);
     603                 :         35 :           GEN p = Q_primpart( RgM_mul(P, K) );
     604                 :         35 :           vectrunc_append(V, Qevproj_init(p));
     605 [ +  - ][ +  + ]:         35 :           if (lg(K) == 2 || isint1(gel(E,k)))
     606                 :            :           { /* simple subspace */
     607                 :         28 :             swap(gel(V,first), gel(V, lg(V)-1));
     608                 :         28 :             first++;
     609                 :            :           }
     610                 :            :         }
     611         [ +  - ]:          7 :         if (j < first) j = first;
     612                 :            :       }
     613                 :            :     }
     614         [ +  + ]:         21 :     if (first >= lg(V)) return V;
     615                 :            :   }
     616                 :          0 :   pari_err_BUG("subspaces not found");
     617                 :         14 :   return NULL;
     618                 :            : }
     619                 :            : GEN
     620                 :         14 : mssplit(GEN W, GEN H)
     621                 :            : {
     622                 :         14 :   pari_sp av = avma;
     623                 :         14 :   checkms(W);
     624         [ -  + ]:         14 :   if (!msk_get_sign(W))
     625                 :          0 :     pari_err_DOMAIN("mssplit","abs(sign)","!=",gen_1,gen_0);
     626                 :         14 :   H = Qevproj_init0(H);
     627                 :         14 :   return gerepilecopy(av, mssplit_i(W,H));
     628                 :            : }
     629                 :            : 
     630                 :            : /* proV = Qevproj_init of a Hecke simple subspace, return [ a_n, n <= B ] */
     631                 :            : static GEN
     632                 :          7 : msqexpansion_i(GEN W, GEN proV, ulong B)
     633                 :            : {
     634                 :          7 :   ulong p, N = ms_get_N(W), sqrtB;
     635                 :          7 :   long i, d, k = msk_get_weight(W);
     636                 :            :   forprime_t S;
     637                 :          7 :   GEN T1 = NULL, T2 = NULL, TV = NULL, ch, v, dTiv, Tiv, dinv, ciM, iM, L;
     638      [ -  -  + ]:          7 :   switch(B)
     639                 :            :   {
     640                 :          0 :     case 0: return cgetg(1,t_VEC);
     641                 :          0 :     case 1: return mkvec(gen_1);
     642                 :            :   }
     643                 :          7 :   (void)u_forprime_init(&S, 2, ULONG_MAX);
     644         [ +  - ]:          7 :   while ((p = u_forprime_next(&S)))
     645                 :            :   {
     646                 :          7 :     pari_sp av = avma;
     647                 :            :     GEN T;
     648         [ -  + ]:          7 :     if (N % p == 0) continue;
     649 [ -  + ][ #  # ]:          7 :     if (T1 && T2)
     650                 :            :     {
     651                 :          0 :       T = RgM_add(T1,T2);
     652                 :          0 :       T2 = NULL;
     653                 :            :     }
     654                 :            :     else
     655                 :            :     {
     656                 :          7 :       T2 = T1;
     657                 :          7 :       T1 = T = mshecke_i(W, p);
     658                 :            :     }
     659                 :          7 :     TV = Qevproj_apply(T, proV); /* T | V */
     660                 :          7 :     ch = QM_charpoly_ZX(TV);
     661         [ +  - ]:          7 :     if (ZX_is_irred(ch)) break;
     662                 :          0 :     avma = av;
     663                 :          0 :     ch = NULL;
     664                 :            :   }
     665         [ -  + ]:          7 :   if (!ch) pari_err_BUG("q-Expansion not found");
     666                 :            :   /* T generates the Hecke algebra */
     667                 :          7 :   d = degpol(ch);
     668                 :          7 :   v = vec_ei(d, 1); /* take v = e_1 */
     669                 :          7 :   Tiv = cgetg(d+1, t_MAT); /* Tiv[i] = T^(i-1)v */
     670                 :          7 :   gel(Tiv, 1) = v;
     671         [ +  + ]:         14 :   for (i = 2; i <= d; i++) gel(Tiv, i) = RgM_RgC_mul(TV, gel(Tiv,i-1));
     672                 :          7 :   Tiv = Q_remove_denom(Tiv, &dTiv);
     673                 :          7 :   dinv = ZM_inv_denom(Tiv);
     674                 :          7 :   iM = gel(dinv,1);
     675                 :          7 :   ciM = gel(dinv,2);
     676         [ -  + ]:          7 :   if (dTiv) ciM = gdiv(ciM, dTiv);
     677                 :          7 :   L = const_vec(B,NULL);
     678                 :          7 :   sqrtB = (ulong)sqrt(B);
     679         [ +  - ]:          7 :   gel(L,1) = d > 1? mkpolmod(gen_1,ch): gen_1;
     680         [ +  + ]:        210 :   for (p = 2; p <= B; p++)
     681                 :            :   {
     682                 :        203 :     pari_sp av = avma;
     683                 :            :     GEN T, u, Tv, ap, P;
     684                 :            :     ulong m;
     685         [ +  + ]:        203 :     if (gel(L,p)) continue;  /* p not prime */
     686                 :         70 :     T = mshecke_i(W, p);
     687                 :         70 :     Tv = Qevproj_apply_vecei(T, proV, 1); /* Tp.v */
     688                 :            :     /* Write Tp.v = \sum u_i T^i v */
     689                 :         70 :     u = RgC_Rg_div(RgM_RgC_mul(iM, Tv), ciM);
     690                 :         70 :     ap = gerepilecopy(av, RgV_to_RgX(u, 0));
     691         [ +  - ]:         70 :     if (d > 1)
     692                 :         70 :       ap = mkpolmod(ap,ch);
     693                 :            :     else
     694                 :          0 :       ap = simplify_shallow(ap);
     695                 :         70 :     gel(L,p) = ap;
     696                 :         70 :     P = powuu(p,k-1);
     697         [ +  + ]:         70 :     if (p <= sqrtB) {
     698                 :         21 :       ulong pj, oldpj = 1;
     699         [ +  + ]:         84 :       for (pj = p; pj <= B; oldpj=pj, pj *= p)
     700                 :            :       {
     701                 :         63 :         GEN apj = (pj==p)? ap
     702         [ +  + ]:         63 :                          : gsub(gmul(ap,gel(L,oldpj)), gmul(P,gel(L,oldpj/p)));
     703                 :         63 :         gel(L,pj) = apj;
     704         [ +  + ]:        329 :         for (m = B/pj; m > 1; m--)
     705 [ +  + ][ +  + ]:        266 :           if (gel(L,m) && m%p) gel(L,m*pj) = gmul(gel(L,m), apj);
     706                 :            :       }
     707                 :            :     } else {
     708                 :         49 :       gel(L,p) = ap;
     709         [ +  + ]:         84 :       for (m = B/p; m > 1; m--)
     710         [ +  - ]:         35 :         if (gel(L,m)) gel(L,m*p) = gmul(gel(L,m), ap);
     711                 :            :     }
     712                 :            :   }
     713                 :          7 :   return L;
     714                 :            : }
     715                 :            : GEN
     716                 :          7 : msqexpansion(GEN W, GEN proV, ulong B)
     717                 :            : {
     718                 :          7 :   pari_sp av = avma;
     719                 :          7 :   checkms(W);
     720                 :          7 :   proV = Qevproj_init0(proV);
     721                 :          7 :   return gerepilecopy(av, msqexpansion_i(W,proV,B));
     722                 :            : }
     723                 :            : 
     724                 :            : static GEN
     725                 :         98 : Qevproj_star(GEN W, GEN H)
     726                 :            : {
     727                 :         98 :   long s = msk_get_sign(W);
     728         [ +  + ]:         98 :   if (s)
     729                 :            :   { /* project on +/- component */
     730                 :         42 :     GEN star = msk_get_star(W);
     731                 :         42 :     GEN A = gmul(star, H);
     732         [ +  - ]:         42 :     A = (s > 0)? gadd(A, H): gsub(A, H);
     733                 :            :     /* Im(star + sign) = Ker(star - sign) */
     734                 :         42 :     H = QM_image(A);
     735                 :            :   }
     736                 :         98 :   return H;
     737                 :            : }
     738                 :            : static GEN
     739                 :         21 : msnew_trivial(GEN W)
     740                 :            : {
     741                 :         21 :   GEN p1N = ms_get_p1N(W), P = gel(p1N_get_fa(p1N), 1);
     742                 :         21 :   ulong N = ms_get_N(W);
     743                 :         21 :   long i, nP = lg(P)-1;
     744                 :            :   GEN Snew, K, v;
     745                 :            : 
     746                 :         21 :   K = mscuspidal_trivial(W);
     747         [ +  + ]:         21 :   if (uisprime(N)) { Snew = K; goto END; }
     748                 :          7 :   v = cgetg(2*nP + 1, t_COL);
     749         [ +  + ]:         21 :   for (i = 1; i <= nP; i++)
     750                 :            :   {
     751                 :         14 :     pari_sp av = avma, av2;
     752                 :         14 :     ulong M = N / P[i];
     753                 :         14 :     GEN T1, Td, Wi = mskinit(M, 2, 0);
     754                 :         14 :     T1 = getMorphism_trivial(W, Wi, mat2(1,0,0,1));
     755                 :         14 :     Td = getMorphism_trivial(W, Wi, mat2(P[i],0,0,1));
     756                 :         14 :     av2 = avma;
     757                 :         14 :     T1 = ZM_mul(T1, K); /* multiply by K = restrict to Ker delta */
     758                 :         14 :     Td = ZM_mul(Td, K);
     759                 :         14 :     gerepileallsp(av, av2, 2, &T1,&Td);
     760                 :         14 :     gel(v,2*i-1)= T1;
     761                 :         14 :     gel(v,2*i)  = Td;
     762                 :            :   }
     763                 :          7 :   Snew = ZM_mul(K, ZM_ker(matconcat(v)));
     764                 :            : END:
     765                 :         21 :   return Qevproj_star(W, Snew);
     766                 :            : }
     767                 :            : 
     768                 :            : static GEN
     769                 :         42 : msnew_i(GEN W)
     770                 :            : {
     771                 :         42 :   GEN p1N = ms_get_p1N(W), P = gel(p1N_get_fa(p1N), 1);
     772                 :         42 :   ulong p, N = ms_get_N(W);
     773                 :         42 :   long i, nP = lg(P)-1, k = msk_get_weight(W);
     774                 :            :   GEN S, Snew, Sold, pr_S, pr_Sold, v;
     775                 :            :   forprime_t F;
     776         [ +  + ]:         42 :   if (k == 2) return msnew_trivial(W);
     777                 :         21 :   S = gel(mscuspidal_i(W), 2);
     778         [ +  + ]:         21 :   if (uisprime(N)) { Snew = S; goto END; }
     779                 :          7 :   v = cgetg(2*nP + 1, t_VEC);
     780         [ +  + ]:         21 :   for (i = 1; i <= nP; i++)
     781                 :            :   {
     782                 :         14 :     GEN Wi = mskinit(N/P[i], k, 0);
     783                 :         14 :     gel(v,2*i-1) = getMorphism(Wi, W, mat2(1,0,0,1));
     784                 :         14 :     gel(v,2*i)   = getMorphism(Wi, W, mat2(P[i],0,0,1));
     785                 :            :   }
     786                 :          7 :   v = QM_image(matconcat(v)); /* old forms */
     787                 :            :   /* restrict to cuspidal subspace */
     788                 :          7 :   Sold = Q_primpart_basis(intersect(S, v));
     789                 :          7 :   pr_S = Qevproj_init(S);
     790                 :          7 :   pr_Sold = Qevproj_init(Sold);
     791                 :          7 :   Snew = NULL;
     792                 :          7 :   (void)u_forprime_init(&F, 2, ULONG_MAX);
     793         [ +  - ]:         21 :   while ((p = u_forprime_next(&F)))
     794                 :            :   {
     795                 :         21 :     pari_sp av = avma;
     796                 :            :     GEN T, chS, chSold, chSnew;
     797         [ +  + ]:         21 :     if (N % p == 0) continue;
     798                 :          7 :     T = mshecke_i(W, p);
     799                 :          7 :     chS = QM_charpoly_ZX(Qevproj_apply(T, pr_S));
     800                 :          7 :     chSold = QM_charpoly_ZX(Qevproj_apply(T, pr_Sold));
     801                 :          7 :     chSnew = RgX_div(chS, chSold); /* = char T | S^new */
     802         [ +  - ]:          7 :     if (!degpol( ZX_gcd(chSnew, chSold) ))
     803                 :            :     {
     804                 :          7 :       GEN M = RgX_RgM_eval(chSnew, T);
     805                 :          7 :       Snew = QM_ker(M);
     806                 :          7 :       break;
     807                 :            :     }
     808                 :          0 :     avma = av;
     809                 :            :   }
     810                 :            : END:
     811                 :         42 :   return Qevproj_star(W, Snew);
     812                 :            : }
     813                 :            : GEN
     814                 :         42 : msnew(GEN W)
     815                 :            : {
     816                 :         42 :   pari_sp av = avma;
     817                 :         42 :   checkms(W);
     818                 :         42 :   return gerepilecopy(av, Qevproj_init( msnew_i(W)) );
     819                 :            : }
     820                 :            : 
     821                 :            : /* Solve the Manin relations for a congruence subgroup \Gamma by constructing
     822                 :            :  * a well-formed fundamental domain for the action of \Gamma on upper half
     823                 :            :  * space. See
     824                 :            :  * Pollack and Stevens, Overconvergent modular symbols and p-adic L-functions
     825                 :            :  * Annales scientifiques de l'ENS 44, fascicule 1 (2011), 1-42
     826                 :            :  * http://math.bu.edu/people/rpollack/Papers/Overconvergent_modular_symbols_and_padic_Lfunctions.pdf
     827                 :            :  *
     828                 :            :  * FIXME: Implemented for \Gamma = \Gamma_0(N) only. */
     829                 :            : 
     830                 :            : #if 0 /* Pollack-Stevens shift their paths so as to solve equations of the
     831                 :            :          form f(z+1) - f(z) = g. We don't (to avoid mistakes) so we will
     832                 :            :          have to solve eqs of the form f(z-1) - f(z) = g */
     833                 :            : /* c = a/b; as a t_VECSMALL [a,b]; return c-1 as a t_VECSMALL */
     834                 :            : static GEN
     835                 :            : Shift_left_cusp(GEN c) { long a=c[1], b=c[2]; return mkvecsmall2(a - b, b); }
     836                 :            : /* c = a/b; as a t_VECSMALL [a,b]; return c+1 as a t_VECSMALL */
     837                 :            : static GEN
     838                 :            : Shift_right_cusp(GEN c) { long a=c[1], b=c[2]; return mkvecsmall2(a + b, b); }
     839                 :            : /*Input: path = [r,s] (thought of as a geodesic between these points)
     840                 :            :  *Output: The path shifted by one to the left, i.e. [r-1,s-1] */
     841                 :            : static GEN
     842                 :            : Shift_left(GEN path)
     843                 :            : {
     844                 :            :   GEN r = gel(path,1), s = gel(path,2);
     845                 :            :   return mkvec2(Shift_left_cusp(r), Shift_left_cusp(s)); }
     846                 :            : /*Input: path = [r,s] (thought of as a geodesic between these points)
     847                 :            :  *Output: The path shifted by one to the right, i.e. [r+1,s+1] */
     848                 :            : GEN
     849                 :            : Shift_right(GEN path)
     850                 :            : {
     851                 :            :   GEN r = gel(path,1), s = gel(path,2);
     852                 :            :   return mkvec2(Shift_right_cusp(r), Shift_right_cusp(s)); }
     853                 :            : #endif
     854                 :            : 
     855                 :            : /* linked lists */
     856                 :            : typedef struct list_t { GEN data; struct list_t *next; } list_t;
     857                 :            : static list_t *
     858                 :      13825 : list_new(GEN x)
     859                 :            : {
     860                 :      13825 :   list_t *L = (list_t*)stack_malloc(sizeof(list_t));
     861                 :      13825 :   L->data = x;
     862                 :      13825 :   L->next = NULL; return L;
     863                 :            : }
     864                 :            : static void
     865                 :      13629 : list_insert(list_t *L, GEN x)
     866                 :            : {
     867                 :      13629 :   list_t *l = list_new(x);
     868                 :      13629 :   l->next = L->next;
     869                 :      13629 :   L->next = l;
     870                 :      13629 : }
     871                 :            : 
     872                 :            : /*Input: N > 1, p1N = P^1(Z/NZ)
     873                 :            :  *Output: a connected fundamental domain for the action of \Gamma_0(N) on
     874                 :            :  *  upper half space.  When \Gamma_0(N) is torsion free, the domain has the
     875                 :            :  *  property that all of its vertices are cusps.  When \Gamma_0(N) has
     876                 :            :  *  three-torsion, 2 extra triangles need to be added.
     877                 :            :  *
     878                 :            :  * The domain is constructed by beginning with the triangle with vertices 0,1
     879                 :            :  * and oo.  Each adjacent triangle is successively tested to see if it contains
     880                 :            :  * points not \Gamma_0(N) equivalent to some point in our region.  If a
     881                 :            :  * triangle contains new points, it is added to the region.  This process is
     882                 :            :  * continued until the region can no longer be extended (and still be a
     883                 :            :  * fundamental domain) by added an adjacent triangle.  The list of cusps
     884                 :            :  * between 0 and 1 are then returned
     885                 :            :  *
     886                 :            :  * Precisely, the function returns a list such that the elements of the list
     887                 :            :  * with odd index are the cusps in increasing order.  The even elements of the
     888                 :            :  * list are either an "x" or a "t".  A "t" represents that there is an element
     889                 :            :  * of order three such that its fixed point is in the triangle directly
     890                 :            :  * adjacent to the our region with vertices given by the cusp before and after
     891                 :            :  * the "t".  The "x" represents that this is not the case. */
     892                 :            : enum { type_X, type_DO /* ? */, type_T };
     893                 :            : static GEN
     894                 :        196 : form_list_of_cusps(ulong N, GEN p1N)
     895                 :            : {
     896                 :        196 :   pari_sp av = avma;
     897                 :        196 :   long i, position, nbC = 2;
     898                 :            :   GEN v, L;
     899                 :            :   list_t *C, *c;
     900                 :            :   /* Let t be the index of a class in PSL2(Z) / \Gamma in our fixed enumeration
     901                 :            :    * v[t] != 0 iff it is the class of z tau^r for z a previous alpha_i
     902                 :            :    * or beta_i.
     903                 :            :    * For \Gamma = \Gamma_0(N), the enumeration is given by p1_index.
     904                 :            :    * We write cl(gamma) = the class of gamma mod \Gamma */
     905                 :        196 :   v = const_vecsmall(p1_size(p1N), 0);
     906                 :        196 :   i = p1_index( 0, 1, p1N); v[i] = 1;
     907                 :        196 :   i = p1_index( 1,-1, p1N); v[i] = 2;
     908                 :        196 :   i = p1_index(-1, 0, p1N); v[i] = 3;
     909                 :            :   /* the value is unused [debugging]: what matters is whether it is != 0 */
     910                 :        196 :   position = 4;
     911                 :            :   /* at this point, Fund = R, v contains the classes of Id, tau, tau^2 */
     912                 :            : 
     913                 :        196 :   C  = list_new(mkvecsmall3(0,1, type_X));
     914                 :        196 :   list_insert(C, mkvecsmall3(1,1,type_DO));
     915                 :            :   /* C is a list of triples[a,b,t], where c = a/b is a cusp, and t is the type
     916                 :            :    * of the path between c and the PREVIOUS cusp in the list, coded as
     917                 :            :    *   type_DO = "?", type_X = "x", type_T = "t"
     918                 :            :    * Initially, C = [0/1,"?",1/1]; */
     919                 :            : 
     920                 :            :   /* loop through the current set of cusps C and check to see if more cusps
     921                 :            :    * should be added */
     922                 :            :   for (;;)
     923                 :            :   {
     924                 :        980 :     int done = 1;
     925         [ +  - ]:      73304 :     for (c = C; c; c = c->next)
     926                 :            :     {
     927                 :            :       GEN cusp1, cusp2, gam;
     928                 :            :       long pos, b1, b2, b;
     929                 :            : 
     930         [ +  + ]:      73304 :       if (!c->next) break;
     931                 :      72324 :       cusp1 = c->data; /* = a1/b1 */
     932                 :      72324 :       cusp2 = (c->next)->data; /* = a2/b2 */
     933         [ +  + ]:      72324 :       if (cusp2[3] != type_DO) continue;
     934                 :            : 
     935                 :            :       /* gam (oo -> 0) = (cusp2 -> cusp1), gam in PSL2(Z) */
     936                 :      27062 :       gam = path_to_zm(mkpath(cusp2, cusp1)); /* = [a2,a1;b2,b1] */
     937                 :            :       /* we have normalized the cusp representation so that a1 b2 - a2 b1 = 1 */
     938                 :      27062 :       b1 = coeff(gam,2,1); b2 = coeff(gam,2,2);
     939                 :            :       /* gam.1  = (a1 + a2) / (b1 + b2) */
     940                 :      27062 :       b = b1 + b2;
     941                 :            :       /* Determine whether the adjacent triangle *below* (cusp1->cusp2)
     942                 :            :        * should be added */
     943                 :      27062 :       pos = p1_index(b1,b2, p1N); /* did we see cl(gam) before ? */
     944         [ +  + ]:      27062 :       if (v[pos])
     945                 :      13587 :         cusp2[3] = type_X; /* NO */
     946                 :            :       else
     947                 :            :       { /* YES */
     948                 :            :         ulong B1, B2;
     949                 :      13475 :         v[pos] = position;
     950                 :      13475 :         i = p1_index(-(b1+b2), b1, p1N); v[i] = position+1;
     951                 :      13475 :         i = p1_index(b2, -(b1+b2), p1N); v[i] = position+2;
     952                 :            :         /* add cl(gam), cl(gam*TAU), cl(gam*TAU^2) to v */
     953                 :      13475 :         position += 3;
     954                 :            :         /* gam tau gam^(-1) in \Gamma ? */
     955                 :      13475 :         B1 = smodss(b1, N);
     956                 :      13475 :         B2 = smodss(b2, N);
     957         [ +  + ]:      13475 :         if ((Fl_sqr(B2,N) + Fl_sqr(B1,N) + Fl_mul(B1,B2,N)) % N == 0)
     958                 :         42 :           cusp2[3] = type_T;
     959                 :            :         else
     960                 :            :         {
     961                 :      13433 :           long a1 = coeff(gam, 1,1), a2 = coeff(gam, 1,2);
     962                 :      13433 :           long a = a1 + a2; /* gcd(a,b) = 1 */
     963                 :      13433 :           list_insert(c, mkvecsmall3(a,b,type_DO));
     964                 :      13433 :           c = c->next;
     965                 :      13433 :           nbC++;
     966                 :      13433 :           done = 0;
     967                 :            :         }
     968                 :            :       }
     969                 :            :     }
     970         [ +  + ]:        980 :     if (done) break;
     971                 :        784 :   }
     972                 :        196 :   L = cgetg(nbC+1, t_VEC); i = 1;
     973         [ +  + ]:      14021 :   for (c = C; c; c = c->next) gel(L,i++) = c->data;
     974                 :        196 :   return gerepilecopy(av, L);
     975                 :            : }
     976                 :            : 
     977                 :            : /* M in PSL2(Z). Return index of M in P1^(Z/NZ) = Gamma0(N) \ PSL2(Z),
     978                 :            :  * and M0 in Gamma_0(N) such that M = M0 * M', where M' = chosen
     979                 :            :  * section( PSL2(Z) -> P1^(Z/NZ) ). */
     980                 :            : static GEN
     981                 :       3052 : Gamma0N_decompose(GEN W, GEN M, long *index)
     982                 :            : {
     983                 :       3052 :   GEN p1N = gel(W,1), W3 = gel(W,3), section = ms_get_section(W);
     984                 :       3052 :   ulong N = p1N_get_N(p1N);
     985                 :       3052 :   ulong c = umodiu(gcoeff(M,2,1), N);
     986                 :       3052 :   ulong d = umodiu(gcoeff(M,2,2), N);
     987                 :       3052 :   long ind = p1_index(c, d, p1N); /* as an elt of P1(Z/NZ) */
     988                 :       3052 :   *index = W3[ind]; /* as an elt of F, E2, ... */
     989                 :       3052 :   return ZM_zm_mul(M, sl2_inv(gel(section,ind)));
     990                 :            : }
     991                 :            : /* same for a path. Return [[ind], M] */
     992                 :            : static GEN
     993                 :      28042 : path_Gamma0N_decompose(GEN W, GEN path)
     994                 :            : {
     995                 :      28042 :   GEN p1N = gel(W,1);
     996                 :      28042 :   GEN p1index_to_ind = gel(W,3);
     997                 :      28042 :   GEN section = ms_get_section(W);
     998                 :      28042 :   GEN M = path_to_zm(path);
     999                 :      28042 :   long p1index = p1_index(cc(M), dd(M), p1N);
    1000                 :      28042 :   long ind = p1index_to_ind[p1index];
    1001                 :      28042 :   GEN M0 = ZM_zm_mul(zm_to_ZM(M), sl2_inv(gel(section,p1index)));
    1002                 :      28042 :   return mkvec2(mkvecsmall(ind), M0);
    1003                 :            : }
    1004                 :            : 
    1005                 :            : /* Input:  [v1, ..., vn]
    1006                 :            :  * Output: [vn, ..., v1]. Shallow function */
    1007                 :            : static GEN
    1008                 :      41083 : Reverse(GEN v)
    1009                 :            : {
    1010                 :            :   long i, l;
    1011                 :      41083 :   GEN w = cgetg_copy(v, &l);
    1012         [ +  + ]:     123249 :   for (i = 1; i < l; i++) gel(w, i) = gel(v, l-i);
    1013                 :      41083 :   return w;
    1014                 :            : }
    1015                 :            : 
    1016                 :            : /*Form generators of H_1(X_0(N),{cusps},Z)
    1017                 :            : *
    1018                 :            : *Input: N = integer > 1, p1N = P^1(Z/NZ)
    1019                 :            : *Output: [cusp_list,E,F,T2,T3,E1] where
    1020                 :            : *  cusps_list = list of cusps describing fundamental domain of
    1021                 :            : *    \Gamma_0(N).
    1022                 :            : *  E = list of paths in the boundary of the fundamental domains and oriented
    1023                 :            : *    clockwise such that they do not contain a point
    1024                 :            : *    fixed by an element of order 2 and they are not an edge of a
    1025                 :            : *    triangle containing a fixed point of an element of order 3
    1026                 :            : *  F = list of paths in the interior of the domain with each
    1027                 :            : *    orientation appearing separately
    1028                 :            : * T2 = list of paths in the boundary of domain containing a point fixed
    1029                 :            : *    by an element of order 2 (oriented clockwise)
    1030                 :            : * T3 = list of paths in the boundard of domain which are the edges of
    1031                 :            : *    some triangle containing a fixed point of a matrix of order 3 (both
    1032                 :            : *    orientations appear)
    1033                 :            : * E1 = a sublist of E such that every path in E is \Gamma_0(N)-equivalent to
    1034                 :            : *    either an element of E1 or the flip (reversed orientation) of an element
    1035                 :            : *    of E1.
    1036                 :            : * (Elements of T2 are \Gamma_0(N)-equivalent to their own flip.)
    1037                 :            : *
    1038                 :            : * sec = a list from 1..#p1N of matrices describing a section of the map
    1039                 :            : *   SL_2(Z) to P^1(Z/NZ) given by [a,b;c,d]-->[c,d].
    1040                 :            : *   Given our fixed enumeration of P^1(Z/NZ), the j-th element of the list
    1041                 :            : *   represents the image of the j-th element of P^1(Z/NZ) under the section. */
    1042                 :            : 
    1043                 :            : /* insert path in set T */
    1044                 :            : static void
    1045                 :      40929 : set_insert(hashtable *T, GEN path)
    1046                 :      40929 : { hash_insert(T, path,  (void*)(T->nb + 1)); }
    1047                 :            : 
    1048                 :            : static GEN
    1049                 :       1764 : hash_to_vec(hashtable *h)
    1050                 :            : {
    1051                 :       1764 :   GEN v = cgetg(h->nb + 1, t_VEC);
    1052                 :            :   ulong i;
    1053         [ +  + ]:     310632 :   for (i = 0; i < h->len; i++)
    1054                 :            :   {
    1055                 :     308868 :     hashentry *e = h->table[i];
    1056         [ +  + ]:     376782 :     while (e)
    1057                 :            :     {
    1058                 :      67914 :       GEN key = (GEN)e->key;
    1059                 :      67914 :       long index = (long)e->val;
    1060                 :      67914 :       gel(v, index) = key;
    1061                 :      67914 :       e = e->next;
    1062                 :            :     }
    1063                 :            :   }
    1064                 :       1764 :   return v;
    1065                 :            : }
    1066                 :            : 
    1067                 :            : static long
    1068                 :      20853 : path_to_p1_index(GEN path, GEN p1N)
    1069                 :            : {
    1070                 :      20853 :   GEN M = path_to_zm(path);
    1071                 :      20853 :   return p1_index(cc(M), dd(M), p1N);
    1072                 :            : }
    1073                 :            : 
    1074                 :            : /* Pollack-Stevens sets */
    1075                 :            : typedef struct PS_sets_t {
    1076                 :            :   hashtable *F, *T2, *T31, *T32, *E1, *E2;
    1077                 :            :   GEN E2_in_terms_of_E1, stdE1;
    1078                 :            : } PS_sets_t;
    1079                 :            : 
    1080                 :            : static hashtable *
    1081                 :       1176 : set_init(long max)
    1082                 :       1176 : { return hash_create(max, (ulong(*)(void*))&hash_GEN,
    1083                 :            :                           (int(*)(void*,void*))&gidentical, 1); }
    1084                 :            : static void
    1085                 :      13902 : insert_E(GEN path, PS_sets_t *S, GEN p1N)
    1086                 :            : {
    1087                 :      13902 :   GEN rev = Reverse(path);
    1088                 :      13902 :   long std = path_to_p1_index(rev, p1N);
    1089                 :      13902 :   GEN v = gel(S->stdE1, std);
    1090         [ +  + ]:      13902 :   if (v)
    1091                 :            :   { /* [s, p1], where E1[s] = the path p1 \equiv Reverse(path) mod \Gamma */
    1092                 :       6951 :     GEN gamma, p1 = gel(v,2);
    1093                 :       6951 :     long r, s = itos(gel(v,1));
    1094                 :            : 
    1095                 :       6951 :     set_insert(S->E2, path);
    1096                 :       6951 :     r = S->E2->nb;
    1097         [ -  + ]:       6951 :     if (gel(S->E2_in_terms_of_E1, r) != gen_0) pari_err_BUG("insert_E");
    1098                 :            : 
    1099                 :       6951 :     gamma = gamma_equiv_matrix(rev, p1);
    1100                 :            :     /* E2[r] + gamma * E1[s] = 0 */
    1101                 :       6951 :     gel(S->E2_in_terms_of_E1, r) = mkvec2(utoipos(s),
    1102                 :            :                                           to_famat_shallow(gamma,gen_m1));
    1103                 :            :   }
    1104                 :            :   else
    1105                 :            :   {
    1106                 :       6951 :     set_insert(S->E1, path);
    1107                 :       6951 :     std = path_to_p1_index(path, p1N);
    1108                 :       6951 :     gel(S->stdE1, std) = mkvec2(utoipos(S->E1->nb), path);
    1109                 :            :   }
    1110                 :      13902 : }
    1111                 :            : 
    1112                 :            : static GEN
    1113                 :        784 : cusp_infinity() { return mkvecsmall2(1,0); }
    1114                 :            : 
    1115                 :            : static void
    1116                 :        196 : form_E_F_T(ulong N, GEN p1N, GEN *pC, PS_sets_t *S)
    1117                 :            : {
    1118                 :        196 :   GEN C, cusp_list = form_list_of_cusps(N, p1N);
    1119                 :        196 :   long nbgen = lg(cusp_list)-1, nbmanin = p1_size(p1N), r, s, i;
    1120                 :            :   hashtable *F, *T2, *T31, *T32, *E1, *E2;
    1121                 :            : 
    1122                 :        196 :   *pC = C = cgetg(nbgen+1, t_VEC);
    1123         [ +  + ]:      14021 :   for (i = 1; i <= nbgen; i++)
    1124                 :            :   {
    1125                 :      13825 :     GEN c = gel(cusp_list,i);
    1126                 :      13825 :     gel(C,i) = mkvecsmall2(c[1], c[2]);
    1127                 :            :   }
    1128                 :        196 :   S->F  = F  = set_init(nbmanin);
    1129                 :        196 :   S->E1 = E1 = set_init(nbgen);
    1130                 :        196 :   S->E2 = E2 = set_init(nbgen);
    1131                 :        196 :   S->T2 = T2 = set_init(nbgen);
    1132                 :        196 :   S->T31 = T31 = set_init(nbgen);
    1133                 :        196 :   S->T32 = T32 = set_init(nbgen);
    1134                 :            : 
    1135                 :            :   /* T31 represents the three torsion paths going from left to right */
    1136                 :            :   /* T32 represents the three torsion paths going from right to left */
    1137         [ +  + ]:      13825 :   for (r = 1; r < nbgen; r++)
    1138                 :            :   {
    1139                 :      13629 :     GEN c2 = gel(cusp_list,r+1);
    1140         [ +  + ]:      13629 :     if (c2[3] == type_T)
    1141                 :            :     {
    1142                 :         42 :       GEN c1 = gel(cusp_list,r), path = mkpath(c1,c2), path2 = Reverse(path);
    1143                 :         42 :       set_insert(T31, path);
    1144                 :         42 :       set_insert(T32, path2);
    1145                 :            :     }
    1146                 :            :   }
    1147                 :            : 
    1148                 :            :   /* to record relations between E2 and E1 */
    1149                 :        196 :   S->E2_in_terms_of_E1 = zerovec(nbgen);
    1150                 :        196 :   S->stdE1 = const_vec(nbmanin, NULL);
    1151                 :            : 
    1152                 :            :   /* Assumption later: path [oo,0] is E1[1], path [1,oo] is E2[1] */
    1153                 :            :   {
    1154                 :        196 :     GEN oo = cusp_infinity();
    1155                 :        196 :     GEN p1 = mkpath(oo, mkvecsmall2(0,1)); /* [oo, 0] */
    1156                 :        196 :     GEN p2 = mkpath(mkvecsmall2(1,1), oo); /* [1, oo] */
    1157                 :        196 :     insert_E(p1, S, p1N);
    1158                 :        196 :     insert_E(p2, S, p1N);
    1159                 :            :   }
    1160                 :            : 
    1161         [ +  + ]:      13825 :   for (r = 1; r < nbgen; r++)
    1162                 :            :   {
    1163                 :      13629 :     GEN c1 = gel(cusp_list,r);
    1164         [ +  + ]:    3264695 :     for (s = r+1; s <= nbgen; s++)
    1165                 :            :     {
    1166                 :    3251066 :       pari_sp av = avma;
    1167                 :    3251066 :       GEN c2 = gel(cusp_list,s), path;
    1168                 :    3251066 :       GEN d = subii(mulss(c1[1],c2[2]), mulss(c1[2],c2[1]));
    1169                 :    3251066 :       avma = av;
    1170         [ +  + ]:    3251066 :       if (!is_pm1(d)) continue;
    1171                 :            : 
    1172                 :      27062 :       path = mkpath(c1,c2);
    1173         [ +  + ]:      27062 :       if (r+1 == s)
    1174                 :            :       {
    1175                 :      13629 :         GEN w = path;
    1176                 :      13629 :         ulong hash = T31->hash(w); /* T31, T32 use the same hash function */
    1177 [ +  + ][ +  - ]:      13629 :         if (!hash_search2(T31, w, hash) && !hash_search2(T32, w, hash))
    1178                 :            :         {
    1179         [ +  + ]:      13587 :           if (gamma_equiv(path, Reverse(path), N))
    1180                 :         77 :             set_insert(T2, path);
    1181                 :            :           else
    1182                 :      13510 :             insert_E(path, S, p1N);
    1183                 :            :         }
    1184                 :            :       } else {
    1185                 :      13433 :         set_insert(F, mkvec2(path, mkvecsmall2(r,s)));
    1186                 :      13433 :         set_insert(F, mkvec2(Reverse(path), mkvecsmall2(s,r)));
    1187                 :            :       }
    1188                 :            :     }
    1189                 :            :   }
    1190                 :        196 :   setlg(S->E2_in_terms_of_E1, E2->nb+1);
    1191                 :        196 : }
    1192                 :            : 
    1193                 :            : /* v = \sum n_i g_i, return \sum n_i g_i^(-1) */
    1194                 :            : static GEN
    1195                 :       2555 : ZGl2Q_star(GEN v)
    1196                 :            : {
    1197                 :            :   long i, l;
    1198                 :            :   GEN w, G;
    1199         [ +  + ]:       2555 :   if (typ(v) == t_INT) return v;
    1200                 :       1589 :   G = gel(v,1);
    1201                 :       1589 :   w = cgetg_copy(G, &l);
    1202         [ +  + ]:       4081 :   for (i = 1; i < l; i++)
    1203                 :            :   {
    1204                 :       2492 :     GEN g = gel(G,i);
    1205         [ +  + ]:       2492 :     if (typ(g) == t_MAT) g = SL2_inv(g);
    1206                 :       2492 :     gel(w,i) = g;
    1207                 :            :   }
    1208                 :       2555 :   return ZG_normalize(mkmat2(w, gel(v,2)));
    1209                 :            : }
    1210                 :            : static GEN
    1211                 :        903 : ZGl2QC_star(GEN v)
    1212                 :            : {
    1213                 :            :   long i, l;
    1214                 :        903 :   GEN w = cgetg_copy(v, &l);
    1215         [ +  + ]:       3234 :   for (i = 1; i < l; i++) gel(w,i) = ZGl2Q_star(gel(v,i));
    1216                 :        903 :   return w;
    1217                 :            : }
    1218                 :            : 
    1219                 :            : /* Input: h = set of unimodular paths, p1N = P^1(Z/NZ) = Gamma_0(N)\PSL2(Z)
    1220                 :            :  * Output: Each path is converted to a matrix and then an element of P^1(Z/NZ)
    1221                 :            :  * Append the matrix to W[12], append the index that represents
    1222                 :            :  * these elements of P^1 (the classes mod Gamma_0(N) via our fixed
    1223                 :            :  * enumeration to W[2]. */
    1224                 :            : static void
    1225                 :       1176 : paths_decompose(GEN W, hashtable *h, int flag)
    1226                 :            : {
    1227                 :       1176 :   GEN p1N = ms_get_p1N(W), section = ms_get_section(W);
    1228                 :       1176 :   GEN v = hash_to_vec(h);
    1229                 :       1176 :   long i, l = lg(v);
    1230         [ +  + ]:      42105 :   for (i = 1; i < l; i++)
    1231                 :            :   {
    1232                 :      40929 :     GEN e = gel(v,i);
    1233         [ +  + ]:      40929 :     GEN M = path_to_zm(flag? gel(e,1): e);
    1234                 :      40929 :     long index = p1_index(cc(M), dd(M), p1N);
    1235                 :      40929 :     vecsmalltrunc_append(gel(W,2), index);
    1236                 :      40929 :     gel(section, index) = M;
    1237                 :            :   }
    1238                 :       1176 : }
    1239                 :            : static void
    1240                 :        196 : fill_W2_W12(GEN W, PS_sets_t *S)
    1241                 :            : {
    1242                 :        196 :   GEN p1N = gel(W,1);
    1243                 :        196 :   long n = p1_size(p1N);
    1244                 :        196 :   gel(W, 2) = vecsmalltrunc_init(n+1);
    1245                 :        196 :   gel(W,12) = cgetg(n+1, t_VEC);
    1246                 :            :   /* F contains [path, [index cusp1, index cusp2]]. Others contain paths only */
    1247                 :        196 :   paths_decompose(W, S->F, 1);
    1248                 :        196 :   paths_decompose(W, S->E2, 0);
    1249                 :        196 :   paths_decompose(W, S->T32, 0);
    1250                 :        196 :   paths_decompose(W, S->E1, 0);
    1251                 :        196 :   paths_decompose(W, S->T2, 0);
    1252                 :        196 :   paths_decompose(W, S->T31, 0);
    1253                 :        196 : }
    1254                 :            : 
    1255                 :            : /* x t_VECSMALL, corresponds to a map x(i) = j, where 1 <= j <= max for all i
    1256                 :            :  * Return y s.t. y[j] = i or 0 (not in image) */
    1257                 :            : static GEN
    1258                 :        392 : reverse_list(GEN x, long max)
    1259                 :            : {
    1260                 :        392 :   GEN y = const_vecsmall(max, 0);
    1261                 :        392 :   long r, lx = lg(x);
    1262         [ +  + ]:      48391 :   for (r = 1; r < lx; r++) y[ x[r] ] = r;
    1263                 :        392 :   return y;
    1264                 :            : }
    1265                 :            : 
    1266                 :            : /* go from C[a] to C[b]; return the indices of paths
    1267                 :            :  * E.g. if a < b
    1268                 :            :  *   (C[a]->C[a+1], C[a+1]->C[a+2], ... C[b-1]->C[b])
    1269                 :            :  * (else reverse direction)
    1270                 :            :  * = b - a paths */
    1271                 :            : static GEN
    1272                 :      26474 : F_indices(GEN W, long a, long b)
    1273                 :            : {
    1274                 :      26474 :   GEN v = cgetg(labs(b-a) + 1, t_VEC);
    1275                 :      26474 :   long s, k = 1;
    1276         [ +  + ]:      26474 :   if (a < b) {
    1277                 :      13237 :     GEN index_forward = gel(W,13);
    1278         [ +  + ]:     111951 :     for (s = a; s < b; s++) gel(v,k++) = gel(index_forward,s);
    1279                 :            :   } else {
    1280                 :      13237 :     GEN index_backward = gel(W,14);
    1281         [ +  + ]:     111951 :     for (s = a; s > b; s--) gel(v,k++) = gel(index_backward,s);
    1282                 :            :   }
    1283                 :      26474 :   return v;
    1284                 :            : }
    1285                 :            : /* go from C[a] to C[b] via oo; return the indices of paths
    1286                 :            :  * E.g. if a < b
    1287                 :            :  *   (C[a]->C[a-1], ... C[2]->C[1],
    1288                 :            :  *    C[1]->oo, oo-> C[end],
    1289                 :            :  *    C[end]->C[end-1], ... C[b+1]->C[b])
    1290                 :            :  *  a-1 + 2 + end-(b+1)+1 = end - b + a + 1 paths  */
    1291                 :            : static GEN
    1292                 :        392 : F_indices_oo(GEN W, long end, long a, long b)
    1293                 :            : {
    1294                 :        392 :   GEN index_oo = gel(W,15);
    1295                 :        392 :   GEN v = cgetg(end-labs(b-a)+1 + 1, t_VEC);
    1296                 :        392 :   long s, k = 1;
    1297                 :            : 
    1298         [ +  + ]:        392 :   if (a < b) {
    1299                 :        196 :     GEN index_backward = gel(W,14);
    1300         [ -  + ]:        196 :     for (s = a; s > 1; s--) gel(v,k++) = gel(index_backward,s);
    1301                 :        196 :     gel(v,k++) = gel(index_backward,1); /* C[1] -> oo */
    1302                 :        196 :     gel(v,k++) = gel(index_oo,2); /* oo -> C[end] */
    1303         [ +  + ]:       5740 :     for (s = end; s > b; s--) gel(v,k++) = gel(index_backward,s);
    1304                 :            :   } else {
    1305                 :        196 :     GEN index_forward = gel(W,13);
    1306         [ +  + ]:       5740 :     for (s = a; s < end; s++) gel(v,k++) = gel(index_forward,s);
    1307                 :        196 :     gel(v,k++) = gel(index_forward,end); /* C[end] -> oo */
    1308                 :        196 :     gel(v,k++) = gel(index_oo,1); /* oo -> C[1] */
    1309         [ -  + ]:        196 :     for (s = 1; s < b; s++) gel(v,k++) = gel(index_forward,s);
    1310                 :            :   }
    1311                 :        392 :   return v;
    1312                 :            : }
    1313                 :            : /* index of oo -> C[1], oo -> C[end] */
    1314                 :            : static GEN
    1315                 :        196 : indices_oo(GEN W, GEN C)
    1316                 :            : {
    1317                 :        196 :   long end = lg(C)-1;
    1318                 :        196 :   GEN w, v = cgetg(2+1, t_VEC), oo = cusp_infinity();
    1319                 :        196 :   w = mkpath(oo, gel(C,1)); /* oo -> C[1]=0 */
    1320                 :        196 :   gel(v,1) = path_Gamma0N_decompose(W, w);
    1321                 :        196 :   w = mkpath(oo, gel(C,end)); /* oo -> C[end]=1 */
    1322                 :        196 :   gel(v,2) = path_Gamma0N_decompose(W, w);
    1323                 :        196 :   return v;
    1324                 :            : }
    1325                 :            : 
    1326                 :            : /* index of C[1]->C[2], C[2]->C[3], ... C[end-1]->C[end], C[end]->oo
    1327                 :            :  * Recall that C[1] = 0, C[end] = 1 */
    1328                 :            : static GEN
    1329                 :        196 : indices_forward(GEN W, GEN C)
    1330                 :            : {
    1331                 :        196 :   long s, k = 1, end = lg(C)-1;
    1332                 :        196 :   GEN v = cgetg(end+1, t_VEC);
    1333         [ +  + ]:      14021 :   for (s = 1; s <= end; s++)
    1334                 :            :   {
    1335         [ +  + ]:      13825 :     GEN w = mkpath(gel(C,s), s == end? cusp_infinity(): gel(C,s+1));
    1336                 :      13825 :     gel(v,k++) = path_Gamma0N_decompose(W, w);
    1337                 :            :   }
    1338                 :        196 :   return v;
    1339                 :            : }
    1340                 :            : /* index of C[1]->oo, C[2]->C[1], ... C[end]->C[end-1] */
    1341                 :            : static GEN
    1342                 :        196 : indices_backward(GEN W, GEN C)
    1343                 :            : {
    1344                 :        196 :   long s, k = 1, end = lg(C)-1;
    1345                 :        196 :   GEN v = cgetg(end+1, t_VEC);
    1346         [ +  + ]:      14021 :   for (s = 1; s <= end; s++)
    1347                 :            :   {
    1348         [ +  + ]:      13825 :     GEN w = mkpath(gel(C,s), s == 1? cusp_infinity(): gel(C,s-1));
    1349                 :      13825 :     gel(v,k++) = path_Gamma0N_decompose(W, w);
    1350                 :            :   }
    1351                 :        196 :   return v;
    1352                 :            : }
    1353                 :            : 
    1354                 :            : /* N = integer > 1. Returns data describing Delta_0 = Z[P^1(Q)]_0 seen as
    1355                 :            :  * a Gamma_0(N) - module. */
    1356                 :            : static GEN
    1357                 :        196 : msinit_N(ulong N)
    1358                 :            : {
    1359                 :        196 :   GEN p1N = create_p1mod(N);
    1360                 :            :   GEN C, vecF, vecT2, vecT31;
    1361                 :            :   ulong r, s, width;
    1362                 :        196 :   long nball, nbgen, nbp1N = p1_size(p1N);
    1363                 :        196 :   GEN TAU = mkmat2(mkcol2(gen_0,gen_1), mkcol2(gen_m1,gen_m1)); /*[0,-1;1,-1]*/
    1364                 :            :   GEN W, W2, singlerel, annT2, annT31;
    1365                 :            :   GEN F_index;
    1366                 :            :   hashtable *F, *T2, *T31, *T32, *E1, *E2;
    1367                 :            :   PS_sets_t S;
    1368                 :            : 
    1369                 :        196 :   form_E_F_T(N,p1N, &C, &S);
    1370                 :        196 :   E1  = S.E1;
    1371                 :        196 :   E2  = S.E2;
    1372                 :        196 :   T31 = S.T31;
    1373                 :        196 :   T32 = S.T32;
    1374                 :        196 :   F   = S.F;
    1375                 :        196 :   T2  = S.T2;
    1376                 :        196 :   nbgen = lg(C)-1;
    1377                 :            : 
    1378                 :        196 :   W = cgetg(17, t_VEC);
    1379                 :        196 :   gel(W,1) = p1N;
    1380                 :            : 
    1381                 :            :  /* Put our paths in the order: F,E2,T32,E1,T2,T31
    1382                 :            :   * W2[j] associates to the j-th element of this list its index in P1. */
    1383                 :        196 :   fill_W2_W12(W, &S);
    1384                 :        196 :   W2 = gel(W, 2);
    1385                 :        196 :   nball = lg(W2)-1;
    1386                 :        196 :   gel(W,3) = reverse_list(W2, nbp1N);
    1387                 :            : 
    1388                 :        196 :   gel(W,5) = vecslice(gel(W,2), F->nb + E2->nb + T32->nb + 1, nball);
    1389                 :        196 :   gel(W,4) = reverse_list(gel(W,5), nbp1N);
    1390                 :        196 :   gel(W,13) = indices_forward(W, C);
    1391                 :        196 :   gel(W,14) = indices_backward(W, C);
    1392                 :        196 :   gel(W,15) = indices_oo(W, C);
    1393                 :        196 :   gel(W,11) = mkvecsmall5(F->nb,
    1394                 :        196 :                           F->nb + E2->nb,
    1395                 :        196 :                           F->nb + E2->nb + T32->nb,
    1396                 :        196 :                           F->nb + E2->nb + T32->nb + E1->nb,
    1397                 :        196 :                           F->nb + E2->nb + T32->nb + E1->nb + T2->nb);
    1398                 :            : 
    1399                 :            :   /* relations between T32 and T31 [not stored!]
    1400                 :            :    * T32[i] = - T31[i] */
    1401                 :            : 
    1402                 :            :   /* relations of F */
    1403                 :        196 :   width = E1->nb + T2->nb + T31->nb;
    1404                 :            :   /* F_index[r] = [index_1, ..., index_k], where index_i is the p1_index()
    1405                 :            :    * of the elementary unimodular path between 2 consecutive cusps
    1406                 :            :    * [in E1,E2,T2,T31 or T32] */
    1407                 :        196 :   F_index = cgetg(F->nb+1, t_VEC);
    1408                 :        196 :   vecF = hash_to_vec(F);
    1409         [ +  + ]:      27062 :   for (r = 1; r <= F->nb; r++)
    1410                 :            :   {
    1411                 :      26866 :     GEN w = gel(gel(vecF,r), 2);
    1412                 :      26866 :     long a = w[1], b = w[2], d = labs(b - a);
    1413                 :            :     /* c1 = cusp_list[a],  c2 = cusp_list[b], ci != oo */
    1414                 :      53732 :     gel(F_index,r) = (nbgen-d >= d-1)? F_indices(W, a,b)
    1415         [ +  + ]:      26866 :                                      : F_indices_oo(W, lg(C)-1,a,b);
    1416                 :            :   }
    1417                 :            : 
    1418                 :        196 :   singlerel = cgetg(width+1, t_VEC);
    1419                 :            :   /* form the single boundary relation */
    1420         [ +  + ]:       7147 :   for (s = 1; s <= E2->nb; s++)
    1421                 :            :   {
    1422                 :       6951 :     GEN data = gel(S.E2_in_terms_of_E1,s);
    1423                 :       6951 :     long c = itos(gel(data,1));
    1424                 :       6951 :     GEN u = gel(data,2); /* E2[s] = u * E1[c], u = - [gamma] */
    1425                 :       6951 :     GEN gamma = gcoeff(u,1,1);
    1426                 :       6951 :     gel(singlerel, c) = mkmat2(mkcol2(gen_1,gamma),mkcol2(gen_1,gen_m1));
    1427                 :            :   }
    1428         [ +  + ]:        315 :   for (r = E1->nb + 1; r <= width; r++) gel(singlerel, r) = gen_1;
    1429                 :            : 
    1430                 :            :   /* form the 2-torsion relations */
    1431                 :        196 :   annT2 = cgetg(T2->nb+1, t_VEC);
    1432                 :        196 :   vecT2 = hash_to_vec(T2);
    1433         [ +  + ]:        273 :   for (r = 1; r <= T2->nb; r++)
    1434                 :            :   {
    1435                 :         77 :     GEN w = gel(vecT2,r);
    1436                 :         77 :     GEN gamma = gamma_equiv_matrix(Reverse(w), w);
    1437                 :         77 :     gel(annT2, r) = mkmat2(mkcol2(gen_1,gamma), mkcol2(gen_1,gen_1));
    1438                 :            :   }
    1439                 :            : 
    1440                 :            :   /* form the 3-torsion relations */
    1441                 :        196 :   annT31 = cgetg(T31->nb+1, t_VEC);
    1442                 :        196 :   vecT31 = hash_to_vec(T31);
    1443         [ +  + ]:        238 :   for (r = 1; r <= T31->nb; r++)
    1444                 :            :   {
    1445                 :         42 :     GEN M = zm_to_ZM( path_to_zm( Reverse(gel(vecT31,r)) ) );
    1446                 :         42 :     GEN gamma = ZM_mul(ZM_mul(M, TAU), SL2_inv(M));
    1447                 :         42 :     gel(annT31, r) = mkmat2(mkcol3(gen_1,gamma,ZM_sqr(gamma)),
    1448                 :            :                             mkcol3(gen_1,gen_1,gen_1));
    1449                 :            :   }
    1450                 :        196 :   gel(W,6) = F_index;
    1451                 :        196 :   gel(W,7) = S.E2_in_terms_of_E1;
    1452                 :        196 :   gel(W,8) = annT2;
    1453                 :        196 :   gel(W,9) = annT31;
    1454                 :        196 :   gel(W,10)= singlerel;
    1455                 :        196 :   gel(W,16)= inithashcusps(p1N);
    1456                 :        196 :   return W;
    1457                 :            : }
    1458                 :            : static GEN
    1459                 :      66080 : cusp_to_frac(GEN c) { return gdivgs(stoi(c[1]), c[2]); }
    1460                 :            : static GEN
    1461         [ +  + ]:         28 : cusp_to_P1Q(GEN c) { return c[2]? gdivgs(stoi(c[1]), c[2]): mkoo(); }
    1462                 :            : GEN
    1463                 :          7 : mspathgens(GEN W)
    1464                 :            : {
    1465                 :          7 :   pari_sp av = avma;
    1466                 :            :   long i,j, l, nbE1, nbT2, nbT31;
    1467                 :            :   GEN R, r, g, section, gen, annT2, annT31, singlerel;
    1468                 :          7 :   checkms(W); W = get_ms(W);
    1469                 :          7 :   section = ms_get_section(W);
    1470                 :          7 :   gen = ms_get_genindex(W);
    1471                 :          7 :   l = lg(gen);
    1472                 :          7 :   g = cgetg(l,t_VEC);
    1473         [ +  + ]:         21 :   for (i=1; i<l; i++)
    1474                 :            :   {
    1475                 :         14 :     GEN p = gel(section,gen[i]);
    1476                 :         14 :     gel(g,i) = mkvec2(cusp_to_P1Q(gel(p,1)), cusp_to_P1Q(gel(p,2)));
    1477                 :            :   }
    1478                 :          7 :   nbE1 = ms_get_nbE1(W);
    1479                 :          7 :   annT2 = gel(W,8); nbT2 = lg(annT2)-1;
    1480                 :          7 :   annT31 = gel(W,9);nbT31 = lg(annT31)-1;
    1481                 :          7 :   singlerel = gel(W,10);
    1482                 :          7 :   R = cgetg(nbT2+nbT31+2, t_VEC);
    1483                 :          7 :   l = lg(singlerel);
    1484                 :          7 :   r = cgetg(l, t_VEC);
    1485         [ +  + ]:         14 :   for (i = 1; i <= nbE1; i++)
    1486                 :          7 :     gel(r,i) = mkvec2(gel(singlerel, i), stoi(i));
    1487         [ +  + ]:         14 :   for (; i < l; i++)
    1488                 :          7 :     gel(r,i) = mkvec2(gen_1, stoi(i));
    1489                 :          7 :   gel(R,1) = r; j = 2;
    1490         [ +  + ]:         14 :   for (i = 1; i <= nbT2; i++,j++)
    1491                 :          7 :     gel(R,j) = mkvec( mkvec2(gel(annT2,i), stoi(i + nbE1)) );
    1492         [ -  + ]:          7 :   for (i = 1; i <= nbT31; i++,j++)
    1493                 :          0 :     gel(R,j) = mkvec( mkvec2(gel(annT31,i), stoi(i + nbE1 + nbT2)) );
    1494                 :          7 :   return gerepilecopy(av, mkvec2(g,R));
    1495                 :            : }
    1496                 :            : 
    1497                 :            : /* Modular symbols in weight k: Hom_Gamma(Delta, Q[x,y]_{k-2}) */
    1498                 :            : /* A symbol phi is represented by the {phi(g_i)}, {phi(g'_i)}, {phi(g''_i)}
    1499                 :            :  * where the {g_i, g'_i, g''_i} are the Z[\Gamma]-generators of Delta,
    1500                 :            :  * g_i corresponds to E1, g'_i to T2, g''_i to T31.
    1501                 :            :  */
    1502                 :            : 
    1503                 :            : /* FIXME: export. T^1, ..., T^n */
    1504                 :            : static GEN
    1505                 :       6496 : RgX_powers(GEN T, long n)
    1506                 :            : {
    1507                 :       6496 :   GEN v = cgetg(n+1, t_VEC);
    1508                 :            :   long i;
    1509                 :       6496 :   gel(v, 1) = T;
    1510         [ +  + ]:      18872 :   for (i = 1; i < n; i++) gel(v,i+1) = RgX_mul(gel(v,i), T);
    1511                 :       6496 :   return v;
    1512                 :            : }
    1513                 :            : 
    1514                 :            : /* g = [a,b;c,d]. Return (X^{k-2} | g)(X,Y)[X = 1]. */
    1515                 :            : static GEN
    1516                 :        112 : voo_act_Gl2Q(GEN g, long k)
    1517                 :            : {
    1518                 :        112 :   GEN c = gcoeff(g,2,1), d = gcoeff(g,2,2);
    1519                 :        112 :   return RgX_to_RgC(gpowgs(deg1pol_shallow(gneg(c), d, 0), k-2), k-1);
    1520                 :            : }
    1521                 :            : 
    1522                 :            : /* g = [a,b;c,d]. Return (P | g)(X,Y)[X = 1] = P(dX - cY, -b X + aY)[X = 1],
    1523                 :            :  * for P = X^{k-2}, X_^{k-3}Y, ..., Y^{k-2} */
    1524                 :            : GEN
    1525                 :       3248 : RgX_act_Gl2Q(GEN g, long k)
    1526                 :            : {
    1527                 :       3248 :   GEN a = gcoeff(g,1,1), b = gcoeff(g,1,2);
    1528                 :       3248 :   GEN c = gcoeff(g,2,1), d = gcoeff(g,2,2);
    1529                 :       3248 :   GEN V1 = RgX_powers(deg1pol_shallow(gneg(c), d, 0), k-2);
    1530                 :       3248 :   GEN V2 = RgX_powers(deg1pol_shallow(a, gneg(b), 0), k-2);
    1531                 :       3248 :   GEN V = cgetg(k, t_MAT);
    1532                 :            :   long i;
    1533                 :       3248 :   gel(V,1)   = RgX_to_RgC(gel(V1, k-2), k-1);
    1534         [ +  + ]:       9436 :   for (i = 1; i < k-2; i++)
    1535                 :            :   {
    1536                 :       6188 :     GEN v1 = gel(V1, k-2-i); /* (d-cY)^(k-2-i) */
    1537                 :       6188 :     GEN v2 = gel(V2, i); /* (-b+aY)^i */
    1538                 :       6188 :     gel(V,i+1) = RgX_to_RgC(RgX_mul(v1,v2), k-1);
    1539                 :            :   }
    1540                 :       3248 :   gel(V,k-1) = RgX_to_RgC(gel(V2, k-2), k-1);
    1541                 :       3248 :   return V; /* V[i+1] = X^i | g */
    1542                 :            : }
    1543                 :            : /* z in Z[\Gamma], return the matrix of z acting on (X^{k-2},...,Y^{k-2}) */
    1544                 :            : GEN
    1545                 :       2555 : RgX_act_ZGl2Q(GEN z, long k)
    1546                 :            : {
    1547                 :            :   long l, j;
    1548                 :       2555 :   GEN S = NULL, G, E;
    1549         [ -  + ]:       2555 :   if (typ(z) == t_INT) return matid(k-1);
    1550                 :       2555 :   G = gel(z,1); l = lg(G);
    1551                 :       2555 :   E = gel(z,2);
    1552         [ +  + ]:       6013 :   for (j = 1; j < l; j++)
    1553                 :            :   {
    1554                 :       3458 :     GEN M, g = gel(G,j), n = gel(E,j);
    1555         [ +  + ]:       3458 :     if (typ(g) == t_INT)
    1556                 :        210 :       M = scalarmat_shallow(n, k-1);
    1557                 :            :     else
    1558                 :            :     {
    1559                 :       3248 :       M = RgX_act_Gl2Q(g, k);
    1560         [ +  + ]:       3248 :       if (is_pm1(n))
    1561         [ +  + ]:       2282 :       { if (signe(n) < 0) M = RgM_neg(M);
    1562                 :            :       } else
    1563                 :        966 :         M = RgM_Rg_mul(M, n);
    1564                 :            :     }
    1565         [ +  + ]:       3458 :     S = j == 1? M: RgM_add(S, M);
    1566                 :            :   }
    1567                 :       2555 :   return S;
    1568                 :            : }
    1569                 :            : /* Given a vector of elements in Z[G], return it as vector of operators on V
    1570                 :            :  * (given by t_MAT) */
    1571                 :            : static GEN
    1572                 :        903 : ZGl2QC_to_act(GEN v, long k)
    1573                 :            : {
    1574                 :            :   long i, l;
    1575                 :        903 :   GEN w = cgetg_copy(v, &l);
    1576         [ +  + ]:       3234 :   for (i = 1; i < l; i++) gel(w,i) = RgX_act_ZGl2Q(gel(v,i), k);
    1577                 :        903 :   return w;
    1578                 :            : }
    1579                 :            : 
    1580                 :            : /* For all V[i] in Z[\Gamma], find the P such that  P . V[i]^* = 0;
    1581                 :            :  * write P in basis X^{k-2}, ..., Y^{k-2} */
    1582                 :            : static GEN
    1583                 :        126 : ZGV_tors(GEN V, long k)
    1584                 :            : {
    1585                 :        126 :   long i, l = lg(V);
    1586                 :        126 :   GEN v = cgetg(l, t_VEC);
    1587         [ +  + ]:        175 :   for (i = 1; i < l; i++)
    1588                 :            :   {
    1589                 :         49 :     GEN a = ZGl2Q_star(gel(V,i));
    1590                 :         49 :     gel(v,i) = ZM_ker(RgX_act_ZGl2Q(a,k));
    1591                 :            :   }
    1592                 :        126 :   return v;
    1593                 :            : }
    1594                 :            : 
    1595                 :            : static long
    1596                 :   90420680 : set_from_index(GEN W11, long i)
    1597                 :            : {
    1598         [ +  + ]:   90420680 :   if (i <= W11[1]) return 1;
    1599         [ +  + ]:   78064448 :   if (i <= W11[2]) return 2;
    1600         [ +  + ]:   48141989 :   if (i <= W11[3]) return 3;
    1601         [ +  + ]:   48098848 :   if (i <= W11[4]) return 4;
    1602         [ +  + ]:    3515603 :   if (i <= W11[5]) return 5;
    1603                 :   90420680 :   return 6;
    1604                 :            : }
    1605                 :            : 
    1606                 :            : static void
    1607                 :       4634 : treat_index(GEN W, GEN M, long index, int negate, GEN v)
    1608                 :            : {
    1609                 :       4634 :   GEN W11 = gel(W,11);
    1610                 :       4634 :   long shift = W11[3]; /* #F + #E2 + T32 */
    1611   [ +  +  +  + ]:       4634 :   switch(set_from_index(W11, index))
    1612                 :            :   {
    1613                 :            :     case 1: /*F*/
    1614                 :            :     {
    1615                 :        791 :       GEN F_index = gel(W,6), ind = gel(F_index, index);
    1616                 :        791 :       long j, l = lg(ind);
    1617         [ +  + ]:       2373 :       for (j = 1; j < l; j++)
    1618                 :            :       {
    1619                 :       1582 :         GEN IND = gel(ind,j), M0 = gel(IND,2);
    1620                 :       1582 :         long index = mael(IND,1,1);
    1621                 :       1582 :         treat_index(W, ZM_mul(M,M0), index, negate, v);
    1622                 :            :       }
    1623                 :        791 :       break;
    1624                 :            :     }
    1625                 :            : 
    1626                 :            :     case 2: /*E2, E2[r] + gamma * E1[s] = 0 */
    1627                 :            :     {
    1628                 :        819 :       long r = index - W11[1];
    1629                 :        819 :       GEN E2_in_terms_of_E1= gel(W,7), z = gel(E2_in_terms_of_E1, r);
    1630                 :        819 :       long s = itou(gel(z,1));
    1631                 :            : 
    1632                 :        819 :       index = s;
    1633                 :        819 :       M = G_ZG_mul(M, gel(z,2)); /* M * (-gamma) */
    1634                 :       1981 :       gel(v, index) = negate? ZG_sub(gel(v, index), M)
    1635         [ +  + ]:        819 :                             : ZG_add(gel(v, index), M);
    1636                 :        819 :       break;
    1637                 :            :     }
    1638                 :            : 
    1639                 :            :     case 3: /*T32, T32[i] = -T31[i] */
    1640                 :            :     {
    1641                 :         42 :       long T3shift = W11[5] - W11[2]; /* #T32 + #E1 + #T2 */
    1642                 :         42 :       index += T3shift;
    1643                 :         42 :       index -= shift;
    1644         [ +  + ]:         42 :       gel(v, index) = ZG_add(gel(v, index),
    1645                 :            :                              to_famat_shallow(M,negate?gen_1:gen_m1));
    1646                 :         42 :       break;
    1647                 :            :     }
    1648                 :            :     default: /*E1,T2,T31*/
    1649                 :       2982 :       index -= shift;
    1650         [ +  + ]:       2982 :       gel(v, index) = ZG_add(gel(v, index),
    1651                 :            :                              to_famat_shallow(M,negate?gen_m1:gen_1));
    1652                 :       2982 :       break;
    1653                 :            :   }
    1654                 :       4634 : }
    1655                 :            : static void
    1656                 :   90416046 : treat_index_trivial(GEN W, long index, int negate, GEN v)
    1657                 :            : {
    1658                 :   90416046 :   GEN W11 = gel(W,11);
    1659                 :   90416046 :   long shift = W11[3]; /* #F + #E2 + T32 */
    1660   [ +  +  +  +  :   90416046 :   switch(set_from_index(W11, index))
                      - ]
    1661                 :            :   {
    1662                 :            :     case 1: /*F*/
    1663                 :            :     {
    1664                 :   12355441 :       GEN F_index = gel(W,6), ind = gel(F_index, index);
    1665                 :   12355441 :       long j, l = lg(ind);
    1666         [ +  + ]:   80209619 :       for (j = 1; j < l; j++)
    1667                 :            :       {
    1668                 :   67854178 :         GEN IND = gel(ind,j);
    1669                 :   67854178 :         treat_index_trivial(W, mael(IND,1,1), negate, v);
    1670                 :            :       }
    1671                 :   12355441 :       break;
    1672                 :            :     }
    1673                 :            : 
    1674                 :            :     case 2: /*E2, E2[r] + gamma * E1[s] = 0 */
    1675                 :            :     {
    1676                 :   29921640 :       long r = index - W11[1];
    1677                 :   29921640 :       GEN E2_in_terms_of_E1= gel(W,7), z = gel(E2_in_terms_of_E1, r);
    1678                 :   29921640 :       long s = itou(gel(z,1));
    1679                 :   29921640 :       index = s;
    1680                 :   61012553 :       gel(v, index) = negate? addiu(gel(v, index), 1)
    1681         [ +  + ]:   29921640 :                             : subiu(gel(v, index), 1);
    1682                 :   29921640 :       break;
    1683                 :            :     }
    1684                 :            : 
    1685                 :            :     case 3: case 5: case 6: /*T32,T2,T31*/
    1686                 :    3558527 :       break;
    1687                 :            : 
    1688                 :            :     case 4: /*E1*/
    1689                 :   44580438 :       index -= shift;
    1690                 :   93577890 :       gel(v, index) = negate? subiu(gel(v, index), 1)
    1691         [ +  + ]:   44580438 :                             : addiu(gel(v, index), 1);
    1692                 :   44580438 :       break;
    1693                 :            :   }
    1694                 :   90416046 : }
    1695                 :            : 
    1696                 :            : /* cusp q a t_INT/t_FRAC, expresses {1/0 -> q} as \sum x_i g_i, see mspathlog.
    1697                 :            :  * Set v := [x_i]  ([-x_i] if negate) */
    1698                 :            : static void
    1699                 :       1533 : Q_log(GEN v, GEN W, int negate, GEN q)
    1700                 :            : {
    1701                 :       1533 :   GEN PQ = ZV_allpnqn( gboundcf(q, 0) );
    1702                 :       1533 :   GEN P = gel(PQ,1), Q = gel(PQ,2), a,b,c,d;
    1703                 :       1533 :   long i, lx = lg(P);
    1704                 :            : 
    1705                 :       1533 :   a = gen_1;
    1706                 :       1533 :   c = gen_0;
    1707         [ +  + ]:       4585 :   for (i = 1; i < lx; i++, c = d, a = b)
    1708                 :            :   {
    1709                 :            :     long index;
    1710                 :            :     GEN M;
    1711                 :       3052 :     b = gel(P,i);
    1712                 :       3052 :     d = gel(Q,i);
    1713         [ +  + ]:       3052 :     if (!odd(i)) { a = negi(a); c = negi(c); }
    1714                 :       3052 :     M = Gamma0N_decompose(W, mkmat2(mkcol2(a,c), mkcol2(b,d)), &index);
    1715                 :       3052 :     treat_index(W, M, index, negate, v);
    1716                 :            :   }
    1717                 :       1533 : }
    1718                 :            : /* Q_log for cusp [a,b] in case the action is trivial, q = a/b */
    1719                 :            : static void
    1720                 :    1981371 : Q_log_trivial(GEN v, GEN W, int negate, GEN q)
    1721                 :            : {
    1722                 :    1981371 :   GEN Q, W3 = gel(W,3), p1N = gel(W,1);
    1723                 :    1981371 :   ulong c,d, N = p1N_get_N(p1N);
    1724                 :            :   long i, lx;
    1725                 :            : 
    1726                 :    1981371 :   Q = modulartosym_init(N, q);
    1727                 :    1981371 :   lx = lg(Q);
    1728                 :    1981371 :   c = 0;
    1729         [ +  + ]:   22626527 :   for (i = 1; i < lx; i++, c = d)
    1730                 :            :   {
    1731                 :            :     long index;
    1732                 :   20645156 :     d = Q[i];
    1733 [ +  + ][ +  + ]:   20645156 :     if (c && !odd(i)) c = N - c;
    1734                 :   20645156 :     index = W3[ p1_index(c,d,p1N) ];
    1735                 :   20645156 :     treat_index_trivial(W, index, negate, v);
    1736                 :            :   }
    1737                 :    1981371 : }
    1738                 :            : static void
    1739                 :         84 : P1Q_log(GEN v, GEN W, int negate, GEN c)
    1740                 :            : {
    1741      [ +  +  - ]:         84 :   switch(typ(c))
    1742                 :            :   {
    1743                 :         14 :     case t_INFINITY: break;
    1744                 :            :     case t_INT: case t_FRAC:
    1745                 :         70 :       Q_log(v, W, negate, c);
    1746                 :         70 :       break;
    1747                 :            :     default:
    1748                 :          0 :       pari_err_TYPE("mspathlog",c);
    1749                 :            :   }
    1750                 :         84 : }
    1751                 :            : /* same with trivial action */
    1752                 :            : static void
    1753                 :         42 : P1Q_log_trivial(GEN v, GEN W, int negate, GEN c)
    1754                 :            : {
    1755      [ -  +  - ]:         42 :   switch(typ(c))
    1756                 :            :   {
    1757                 :          0 :     case t_INFINITY: break;
    1758                 :            :     case t_INT: case t_FRAC:
    1759                 :         42 :       Q_log_trivial(v, W, negate, c);
    1760                 :         42 :       break;
    1761                 :            :     default:
    1762                 :          0 :       pari_err_TYPE("mspathlog",c);
    1763                 :            :   }
    1764                 :         42 : }
    1765                 :            : 
    1766                 :            : /* cusp in P^1(Q): expresses {oo -> cusp} as \sum x_i g_i, see mspathlog */
    1767                 :            : static void
    1768                 :       1806 : cusplog(GEN v, GEN W, int negate, GEN cusp)
    1769         [ +  + ]:       1806 : { if (cusp[2]) Q_log(v,W,negate,cusp_to_frac(cusp)); }
    1770                 :            : 
    1771                 :            : /* cusplog in case the action is trivial */
    1772                 :            : static void
    1773                 :      66318 : cusplog_trivial(GEN v, GEN W, int negate, GEN cusp)
    1774         [ +  + ]:      66318 : { if (cusp[2]) Q_log_trivial(v,W,negate,cusp_to_frac(cusp)); }
    1775                 :            : 
    1776                 :            : /* Expresses path p as \sum x_i g_i, where the g_i are our distinguished
    1777                 :            :  * generators and x_i \in Z[\Gamma]. Returns [x_1,...,x_n] */
    1778                 :            : static GEN
    1779                 :        903 : mspathlog_i(GEN W, GEN p)
    1780                 :            : {
    1781                 :            :   GEN v;
    1782                 :        903 :   W = get_ms(W);
    1783                 :        903 :   v = zerovec(ms_get_nbgen(W));
    1784                 :        903 :   cusplog(v, W, 0, gel(p,2));
    1785                 :        903 :   cusplog(v, W, 1, gel(p,1));
    1786                 :        903 :   return v;
    1787                 :            : }
    1788                 :            : /* in case the action is trivial: v += mspathlog(path) */
    1789                 :            : static void
    1790                 :      33159 : mspathlog_i_trivial(GEN v, GEN W, GEN path)
    1791                 :            : {
    1792                 :      33159 :   cusplog_trivial(v, W, 0, gel(path,2));
    1793                 :      33159 :   cusplog_trivial(v, W, 1, gel(path,1));
    1794                 :      33159 : }
    1795                 :            : 
    1796                 :            : GEN
    1797                 :         42 : mspathlog(GEN W, GEN p)
    1798                 :            : {
    1799                 :         42 :   pari_sp av = avma;
    1800                 :            :   GEN v;
    1801                 :         42 :   checkms(W); W = get_ms(W);
    1802 [ +  - ][ -  + ]:         42 :   if (typ(p) != t_VEC || lg(p) != 3) pari_err_TYPE("mspathlog",p);
    1803                 :         42 :   v = zerovec(ms_get_nbgen(W));
    1804                 :         42 :   P1Q_log(v,W,0,gel(p,2));
    1805                 :         42 :   P1Q_log(v,W,1,gel(p,1));
    1806                 :         42 :   return gerepilecopy(av, v);
    1807                 :            : }
    1808                 :            : static GEN
    1809                 :         21 : mspathlog_trivial(GEN W, GEN p)
    1810                 :            : {
    1811                 :         21 :   pari_sp av = avma;
    1812                 :            :   GEN v;
    1813                 :         21 :   W = get_ms(W);
    1814 [ +  - ][ -  + ]:         21 :   if (typ(p) != t_VEC || lg(p) != 3) pari_err_TYPE("mspathlog",p);
    1815                 :         21 :   v = zerovec(ms_get_nbgen(W));
    1816                 :         21 :   P1Q_log_trivial(v,W,0,gel(p,2));
    1817                 :         21 :   P1Q_log_trivial(v,W,1,gel(p,1));
    1818                 :         21 :   return gerepilecopy(av, v);
    1819                 :            : }
    1820                 :            : 
    1821                 :            : /** HECKE OPERATORS **/
    1822                 :            : /* [a,b;c,d] * cusp */
    1823                 :            : static GEN
    1824                 :      68124 : cusp_mul(GEN cusp, long a, long b, long c, long d)
    1825                 :            : {
    1826                 :      68124 :   long x = cusp[1], y = cusp[2];
    1827                 :      68124 :   return mkvecsmall2(a*x+b*y, c*x+d*y);
    1828                 :            : }
    1829                 :            : static GEN
    1830                 :      34062 : path_mul(GEN path, long a, long b, long c, long d)
    1831                 :            : {
    1832                 :      34062 :   GEN c1 = cusp_mul(gel(path,1), a,b,c,d);
    1833                 :      34062 :   GEN c2 = cusp_mul(gel(path,2), a,b,c,d);
    1834                 :      34062 :   return mkpath(c1,c2);
    1835                 :            : }
    1836                 :            : /* f in Gl2(Q), act on path (zm) */
    1837                 :            : static GEN
    1838                 :      34062 : Gl2Q_act_path(GEN f, GEN path)
    1839                 :      34062 : { return path_mul(path, coeff(f,1,1),coeff(f,1,2),coeff(f,2,1),coeff(f,2,2)); }
    1840                 :            : 
    1841                 :            : static GEN
    1842                 :    1936193 : init_act_trivial(GEN W) { return zerocol(ms_get_nbE1(W)); }
    1843                 :            : 
    1844                 :            : /* map from WW1 -> WW2, weight 2, trivial action. v a Gl2_Q or a t_VEC
    1845                 :            :  * of Gl2_Q (\sum v[i] in Z[Gl2(Q)]). Return the matrix associated to the
    1846                 :            :  * action of v. */
    1847                 :            : static GEN
    1848                 :        385 : getMorphism_trivial(GEN WW1, GEN WW2, GEN v)
    1849                 :            : {
    1850                 :        385 :   GEN W1 = get_ms(WW1), W2 = get_ms(WW2);
    1851                 :        385 :   GEN section = ms_get_section(W1), gen = gel(W1,5);
    1852                 :        385 :   long i, j, lv, nbE1 = ms_get_nbE1(W1);
    1853                 :        385 :   GEN T = cgetg(nbE1+1, t_MAT);
    1854         [ +  + ]:        385 :   if (typ(v) != t_VEC) v = mkvec(v);
    1855                 :        385 :   lv = lg(v);
    1856         [ +  + ]:      19866 :   for (i = 1; i <= nbE1; i++)
    1857                 :            :   {
    1858                 :      19481 :     long e = gen[i]; /* path-index of E1-element */
    1859                 :      19481 :     GEN w = gel(section, e); /* path_to_zm() */
    1860                 :      19481 :     GEN t = init_act_trivial(W2);
    1861         [ +  + ]:      52640 :     for (j = 1; j < lv; j++)
    1862                 :      33159 :       mspathlog_i_trivial(t, W2, Gl2Q_act_path(gel(v,j), w));
    1863                 :      19481 :     gel(T,i) = t;
    1864                 :            :   }
    1865                 :        385 :   return T;
    1866                 :            : }
    1867                 :            : 
    1868                 :            : /* f zm/ZM in Gl_2(Q), acts from the left on Delta, which is generated by
    1869                 :            :  * (g_i) as Z[Gamma1]-module, and by (G_i) as Z[Gamma2]-module.
    1870                 :            :  * We have f.G_j = \sum_i \lambda_{i,j} g_i,   \lambda_{i,j} in Z[Gamma1]
    1871                 :            :  *
    1872                 :            :  * For phi in Hom_Gamma1(D,V), g in D, phi | f is in Hom_Gamma2(D,V) and
    1873                 :            :  *  (phi | f)(G_j) = phi(f.G_j) | f
    1874                 :            :  *                 = phi( \sum_i \lambda_{i,j} g_i ) | f
    1875                 :            :  *                 = \sum_i phi(g_i) | (\lambda_{i,j}^* f)
    1876                 :            :  *                 = \sum_i phi(g_i) | \mu_{i,j}
    1877                 :            :  * Return the \mu_{i,j} matrix as operators on V (t_MAT) */
    1878                 :            : static GEN
    1879                 :        350 : init_dual_act(GEN f, GEN W1, GEN W2)
    1880                 :            : {
    1881                 :        350 :   GEN section = ms_get_section(W2), gen = ms_get_genindex(W2);
    1882                 :        350 :   long j, dim = lg(gen)-1, k = msk_get_weight(W1);
    1883                 :        350 :   GEN T = cgetg(dim+1, t_MAT), F;
    1884         [ -  + ]:        350 :   if (typ(gel(f,1)) == t_VEC)
    1885                 :            :   {
    1886                 :          0 :     F = f;
    1887                 :          0 :     f = ZM_to_zm(F);
    1888                 :            :   }
    1889                 :            :   else
    1890                 :        350 :     F = zm_to_ZM(f);
    1891                 :            :   /* f zm = F ZM */
    1892         [ +  + ]:       1253 :   for (j = 1; j <= dim; j++)
    1893                 :            :   {
    1894                 :        903 :     pari_sp av = avma;
    1895                 :        903 :     GEN w = gel(section, gen[j]); /* path_to_zm( E1/T2/T3 element ) */
    1896                 :        903 :     GEN l = mspathlog_i(W1, Gl2Q_act_path(f, w)); /* lambda_{i,j} */
    1897                 :        903 :     l = ZGl2QC_star(l); /* lambda_{i,j}^* */
    1898                 :        903 :     l = ZGC_G_mul(l, F); /* mu_{i,j} */
    1899                 :        903 :     gel(T,j) = gerepilecopy(av, ZGl2QC_to_act(l, k)); /* as operators on V */
    1900                 :            :   }
    1901                 :        350 :   return T;
    1902                 :            : }
    1903                 :            : /* phi in Hom_Gamma1(Delta, V), return the matrix whose colums are the
    1904                 :            :  *   \sum_i phi(g_i) | \mu_{i,j} = (phi|f)(G_j),
    1905                 :            :  * see init_dual_act. */
    1906                 :            : static GEN
    1907                 :       1645 : dual_act(GEN phi, GEN mu)
    1908                 :            : {
    1909                 :       1645 :   long l = lg(mu), a, j;
    1910                 :       1645 :   GEN ind = gel(phi,2), pols = gel(phi,3);
    1911                 :       1645 :   GEN v = cgetg(l, t_MAT);
    1912         [ +  + ]:       6097 :   for (j = 1; j < l; j++)
    1913                 :            :   {
    1914                 :       4452 :     GEN T = NULL;
    1915         [ +  + ]:      14112 :     for (a = 1; a < lg(ind); a++)
    1916                 :            :     {
    1917                 :       9660 :       long i = ind[a];
    1918                 :       9660 :       GEN t = gel(pols, a); /* phi(G_i) */
    1919                 :       9660 :       t = RgM_RgC_mul(gcoeff(mu,i,j), t);
    1920         [ +  + ]:       9660 :       T = T? RgC_add(T, t): t;
    1921                 :            :     }
    1922                 :       4452 :     gel(v,j) = T;
    1923                 :            :   }
    1924                 :       1645 :   return v;
    1925                 :            : }
    1926                 :            : 
    1927                 :            : /* phi in Hom(Delta, V), phi(G_k) = vecT[k]. Write phi as
    1928                 :            :  *   \sum_{i,j} mu_{i,j} phi_{i,j}, mu_{i,j} in Q */
    1929                 :            : static GEN
    1930                 :        840 : getMorphism_basis(GEN W, GEN vecT)
    1931                 :            : {
    1932                 :        840 :   GEN basis = msk_get_basis(W);
    1933                 :        840 :   long i, j, r, lvecT = lg(vecT), dim = lg(basis)-1;
    1934                 :        840 :   GEN st = msk_get_st(W);
    1935                 :        840 :   GEN link = msk_get_link(W);
    1936                 :        840 :   GEN invphiblock = msk_get_invphiblock(W);
    1937                 :        840 :   long s = st[1], t = st[2];
    1938                 :        840 :   GEN R = zerovec(dim), Q, Ls, T0, T1, Ts, mu_st;
    1939         [ +  + ]:       2268 :   for (r = 2; r < lvecT; r++)
    1940                 :            :   {
    1941                 :            :     GEN Tr, L;
    1942         [ +  + ]:       1428 :     if (r == s) continue;
    1943                 :        588 :     Tr = gel(vecT,r); /* Phi(G_r), r != 1,s */
    1944                 :        588 :     L = gel(link, r);
    1945                 :        588 :     Q = ZC_apply_dinv(gel(invphiblock,r), Tr);
    1946                 :            :     /* write Phi(G_r) as sum_{a,b} mu_{a,b} Phi_{a,b}(G_r) */
    1947         [ +  + ]:       2478 :     for (j = 1; j < lg(L); j++) gel(R, L[j]) = gel(Q,j);
    1948                 :            :   }
    1949                 :        840 :   Ls = gel(link, s);
    1950                 :        840 :   T1 = gel(vecT,1); /* Phi(G_1) */
    1951                 :        840 :   gel(R, Ls[t]) = mu_st = gel(T1, 1);
    1952                 :            : 
    1953                 :        840 :   T0 = NULL;
    1954         [ +  + ]:       2268 :   for (i = 2; i < lg(link); i++)
    1955                 :            :   {
    1956                 :            :     GEN L;
    1957         [ +  + ]:       1428 :     if (i == s) continue;
    1958                 :        588 :     L = gel(link,i);
    1959         [ +  + ]:       2478 :     for (j =1 ; j < lg(L); j++)
    1960                 :            :     {
    1961                 :       1890 :       long n = L[j]; /* phi_{i,j} = basis[n] */
    1962                 :       1890 :       GEN mu_ij = gel(R, n);
    1963                 :       1890 :       GEN phi_ij = gel(basis, n), pols = gel(phi_ij,3);
    1964                 :       1890 :       GEN z = RgC_Rg_mul(gel(pols, 3), mu_ij);
    1965         [ +  + ]:       1890 :       T0 = T0? RgC_add(T0, z): z; /* += mu_{i,j} Phi_{i,j} (G_s) */
    1966                 :            :     }
    1967                 :            :   }
    1968                 :        840 :   Ts = gel(vecT,s); /* Phi(G_s) */
    1969         [ +  + ]:        840 :   if (T0) Ts = RgC_sub(Ts, T0);
    1970                 :            :   /* solve \sum_{j!=t} mu_{s,j} Phi_{s,j}(G_s) = Ts */
    1971                 :        840 :   Q = ZC_apply_dinv(gel(invphiblock,s), Ts);
    1972         [ +  + ]:       2324 :   for (j = 1; j < t; j++) gel(R, Ls[j]) = gel(Q,j);
    1973                 :            :   /* avoid mu_{s,t} */
    1974         [ +  + ]:       1260 :   for (j = t; j < lg(Q); j++) gel(R, Ls[j+1]) = gel(Q,j);
    1975                 :        840 :   return R;
    1976                 :            : }
    1977                 :            : 
    1978                 :            : /* a = s(g_i) for some modular symbol s; b in Z[G]
    1979                 :            :  * return s(b.g_i) = b^* . s(g_i) */
    1980                 :            : static GEN
    1981                 :        119 : ZGl2Q_act_s(GEN b, GEN a, long k)
    1982                 :            : {
    1983         [ +  + ]:        119 :   if (typ(b) == t_INT)
    1984                 :            :   {
    1985         [ +  + ]:         42 :     if (!signe(b)) return gen_0;
    1986      [ +  -  - ]:         14 :     switch(typ(a))
    1987                 :            :     {
    1988                 :            :       case t_POL:
    1989                 :         14 :         a = RgX_to_RgC(a, k-1); /*fall through*/
    1990                 :            :       case t_COL:
    1991                 :         14 :         a = RgC_Rg_mul(a,b);
    1992                 :         14 :         break;
    1993                 :         14 :       default: a = scalarcol_shallow(b,k-1);
    1994                 :            :     }
    1995                 :            :   }
    1996                 :            :   else
    1997                 :            :   {
    1998                 :         77 :     b = RgX_act_ZGl2Q(ZGl2Q_star(b), k);
    1999      [ +  +  - ]:         77 :     switch(typ(a))
    2000                 :            :     {
    2001                 :            :       case t_POL:
    2002                 :         63 :         a = RgX_to_RgC(a, k-1); /*fall through*/
    2003                 :            :       case t_COL:
    2004                 :         77 :         a = RgM_RgC_mul(b,a);
    2005                 :         77 :         break;
    2006                 :          0 :       default: a = RgC_Rg_mul(gel(b,1),a);
    2007                 :            :     }
    2008                 :            :   }
    2009                 :        119 :   return a;
    2010                 :            : }
    2011                 :            : 
    2012                 :            : static int
    2013                 :         21 : checksymbol(GEN W, GEN s)
    2014                 :            : {
    2015                 :            :   GEN t, annT2, annT31, singlerel;
    2016                 :            :   long i, k, l, nbE1, nbT2, nbT31;
    2017                 :         21 :   k = msk_get_weight(W);
    2018                 :         21 :   W = get_ms(W);
    2019                 :         21 :   nbE1 = ms_get_nbE1(W);
    2020                 :         21 :   singlerel = gel(W,10);
    2021                 :         21 :   l = lg(singlerel);
    2022         [ -  + ]:         21 :   if (k == 2)
    2023                 :            :   {
    2024         [ #  # ]:          0 :     for (i = nbE1+1; i < l; i++)
    2025         [ #  # ]:          0 :       if (!gequal0(gel(s,i))) return 0;
    2026                 :          0 :     return 1;
    2027                 :            :   }
    2028                 :         21 :   annT2 = gel(W,8); nbT2 = lg(annT2)-1;
    2029                 :         21 :   annT31 = gel(W,9);nbT31 = lg(annT31)-1;
    2030                 :         21 :   t = NULL;
    2031         [ +  + ]:         84 :   for (i = 1; i < l; i++)
    2032                 :            :   {
    2033                 :         63 :     GEN a = gel(s,i);
    2034                 :         63 :     a = ZGl2Q_act_s(gel(singlerel,i), a, k);
    2035         [ +  + ]:         63 :     t = t? gadd(t, a): a;
    2036                 :            :   }
    2037         [ +  + ]:         21 :   if (!gcmp0(t)) return 0;
    2038         [ -  + ]:         14 :   for (i = 1; i <= nbT2; i++)
    2039                 :            :   {
    2040                 :          0 :     GEN a = gel(s,i + nbE1);
    2041                 :          0 :     a = ZGl2Q_act_s(gel(annT2,i), a, k);
    2042         [ #  # ]:          0 :     if (!gcmp0(a)) return 0;
    2043                 :            :   }
    2044         [ +  + ]:         28 :   for (i = 1; i <= nbT31; i++)
    2045                 :            :   {
    2046                 :         14 :     GEN a = gel(s,i + nbE1 + nbT2);
    2047                 :         14 :     a = ZGl2Q_act_s(gel(annT31,i), a, k);
    2048         [ -  + ]:         14 :     if (!gcmp0(a)) return 0;
    2049                 :            :   }
    2050                 :         21 :   return 1;
    2051                 :            : }
    2052                 :            : long
    2053                 :         28 : msissymbol(GEN W, GEN s)
    2054                 :            : {
    2055                 :            :   long k, nbgen;
    2056                 :         28 :   checkms(W);
    2057                 :         28 :   k = msk_get_weight(W);
    2058                 :         28 :   nbgen = ms_get_nbgen(W);
    2059      [ +  +  - ]:         28 :   switch(typ(s))
    2060                 :            :   {
    2061                 :            :     case t_VEC: /* values s(g_i) */
    2062         [ -  + ]:         21 :       if (lg(s)-1 != nbgen) return 0;
    2063                 :         21 :       break;
    2064                 :            :     case t_COL:
    2065         [ -  + ]:          7 :       if (k == 2) /* on the dual basis of (g_i) */
    2066                 :            :       {
    2067         [ #  # ]:          0 :         if (lg(s)-1 != nbgen) return 0;
    2068                 :            :       }
    2069                 :            :       else
    2070                 :            :       {
    2071                 :          7 :         GEN basis = msk_get_basis(W);
    2072                 :          7 :         return (lg(s) == lg(basis));
    2073                 :            :       }
    2074                 :          0 :       break;
    2075                 :          0 :     default: return 0;
    2076                 :            :   }
    2077                 :         28 :   return checksymbol(W,s);
    2078                 :            : }
    2079                 :            : #if DEBUG
    2080                 :            : /* phi_i(G_j) */
    2081                 :            : static GEN
    2082                 :            : eval_phii_Gj(GEN W, long i, long j)
    2083                 :            : {
    2084                 :            :   GEN basis = msk_get_basis(W), b = gel(basis,i);
    2085                 :            :   GEN ind = gel(b,2), pols = gel(b,3);
    2086                 :            :   long s;
    2087                 :            :   for (s = 1; s < lg(ind); s++)
    2088                 :            :     if (ind[s] == j) return gel(pols,s);
    2089                 :            :   return zerocol(lg(gel(pols,1))-1);
    2090                 :            : }
    2091                 :            : /* check that \sum d_i phi_i(G_j)  = T_j for all j */
    2092                 :            : static void
    2093                 :            : checkdec(GEN W, GEN D, GEN T)
    2094                 :            : {
    2095                 :            :   long i, j;
    2096                 :            :   if (!checksymbol(W,T)) pari_err_BUG("checkdec");
    2097                 :            :   for (j = 1; j < lg(T); j++)
    2098                 :            :   {
    2099                 :            :     GEN S = gen_0;
    2100                 :            :     for (i = 1; i < lg(D); i++)
    2101                 :            :     {
    2102                 :            :       GEN d = gel(D,i);
    2103                 :            :       if (gcmp0(d)) continue;
    2104                 :            :       S = gadd(S, gmul(d, eval_phii_Gj(W, i, j)));
    2105                 :            :     }
    2106                 :            :     /* S = \sum_i d_i phi_i(G_j) */
    2107                 :            :     if (!gequal(S, gel(T,j)))
    2108                 :            :       pari_warn(warner, "checkdec j = %ld\n\tS = %Ps\n\tT = %Ps", j,S,gel(T,j));
    2109                 :            :   }
    2110                 :            : }
    2111                 :            : #endif
    2112                 :            : 
    2113                 :            : /* map op: Hom(W1,V) -> Hom(W2,V), given by \sum v[i], v[i] in Gl2(Q) */
    2114                 :            : static GEN
    2115                 :        182 : getMorphism(GEN W1, GEN W2, GEN v)
    2116                 :            : {
    2117                 :        182 :   GEN basis1 = msk_get_basis(W1);
    2118                 :        182 :   long i, a, lv, dim1 = lg(basis1)-1;
    2119                 :        182 :   GEN M = cgetg(dim1+1, t_MAT), act;
    2120         [ +  + ]:        182 :   if (typ(v) != t_VEC) v = mkvec(v);
    2121                 :        182 :   lv = lg(v); act = cgetg(lv, t_MAT);
    2122         [ +  + ]:        532 :   for (i = 1; i < lv; i++) gel(act,i) = init_dual_act(gel(v,i), W1, W2);
    2123         [ +  + ]:        966 :   for (a = 1; a <= dim1; a++)
    2124                 :            :   {
    2125                 :        784 :     pari_sp av = avma;
    2126                 :        784 :     GEN phi = gel(basis1, a), D, T = NULL;
    2127         [ +  + ]:       2429 :     for (i = 1; i < lv; i++)
    2128                 :            :     {
    2129                 :       1645 :       GEN t = dual_act(phi, gel(act, i));
    2130         [ +  + ]:       1645 :       T = T? gerepileupto(av, RgM_add(T,t)): t;
    2131                 :            :     }
    2132                 :            :     /* T = (phi|op)(G_1,...,G_d2) */
    2133                 :        784 :     D = getMorphism_basis(W2, T);
    2134                 :            : #if DEBUG
    2135                 :            :     checkdec(W2,D,T);
    2136                 :            : #endif
    2137                 :        784 :     gel(M,a) = gerepilecopy(av, D);
    2138                 :            :   }
    2139                 :        182 :   return M;
    2140                 :            : }
    2141                 :            : 
    2142                 :            : static GEN
    2143                 :        259 : Tp_matrices(ulong p)
    2144                 :            : {
    2145                 :        259 :   GEN v = cgetg(p+2, t_VEC);
    2146                 :            :   ulong i;
    2147         [ +  + ]:       1505 :   for (i = 1; i <= p; i++) gel(v,i) = mat2(1, i-1, 0, p);
    2148                 :        259 :   gel(v,i) = mat2(p, 0, 0, 1);
    2149                 :        259 :   return v;
    2150                 :            : }
    2151                 :            : static GEN
    2152                 :         28 : Up_matrices(ulong p)
    2153                 :            : {
    2154                 :         28 :   GEN v = cgetg(p+1, t_VEC);
    2155                 :            :   ulong i;
    2156         [ +  + ]:        294 :   for (i = 1; i <= p; i++) gel(v,i) = mat2(1, i-1, 0, p);
    2157                 :         28 :   return v;
    2158                 :            : }
    2159                 :            : static GEN
    2160                 :        511 : msendo(GEN W, GEN v)
    2161                 :            : {
    2162                 :       1022 :   return msk_get_weight(W) == 2? getMorphism_trivial(W,W,v)
    2163         [ +  + ]:        511 :                                : getMorphism(W, W, v);
    2164                 :            : }
    2165                 :            : static GEN
    2166                 :        105 : endo_project(GEN W, GEN e, GEN H)
    2167                 :            : {
    2168         [ -  + ]:        105 :   if (H) H = Qevproj_init0(H);
    2169         [ +  + ]:        105 :   else if (msk_get_sign(W))
    2170                 :         21 :     H = msk_get_starproj(W);
    2171         [ +  + ]:        105 :   if (H) e = Qevproj_apply(e, H);
    2172                 :        105 :   return e;
    2173                 :            : }
    2174                 :            : static GEN
    2175                 :        287 : mshecke_i(GEN W, ulong p)
    2176                 :            : {
    2177         [ +  + ]:        287 :   GEN v = ms_get_N(W) % p? Tp_matrices(p): Up_matrices(p);
    2178                 :        287 :   return msendo(W,v);
    2179                 :            : }
    2180                 :            : GEN
    2181                 :         70 : mshecke(GEN W, long p, GEN H)
    2182                 :            : {
    2183                 :         70 :   pari_sp av = avma;
    2184                 :            :   GEN T;
    2185                 :         70 :   checkms(W);
    2186         [ -  + ]:         70 :   if (p <= 1) pari_err_PRIME("mshecke",stoi(p));
    2187                 :         70 :   T = mshecke_i(W,p);
    2188                 :         70 :   return gerepilecopy(av, endo_project(W,T,H));
    2189                 :            : }
    2190                 :            : 
    2191                 :            : static GEN
    2192                 :         35 : msatkinlehner_i(GEN W, long Q)
    2193                 :            : {
    2194                 :         35 :   long w, z, d, N = ms_get_N(W), M = N / Q;
    2195                 :            :   GEN v;
    2196         [ +  + ]:         35 :   if (Q == 1) return matid(msk_get_dim(W));
    2197                 :         28 :   d = cbezout(Q, -M, &w, &z);
    2198         [ +  + ]:         28 :   if (N % Q) pari_err_DOMAIN("msatkinlehner","N % Q","!=",gen_0,stoi(Q));
    2199         [ -  + ]:         21 :   if (d != 1) pari_err_DOMAIN("msatkinlehner","gcd(Q,N/Q)","!=",gen_1,stoi(Q));
    2200         [ +  + ]:         21 :   v = (Q == N)? mat2(0,1,-N,0): mat2(Q,1,N*z,Q*w);
    2201                 :         28 :   return msendo(W,v);
    2202                 :            : }
    2203                 :            : GEN
    2204                 :         35 : msatkinlehner(GEN W, long Q, GEN H)
    2205                 :            : {
    2206                 :         35 :   pari_sp av = avma;
    2207                 :            :   GEN w;
    2208                 :         35 :   checkms(W);
    2209         [ -  + ]:         35 :   if (Q <= 0) pari_err_DOMAIN("msatkinlehner","Q","<=",gen_0,stoi(Q));
    2210                 :         35 :   w = msatkinlehner_i(W,Q);
    2211                 :         28 :   return gerepilecopy(av, endo_project(W,w,H));
    2212                 :            : }
    2213                 :            : 
    2214                 :            : static GEN
    2215                 :        203 : msstar_i(GEN W)
    2216                 :            : {
    2217                 :        203 :   GEN v = mat2(-1,0,0,1);
    2218                 :        203 :   return msendo(W,v);
    2219                 :            : }
    2220                 :            : GEN
    2221                 :          7 : msstar(GEN W, GEN H)
    2222                 :            : {
    2223                 :          7 :   pari_sp av = avma;
    2224                 :            :   GEN s;
    2225                 :          7 :   checkms(W);
    2226                 :          7 :   s = msstar_i(W);
    2227                 :          7 :   return gerepilecopy(av, endo_project(W,s,H));
    2228                 :            : }
    2229                 :            : 
    2230                 :            : #if 0
    2231                 :            : /* is \Gamma_0(N) cusp1 = \Gamma_0(N) cusp2 ? */
    2232                 :            : static int
    2233                 :            : iscuspeq(ulong N, GEN cusp1, GEN cusp2)
    2234                 :            : {
    2235                 :            :   long p1, q1, p2, q2, s1, s2, d;
    2236                 :            :   p1 = cusp1[1]; p2 = cusp2[1];
    2237                 :            :   q1 = cusp1[2]; q2 = cusp2[2];
    2238                 :            :   d = Fl_mul(smodss(q1,N),smodss(q2,N), N);
    2239                 :            :   d = ugcd(d, N);
    2240                 :            : 
    2241                 :            :   s1 = q1 > 2? Fl_inv(smodss(p1,q1), q1): 1;
    2242                 :            :   s2 = q2 > 2? Fl_inv(smodss(p2,q2), q2): 1;
    2243                 :            :   return Fl_mul(s1,q2,d) == Fl_mul(s2,q1,d);
    2244                 :            : }
    2245                 :            : #endif
    2246                 :            : 
    2247                 :            : static GEN
    2248                 :         35 : mscuspidal_trivial(GEN W0)
    2249                 :            : {
    2250                 :         35 :   pari_sp av = avma;
    2251                 :         35 :   GEN W = get_ms(W0);
    2252                 :         35 :   GEN section = ms_get_section(W), gen = ms_get_genindex(W);
    2253                 :         35 :   GEN S = ms_get_hashcusps(W);
    2254                 :         35 :   long j, nbE1 = ms_get_nbE1(W), ncusp = gel(S,1)[2];
    2255                 :         35 :   GEN T = zeromatcopy(ncusp,nbE1);
    2256         [ +  + ]:       2492 :   for (j = 1; j <= nbE1; j++)
    2257                 :            :   {
    2258                 :       2457 :     long e = gen[j]; /* path-index of E1-element */
    2259                 :       2457 :     GEN t = gel(section, e); /* path_to_zm() */
    2260                 :       2457 :     long i1 = cusp_index(gel(t,1), S);
    2261                 :       2457 :     long i2 = cusp_index(gel(t,2), S);
    2262         [ +  + ]:       2457 :     if (i1 != i2)
    2263                 :            :     {
    2264                 :       1694 :       gcoeff(T, i1, j) = gen_1;
    2265                 :       1694 :       gcoeff(T, i2, j) = gen_m1;
    2266                 :            :     }
    2267                 :            :   }
    2268                 :         35 :   return gerepilecopy(av, ZM_ker(T));
    2269                 :            : }
    2270                 :            : 
    2271                 :            : /* return E_c(r) */
    2272                 :            : static GEN
    2273                 :        112 : get_Ec_r(GEN c, long k)
    2274                 :            : {
    2275                 :        112 :   long p = c[1], q = c[2], u, v;
    2276                 :            :   GEN gr;
    2277                 :        112 :   (void)cbezout(p, q, &u, &v);
    2278                 :        112 :   gr = mat2(p, -v, q, u); /* g . (1:0) = (p:q) */
    2279                 :        112 :   return voo_act_Gl2Q(zm_to_ZM(sl2_inv(gr)), k);
    2280                 :            : }
    2281                 :            : /* returns the modular symbol associated to the cusp c := p/q via the rule
    2282                 :            :  * E_c(path from a to b in Delta_0) := E_c(b) - E_c(a), where
    2283                 :            :  * E_c(r) := 0 if r != c mod Gamma
    2284                 :            :  *           v_oo | gamma_r^(-1)
    2285                 :            :  * where v_oo is stable by T = [1,1;0,1] (i.e x^(k-2)) and
    2286                 :            :  * gamma_r . (1:0) = r, for some gamma_r in SL_2(Z) * */
    2287                 :            : GEN
    2288                 :         56 : Eisenstein_symbol(GEN W, GEN c)
    2289                 :            : {
    2290                 :         56 :   GEN section = ms_get_section(W), gen = ms_get_genindex(W);
    2291                 :         56 :   GEN S = ms_get_hashcusps(W);
    2292                 :         56 :   long j, ic = cusp_index(c, S), l = lg(gen), k = msk_get_weight(W);
    2293                 :         56 :   GEN vecT = cgetg(l, t_COL);
    2294         [ +  + ]:        210 :   for (j = 1; j < l; j++)
    2295                 :            :   {
    2296                 :        154 :     GEN vj = NULL, g = gel(section, gen[j]); /* path_to_zm(generator) */
    2297                 :        154 :     GEN c1 = gel(g,1), c2 = gel(g,2);
    2298                 :        154 :     long i1 = cusp_index(c1, S);
    2299                 :        154 :     long i2 = cusp_index(c2, S);
    2300         [ +  + ]:        154 :     if (i1 == ic) vj = get_Ec_r(c1, k);
    2301         [ +  + ]:        154 :     if (i2 == ic)
    2302                 :            :     {
    2303                 :         56 :       GEN s = get_Ec_r(c2, k);
    2304         [ +  + ]:         56 :       vj = vj? gsub(vj, s): gneg(s);
    2305                 :            :     }
    2306         [ +  + ]:        154 :     if (!vj) vj = zerocol(k-1);
    2307                 :        154 :     gel(vecT, j) = vj;
    2308                 :            :   }
    2309                 :         56 :   return getMorphism_basis(W, vecT);
    2310                 :            : }
    2311                 :            : 
    2312                 :            : static GEN
    2313                 :         14 : EC_subspace_trivial(GEN W)
    2314                 :            : {
    2315                 :         14 :   GEN M, ch, chC, chE, T, TC, C = mscuspidal_trivial(W);
    2316                 :         14 :   long N = ms_get_N(W);
    2317                 :            :   ulong p;
    2318                 :            :   forprime_t S;
    2319                 :         14 :   (void)u_forprime_init(&S, 2, ULONG_MAX);
    2320         [ +  - ]:         14 :   while ((p = u_forprime_next(&S)))
    2321         [ +  - ]:         14 :     if (N % p) break;
    2322                 :         14 :   T = mshecke_i(W, p);
    2323                 :         14 :   ch = QM_charpoly_ZX(T);
    2324                 :         14 :   TC = Qevproj_apply(T, Qevproj_init(C)); /* T_p | TC */
    2325                 :         14 :   chC = QM_charpoly_ZX(TC);
    2326                 :         14 :   chE = RgX_div(ch, chC); /* charpoly(T_p | E_k), coprime to chC */
    2327                 :         14 :   M = RgX_RgM_eval(chE, T);
    2328                 :         14 :   return mkvec2(Qevproj_star(W, QM_ker(M)), C);
    2329                 :            : }
    2330                 :            : 
    2331                 :            : static GEN
    2332                 :          7 : mseisenstein_trivial(GEN W)
    2333                 :            : {
    2334                 :          7 :   GEN ES = EC_subspace_trivial(W);
    2335                 :          7 :   return gel(ES,1);
    2336                 :            : }
    2337                 :            : static GEN
    2338                 :         28 : mseisenstein_i(GEN W)
    2339                 :            : {
    2340                 :            :   GEN S, cusps, M;
    2341                 :            :   long i, l;
    2342         [ +  + ]:         28 :   if (msk_get_weight(W) == 2) return mseisenstein_trivial(W);
    2343                 :         21 :   S = ms_get_hashcusps(W);
    2344                 :         21 :   cusps = gel(S,3);
    2345                 :         21 :   l = lg(cusps);
    2346                 :         21 :   M = cgetg(l, t_MAT);
    2347         [ +  + ]:         77 :   for (i = 1; i < l; i++) gel(M,i) = Eisenstein_symbol(W, gel(cusps,i));
    2348                 :         28 :   return Qevproj_star(W, QM_image(M));
    2349                 :            : }
    2350                 :            : GEN
    2351                 :          7 : mseisenstein(GEN W)
    2352                 :            : {
    2353                 :          7 :   pari_sp av = avma;
    2354                 :          7 :   checkms(W);
    2355                 :          7 :   return gerepilecopy(av, Qevproj_init(mseisenstein_i(W)));
    2356                 :            : }
    2357                 :            : 
    2358                 :            : static GEN
    2359                 :         28 : mscuspidal_i(GEN W)
    2360                 :            : {
    2361                 :            :   GEN E;
    2362                 :            :   GEN M, T, TE, ch, chS, chE;
    2363                 :            :   forprime_t S;
    2364                 :            :   ulong p, N;
    2365         [ +  + ]:         28 :   if (msk_get_weight(W) == 2) return EC_subspace_trivial(W);
    2366                 :         21 :   E = mseisenstein_i(W);
    2367                 :         21 :   N = ms_get_N(W);
    2368                 :         21 :   (void)u_forprime_init(&S, 2, ULONG_MAX);
    2369         [ +  - ]:         42 :   while ((p = u_forprime_next(&S)))
    2370         [ +  + ]:         42 :     if (N % p) break;
    2371                 :         21 :   T = mshecke_i(W, p);
    2372                 :         21 :   ch = QM_charpoly_ZX(T);
    2373                 :         21 :   TE = Qevproj_apply(T, Qevproj_init(E)); /* T_p | E */
    2374                 :         21 :   chE = QM_charpoly_ZX(TE);
    2375                 :         21 :   chS = RgX_div(ch, chE); /* charpoly(T_p | S_k), coprime to chE */
    2376                 :         21 :   M = RgX_RgM_eval(chS, T);
    2377                 :         28 :   return mkvec2(E, Qevproj_star(W, QM_ker(M)));
    2378                 :            : }
    2379                 :            : GEN
    2380                 :          7 : mscuspidal(GEN W)
    2381                 :            : {
    2382                 :          7 :   pari_sp av = avma;
    2383                 :            :   GEN v;
    2384                 :          7 :   checkms(W);
    2385                 :          7 :   v = mscuspidal_i(W);
    2386                 :          7 :   gel(v,1) = Qevproj_init(gel(v,1));
    2387                 :          7 :   gel(v,2) = Qevproj_init(gel(v,2));
    2388                 :          7 :   return gerepilecopy(av, v);
    2389                 :            : }
    2390                 :            : 
    2391                 :            : /** INIT ELLSYM STRUCTURE **/
    2392                 :            : /* V a vector of ZM. If all of them have 0 last row, return NULL.
    2393                 :            :  * Otherwise return [m,i,j], where m = V[i][last,j] contains the value
    2394                 :            :  * of smallest absolute value */
    2395                 :            : static GEN
    2396                 :         84 : RgMV_find_non_zero_last_row(long offset, GEN V)
    2397                 :            : {
    2398                 :         84 :   long i, lasti = 0, lastj = 0, lV = lg(V);
    2399                 :         84 :   GEN m = NULL;
    2400         [ +  + ]:        189 :   for (i = 1; i < lV; i++)
    2401                 :            :   {
    2402                 :        105 :     GEN M = gel(V,i);
    2403                 :        105 :     long j, n, l = lg(M);
    2404         [ +  + ]:        105 :     if (l == 1) continue;
    2405                 :         84 :     n = nbrows(M);
    2406         [ +  + ]:        343 :     for (j = 1; j < l; j++)
    2407                 :            :     {
    2408                 :        259 :       GEN a = gcoeff(M, n, j);
    2409 [ +  + ][ +  + ]:        259 :       if (!gcmp0(a) && (!m || absi_cmp(a, m) < 0))
                 [ +  + ]
    2410                 :            :       {
    2411                 :         91 :         m = a; lasti = i; lastj = j;
    2412         [ -  + ]:         91 :         if (is_pm1(m)) goto END;
    2413                 :            :       }
    2414                 :            :     }
    2415                 :            :   }
    2416                 :            : END:
    2417         [ +  + ]:         84 :   if (!m) return NULL;
    2418                 :         84 :   return mkvec2(m, mkvecsmall2(lasti+offset, lastj));
    2419                 :            : }
    2420                 :            : /* invert the d_oo := (\gamma_oo - 1) operator, acting on
    2421                 :            :  * [x^(k-2), ..., y^(k-2)] */
    2422                 :            : static GEN
    2423                 :         63 : Delta_inv(GEN doo, long k)
    2424                 :            : {
    2425                 :         63 :   GEN M = RgX_act_ZGl2Q(doo, k);
    2426                 :         63 :   M = RgM_minor(M, k-1, 1); /* 1st column and last row are 0 */
    2427                 :         63 :   return ZM_inv_denom(M);
    2428                 :            : }
    2429                 :            : /* The ZX P = \sum a_i x^i y^{k-2-i} is given by the ZV [a_0, ..., a_k-2]~,
    2430                 :            :  * return Q and d such that P = doo Q + d y^k-2, where d in Z and Q */
    2431                 :            : static GEN
    2432                 :        259 : doo_decompose(GEN dinv, GEN P, GEN *pd)
    2433                 :            : {
    2434                 :        259 :   long l = lg(P); *pd = gel(P, l-1);
    2435                 :        259 :   P = vecslice(P, 1, l-2);
    2436                 :        259 :   return shallowconcat(gen_0, ZC_apply_dinv(dinv, P));
    2437                 :            : }
    2438                 :            : 
    2439                 :            : static GEN
    2440                 :        259 : get_phi_ij(long i,long j,long n, long s,long t,GEN P_st,GEN Q_st,GEN d_st,
    2441                 :            :            GEN P_ij, GEN lP_ij, GEN dinv)
    2442                 :            : {
    2443                 :            :   GEN ind, pols;
    2444 [ +  + ][ +  + ]:        259 :   if (i == s && j == t)
    2445                 :            :   {
    2446                 :         63 :     ind = mkvecsmall(1);
    2447                 :         63 :     pols = mkvec(scalarcol_shallow(gen_1, lg(P_st)-1)); /* x^{k-2} */
    2448                 :            :   }
    2449                 :            :   else
    2450                 :            :   {
    2451                 :        196 :     GEN d_ij, Q_ij = doo_decompose(dinv, lP_ij, &d_ij);
    2452                 :        196 :     GEN a = ZC_Z_mul(P_ij, d_st);
    2453                 :        196 :     GEN b = ZC_Z_mul(P_st, negi(d_ij));
    2454                 :        196 :     GEN c = RgC_sub(RgC_Rg_mul(Q_ij, d_st), RgC_Rg_mul(Q_st, d_ij));
    2455         [ +  + ]:        196 :     if (i == s) { /* j != t */
    2456                 :        126 :       ind = mkvecsmall2(1, s);
    2457                 :        126 :       pols = mkvec2(c, ZC_add(a, b));
    2458                 :            :     } else {
    2459                 :         70 :       ind = mkvecsmall3(1, i, s);
    2460                 :        196 :       pols = mkvec3(c, a, b); /* image of g_1, g_i, g_s */
    2461                 :            :     }
    2462                 :            :   }
    2463                 :        259 :   return mkvec3(mkvecsmall3(i,j,n), ind, pols);
    2464                 :            : }
    2465                 :            : 
    2466                 :            : static GEN
    2467                 :        133 : mskinit_trivial(GEN WN)
    2468                 :            : {
    2469                 :        133 :   long dim = ms_get_nbE1(WN);
    2470                 :        133 :   return mkvec3(WN, gen_0, mkvec2(gen_0,mkvecsmall2(2, dim)));
    2471                 :            : }
    2472                 :            : /* sum of #cols of the matrices contained in V */
    2473                 :            : static long
    2474                 :        126 : RgMV_dim(GEN V)
    2475                 :            : {
    2476                 :        126 :   long l = lg(V), d = 0, i;
    2477         [ +  + ]:        175 :   for (i = 1; i < l; i++) d += lg(gel(V,i)) - 1;
    2478                 :        126 :   return d;
    2479                 :            : }
    2480                 :            : static GEN
    2481                 :         63 : mskinit_nontrivial(GEN WN, long k)
    2482                 :            : {
    2483                 :         63 :   GEN annT2 = gel(WN,8), annT31 = gel(WN,9), singlerel = gel(WN,10);
    2484                 :            :   GEN link, basis, monomials, invphiblock;
    2485                 :         63 :   long nbE1 = ms_get_nbE1(WN);
    2486                 :         63 :   GEN dinv = Delta_inv(ZG_neg( ZGl2Q_star(gel(singlerel,1)) ), k);
    2487                 :         63 :   GEN p1 = cgetg(nbE1+1, t_VEC), remove;
    2488                 :         63 :   GEN p2 = ZGV_tors(annT2, k);
    2489                 :         63 :   GEN p3 = ZGV_tors(annT31, k);
    2490                 :         63 :   GEN gentor = shallowconcat(p2, p3);
    2491                 :            :   GEN P_st, lP_st, Q_st, d_st;
    2492                 :            :   long n, i, dim, s, t, u;
    2493                 :         63 :   gel(p1, 1) = cgetg(1,t_MAT); /* dummy */
    2494         [ +  + ]:         98 :   for (i = 2; i <= nbE1; i++) /* skip 1st element = (\gamma_oo-1)g_oo */
    2495                 :            :   {
    2496                 :         35 :     GEN z = gel(singlerel, i);
    2497                 :         35 :     gel(p1, i) = RgX_act_ZGl2Q(ZGl2Q_star(z), k);
    2498                 :            :   }
    2499                 :         63 :   remove = RgMV_find_non_zero_last_row(nbE1, gentor);
    2500         [ +  + ]:         63 :   if (!remove) remove = RgMV_find_non_zero_last_row(0, p1);
    2501         [ -  + ]:         63 :   if (!remove) pari_err_BUG("msinit [no y^k-2]");
    2502                 :         63 :   remove = gel(remove,2); /* [s,t] */
    2503                 :         63 :   s = remove[1];
    2504                 :         63 :   t = remove[2];
    2505                 :            :   /* +1 because of = x^(k-2), but -1 because of Manin relation */
    2506                 :         63 :   dim = (k-1)*(nbE1-1) + RgMV_dim(p2) + RgMV_dim(p3);
    2507                 :            :   /* Let (g_1,...,g_d) be the Gamma-generators of Delta, g_1 = g_oo.
    2508                 :            :    * We describe modular symbols by the collection phi(g_1), ..., phi(g_d)
    2509                 :            :    * \in V := Q[x,y]_{k-2}, with right Gamma action.
    2510                 :            :    * For each i = 1, .., d, let V_i \subset V be the Q-vector space of
    2511                 :            :    * allowed values for phi(g_i): with basis (P^{i,j}) given by the monomials
    2512                 :            :    * x^(j-1) y^{k-2-(j-1)}, j = 1 .. k-1
    2513                 :            :    * (g_i in E_1) or the solution of the torsion equations (1 + gamma)P = 0
    2514                 :            :    * (g_i in T2) or (1 + gamma + gamma^2)P = 0 (g_i in T31).
    2515                 :            :    *
    2516                 :            :    * The Manin relation (singlerel) is of the form \sum_i \lambda_i g_i = 0,
    2517                 :            :    * where \lambda_i = 1 if g_i in T2 or T31, and \lambda_i = (1 - \gamma_i)
    2518                 :            :    * for g_i in E1.
    2519                 :            :    *
    2520                 :            :    * If phi \in Hom_Gamma(Delta, V), it is defined by phi(g_i) := P_i in V
    2521                 :            :    * with \sum_i P_i . \lambda_i^* = 0, where (\sum n_i g_i)^* :=
    2522                 :            :    * \sum n_i \gamma_i^(-1).
    2523                 :            :    *
    2524                 :            :    * We single out gamma_1 / g_1 (g_oo in Pollack-Stevens paper) and
    2525                 :            :    * write P_{i,j} \lambda_i^* =  Q_{i,j} (\gamma_1 - 1)^* + d_{i,j} y^{k-2}
    2526                 :            :    * where d_{i,j} is a scalar and Q_{i,j} in V; we normalize Q_{i,j} to
    2527                 :            :    * that the coefficient of x^{k-2} is 0.
    2528                 :            :    *
    2529                 :            :    * There exist (s,t) such that d_{s,t} != 0.
    2530                 :            :    * A Q-basis of the (dual) space of modular symbols is given by the
    2531                 :            :    * functions phi_{i,j}, 2 <= i <= d, 1 <= j <= k-1, mapping
    2532                 :            :    *  g_1 -> d_{s,t} Q_{i,j} - d_{i,j} Q_{s,t} + [(i,j)=(s,t)] x^{k-2}
    2533                 :            :    * If i != s
    2534                 :            :    *   g_i -> d_{s,t} P_{i,j}
    2535                 :            :    *   g_s -> - d_{i,j} P_{s,t}
    2536                 :            :    * If i = s, j != t
    2537                 :            :    *   g_i -> d_{s,t} P_{i,j} - d_{i,j} P_{s,t}
    2538                 :            :    * And everything else to 0. */
    2539                 :         63 :   monomials = matid(k-1); /* represent the monomials x^{k-2}, ... , y^{k-2} */
    2540         [ +  + ]:         63 :   if (s <= nbE1) /* in E1 */
    2541                 :            :   {
    2542                 :         21 :     P_st = gel(monomials, t);
    2543                 :         21 :     lP_st = gmael(p1, s, t); /* P_{s,t} lambda_s^* */
    2544                 :            :   }
    2545                 :            :   else /* in T2, T31 */
    2546                 :            :   {
    2547                 :         42 :     P_st = gmael(gentor, s - nbE1, t);
    2548                 :         42 :     lP_st = P_st;
    2549                 :            :   }
    2550                 :         63 :   Q_st = doo_decompose(dinv, lP_st, &d_st);
    2551                 :         63 :   basis = cgetg(dim+1, t_VEC);
    2552                 :         63 :   link = cgetg(nbE1 + lg(gentor), t_VEC);
    2553                 :         63 :   gel(link,1) = cgetg(1,t_VECSMALL); /* dummy */
    2554                 :         63 :   n = 1;
    2555         [ +  + ]:         98 :   for (i = 2; i <= nbE1; i++)
    2556                 :            :   {
    2557                 :         35 :     GEN L = cgetg(k, t_VECSMALL);
    2558                 :            :     long j;
    2559                 :            :     /* link[i][j] = n gives correspondance between phi_{i,j} and basis[n] */
    2560                 :         35 :     gel(link,i) = L;
    2561         [ +  + ]:        140 :     for (j = 1; j < k; j++)
    2562                 :            :     {
    2563                 :        105 :       GEN lP_ij = gmael(p1, i, j); /* P_{i,j} lambda_i^* */
    2564                 :        105 :       GEN P_ij = gel(monomials,j);
    2565                 :        105 :       L[j] = n;
    2566                 :        105 :       gel(basis, n) = get_phi_ij(i,j,n, s,t, P_st, Q_st, d_st, P_ij, lP_ij, dinv);
    2567                 :        105 :       n++;
    2568                 :            :     }
    2569                 :            :   }
    2570         [ +  + ]:        112 :   for (u = 1; u < lg(gentor); u++,i++)
    2571                 :            :   {
    2572                 :         49 :     GEN V = gel(gentor,u);
    2573                 :         49 :     long j, lV = lg(V);
    2574                 :         49 :     GEN L = cgetg(lV, t_VECSMALL);
    2575                 :         49 :     gel(link,i) = L;
    2576         [ +  + ]:        203 :     for (j = 1; j < lV; j++)
    2577                 :            :     {
    2578                 :        154 :       GEN lP_ij = gel(V, j); /* P_{i,j} lambda_i^* = P_{i,j} */
    2579                 :        154 :       GEN P_ij = lP_ij;
    2580                 :        154 :       L[j] = n;
    2581                 :        154 :       gel(basis, n) = get_phi_ij(i,j,n, s,t, P_st, Q_st, d_st, P_ij, lP_ij, dinv);
    2582                 :        154 :       n++;
    2583                 :            :     }
    2584                 :            :   }
    2585                 :         63 :   invphiblock = cgetg(lg(link), t_VEC);
    2586                 :         63 :   gel(invphiblock,1) = cgetg(1, t_MAT); /* dummy */
    2587         [ +  + ]:        147 :   for (i = 2; i < lg(link); i++)
    2588                 :            :   {
    2589                 :         84 :     GEN M, inv, B = gel(link,i);
    2590                 :         84 :     long j, lB = lg(B);
    2591         [ +  + ]:         84 :     if (i == s) { B = vecsplice(B, t); lB--; } /* remove phi_st */
    2592                 :         84 :     M = cgetg(lB, t_MAT);
    2593         [ +  + ]:        280 :     for (j = 1; j < lB; j++)
    2594                 :            :     {
    2595                 :        196 :       GEN phi_ij = gel(basis, B[j]), pols = gel(phi_ij,3);
    2596                 :        196 :       gel(M, j) = gel(pols, 2); /* phi_ij(g_i) */
    2597                 :            :     }
    2598 [ +  + ][ +  + ]:         84 :     if (i <= nbE1 && i != s) /* maximal rank k-1 */
    2599                 :         14 :       inv = ZM_inv_denom(M);
    2600                 :            :     else /* i = s (rank k-2) or from torsion: rank k/3 or k/2 */
    2601                 :         70 :       inv = Qevproj_init(M);
    2602                 :         84 :     gel(invphiblock,i) = inv;
    2603                 :            :   }
    2604                 :         63 :   return mkvec3(WN, gen_0, mkvec5(basis, mkvecsmall2(k, dim), mkvecsmall2(s,t),
    2605                 :            :                                   link, invphiblock));
    2606                 :            : }
    2607                 :            : static GEN
    2608                 :        196 : add_star(GEN W, long sign)
    2609                 :            : {
    2610                 :        196 :   GEN s = msstar_i(W);
    2611                 :        196 :   GEN K = QM_ker(gsubgs(s, sign));
    2612                 :        196 :   gel(W,2) = mkvec3(stoi(sign), s, Qevproj_init(K));
    2613                 :        196 :   return W;
    2614                 :            : }
    2615                 :            : /* WN = msinit_N(N) */
    2616                 :            : static GEN
    2617                 :        196 : mskinit(ulong N, long k, long sign)
    2618                 :            : {
    2619                 :        196 :   GEN WN = msinit_N(N);
    2620                 :        196 :   GEN W = k == 2? mskinit_trivial(WN)
    2621         [ +  + ]:        196 :                 : mskinit_nontrivial(WN, k);
    2622                 :        196 :   return add_star(W, sign);
    2623                 :            : }
    2624                 :            : GEN
    2625                 :         98 : msinit(GEN N, GEN K, long sign)
    2626                 :            : {
    2627                 :         98 :   pari_sp av = avma;
    2628                 :            :   GEN W;
    2629                 :            :   long k;
    2630         [ -  + ]:         98 :   if (typ(N) != t_INT) pari_err_TYPE("msinit", N);
    2631         [ -  + ]:         98 :   if (typ(K) != t_INT) pari_err_TYPE("msinit", K);
    2632                 :         98 :   k = itos(K);
    2633         [ -  + ]:         98 :   if (k < 2) pari_err_DOMAIN("msinit","k", "<", gen_2,K);
    2634         [ -  + ]:         98 :   if (odd(k)) pari_err_IMPL("msinit [odd weight]");
    2635         [ -  + ]:         98 :   if (signe(N) <= 0) pari_err_DOMAIN("msinit","N", "<=", gen_0,N);
    2636         [ -  + ]:         98 :   if (equali1(N)) pari_err_IMPL("msinit [ N = 1 ]");
    2637                 :         98 :   W = mskinit(itou(N), k, sign);
    2638                 :         98 :   return gerepilecopy(av, W);
    2639                 :            : }
    2640                 :            : 
    2641                 :            : /* W = msinit, xpm modular symbol attached to elliptic curve E;
    2642                 :            :  * c t_FRAC or +oo; image of <0->c> */
    2643                 :            : GEN
    2644                 :    1916712 : Q_xpm(GEN W, GEN xpm, GEN c)
    2645                 :            : {
    2646                 :    1916712 :   pari_sp av = avma;
    2647                 :            :   GEN v, oo_0;
    2648                 :            :   long index;
    2649                 :    1916712 :   W = get_ms(W);
    2650                 :    1916712 :   v = init_act_trivial(W);
    2651                 :    1916712 :   oo_0 = gmael(W,15,1);
    2652                 :    1916712 :   index = gel(oo_0,1)[1];
    2653                 :    1916712 :   Q_log_trivial(v, W, 0, c); /* oo -> (a:b), c = a/b */
    2654                 :    1916712 :   treat_index_trivial(W, index, 1, v); /* - (oo->0) to correct to 0 -> (a:b) */
    2655                 :    1916712 :   return gerepileuptoint(av, RgV_dotproduct(xpm,v));
    2656                 :            : }
    2657                 :            : 
    2658                 :            : /* Evaluate symbol s on mspathlog B (= sum p_i g_i, p_i in Z[G]) */
    2659                 :            : static GEN
    2660                 :         42 : mseval_by_values(GEN W, GEN s, GEN p)
    2661                 :            : {
    2662                 :         42 :   long i, l, k = msk_get_weight(W);
    2663                 :            :   GEN A, B;
    2664                 :            : 
    2665         [ +  + ]:         42 :   if (k == 2)
    2666                 :            :   { /* trivial represention: don't bother with Z[G] */
    2667                 :         21 :     B = mspathlog_trivial(W,p);
    2668                 :         21 :     return RgV_dotproduct(s,B);
    2669                 :            :   }
    2670                 :            : 
    2671                 :         21 :   A = cgetg_copy(s,&l);
    2672                 :         21 :   B = mspathlog(W,p);
    2673         [ +  + ]:         63 :   for (i=1; i<l; i++) gel(A,i) = ZGl2Q_act_s(gel(B,i), gel(s,i), k);
    2674                 :         42 :   return RgV_sum(A);
    2675                 :            : }
    2676                 :            : /* evaluate symbol s on path p */
    2677                 :            : GEN
    2678                 :         77 : mseval(GEN W, GEN s, GEN p)
    2679                 :            : {
    2680                 :         77 :   pari_sp av = avma;
    2681                 :         77 :   long i, k, l, nbgen, v = 0;
    2682                 :            :   GEN e;
    2683                 :         77 :   checkms(W);
    2684                 :         77 :   k = msk_get_weight(W);
    2685                 :         77 :   nbgen = ms_get_nbgen(W);
    2686      [ +  +  - ]:         77 :   switch(typ(s))
    2687                 :            :   {
    2688                 :            :     case t_VEC: /* values s(g_i) */
    2689         [ -  + ]:          7 :       if (lg(s)-1 != nbgen) pari_err_TYPE("mseval",s);
    2690         [ +  - ]:          7 :       if (!p) return gcopy(s);
    2691                 :          0 :       v = gvar(s);
    2692                 :          0 :       break;
    2693                 :            :     case t_COL:
    2694         [ +  + ]:         70 :       if (k == 2) /* on the dual basis of (g_i) */
    2695                 :            :       {
    2696         [ -  + ]:         28 :         if (lg(s)-1 != ms_get_nbE1(W)) pari_err_TYPE("mseval",s);
    2697         [ +  + ]:         28 :         if (!p) return gtrans(s);
    2698                 :            :       }
    2699                 :            :       else
    2700                 :            :       { /* on the basis phi_{i,j} */
    2701                 :         42 :         GEN basis = msk_get_basis(W);
    2702                 :         42 :         l = lg(basis);
    2703         [ -  + ]:         42 :         if (lg(s) != l) pari_err_TYPE("mseval",s);
    2704                 :         42 :         e = const_vec(nbgen, gen_0);
    2705         [ +  + ]:        252 :         for (i=1; i<l; i++)
    2706                 :            :         {
    2707                 :        210 :           GEN phi, ind, pols, c = gel(s,i);
    2708                 :            :           long j, m;
    2709         [ +  + ]:        210 :           if (gequal0(c)) continue;
    2710                 :        133 :           phi = gel(basis,i);
    2711                 :        133 :           ind = gel(phi,2); m = lg(ind);
    2712                 :        133 :           pols = gel(phi,3);
    2713         [ +  + ]:        427 :           for (j=1; j<m; j++) {
    2714                 :        294 :             long t = ind[j];
    2715                 :        294 :             gel(e,t) = gadd(gel(e,t), gmul(c, gel(pols,j)));
    2716                 :            :           }
    2717                 :            :         }
    2718                 :         42 :         s = e;
    2719                 :            :       }
    2720                 :         63 :       break;
    2721                 :          0 :     default: pari_err_TYPE("mseval",s);
    2722                 :            :   }
    2723         [ +  + ]:         63 :   if (p)
    2724                 :            :   {
    2725                 :         42 :     s = mseval_by_values(W,s,p);
    2726 [ +  + ][ +  + ]:         42 :     if (k != 2 && is_vec_t(typ(s))) s = RgV_to_RgX(s, v);
    2727                 :            :   }
    2728                 :            :   else
    2729                 :            :   {
    2730                 :         21 :     l = lg(s);
    2731         [ +  + ]:         77 :     for (i = 1; i < l; i++)
    2732                 :            :     {
    2733                 :         56 :       GEN c = gel(s,i);
    2734         [ +  - ]:         56 :       if (!isintzero(c)) gel(s,i) = RgV_to_RgX(gel(s,i), v);
    2735                 :            :     }
    2736                 :            :   }
    2737                 :         77 :   return gerepilecopy(av, s);
    2738                 :            : }
    2739                 :            : 
    2740                 :            : #if 0
    2741                 :            : /* <0->oo>: unused */
    2742                 :            : static GEN
    2743                 :            : xpmoo(GEN W, GEN xpm)
    2744                 :            : {
    2745                 :            :   pari_sp av = avma;
    2746                 :            :   GEN v = init_act_trivial(W);
    2747                 :            :   GEN oo_0 = gmael(W,15,1);
    2748                 :            :   long index = gel(oo_0,1)[1];
    2749                 :            :   treat_index_trivial(W, index, 1, v); /* - (oo->0) to correct to 0 -> (a:b) */
    2750                 :            :   return gerepileuptoint(av, ZV_dotproduct(xpm,v));
    2751                 :            : }
    2752                 :            : #endif
    2753                 :            : 
    2754                 :            : static GEN
    2755                 :         70 : twistcurve(GEN e, GEN D)
    2756                 :            : {
    2757                 :         70 :   GEN D2 = sqri(D);
    2758                 :         70 :   GEN a4 = mulii(mulsi(-27, D2), ell_get_c4(e));
    2759                 :         70 :   GEN a6 = mulii(mulsi(-54, mulii(D, D2)), ell_get_c6(e));
    2760                 :         70 :   return ellinit(mkvec2(a4,a6),NULL,0);
    2761                 :            : }
    2762                 :            : 
    2763                 :            : /* sum_{a < |D|} (D/a)*xpm(E,a/D) */
    2764                 :            : static GEN
    2765                 :        175 : get_X(GEN W, GEN xpm, long D)
    2766                 :            : {
    2767                 :        175 :   ulong a, d = (ulong)labs(D);
    2768                 :        175 :   GEN t = gen_0;
    2769                 :        175 :   GEN nc = icopy(gen_1), c = mkfrac(nc, utoipos(d));
    2770         [ +  + ]:        728 :   for (a=1; a < d; a++)
    2771                 :            :   {
    2772                 :        553 :     long s = kross(D,a);
    2773                 :            :     GEN x;
    2774         [ +  + ]:        553 :     if (!s) continue;
    2775                 :        420 :     nc[2] = a; x = Q_xpm(W, xpm, c);
    2776         [ +  + ]:        420 :     t = (s > 0)? addii(t, x): subii(t, x);
    2777                 :            :   }
    2778                 :        175 :   return t;
    2779                 :            : }
    2780                 :            : /* quotient of the Neron periods of E^(d) and E, divided by sqrt(d) */
    2781                 :            : static long
    2782                 :         70 : get_alpha_d(GEN E, long d)
    2783                 :            : {
    2784         [ +  + ]:         70 :   if (odd(d)) return 1;
    2785         [ +  + ]:         28 :   if (!mpodd(ell_get_c4(E))) return 2; /* additive reduction at 2 */
    2786                 :            :   /* reduction not additive */
    2787 [ +  - ][ -  + ]:         70 :   return (d % 8 == 0 && !mpodd(ell_get_a1(E)))? 2: 1;
    2788                 :            : }
    2789                 :            : /* write L(E,1) = Q*w1, return the rational Q */
    2790                 :            : static GEN
    2791                 :         70 : get_Q(GEN E)
    2792                 :            : {
    2793                 :            :   GEN L, N, tam, T, n, w1;
    2794                 :            :   long ex, t, t2;
    2795                 :         70 :   E = ellanal_globalred_all(E, NULL, &N, &tam);
    2796                 :         70 :   T = elltors(E); t = itos(gel(T,1)); t2 = t*t;
    2797                 :         70 :   w1 = gel(ellR_omega(E,DEFAULTPREC), 1);
    2798                 :            : 
    2799                 :            :   /* |Sha| = n^2 */
    2800                 :         70 :   L = ellL1(E, 0, DEFAULTPREC);
    2801                 :         70 :   n = sqrtr(divrr(mulru(L, t2), mulri(w1,tam)));
    2802                 :         70 :   n = grndtoi(n, &ex);
    2803         [ -  + ]:         70 :   if (ex > -5) pari_err_BUG("msfromell (can't compute analytic |Sha|)");
    2804                 :         70 :   return gdivgs(mulii(tam,sqri(n)), t2);
    2805                 :            : }
    2806                 :            : 
    2807                 :            : /* return C such that C*L(E,1)_{xpm} = L(E,1) / w1 */
    2808                 :            : static GEN
    2809                 :         70 : ell_get_scale(GEN E, GEN W, GEN xpm, long s)
    2810                 :            : {
    2811                 :         70 :   GEN Q, X = NULL;
    2812                 :            :   long d;
    2813                 :            : 
    2814                 :            :   /* find D = s*d such that twist by D has rank 0 */
    2815         [ +  - ]:        462 :   for (d = 1; d < LONG_MAX; d++)
    2816                 :            :   {
    2817                 :        462 :     pari_sp av = avma;
    2818         [ -  + ]:        462 :     if (s < 0)
    2819         [ #  # ]:          0 :     { if (!unegisfundamental(d)) continue; }
    2820                 :            :     else
    2821         [ +  + ]:        462 :     { if (!uposisfundamental(d)) continue; }
    2822         [ -  + ]:        175 :     X = get_X(W, xpm, s < 0? -d: d);
    2823         [ +  + ]:        175 :     if (signe(X)) break;
    2824                 :        105 :     avma = av;
    2825                 :            :   }
    2826         [ -  + ]:         70 :   if (d == LONG_MAX) pari_err_BUG("msfromell (no suitable twist)");
    2827         [ -  + ]:         70 :   if (s < 0) d = -d;
    2828                 :         70 :   Q = get_Q(twistcurve(E, stoi(d)));
    2829                 :         70 :   return gdiv(gmulsg(get_alpha_d(E,d), Q), X);
    2830                 :            : }
    2831                 :            : 
    2832                 :            : GEN
    2833                 :         70 : msfromell(GEN E, long sign)
    2834                 :            : {
    2835                 :         70 :   pari_sp av = avma;
    2836                 :            :   GEN cond;
    2837                 :            :   GEN W, K, x, scale;
    2838                 :            :   ulong p, N;
    2839                 :            :   forprime_t T;
    2840                 :            : 
    2841         [ -  + ]:         70 :   if (labs(sign) != 1)
    2842                 :          0 :     pari_err_DOMAIN("msfromell","abs(sign)","!=",gen_1,stoi(sign));
    2843                 :         70 :   E = ellminimalmodel(E, NULL);
    2844                 :         70 :   cond = gel(ellglobalred(E), 1);
    2845                 :         70 :   N = itou(cond);
    2846                 :         70 :   W = mskinit(N, 2, sign);
    2847                 :            :   /* linear form = 0 on Im(S - sign) */
    2848                 :         70 :   K = keri(shallowtrans(gsubgs(msk_get_star(W), sign)));
    2849                 :            : 
    2850                 :            :   /* loop for p <= count_Manin_symbols(N) / 6 would be enough */
    2851                 :         70 :   (void)u_forprime_init(&T, 2, ULONG_MAX);
    2852         [ +  - ]:         98 :   while( (p = u_forprime_next(&T)) )
    2853                 :            :   {
    2854                 :            :     GEN Tp, ap, M, K2;
    2855         [ +  + ]:         98 :     if (N % p == 0) continue;
    2856                 :         77 :     Tp = mshecke_i(W, p);
    2857                 :         77 :     ap = ellap(E, utoipos(p));
    2858                 :         77 :     M = RgM_Rg_add_shallow(Tp, negi(ap));
    2859                 :         77 :     K2 = keri( ZM_mul(shallowtrans(M), K) );
    2860         [ +  - ]:         77 :     if (lg(K2) < lg(K)) K = ZM_mul(K, K2);
    2861         [ +  + ]:         77 :     if (lg(K2)-1 == 1) break;
    2862                 :            :   }
    2863         [ -  + ]:         70 :   if (!p) pari_err_BUG("msfromell: ran out of primes");
    2864                 :            :   /* linear form = 0 on all Im(Tp - ap) and Im(S - sign) */
    2865                 :         70 :   x = Q_primpart(gel(K,1));
    2866                 :         70 :   scale = ell_get_scale(E, W, x, sign);
    2867                 :         70 :   return gerepilecopy(av, mkvec2(W, RgC_Rg_mul(x, scale)));
    2868                 :            : }

Generated by: LCOV version 1.9