RPAT/RPAT.py at main · TotallyAm/RPAT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
##############################################
##Rocket Performance Analysis Tool (RPAT) v3##
########## Created by TotallyAm ##############
##############################################


## version 0.90 - MECO


import json
import os
import time

from scripts.ansi import *
from config import DEBUG_MODE, GRAPH, TRAJECTORY_TARGETS_PATH

print(D_GRAY("----------------------------------------"))
print(GRAY("Rocket Performance Analysis Tool (RPAT)"))
print(GRAY("         Created by TotallyAm"))
print(D_GRAY("----------------------------------------"))

from scripts.graphing import graph
from scripts.payload import trajectories
from scripts.user_input import rocket

def load_targets():
    path = os.path.join(os.path.dirname(__file__), TRAJECTORY_TARGETS_PATH)

    with open(path, "r") as file:
        return json.load(file)


try:
    trajectory_targets = load_targets()
except Exception as error:
    print("[Error] Failed to load trajectory_targets.json:", error)
    trajectory_targets = {}


print(GRAY(f"\nEvaluating {rocket.rocket_name}....."))

start_time = time.perf_counter()

leo_payload = trajectories(rocket, trajectory_targets)

if DEBUG_MODE and not GRAPH:
    delta_time = time.perf_counter() - start_time
    print(f"\nThis program took {(delta_time * 1000):.0f} ms to complete.")
elif GRAPH:
    graph(rocket, leo_payload)