untested cleanup

generators/refac/lorenz.py

@@ -0,0 +1,59 @@
+import numpy as np
+import sys
+import os
+sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), '..'))
+from scipy.integrate import odeint
+import matplotlib.pyplot as plt
+from mpl_toolkits.mplot3d import Axes3D
+from Program import Program
+from Command import Command
+
+
+# Lorenz paramters and initial conditions
+sigma, beta, rho = 10, 2.667, 28
+u0, v0, w0 = 0, 1, 1.05
+
+# Maximum time point and total number of time points
+tmax, n = 100, 10000
+
+def lorenz(X, t, sigma, beta, rho):
+    """The Lorenz equations."""
+    u, v, w = X
+    up = -sigma*(u - v)
+    vp = rho*u - v - u*w
+    wp = -beta*w + u*v
+    return up, vp, wp
+
+# Integrate the Lorenz equations on the time grid t
+t = np.linspace(0, tmax, n)
+f = odeint(lorenz, (u0, v0, w0), t, args=(sigma, beta, rho))
+x, y, z = f.T
+
+pts=[]
+
+
+_x = x+y
+_y=z
+
+pts=(_x,_y)
+
+print(pts)
+plt.plot(_x,_y)
+# Plot the Lorenz attractor using a Matplotlib 3D projection
+fig = plt.figure()
+ax = fig.gca(projection='3d')
+plt.show()
+
+
+def lorenz(plt):
+
+	list=hilbert(min(plt.winsize[0],plt.winsize[1]),penthickness)
+	return Program([Command('IN'),Command('SP1'),Command('PU',*list[0])]\
+	+[Command('PD',*p) for p in plt]\
+	+[Command('PU')])
+	
+
+# Remove all the axis clutter, leaving just the curve.
+#ax.set_axis_off()
+
+