Update .gitignore

.gitignore

@@ -1,5 +1,4 @@
 Vivado
-simulation
 hardware/PhasedArray4x4
 hardware/PhasedArray4x4 (without SR)
 Vitis/.metadata

simulation/simulate.py

@@ -0,0 +1,53 @@
+import numpy as np
+import matplotlib.pyplot as plt
+from matplotlib.colors import LinearSegmentedColormap
+from matplotlib.animation import FuncAnimation
+
+source = 10
+#sources = [-3*source, -2*source, -source, 0, source, 2*source, 3*source]
+sources = np.linspace(-20.0, 20.0, 5)
+axlim = max(sources)*2 + 1
+N = 1000
+f = 1
+
+x = np.linspace(-axlim, axlim, N)
+y = np.linspace(-2*axlim, 0, N)
+X, Y = np.meshgrid(x, y)
+
+
+norm = plt.Normalize(-5, 5)
+cmap = LinearSegmentedColormap.from_list('', ['blue', 'black', 'red'])
+fig, ax = plt.subplots()
+plt.gca().set_axis_off()
+plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, hspace = 0, wspace = 0)
+plt.margins(0,0)
+
+w = 0
+h = 30
+x = np.linspace(min(sources), max(sources), len(sources))
+pos = np.sqrt((-w - x)**2 + h**2)/8.575
+
+def update(p):
+    ax.cla()
+    Z = 0
+    
+    for i in range(len(sources)):
+        Z += np.sin(np.pi*f*(np.sqrt((X - sources[i])**2 + Y**2)/8.575 - p/10 - pos[i]))
+
+    ax.imshow(Z,
+              cmap = cmap,
+              norm = norm)
+
+    for s in sources:
+        ax.plot(N/2*(1 - s/axlim), N-1, 'ro')
+
+    ax.plot(N/2*(1 - w/axlim), N/2*(2-h/axlim), 'ro')
+
+    #ax.set_title(f'f = {40} KHz')
+    ax.set_aspect('equal')
+    ax.axis('off')
+
+ani = FuncAnimation(fig = fig, func = update, frames=60, interval = 3/8.575)
+ani.save('C:/Ultrasonic_phased_array/simulation/animation.gif', writer='imagemagick', fps=30)
+
+plt.show()