#include "pq_defs.h"
#include "pcp_vars.h"

void print_Magma_word ();

int Magma_countcall = 0;

/* print the power and commutator relations in Magma format so 
   that the file can be used as input to the Magma pquotient command */

void Magma_presentation (file, pcp)
FILE_TYPE file;
struct pcp_vars *pcp;
{
#include "define_y.h"

   register int i;
   register int j;
   register int k;
   register int l;
   register int p1;
   register int p2;
   register int weight;

   fprintf (file, "F< ");
   for (i = 1; i < pcp->lastg; ++i) {
      fprintf (file, "x%d, ", i); 
      if (i % 15 == 0) 
         fprintf (file, "\n");
   }
   fprintf (file, "x%d> := FreeGroup (%d);\n", i, pcp->lastg); 
   fprintf (file, "Q := quo < F | \n");

   k = y[pcp->clend + pcp->cc - 1];

   for (i = 1; i <= k; i++) {
      p2 = y[pcp->ppower + i];
      fprintf (file, " x%d^%d =", i, pcp->p);
      print_Magma_word (file, p2, pcp);
      fprintf (file, ",\n");
   }

   /*
     for (i = k + 1; i <= pcp->lastg; ++i) 
     fprintf (file, " x%d^%d = 1,\n", i, pcp->p);
     */

   for (i = 2; i <= k; i++) {
      weight = WT(y[pcp->structure + i]);
      p1 = y[pcp->ppcomm + i];
      l = MIN(i - 1, y[pcp->clend + pcp->cc - weight]);
      for (j = 1; j <= l; j++) {
         p2 = y[p1 + j];
         fprintf (file, "(x%d, x%d) =", i, j);
         print_Magma_word (file, p2, pcp);
         if (i == k && j == l)
            fprintf (file, ">;\n");
         else 
            fprintf (file, ",\n");
      }
   }

   /* 
     for (i = k + 1; i <= pcp->lastg; ++i) {
     for (j = 1; j <= k; ++j) {
     fprintf (file, "(x%d, x%d) = 1", i, j);
     if (i == pcp->lastg && j == k)
     fprintf (file, ">;\n");
     else
     fprintf (file, ",\n");
     }
     }
     */

   fprintf (file, "G := pQuotient (Q, %d, %d: ", pcp->p, pcp->cc);
   if (pcp->extra_relations != 0) 
      fprintf (file, "Exponent := %d, ", pcp->extra_relations);
   fprintf (file, "Print := 0);\n");
}

/* write Magma library file in form suitable for reading into Magma */

void write_Magma_library (file, pcp)
FILE_TYPE file;
struct pcp_vars *pcp;
{
   if (Magma_countcall == 0)
      fprintf (file, "GroupList := [];\n");

   Magma_countcall++;

   Magma_presentation (file, pcp);

   fprintf (file, "GroupList[#GroupList + 1] := G;\n");

   /* special lines -- replace by whatever computation is desired */
   /*
     fprintf (file, "j = jennings series (g);\n");
     fprintf (file, "print j;\n");
     */
}

/* print out a word of a pcp presentation in Magma format */

void print_Magma_word (file, ptr, pcp)
FILE_TYPE file;
int ptr;
struct pcp_vars *pcp;
{
#include "define_y.h"

   register int gen, exp;
   register int i;
   register int count;
   register int length;

   if (ptr == 0)
      fprintf (file, " 1");
   else if (ptr > 0)
      fprintf (file, " x%d", ptr);
   else {
      length = 0; 
      ptr = -ptr + 1;
      count = y[ptr];
      for (i = 1; i <= count; i++) {
         exp = FIELD1 (y[ptr + i]);
         gen = FIELD2 (y[ptr + i]);
         fprintf (file, " x%d", gen);
         length += 4;
         if (exp != 1) {
            fprintf (file, "^%d", exp);
            length += 3;
         }
         if (i != count) {
            fprintf (file, " *");
            length += 2;
            if (length >= 70) {
               length = 0;
               fprintf (file, "\n");
            }
         }
      }
   }
}