Напиши код на python с использованием PyQt для создание декстопного приложения для построение траектории полета снаряда при помощи системы дифференциальных уравнений для внешней баллистики. Данные задает пользователь в специальных полях, график траектории приводится в специальном поле. Для трехмерного пространства
python
from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton, QLineEdit, QLabel, QTextEdit
from PyQt5.QtGui import QPainter, QColor, QPen
from PyQt5.QtCore import Qt
import sys
import numpy as np
from scipy.integrate import odeint
import matplotlib.pyplot as plt
from mpl_toolkits import mplot3d
class Trajectory:
def __init__(self, x0, y0, z0, v0, theta, phi, h, m):
self.x0 = x0
self.y0 = y0
self.z0 = z0
self.v0 = v0
self.theta = theta
self.phi = phi
self.h = h
self.m = m
self.g = 9.81
def motion(self, t, y):
x, y, z, vx, vy, vz = y
dxdt = vx
dydt = vy
dzdt = vz
v = np.sqrt(vx**2 + vy**2 + vz**2)
f_gravity = np.array([0, 0, -self.m*self.g])
f_air_resistance = -0.5*self.h*np.pi*v**2*np.array([vx, vy, vz])
f = f_gravity + f_air_resistance
dvxdt, dvydt, dvzdt = f/self.m
return [dxdt, dydt, dzdt, dvxdt, dvydt, dvzdt]
def solve(self, t_max):
t = np.linspace(0, t_max, 1000)
initial_conditions = np.array([self.x0, self.y0, self.z0, self.v0*np.sin(self.theta)*np.cos(self.phi),
self.v0*np.sin(self.theta)*np.sin(self.phi), self.v0*np.cos(self.theta)])
solution = odeint(self.motion, initial_conditions, t)
return solution
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setWindowTitle('Ballistic Trajectory')
self.setGeometry(100, 100, 1000, 800)
self.label_x0 = QLabel('Initial x-coordinate (m):', self)
self.label_x0.move(50, 50)
self.label_y0 = QLabel('Initial y-coordinate (m):', self)
self.label_y0.move(50, 100)
self.label_z0 = QLabel('Initial z-coordinate (m):', self)
self.label_z0.move(50, 150)
self.label_v0 = QLabel('Initial velocity (m/s):', self)
self.label_v0.move(50, 200)
self.label_theta = QLabel('Launch angle (degrees):', self)
self.label_theta.move(50, 250)
self.label_phi = QLabel('Azimuth angle (degrees):', self)
self.label_phi.move(50, 300)
self.label_h = QLabel('Cross-sectional area (m^2):', self)
self.label_h.move(50, 350)
self.label_m = QLabel('Mass of the projectile (kg):', self)
self.label_m.move(50, 400)
self.text_x0 = QLineEdit('', self)
self.text_x0.move(300, 50)
self.text_x0.resize(200, 30)
self.text_y0 = QLineEdit('', self)
self.text_y0.move(300, 100)
self.text_y0.resize(200, 30)
self.text_z0 = QLineEdit('', self)
self.text_z0.move(300, 150)
self.text_z0.resize(200, 30)
self.text_v0 = QLineEdit('', self)
self.text_v0.move(300, 200)
self.text_v0.resize(200, 30)
self.text_theta = QLineEdit('', self)
self.text_theta.move(300, 250)
self.text_theta.resize(200, 30)
self.text_phi = QLineEdit('', self)
self.text_phi.move(300, 300)
self.text_phi.resize(200, 30)
self.text_h = QLineEdit('', self)
self.text_h.move(300, 350)
self.text_h.resize(200, 30)
self.text_m = QLineEdit('', self)
self.text_m.move(300, 400)
self.text_m.resize(200, 30)
self.button = QPushButton('Calculate', self)
self.button.move(50, 500)
self.button.resize(200, 50)
self.button.clicked.connect(self.calculate)
self.output = QTextEdit('', self)
self.output.move(550, 50)
self.output.resize(400, 700)
def calculate(self):
x0 = float(self.text_x0.text())
y0 = float(self.text_y0.text())
z0 = float(self.text_z0.text())
v0 = float(self.text_v0.text())
theta = np.deg2rad(float(self.text_theta.text()))
phi = np.deg2rad(float(self.text_phi.text()))
h = float(self.text_h.text())
m = float(self.text_m.text())
t_max = 2*v0*np.sin(theta)/9.81
ball = Trajectory(x0, y0, z0, v0, theta, phi, h, m)
solution = ball.solve(t_max)
fig = plt.figure(figsize=(6, 6))
ax = plt.axes(projection='3d')
x, y, z = solution[:, 0], solution[:, 1], solution[:, 2]
ax.plot3D(x, y, z)
ax.set_xlabel('x (m)')
ax.set_ylabel('y (m)')
ax.set_zlabel('z (m)')
plt.show()
if __name__ == '__main__':
app = QApplication(sys.argv)
window = MainWindow()
window.show()
sys.exit(app.exec_())
Класс `Trajectory` определяет положение и скорость снаряда как функции времени с помощью системы дифференциальных уравнений, учитывая гравитационную силу и силу сопротивления воздуха. Решение этой системы дифференциальных уравнений находится при помощи функции `solve`, используя метод `odeint` из модуля `scipy.integrate`.
Класс `MainWindow` определяет главное окно приложения. Оно содержит несколько `QLabel`, `QLineEdit` и `QPushButton`. `QLineEdit` позволяет пользователю задавать начальные условия для траектории снаряда, включая начальные координаты, начальную скорость, угол запуска, угол азимута, площадь поперечного сечения снаряда и его массу. Нажатие кнопки `QPushButton` запускает функцию `calculate`, которая вычисляет решение системы дифференциальных уравнений, используя класс `Trajectory`, и строит график траектории снаряда в трехмерном пространстве.Нажимая «Регистрация» или «Войти через Google», вы соглашаетесь с Публичной офертой, даете Согласие на обработку персональных данных, а также подтверждаете что вам есть 18 лет
Нажимая «Регистрация» или «Войти через Google», вы соглашаетесь с Публичной офертой, даете Согласие на обработку персональных данных, а также подтверждаете что вам есть 18 лет