#!/usr/bin/python
# 13.6
# Viscosity control problem of MacGregor

import scipy as sp
from pylab import plot, show
from lqg import lqg
from cl import cl

A, dA = sp.array([1, -0.44]), 1
B, dB = sp.array([0.51, 1.21]), 1
F, dF = sp.array([1, -1]), 1
C, dC = sp.array([1, -0.44]),1
k = 1
V, W = 1, 1
dV, dW = 0, 0
u_lqg = []
y_lqg =[]
uy_lqg = []
int1 = 1

for rho in sp.arange(0.001, 3, 0.1):
    R1,dR1,Sc,dSc = lqg(A,dA,B,dB,C,dC,k,rho,V,dV,W,dW,F,dF)
    Nu,dNu,Du,dDu,Ny,dNy,Dy,dDy,yvar,uvar = cl(A,dA,B,dB,C,dC,k,Sc,dSc,R1,dR1,int1)
    uvar = sp.atleast_1d(uvar.squeeze())
    yvar = sp.atleast_1d(yvar.squeeze())
    u_lqg = sp.concatenate((u_lqg, uvar))
    y_lqg = sp.concatenate((y_lqg, yvar))
    if sp.size(uy_lqg) == 0:
        uy_lqg = sp.array([rho, uvar[0], yvar[0]])
    else:
        uy_lqg = sp.vstack((uy_lqg, sp.array([rho, uvar[0], yvar[0]])))

plot(u_lqg, y_lqg)
show()