Added option for logplot

idmsweep.py

@@ -43,22 +43,27 @@ def idmgo(t, a, b, s, v, fmt=None):
     else:
         return res, idm.os(), fmt
 
-def sweep(fn, rngs, axis=None, separate=True):
+def sweep(fn, rngs, axis=None, separate=True, logx=False):
+    if logx:
+        plot = plt.semilogx
+    else:
+        plot = plt.plot
+
     defaults = [x["default"] for x in rngs]
 
     if not separate:
-        plt.plot(*fn(*defaults, fmt='k'))
+        plot(*fn(*defaults, fmt='k'))
 
     for i, r in enumerate(rngs):
         if "values" in r and r["values"] is not None:
             if separate:
-                plt.plot(*fn(*defaults, fmt='k'))
+                plot(*fn(*defaults, fmt='k'))
 
             print r["name"], defaults[i], r["values"]
             args = list(defaults)
             for dot, arg in zip(r["dots"], r["values"]):
                 args[i] = arg
-                plt.plot(*fn(*args, fmt=r["color"]+dot))
+                plot(*fn(*args, fmt=r["color"]+dot))
 
             if axis is not None:
                 plt.axis(axis)
@@ -69,11 +74,10 @@ def sweep(fn, rngs, axis=None, separate=True):
 
 if __name__ == "__main__":
 
-    '''
     ### What happens when you vary a?
     ### ( enhance )
 
-    avals = arange(20)/20. + 0.2
+    avals = linspace(0.5, 0.7, 20)
     dots = ["-"] * len(avals)
 
     arng = {"name": "a",   "color": "", "dots": dots, "default": 1.0, "values": avals }
@@ -85,7 +89,8 @@ if __name__ == "__main__":
     rngs = [trng, arng, brng, srng, vrng]
 
     sweep(idmgo, rngs, axis=[0,0.1,-.05,.45])
-    '''
+    go = linfty(0.5, n=50, tmax=500)
+    sweep(go, rngs)
 
     '''
     ### What happens when you vary IDM parameters?
@@ -102,6 +107,7 @@ if __name__ == "__main__":
     sweep(idmgo, rngs, axis=[0,1,-.2,.2])
     '''
 
+    '''
     ### What happens when you vary IDM parameters?
     dots = (':', '--', '-', '.-')
 
@@ -115,6 +121,7 @@ if __name__ == "__main__":
 
     go = linfty(0.5, n=50, tmax=500)
     sweep(go, rngs)
+    '''
 
     '''
     go = linfty(0.05, n=80, tmax=500, timeplot=True)