Armor/general_tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Armor
2025-01-22 11:10:16 +08:00
parent 06f10fd2f7
commit 0445128fd0
4 changed files with 160 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/KSP_tools/stage_performance_plot.py
+3 -4
View File
@@ -8,14 +8,13 @@ def plot_stage_performance(isp, gross_mass, dry_mass, payload, stage_name):
plt.plot(payload, stage_performance, label=stage_name)
payload_list = [i for i in range(0, 40200, 200)]
plot_stage_performance(450.5, 20830 + 2247, 2247, payload_list, 'Centaur III')
payload_list = [i for i in range(0, 15200, 200)]
# plot_stage_performance(450.5, 20830 + 2247, 2247, payload_list, 'Centaur III')
# plot_stage_performance(438, 21000, 2800, payload_list, 'Long March 3m')
# plot_stage_performance(450, 23500 + 2900, 2900, payload_list, 'Long March 3.35m')
plot_stage_performance(453.8, 54000+4500, 5000, payload_list, 'Centaur V')
plot_stage_performance(348, 112000, 4000, payload_list, 'Falcon 9')
plot_stage_performance(355, 80600, 600, payload_list, 'Neutron S2')
plot_stage_performance(365, 100600, 600, payload_list, 'Neutron S2 MAX')
plot_stage_performance(367, 82000, 2000, payload_list, 'Neutron S2')
# plot_stage_performance(451, 68000, 8000, payload_list, 'Long March 10 S3')
# plot_stage_performance(442.6, 36000, 5100, payload_list, 'Long March 5 S2')
# plot_stage_performance(462, 30710, 3490, payload_list, 'DCSS 5m')
src/KSP_tools/tube_drag_cube_generator.py
+81
View File
@@ -0,0 +1,81 @@
from matplotlib import pyplot as plt, pyplot
def interpolate_y(x1, y1, x2, y2, x):
if x1 == x2:
raise ValueError("x1 and x2 cannot be the same value, as the line would be undefined.")
# 计算插值的y值
y = y1 + (y2 - y1) * (x - x1) / (x2 - x1)
return y
def cal_num_6(size):
if size < 2.5:
return interpolate_y(1.875, 2.363, 2.5, 4.87, size)
elif 2.5 <= size < 3.75:
return interpolate_y(3.75, 11, 2.5, 4.87, size)
elif 3.75 <= size:
return interpolate_y(3.75, 11, 5, 19.48, size)
def cal_cube_parameter(size, height):
# size = 5
# height = 1.25
num0 = size * height
num1 = 0.7778
num2 = size / 2 + 0.03
num3 = num0
num4 = num1
num5 = num2
num6 = cal_num_6(size)
num7 = 0.9796
num8 = interpolate_y(1.875, 0.1449, 5, 0.2207, size)
num9 = cal_num_6(size)
num10 = 0.9796
num11 = interpolate_y(1.875, 0.1449, 5, 0.2207, size)
num12 = num0
num13 = 0.7776
num14 = num2
num15 = num0
num16 = 0.7776
num17 = num2
num18 = 0
num19 = -0.5 * height
num20 = 0
num21 = size
num22 = height
num23 = size
result = [num0, num1, num2, num3, num4, num5, num6, num7, num8, num9, num10, num11, num12, num13, num14, num15,
num16, num17, num18, num19, num21, num22, num23]
return result
def print_cube_content(size, height_list):
print(' DRAG_CUBE')
print(' {')
for i in range(len(height_list)):
print(f' cube = {i}, ', end='')
print(','.join([str(round(item, 5)) for item in cal_cube_parameter(size, height_list[i])]))
print(' }')
if __name__ == '__main__':
# sample_data_list = [6.25, 0.7781, 2.529, 6.25, 0.7781, 2.529, 19.48, 0.9796, 0.2207, 19.48, 0.9796, 0.2207, 6.25,
# 0.7776, 2.529, 6.25, 0.7776, 2.529, 0, -0.625, 0, 5, 1.25, 5]
height_list = [1.875, 2.5, 3.75, 5]
# new_table = cal_cube_parameter(3.35, 5)
# print(sample_data_list)
# print(new_table)
print_cube_content(5.4, height_list)
src/KSP_tools/vessel_simulation_advanced.py
+66 -31
View File
@@ -1,15 +1,17 @@
import math
import load_common_resources
import copy
import time
from matplotlib import pyplot as plt
from tqdm import tqdm
Vc = 299792458
Ga = 9.80665
path = 'CommonResources.cfg'
path = 'data/CommonResources.cfg'
common_resources_dict = load_common_resources.load_common_resources(path)
fuel_density_map = {
'methalox': common_resources_dict['LqdMethane']['density'] * 0.4137 + common_resources_dict['LqdOxygen'][
'density'] * 0.5863,
@@ -72,7 +74,8 @@ class Stage:
for i in range(len(curve)):
try:
if curve_keys[i] <= stage_time < curve_keys[i + 1]:
throttle = curve[curve_keys[i]] + (curve[curve_keys[i+1]] - curve[curve_keys[i]]) / (curve_keys[i+1] - curve_keys[i]) * (stage_time - curve_keys[i])
throttle = curve[curve_keys[i]] + (curve[curve_keys[i + 1]] - curve[curve_keys[i]]) / (
curve_keys[i + 1] - curve_keys[i]) * (stage_time - curve_keys[i])
break
except IndexError:
throttle = curve[curve_keys[-1]]
@@ -95,7 +98,7 @@ class Stage:
self.current_mass = self.fuel_mass + self.dry_mass
self.current_fuel_mass = self.fuel_mass
def simulated_burn_step_by_step(self, stage_time, duration):
def simulated_burn_step_by_step(self, stage_time, duration, print_status):
current_thrust = self.cal_current_thrust(stage_time)
now_thrust = current_thrust
if self.is_empty:
@@ -103,6 +106,7 @@ class Stage:
next_fuel_burned = cal_fuel_burned(self.isp, now_thrust, duration)
if self.current_fuel_mass - next_fuel_burned < 0:
self.is_empty = True
if print_status:
print(f'Part: {self.name} is depleted after {round(stage_time, 3)}s.')
else:
self.current_fuel_mass -= next_fuel_burned
@@ -116,6 +120,7 @@ class Vessel:
self.duration = duration
self.payload = payload
self.name = name
self.print_status = True
def reset_simulation(self):
for stages in self.stage_list:
@@ -154,29 +159,29 @@ class Vessel:
def stage_burn(self, stages, stage_time):
result = []
for stage in stages:
stage_is_empty, stage_mass = stage.simulated_burn_step_by_step(stage_time, self.duration)
stage_is_empty, stage_mass = stage.simulated_burn_step_by_step(stage_time, self.duration, self.print_status)
result.append((stage_is_empty, stage_mass))
return result
@staticmethod
def stage_info(stages):
def stage_info(self, stages):
stage_thrust, stage_fuel, gross_mass, dry_mass = 0, 0, 0, 0
for stage in stages:
stage_thrust += stage.max_thrust
stage_fuel += stage.fuel_mass
gross_mass += stage.total_mass
dry_mass += stage.dry_mass
print(f' Total Vacuum Thrust: {stage_thrust}KN, Burned Mass: {stage_fuel/1000}ton, '
self.vessel_print(f' Total Vacuum Thrust: {stage_thrust}KN, Burned Mass: {stage_fuel / 1000}ton, '
f'Gross Mass: {gross_mass / 1000}ton, '
f'Dry Mass: {dry_mass / 1000}ton')
def run_sim_new(self):
self.vessel_info()
self.reset_simulation()
print('Start Simulation')
self.vessel_print('Start Simulation')
stage_burn_time, timer, dv, stage_dv = 0, 0, 0, 0
now_stage_id = 0
stage_acc_result = {}
start = time.time()
# time based simulation
while True:
@@ -197,25 +202,48 @@ class Vessel:
stage_dv += step_dv
if self.is_depleted(now_stage):
print(f'stage{now_stage_id}:\n burn time: {round(stage_burn_time, 3)}s, dv: {round(stage_dv, 3)}m/s ')
self.vessel_print(
f'stage{now_stage_id}:\n burn time: {round(stage_burn_time, 3)}s, dv: {round(stage_dv, 3)}m/s ')
start_twr = round(stage_acc_result[f'stage{now_stage_id}'][0], 3)
end_twr = round(stage_acc_result[f'stage{now_stage_id}'][-2], 3)
max_twr = round(max(stage_acc_result[f'stage{now_stage_id}']), 3)
print(f' start twr: {start_twr} end twr: {end_twr} max_twr: {max_twr}')
self.vessel_print(f' start twr: {start_twr} end twr: {end_twr} max_twr: {max_twr}')
self.stage_info(now_stage)
stage_burn_time, stage_dv = 0, 0
now_stage_id += 1
timer = round(timer, 3)
dv = round(dv, 3)
print(f'Total Engine Burn Time: {timer}s, Total dv: {dv}m/s')
self.vessel_print(
f'Simulation completed. Total Engine Burn Time: {timer}s, Total dv: {dv}m/s. Time: {round(time.time() - start, 3)}')
return dv
def plot_sim(self, start_payload=500, step=200):
self.print_status = False
total_payload = self.payload
dv_list = []
for payload in tqdm(range(start_payload, total_payload, step)):
self.payload = payload
dv_list.append(self.run_sim_new())
plt.plot([i for i in range(start_payload, total_payload, step)], dv_list)
plt.axhline(y=9600, color='r')
plt.xlabel('payload mass in kg')
plt.ylabel('delta-v in m/s')
plt.show()
self.print_status = True
return
def vessel_info(self):
lift_off_mass = self.get_current_mass() / 1000
lift_off_thrust = self.get_current_thrust()
print(f'Info of {self.name}: ')
print(f' Vessel Lift-off Mass: {lift_off_mass} ton, Vessel Lift-off Vacuum Thrust: {lift_off_thrust} KN')
print(f' Payload Mass: {self.payload} KG')
print('-'*80)
self.vessel_print(f'Info of {self.name}: ')
self.vessel_print(
f' Vessel Lift-off Mass: {lift_off_mass} ton, Vessel Lift-off Vacuum Thrust: {lift_off_thrust} KN')
self.vessel_print(f' Payload Mass: {self.payload} KG')
self.vessel_print('-' * 80)
def vessel_print(self, content):
if self.print_status:
print(content)
if __name__ == '__main__':
@@ -241,27 +269,34 @@ if __name__ == '__main__':
# # cz_10 = Vessel([[cbc, S1], [S2]], name='CZ-10', payload=70000)
# cz_10.run_sim_new()
booster = Stage(1040000, 100000, 1397*14, 338.2, 'kerolox', 'Side Booster')
S1 = Stage(680000, 60000, 1397*7, 338.2, 'kerolox', 'Stage 1')
S2 = Stage(185000, 20000, 2900, 352.3, 'kerolox', 'Stage 2')
S3 = Stage(58500, 11500, 276.324, 451, 'hydrolox', 'Stage 3')
cz_10 = Vessel([[booster, S1], [S2], [S3]], name='CZ-10', payload=27000)
cz_10.run_sim_new()
# booster = Stage(1040000, 100000, 1397*14, 338.2, 'kerolox', 'Side Booster')
# S1 = Stage(680000, 60000, 1397*7, 338.2, 'kerolox', 'Stage 1')
# S2 = Stage(185000, 20000, 2900, 352.3, 'kerolox', 'Stage 2')
# S3 = Stage(58500, 11500, 276.324, 451, 'hydrolox', 'Stage 3')
# cz_10 = Vessel([[booster, S1], [S2], [S3]], name='CZ-10', payload=27000)
# cz_10.run_sim_new()
# S1 = Stage(339000, 30000, 1397*4, 338.2, 'kerolox', 'Stage 1')
# S2 = Stage(48000, 6000, 360, 341.2, 'kerolox', 'Stage 2')
# cz_12 = Vessel([[S1], [S2]], name='CZ-12', payload=10000)
# cz_12.run_sim_new()
booster = Stage(72000*4, 7500*4, 1340*4, 335, 'kerolox', 'CZ-7A Booster')
Stage1 = Stage(144000, 15000, 1340*2, 335, 'kerolox', 'CZ-7A Stage1')
Stage2 = Stage(61500, 8000, 180*4, 342, 'kerolox', 'CZ-7A Stage2')
Stage3 = Stage(18250, 3050, 83*2, 438, 'hydrolox', 'CZ-7A Stage3')
# booster = Stage(72000*4, 7500*4, 1340*4, 335, 'kerolox', 'CZ-7A Booster')
# Stage1 = Stage(144000, 15000, 1340*2, 335, 'kerolox', 'CZ-7A Stage1')
# Stage2 = Stage(61500, 8000, 180*4, 342, 'kerolox', 'CZ-7A Stage2')
# Stage3 = Stage(18250, 3050, 83*2, 438, 'hydrolox', 'CZ-7A Stage3')
#
# cz_7a = Vessel([[booster, Stage1], [Stage2], [Stage3]], name='CZ-7A', payload=8000)
# cz_7a.run_sim_new()
cz_7a = Vessel([[booster, Stage1], [Stage2], [Stage3]], name='CZ-7A', payload=8000)
cz_7a.run_sim_new()
# S1 = Stage(335000, 48000, 7116.12, 329, 'methalox', 'Stage 1')
# S2 = Stage(80000, 2000, 890, 367, 'methalox', 'Stage 2')
# Neutron = Vessel([[S1], [S2]], name='Neutron', payload=15000, duration=0.05)
# Neutron.run_sim_new()
# Neutron.plot_sim(start_payload=13000, step=100)
S1 = Stage(12000, 2500, 80, 335, 'kerolox', 'Service Module')
S2 = Stage(7000, 3800, 50, 330, 'kerolox', 'Ascent Stage')
Lanyue = Vessel([[S1], [S2]], name='Lanyue', payload=200, duration=0.05)
Lanyue.run_sim_new()
print('done')
src/tools/auto_safe.py
+6 -28
View File
@@ -1,31 +1,6 @@
import time
from ctypes import *
from tcping import Ping
def ping_ip(ip_address, request_nums):
"""
ping ip
:param ip_address:
:param request_nums: 请求次数
:return: 丢包率loss和统计结果res
"""
ping = Ping(ip_address, 2077, 3)
ping.ping(request_nums)
res = ping.result.table
ret = ping.result.raw
return_list = list(ret.split('\n'))
loss = return_list[2].split(',')[3].split(' ')[1] # 获取丢包率
return loss, res
def ping_process(host):
# 调用ping_ip方法得到丢包率
loss, res = ping_ip(host, 3)
if float(loss.strip('%')) / 100 <= 0.1: # 0.1为自定义丢包率阈值，可修改
print("ping 不通")
else:
print("ping 通")
from src.tools.ping_ip import ping
def shut_screen_and_lock():
@@ -55,8 +30,8 @@ class AutoSafe:
self.start_monitor = True
def check_connection(self) -> bool:
loss, res = ping_ip(self.target_server, 5)
if float(loss.strip('%')) / 100 <= 0.1:
success_rate = ping(self.target_server, 2077, 5, 10)
if success_rate <= 0.3:
return False
else:
return True
@@ -65,9 +40,12 @@ class AutoSafe:
print("Starting AutoSafe")
while self.start_monitor:
time.sleep(20)
print('checking:', end=' ')
if is_locked() or self.check_connection():
print('ok @', time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
continue
else:
print('failed @', time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
self.safe()
def safe(self):