MultiAsteroids3.py

import numpy as np
from astropy import units as u
from astropy import time
import datetime

from poliastro.bodies import Earth, Sun
from poliastro.twobody import Orbit
f=open("roid_info.txt")
read_lines=f.readlines()
from matplotlib import pylab as plt
import argparse

#then plot
from poliastro.plotting import OrbitPlotter

parser = argparse.ArgumentParser()
parser.add_argument("--asteroids", "-n", help="changes the number of asteroids")
args=parser.parse_args()
print(type(args.asteroids))
if not args.asteroids:
    asteroids=75
else:
    asteroids=int(args.asteroids)

tempo=datetime.datetime(2000,1,1,0,0,0)
#help(Orbit.from_classical)


astratempo = time.Time(tempo, scale='utc')#, value=J2000.000)

#plot Icaurs and Mars
op = OrbitPlotter(bgcolor=(0,0,0), linewidth=0.25, markersize=30, audivision=12)

roid_orbits = []

i=0
#for i in range(asteroids):
line = read_lines[i]
# pull out the e, i, arg_p
line_e = float(line[71:79]) * u.one
line_inc= float(line[60:68]) * u.deg
line_pera= float(line[38:46]) * u.deg
line_semax = float(line[93:103]) * u.AU
line_name=line[167:194]
line_raan= float(line[49:57]) * u.deg
line_ma= float(line[27:35]) * u.deg
#x = Orbit.from_classical(Sun, line_semax, line_e, line_inc, line_raan, line_pera, line_ma, epoch=astratempo)
#x = Orbit.from_classical(Sun, line_semax, line_e, line_inc, line_raan, line_pera, line_ma, time.Time("2095-11-05 12:00", scale='utc'))
x = Orbit.from_classical(
    Sun,
    3.46250 * u.AU, 0.64 * u.one, 7.04 * u.deg,
    50.1350 * u.deg, 12.8007 * u.deg, 63.89 * u.deg,
    time.Time("2005-11-05 12:00", scale='utc')
)
roid_orbits.append(x)


orb = roid_orbits[i]
op.plot(orb)


# Data for Mars at J2000 from JPL HORIZONS



#op.plot(mars)
#op.plot(icarus,label="Icarus")

plt.savefig('asteroidsTest1.png')
plt.close()