#!/usr/bin/python

import pylab as pl
from rowjoin import rowjoin
from polyfuncs import polsize
from t1calc import t1calc
from seshft import seshft
from colsplit import colsplit
from clcoef import clcoef

def left_prm(N, degN, D, degD, job=1, gap=None):
    """ """
    if gap == None:
        gap = 1.0e8
    
    F, degF = rowjoin(D, degD, N, degN)
    Frows, Fbcols = polsize(F, degF)
    Fcols = Fbcols * (degF+1)
    T1 = pl.array([])
    pr = pl.array([]) 
    degT1, T1rows, shft = 0, 0, 0
    S=F
    sel = pl.ones((Frows,1))
    T1bcols =1
    abar = pl.arange(Fbcols, Frows)
    if T1rows:
        T1 = pl.hstack((T1, pl.zeros((T1rows, Frows))))
    if T1.size == 0:
        T1rows = 0
    while T1.size == 0 or T1rows < (Frows - Fbcols):
        if T1.size == 0:
            T1rows = 0
        Srows = Frows*T1bcols     
        T1, T1rows, sel, pr = t1calc(S, Srows, T1, T1rows, sel, pr, Frows, Fbcols, abar, gap)
        T1rows, T1cols = pl.atleast_2d(T1).shape
        if T1.size == 0:
            T1rows = 0
        if T1rows < (Frows - Fbcols):
            if T1.size != 0:
                T1 = pl.hstack((T1, pl.zeros((T1rows,Frows)).squeeze()))
            T1bcols += 1
            degT1 += 1
            shft += Fbcols
            S = seshft(S,F,shft);
            sel = pl.hstack((sel,sel[Srows-Frows:Srows+1]))
            rowvec = pl.arange((T1bcols-1)*Frows+(Fbcols), (T1bcols * Frows))
            abar = pl.hstack((abar, rowvec))

    B, degB, A, degA = colsplit(T1, degT1, Fbcols, Frows-Fbcols)
    B, degB = clcoef(B, degB)
    B = -B;
    A, degA = clcoef(A, degA)
    if job == 1:
        return B, degB, A, degA

    if job == 2:
        S = S[sel!=0,:]
        redSrows, Scols = S.shape
        C = pl.hstack(pl.eye(Fbcols), pl.zeros(Fbcols,Scols-Fbcols))
        T2 = pl.dot(C, pl.inv(S))
        T2 = makezero(T2,gap);
        T2 = move(T2, pl.find(sel), Srows);
        X, degX, Y, degY = colsplit(T2, degT1, Fbcols, Frows - Fbcols);
        X, degX = clcoef(X, degX);
        Y, degY = clcoef(Y, degY);
    elif job == 3:
        Y = S
        degY = sel
        X = degT1
        degX = Fbcols

    else:
        print 'Message from left_prm:no legal job number specified'
        exit()

    return B, degB, A, degA, Y, degY, X, degX


if __name__== "__main__":
    D = pl.array([[1, 0, 0, 0, 0, 0],
                  [0, 1, 0, 1, 0, 0],
                  [0, 0, 1, 1, 1, 0]])

    N = pl.array([[1, 0, 0],
                  [0, 1, 0],
                  [0, 0, 1]])

    dD, dN = 1, 0

    result = left_prm(N, dN, D, dD)
    print result
    pass