1 #!/usr/bin/env python

     2 # 12.2

     3 

     4 import os, sys

     5 sys.path += [os.getcwdu() + os.sep + ".." + os.sep + "python"]

     6 

     7 import pylab as pl

     8 from xdync import xdync

     9 from polyfuncs import polmul

    10 

    11 def gpc_bas(A, dA, B, dB, N, k, rho):

    12     D, dD = pl.array([1, -1]), 1

    13     AD, dAD = pl.convolve(A, D), dA+1

    14     Nu = N+1

    15     zj, dzj = 1, 0

    16     G = pl.zeros((Nu, 1))

    17     H1 = pl.zeros((Nu, k-1+dB))

    18     H2 = pl.zeros((Nu, dA+1))

    19 

    20     for j in range(Nu):

    21         zj = pl.convolve(zj, [0,1])

    22         dzj = dzj + 1

    23         Fj, dFj, Ej, dEj = xdync(zj, dzj, AD, dAD, 1, 0)[:4]

    24         Gj, dGj = polmul(B, dB, Ej, dEj)

    25         m, n = G.shape

    26         G = pl.column_stack((G, pl.zeros((m, dGj-n))))

    27         G[j, :dGj] = Gj[dGj-1::-1]

    28         H1[j,:k-1+dB] = Gj[dGj:dGj+k-1+dB]

    29         H2[j,:dA+1] = Fj

    30     

    31     K = pl.dot(pl.inv(pl.dot(G.T, G) + rho*pl.eye(Nu,Nu)), G.T)

    32     KH1 = pl.dot(K, H1)

    33     KH2 = pl.dot(K, H2)

    34     R1 = pl.concatenate(([1], KH1[0,:]))

    35     dR1 = len(R1)-1

    36     Sc = KH2[0,:]

    37     dSc = len(Sc)-1;

    38     Tc = K[0,:]

    39     dTc = len(Tc)-1;

    40         

    41     return K, KH1, KH2, Tc, dTc, Sc, dSc, R1, dR1