#!/usr/bin/python

# Superposes two polynomial matrices. 

import pylab as pl

from polyfuncs import polsize

def rowjoin(P1, degP1, P2, degP2):

    rP1, cP1 = polsize(P1, degP1)

    rP2, cP2 = polsize(P2, degP2)

    if not cP1 == cP2:

        print 'rowjoin: Inconsistent numbers of columns'

        exit()

    rP = rP1 + rP2

    cP = cP1

    degP = max(degP1, degP2)

    P = pl.atleast_2d(pl.zeros((rP,(degP+1)*cP)))

    P[0:rP1,0:(degP1+1)*cP1] = P1

    P[rP1:rP,0:(degP2+1)*cP2] = P2

    return P, degP

if __name__== "__main__":

    s = """P = pl.array([[1, 2, 3, 4, 1, 2, 3, 4],

           [9, 10, 11, 12, 9, 10, 11, 12],

           [5, 6, 7, 8, 5, 6, 7, 8]])"""

    print s

    exec(s)

    print """rowjoin(P, 3, P, 3)"""

    print rowjoin(P, 3, P, 3)