-
Notifications
You must be signed in to change notification settings - Fork 0
/
2DInto3D.py
65 lines (57 loc) · 3.03 KB
/
2DInto3D.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from PyGMO.problem import base
import rebound
import numpy as np
import math
from simulatorHigh import SpaceSim
from blackBoxHigh import earthToMars
from blackBoxHigh2D import earthToMars2D
MAX_DV= 0.005607457564
def run_2D(n_restarts=5):
from PyGMO import algorithm, island
prob = earthToMars2D()
algo = algorithm.scipy_slsqp(max_iter=500, acc=1e-5)
algo2 = algorithm.mbh(algo, 5, 0.05)
algo2.screen_output = True
isl = island(algo2, prob, 1)
print("Running Monotonic Basin Hopping .... this will take a while.")
isl.evolve(1)
isl.join()
print("Is the solution found a feasible trajectory? " +
str(prob.feasibility_x(isl.population.champion.x)))
return prob._convertToList(isl.population.champion.x)
def run(n_restarts=5):
from PyGMO import algorithm, island, population
prob = earthToMars()
algo = algorithm.scipy_slsqp(max_iter=500, acc=1e-5)
# algo.screen_output = True
algo2 = algorithm.mbh(algo, n_restarts, 0.005)
algo2.screen_output = True
pop = population(prob)
pop.push_back((320.33563525584435, 0.0016748335274261476, 0.0013627675495311467, 0.0))
pop.push_back((301.93091863294785, 0.0016076262089444425, 0.0015896115913838728, 0.0))
pop.push_back((301.93091863294785, 0.0016076262089444425, 0.0015896115913838728, 0.0))
pop.push_back((420.2372419060117, 0.010494326408284994, 0.0044382506954818565, 0.0))
pop.push_back((411.21323621411335, 0.008748839048462907, 0.0033290148214346503, 0.0))
pop.push_back((395.8283718212657, 0.006450877568564355, 0.002069880891910152, 0.0))
pop.push_back((319.95400029222867, 0.0016702166037494744, 0.0013676901851197968, 0.0))
pop.push_back((319.5113399461457, 0.00166499548529299, 0.0013736935829129556, 0.0))
pop.push_back((320.0969905134936, 0.001671977113629641, 0.001365741362825864, 0.0))
pop.push_back((324.8947207784664, 0.0017420256877963634, 0.0013024051696600683, 0.0))
isl = island(algo2,pop)
print("Running Monotonic Basin Hopping .... this will take a while.")
isl.evolve(1)
isl.join()
print("Is the solution found a feasible trajectory? " +
str(prob.feasibility_x(isl.population.champion.x)))
return isl.population.champion.x
x=earthToMars()
vector=run()
deci=x._convertToList(vector)
print deci
sim2=SpaceSim()
sim2.animate(deci)
# import rebound
# import matplotlib.pyplot as plt
# fig = rebound.OrbitPlot(sim)
# plt.show()
# deci = [[0.00018354551497353738, 0.0002897005581203533, 0], [9.385407683672441e-05, -0.0004375546286935724, 0], [-0.00017406053466786356, -0.0004055793819144533, 0], [7.811816626063441e-05, -0.00028869842254392053, 0], [0.000280132941671916, -0.00045467528344872834, 0], [0.00031161406626870487, -0.0004418005074233615, 0], [0.00016912620000403375, -0.00045156036938030775, 0], [0.00043500734938167605, -4.4611940286304056e-05, 0], [0.00023373694896547512, 4.622353180355802e-06, 0], [0.00043504614537196785, -0.00042017445674379463, 0], [0.00016822207354911628, 0.00010574669088542543, 0], [2.1129649656070842e-05, 0.00020199652091584146, 0]]