53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
|
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()
|