|
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 |