ZYM/cumcm2024-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

write orbit

Browse Source
This commit is contained in:
ZhuangYumin
2024-09-07 12:35:43 +08:00
parent ef0ad25434
commit caa0a9e2a1
3 changed files with 144 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
A/4/testplot.py
View File

@@ -5,10 +5,10 @@ kPitch = 1.7
kAlpha = kPitch / (2 * np.pi)
kCriticalRadius = 4.5
theta_max = (kCriticalRadius) / kAlpha + 2*2*np.pi
kPlotingRadius = theta_max * kAlpha
theta_max = (kCriticalRadius) / kAlpha + 2 * 2 * np.pi
kPlotingRadius = theta_max * kAlpha
kCriticalTheta = 2.86 / ((2/3)*kAlpha)
kCriticalTheta = 2.86 / ((2 / 3) * kAlpha)
# 生成角度数组
theta = np.linspace(kCriticalTheta, theta_max, 1000)
@@ -34,36 +34,40 @@ circle_theta = np.linspace(0, 2 * np.pi, 1000)
circle_r = np.full_like(circle_theta, kCriticalRadius)
ax.plot(circle_theta, circle_r, linestyle='--')
point_A_cartesian = (kAlpha*kCriticalTheta*np.cos(kCriticalTheta),kAlpha*kCriticalTheta*np.sin(kCriticalTheta))
point_B_cartesian = (-kAlpha*kCriticalTheta*np.cos(kCriticalTheta),-kAlpha*kCriticalTheta*np.sin(kCriticalTheta))
point_A_cartesian = (kAlpha * kCriticalTheta * np.cos(kCriticalTheta), kAlpha * kCriticalTheta * np.sin(kCriticalTheta))
point_B_cartesian = (-kAlpha * kCriticalTheta * np.cos(kCriticalTheta),
-kAlpha * kCriticalTheta * np.sin(kCriticalTheta))
kPhi = np.arctan(kCriticalTheta)
r = (1/3) * kAlpha * np.sqrt(1 + kCriticalTheta**2)
r = (1 / 3) * kAlpha * np.sqrt(1 + kCriticalTheta**2)
dx, dy = point_A_cartesian[0] - point_B_cartesian[0], point_A_cartesian[1] - point_B_cartesian[1]
angle = np.arctan2(dy, dx)
dx, dy = np.cos(angle - (0.5*np.pi-kPhi)), np.sin(angle - (0.5*np.pi-kPhi))
point_C1_cartesian = (point_A_cartesian[0] - 2*r*dx, point_A_cartesian[1] - 2*r*dy)
point_C2_cartesian = (point_B_cartesian[0] + 1*r*dx, point_B_cartesian[1] + 1*r*dy)
radius_of_C1 = 2*r
radius_of_C2 = 1*r
dx, dy = np.cos(angle - (0.5 * np.pi - kPhi)), np.sin(angle - (0.5 * np.pi - kPhi))
point_C1_cartesian = (point_A_cartesian[0] - 2 * r * dx, point_A_cartesian[1] - 2 * r * dy)
point_C2_cartesian = (point_B_cartesian[0] + 1 * r * dx, point_B_cartesian[1] + 1 * r * dy)
radius_of_C1 = 2 * r
radius_of_C2 = 1 * r
# 定义用于绘制圆的函数
def draw_circle(ax, center, radius, num_points, beg_angle, span_angle):
t = np.linspace(beg_angle, beg_angle+span_angle, num_points)
x = center[0] + radius * np.cos(t)
y = center[1] + radius * np.sin(t)
r, theta = np.sqrt(x**2 + y**2), np.arctan2(y, x)
ax.plot(theta, r)
t = np.linspace(beg_angle, beg_angle + span_angle, num_points)
x = center[0] + radius * np.cos(t)
y = center[1] + radius * np.sin(t)
r, theta = np.sqrt(x**2 + y**2), np.arctan2(y, x)
ax.plot(theta, r)
# 绘制圆C1
draw_circle(ax, point_C1_cartesian, radius_of_C1, 100, angle+0.5*np.pi-kPhi-np.pi, 2*kPhi)
draw_circle(ax, point_C1_cartesian, radius_of_C1, 100, angle + 0.5 * np.pi - kPhi - np.pi, 2 * kPhi)
# 绘制圆C2
draw_circle(ax, point_C2_cartesian, radius_of_C2, 100, angle+0.5*np.pi-kPhi, 2*kPhi)
draw_circle(ax, point_C2_cartesian, radius_of_C2, 100, angle + 0.5 * np.pi - kPhi, 2 * kPhi)
print(f"Total length={6*r*kPhi}")
print(f"kCriticalTheta={kCriticalTheta}")
x_ticks = np.arange(-int(kPlotingRadius)-1, int(kPlotingRadius)+1, 1)
y_ticks = np.arange(-int(kPlotingRadius)-1, int(kPlotingRadius)+1, 1)
x_ticks = np.arange(-int(kPlotingRadius) - 1, int(kPlotingRadius) + 1, 1)
y_ticks = np.arange(-int(kPlotingRadius) - 1, int(kPlotingRadius) + 1, 1)
X, Y = np.meshgrid(x_ticks, y_ticks)
X = X.flatten()
Y = Y.flatten()
@@ -79,4 +83,4 @@ ax.scatter(theta_grid[valid_points], r_grid[valid_points], color='grey', s=10)
plt.title("The Moving Path")
# 显示图像
plt.show()
plt.show()