Skip to content
Snippets Groups Projects
abstable.py 1.8 KiB
Newer Older
from matplotlib import pyplot as plt
from numpy import arange, zeros, linspace, sign

from idmlin import IDMLin


t = 1.5
amin, amax, acount = 0.2, 0.6, 15
bmin, bmax, bcount = 2, 4, 20

arng = linspace(amin, amax, acount)
brng = linspace(bmin, bmax, bcount)
s0 = 2
v0 = 30
re = 0.5

ns = (12, 15, 20, 50)

#f, axarr = plt.subplots(2, len(ns)+1, sharex=True, sharey=True)
f, axarr = plt.subplots(2, 2, sharex=True, sharey=True)

allstab = zeros((len(arng), len(brng)))
alloss = zeros((len(arng), len(brng)))

for i, n in enumerate(ns):
    stab = zeros((len(arng), len(brng)))
    oss = zeros((len(arng), len(brng)))

    def isstable(a, b):
            idm = IDMLin(t=t, a=a, b=b, s0=s0)
            idm.go(v0, re)
            if idm.isloopstable(n):
                return 1, sign(idm.os())
            else:
                return 0, sign(idm.os())

    for ai, a in enumerate(arng):
        for bi, b in enumerate(brng):
            stab[ai, bi], oss[ai, bi] = isstable(a,b)


    axarr[0, i*0].imshow(stab, cmap=plt.cm.Reds, 
                    interpolation='none', 
                    extent=[bmin, bmax, amax, amin], 
                    aspect='auto',
                    )
    axarr[1, i*0].imshow(oss, cmap=plt.cm.Reds, 
                    interpolation='none', 
                    extent=[bmin, bmax, amax, amin], 
                    aspect='auto',
                    )
    allstab += stab
    alloss += oss

axarr[0, -1].imshow(allstab, cmap=plt.cm.Reds, 
                interpolation='none', 
                extent=[bmin, bmax, amax, amin], 
                aspect='auto',
                )
axarr[1, -1].imshow(alloss, cmap=plt.cm.Reds, 
                interpolation='none', 
                extent=[bmin, bmax, amax, amin], 
                aspect='auto',
                )
allstab += stab
alloss += oss

plt.show()