Laboratoire de l'Informatique et du Parallélisme » Pobyso

load "/home/storres/recherche/arithmetique/pobysoPythonSage/src/sageSLZ/sageMatrixOperations.sage"

print "sagePolynomialOperations loading..."

def spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                            pCoefficients, 

                                            knownMonomials, 

                                            protoMatrixRows, 

                                            columnsWidth=0):

    """

    For a given polynomial (under the form of monomials and coefficents lists),

    add the coefficients of the protoMatrix (a list of proto matrix rows).

    Coefficients are added to the protoMatrix row in the order imposed by the 

    monomials discovery list (the knownMonomials list) built as construction

    goes on. 

    As a bonus, data can be printed out for a visual check.

    pMonomials     : the list of the monomials coming form some polynomial;

    pCoefficients  : the list of the corresponding coefficients to add to

                     the protoMatrix in the exact same order as the monomials;

    knownMonomials : the list of the already knonw monomials;

    protoMatrixRows: a list of lists, each one holding the coefficients of the 

                     monomials

    columnWith     : the width, in characters, of the displayed column ; if 0, 

                     do not display anything.

    """

    # We have started with the smaller degrees in the first variable.

    pMonomials.reverse()

    pCoefficients.reverse()

    # New empty proto matrix row.

    protoMatrixRowCoefficients = []

    # We work according to the order of the already known monomials

    # No known monomials yet: add the pMonomials to knownMonomials 

    # and add the coefficients to the proto matrix row.

    if len(knownMonomials) == 0: 

        for pmIdx in xrange(0, len(pMonomials)):

            knownMonomials.append(pMonomials[pmIdx])

            protoMatrixRowCoefficients.append(pCoefficients[pmIdx])

            if columnsWidth != 0:

                monomialAsString = str(pCoefficients[pmIdx]) + " " + \

                                   str(pMonomials[pmIdx])

                print monomialAsString, " " * \

                      (columnsWidth - len(monomialAsString)),

    # There are some known monomials. We search for them in pMonomials and 

    # add their coefficients to the proto matrix row.

    else: 

        for knownMonomialIndex in xrange(0,len(knownMonomials)):

            # We lazily use an exception here since pMonomials.index() function

            # may fail throwing the ValueError exception.

            try:

                indexInPmonomials = \

                    pMonomials.index(knownMonomials[knownMonomialIndex])

                if columnsWidth != 0:

                    monomialAsString = str(pCoefficients[indexInPmonomials]) + \

                        " " + str(knownMonomials[knownMonomialIndex])

                    print monomialAsString, " " * \

                        (columnsWidth - len(monomialAsString)),

                # Add the coefficient to the proto matrix row and delete the \

                # known monomial from the current pMonomial list 

                #(and the corresponding coefficient as well).

                protoMatrixRowCoefficients.append(pCoefficients[indexInPmonomials])

                del pMonomials[indexInPmonomials]

                del pCoefficients[indexInPmonomials]

            # The knownMonomials element is not in pMonomials

            except ValueError:   

                protoMatrixRowCoefficients.append(0)

                if columnsWidth != 0:

                    monomialAsString = "0" + " "+ \

                        str(knownMonomials[knownMonomialIndex])

                    print monomialAsString, " " * \

                        (columnsWidth - len(monomialAsString)),

        # End for knownMonomialKey loop. 

        # We now append the remaining monomials of pMonomials to knownMonomials 

        # and the corresponding coefficients to proto matrix row.

        for pmIdx in xrange(0, len(pMonomials)):

            knownMonomials.append(pMonomials[pmIdx])

            protoMatrixRowCoefficients.append(pCoefficients[pmIdx])

            if columnsWidth != 0:

                monomialAsString = str(pCoefficients[pmIdx]) + " " \

                    + str(pMonomials[pmIdx])

                print monomialAsString, " " * \

                    (columnsWidth - len(monomialAsString)),

        # End for pmIdx loop.

    # Add the new list row elements to the proto matrix.

    protoMatrixRows.append(protoMatrixRowCoefficients)

    if columnsWidth != 0:

        print    

# End spo_add_polynomial_coeffs_to_matrix_row

def spo_expression_as_string(powI, powT, powP, alpha):

    """

    Computes a string version of the i^k + t^l + p^m + N^n expression for

    output.

    """

    expressionAsString =""

    if powI != 0:

        expressionAsString += "i^" + str(powI)

    if powT != 0:

        if len(expressionAsString) != 0:

            expressionAsString += " * "

        expressionAsString += "t^" + str(powT)

    if powP != 0:

        if len(expressionAsString) != 0:

            expressionAsString += " * "

        expressionAsString += "p^" + str(powP)

    if (alpha - powP) != 0 :

        if len(expressionAsString) != 0:

            expressionAsString += " * "

        expressionAsString += "N^" + str(alpha - powP)

    return(expressionAsString)

# End spo_expression_as_string.

def spo_norm(poly, p=2):

    """

    Behaves more or less (no infinity defined) as the norm for the

    univariate polynomials.

    Quoting the Sage documentation:

    Definition: For integer p, the p-norm of a polynomial is the pth root of 

    the sum of the pth powers of the absolute values of the coefficients of 

    the polynomial.

    """

    # TODO: check the arguments (for p see below)..

    norm = 0

    # For infinity norm.

    if p == Infinity:

        for coefficient in poly.coefficients():

            coefficientAbs = coefficient.abs()

            if coefficientAbs > norm:

                norm = coefficientAbs

        return norm

    # TODO: check here the value of p

    # p must be an integer.

    if int(p) != p:

        return None

    # p must be >= 1.

    if p < 1:

        return None

    # For 1 norm.

    if p == 1:

        for coefficient in poly.coefficients():

            norm +=  coefficient.abs()

        return norm

    # For other norms

    for coefficient in poly.coefficients():

        norm +=  coefficient.abs()^p

    return pow(norm, 1/p)

# end spo_norm

def spo_polynomial_to_proto_matrix(p, alpha, N, columnsWidth=0):

    """

    From a (bivariate) polynomial and some other parameters build a proto 

    matrix (an array of "rows") to be converted into a "true" matrix and 

    eventually by reduced by fpLLL.

    The matrix is such as those found in Boneh-Durphee and Stehlé.

    Parameters

    ----------

    p: the (bivariate) polynomial;

    pRing: the ring over which p is defined;

    alpha:

    N:

    columsWidth: if == 0, no information is displayed, otherwise data is 

                 printed in colums of columnsWitdth width.

    """

    pRing = p.parent()

    knownMonomials = []

    protoMatrixRows = []

    polynomialsList = []

    pVariables = p.variables()

    iVariable = pVariables[0]

    tVariable = pVariables[1]

    polynomialAtPower = pRing(1)

    currentPolynomial = pRing(1)

    pIdegree = p.degree(pVariables[0])

    pTdegree = p.degree(pVariables[1])

    currentIdegree = currentPolynomial.degree(iVariable)

    nAtPower = N^alpha

    nAtAlpha = nAtPower

    polExpStr = ""

    # We work from p^0 * N^alpha to p^alpha * N^0

    for pPower in xrange(0, alpha + 1):

        # pPower == 0 is a special case. We introduce all the monomials but one

        # in i and those in t necessary to be able to introduce

        # p. We arbitrary choose to introduce the highest degree monomial in i

        # with p. We also introduce all the mixed i^k * t^l monomials with

        # k < p.degree(i) and l <= p.degree(t).

        # Mixed terms introduction is necessary here before we start "i shifts"

        # in the next iteration.

        if pPower == 0:

            # Notice that i^pIdegree is excluded as the bound of the xrange is

            # pIdegree

            for iPower in xrange(0, pIdegree): 

                for tPower in xrange(0, pTdegree + 1):

                    if columnsWidth != 0:

                        polExpStr = spo_expression_as_string(iPower, 

                                                             tPower, 

                                                             pPower, 

                                                             alpha)

                        print "->", polExpStr

                    currentExpression = iVariable^iPower * \

                                        tVariable^tPower * nAtAlpha

                    # polynomialAtPower == 1 here. Next line should be commented

                    # out but it does not work! Some conversion problem?

                    currentPolynomial = pRing(currentExpression)

                    polynomialsList.append(polExpStr) 

                    pMonomials = currentPolynomial.monomials()

                    pCoefficients = currentPolynomial.coefficients()

                    spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                            pCoefficients, 

                                                            knownMonomials, 

                                                            protoMatrixRows, 

                                                            columnsWidth)

                # End tPower.

            # End for iPower.

        else: # pPower > 0: (p^1..p^alpha)

            # This where we introduce the p^pPower * N^(alpha-pPower)

            # polynomial.

            # This step could technically be fused as the first iteration

            # of the next loop (with iPower starting at 0).

            # We set it apart for clarity.

            if columnsWidth != 0:

                polExpStr = spo_expression_as_string(0, 0, pPower, alpha)

                print "->", polExpStr

            currentPolynomial = polynomialAtPower * nAtPower

            polynomialsList.append(polExpStr) 

            pMonomials = currentPolynomial.monomials()

            pCoefficients = currentPolynomial.coefficients()

            spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                    pCoefficients, 

                                                    knownMonomials, 

                                                    protoMatrixRows, 

                                                    columnsWidth)

            # The i^iPower * p^pPower polynomials: they add i^k monomials to  

            # p^pPower up to k < pIdegree * pPower. This only introduces i^k 

            # monomials since mixed terms (that were introduced at a previous

            # stage) are only shifted to already existing 

            # ones. p^pPower is "shifted" to higher degrees in i as far as 

            # possible, one step short of the degree in i of p^(pPower+1) .

            # These "pure" i^k monomials can only show up with i multiplications.

            for iPower in xrange(1, pIdegree):

                if columnsWidth != 0:

                    polExpStr = spo_expression_as_string(iPower, \

                                                         0,      \

                                                         pPower, \

                                                         alpha)

                    print "->", polExpStr

                currentExpression = i^iPower * nAtPower

                currentPolynomial = pRing(currentExpression) * polynomialAtPower

                polynomialsList.append(polExpStr) 

                pMonomials = currentPolynomial.monomials()

                pCoefficients = currentPolynomial.coefficients()

                spo_add_polynomial_coeffs_to_matrix_row(pMonomials,      \

                                                        pCoefficients,   \

                                                        knownMonomials,  \

                                                        protoMatrixRows, \

                                                        columnsWidth)

            # End for iPower

            # We want now to introduce a t * p^pPower polynomial. But before

            # that we must introduce some mixed monomials.

            # This loop is no triggered before pPower == 2.

            # It introduces a first set of high i degree mixed monomials.

            for iPower in xrange(1, pPower):

                tPower = pPower - iPower + 1

                if columnsWidth != 0:

                    polExpStr = spo_expression_as_string(iPower * pIdegree, 

                                                         tPower, 

                                                         0,  

                                                         alpha)

                    print "->", polExpStr

                currentExpression = i^(iPower * pIdegree) * t^tPower * nAtAlpha

                currentPolynomial = pRing(currentExpression)

                polynomialsList.append(polExpStr) 

                pMonomials = currentPolynomial.monomials()

                pCoefficients = currentPolynomial.coefficients()

                spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                        pCoefficients, 

                                                        knownMonomials, 

                                                        protoMatrixRows, 

                                                        columnsWidth)

            # End for iPower

            #

            # This is the mixed monomials main loop. It introduces:

            # - the missing mixed monomials needed before the 

            #   t^l * p^pPower * N^(alpha-pPower) polynomial;

            # - the t^l * p^pPower * N^(alpha-pPower) itself;

            # - for each of i^k * t^l * p^pPower * N^(alpha-pPower) polynomials:

            #   - the the missing mixed monomials needed  polynomials,

            #   - the i^k * t^l * p^pPower * N^(alpha-pPower) itself.

            # The t^l * p^pPower * N^(alpha-pPower) is introduced when

            # 

            for iShift in xrange(0, pIdegree):

                # When pTdegree == 1, the following loop only introduces 

                # a single new monomial.

                #print "++++++++++"

                for outerTpower in xrange(1, pTdegree + 1):

                    # First one high i degree mixed monomial.

                    iPower = iShift + pPower * pIdegree

                    if columnsWidth != 0:

                        polExpStr = spo_expression_as_string(iPower, 

                                                             outerTpower,

                                                             0,

                                                             alpha)

                        print "->", polExpStr

                    currentExpression = i^iPower * t^outerTpower * nAtAlpha

                    currentPolynomial = pRing(currentExpression)

                    polynomialsList.append(polExpStr) 

                    pMonomials = currentPolynomial.monomials()

                    pCoefficients = currentPolynomial.coefficients()

                    spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                            pCoefficients, 

                                                            knownMonomials, 

                                                            protoMatrixRows, 

                                                            columnsWidth)

                    #print "+++++"

                    # At iShift == 0, the following innerTpower loop adds  

                    # duplicate monomials, since no extra i^l * t^k is needed 

                    # before introducing the  

                    # i^iShift * t^outerPpower * p^pPower * N^(alpha-pPower) 

                    # polynomial.

                    # It introduces smaller i degree monomials than the

                    # one(s) added previously (no pPower multiplication).

                    # Here the exponent of t decreases as that of i increases.

                    # This conditional is not entered before pPower == 1.

                    # The innerTpower loop does not produce anything before

                    # pPower == 2. We keep it anyway for other configuration of

                    # p.

                    if iShift > 0:

                        iPower = pIdegree + iShift

                        for innerTpower in xrange(pPower, 1, -1):

                            if columnsWidth != 0:

                                polExpStr = spo_expression_as_string(iPower, 

                                                                     innerTpower,

                                                                     0,

                                                                     alpha) 

                            currentExpression = \

                                    i^(iPower) * t^(innerTpower) * nAtAlpha

                            currentPolynomial = pRing(currentExpression)

                            polynomialsList.append(polExpStr) 

                            pMonomials = currentPolynomial.monomials()

                            pCoefficients = currentPolynomial.coefficients()

                            spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                                pCoefficients, 

                                                                knownMonomials, 

                                                                protoMatrixRows, 

                                                                columnsWidth)

                            iPower += pIdegree

                        # End for innerTpower

                    # End of if iShift > 0

                    # When iShift == 0, just after each of the  

                    # p^pPower * N^(alpha-pPower) polynomials has 

                    # been introduced (followed by a string of 

                    # i^k * p^pPower * N^(alpha-pPower) polynomials) a

                    # t^l *  p^pPower * N^(alpha-pPower) is introduced here.

                    # 

                    # Eventually, the following section introduces the 

                    # i^iShift * t^outerTpower * p^iPower * N^(alpha-iPower) 

                    # polynomials.

                    if columnsWidth != 0:

                        polExpStr = spo_expression_as_string(iShift, 

                                                             outerTpower,

                                                             pPower,

                                                             alpha)

                        print "->", polExpStr

                    currentExpression = i^iShift * t^outerTpower * nAtPower

                    currentPolynomial = pRing(currentExpression) * polynomialAtPower

                    polynomialsList.append(polExpStr) 

                    pMonomials = currentPolynomial.monomials()

                    pCoefficients = currentPolynomial.coefficients()

                    spo_add_polynomial_coeffs_to_matrix_row(pMonomials, 

                                                            pCoefficients, 

                                                            knownMonomials, 

                                                            protoMatrixRows, 

                                                            columnsWidth)

                # End for outerTpower 

                #print "++++++++++"

            # End for iShift

        polynomialAtPower *= p  

        nAtPower /= N

    # End for pPower loop

    return ((protoMatrixRows, knownMonomials, polynomialsList))

# End spo_polynomial_to_proto_matrix

def spo_proto_to_column_matrix(protoMatrixColumns, \

                               knownMonomialsList, \

                               boundVar1, \

                               boundVar2):

    """

    Create a column (each row holds the coefficients of one monomial) matrix.

    Parameters

    ----------

    protoMatrixColumns: a list of coefficient lists.

    """

    numColumns = len(protoMatrixColumns)

    if numColumns == 0:

        return None

    # The last column holds has the maximum length. 

    numRows = len(protoMatrixColumns[numColumns-1])

    if numColumns == 0:

        return None

    baseMatrix = matrix(ZZ, numRows, numColumns)

    for colIndex in xrange(0, numColumns):

        for rowIndex in xrange(0, len(protoMatrixColumns[colIndex])):

            if protoMatrixColumns[colIndex][rowIndex] != 0:

                baseMatrix[rowIndex, colIndex] = \

                    protoMatrixColumns[colIndex][rowIndex] * \

                    knownMonomialsList[rowIndex](boundVar1, boundVar2)

    return baseMatrix

# End spo_proto_to_column_matrix.

#

def spo_proto_to_row_matrix(protoMatrixRows, \

                            knownMonomialsList, \

                            boundVar1, \

                            boundVar2):

    """

    Create a row (each column holds the evaluation one monomial at boundVar1 and

    boundVar2 values) matrix.

    Parameters

    ----------

    protoMatrixRows: a list of coefficient lists.

    """

    numRows = len(protoMatrixRows)

    if numRows == 0:

        return None

    # The last row is the longest one.

    numColumns = len(protoMatrixRows[numRows-1])

    if numColumns == 0:

        return None

    baseMatrix = matrix(ZZ, numRows, numColumns)

    for rowIndex in xrange(0, numRows):

        for colIndex in xrange(0, len(protoMatrixRows[rowIndex])):

            if protoMatrixRows[rowIndex][colIndex] !=  0:

                baseMatrix[rowIndex, colIndex] = \

                    protoMatrixRows[rowIndex][colIndex] * \

                    knownMonomialsList[colIndex](boundVar1,boundVar2)

            #print rowIndex, colIndex,

            #print protoMatrixRows[rowIndex][colIndex],

            #print knownMonomialsList[colIndex](boundVar1,boundVar2)

    return baseMatrix

# End spo_proto_to_row_matrix.

#

print "\t...sagePolynomialOperations loaded"