Skip to content

Commit

Permalink
Clean up code and break at 100 columns
Browse files Browse the repository at this point in the history
  • Loading branch information
@ennehekma
ennehekma committed Mar 29, 2016
1 parent e9b3f8a commit 74e5aac
Showing 1 changed file with 62 additions and 130 deletions.
192 changes: 62 additions & 130 deletions python/plotcatalog_asr.py
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
'''
Copyright (c) 2014, K. Kumar ([email protected])
Copyright (c) 2014-2016, K. Kumar ([email protected])
Copyright (c) 2016, E.J. Hekma ([email protected])
All rights reserved.
'''

Expand Down Expand Up @@ -51,34 +52,22 @@
# Read and store TLE catalog
###################################################################################################


# years = [2013, 2014, 2015]
# dn = []
# for year in years:
# df1 = pd.DataFrame({'Incidents': [ 'C', 'B','A'],
# year: [1, 1, 1 ],
# }).set_index('Incidents')
# dn.append(df1)
# dn = pd.concat(dn, axis=1)
# print(dn)

sma = []
ecc5 = []
incl5 = []

raan3 = []
ecc3 = []
ecc5 = []
eccentricity = []
sma = []
inclinations3 = []
aop3 = []
inclinations3 = []

# colors = ['k','b','g','r']
# markers = ['.','s','+','D']

order = ['all','SSO','GEO','HEO']
markers = ['.','s','+','D']
colors = ['k','b','g','r']
for x in order:
# print eval(str('tleCatalogFilePath' + order[x]))
# print x
# time.sleep(10)
tleCatalogFilePathNew = eval(str('tleCatalogFilePath' + str(x)))
# print tleCatalogFilePathNew

# Read in catalog and store lines in list.
fileHandle = open(tleCatalogFilePathNew)
catalogLines = fileHandle.readlines()
Expand Down Expand Up @@ -107,8 +96,9 @@
aop.append(inclinationSortedObjects[i].argpo)

smatemp = []
smatemp = [convertMeanMotionToSemiMajorAxis(debrisObject.no/60.0, getgravconst('wgs72')[1])-6373 \
for debrisObject in debrisObjects]
smatemp = [convertMeanMotionToSemiMajorAxis(debrisObject.no/60.0, \
getgravconst('wgs72')[1])-6373 \
for debrisObject in debrisObjects]
smatemp2 = []
smatemp2 = pd.DataFrame(smatemp, columns=[str(x)])
sma.append(smatemp2)
Expand Down Expand Up @@ -140,7 +130,7 @@
inclinations2 = []
inclinations2 = pd.DataFrame(inclinations, columns=[str(x)])
inclinations3.append(inclinations2)
# print 'exit'

sma = pd.concat(sma, axis=1)
ecc5 = pd.concat(ecc5, axis=1)
incl5 = pd.concat(incl5, axis=1)
Expand All @@ -149,19 +139,7 @@
ecc3 = pd.concat(ecc3, axis=1)
aop3 = pd.concat(aop3, axis=1)
inclinations3 = pd.concat(inclinations3, axis=1)
# print raan3['GEO']

# print str('inclinations' + order[x] + 'new' )
# str('inclinations' + order[x] + 'new' ) = []
# str('inclinations' + order[x] + 'new' ) = inclinations
# str('raan' + order[x] + 'new' ) = []
# str('raan' + order[x] + 'new' ) = raan
# str('ecc' + order[x] + 'new' ) = []
# str('ecc' + order[x] + 'new' ) = ecc
# str('aop' + order[x] + 'new' ) = []
# str('aop' + order[x] + 'new' ) = aop
# print raan[1]
# print raanSSOnew[1]

###################################################################################################

###################################################################################################
Expand All @@ -170,43 +148,24 @@

# Set font size for plot labels.
rcParams.update({'font.size': fontSize})
markers = ['.','s','+','D']
colors = ['k','b','g','r']

# Plot distribution of eccentricity [-] against semi-major axis [km].
figure = plt.figure()
axis = figure.add_subplot(111)
plt.xlabel("Semi-major axis altitude [km]")
plt.ylabel("Eccentricity [-]")
plt.ticklabel_format(style='sci', axis='x', scilimits=(0,0))
# print
# print inclinationSortedObjects[1].no
plt.plot([convertMeanMotionToSemiMajorAxis(debrisObject.no/60.0, getgravconst('wgs72')[1]) \
for debrisObject in debrisObjects], \
[debrisObject.ecco for debrisObject in debrisObjects], \
marker='o', markersize=1, color='k', linestyle='none')
# axis.set_xlim(xmax=0.8e4)
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure1_debrisPopulation_eccentricityVsSemiMajorAxis.pdf", \
dpi = figureDPI)
plt.close()

figure = plt.figure()
axis = figure.add_subplot(111)
plt.xlabel("Semi-major axis altitude [km]")
plt.ylabel("Eccentricity [-]")
plt.ticklabel_format(style='sci', axis='x', scilimits=(0,0))
plt.plot(sma['all'],ecc3['all'], \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(sma['SSO'],ecc3['SSO'], \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(sma['GEO'],ecc3['GEO'], \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(sma['HEO'],ecc3['HEO'], \
marker='D', markersize=6, color='r', linestyle='none')
axis.set_xlim(xmax=0.5e5)
plt.plot(sma['all'],ecc3['all'], \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(sma['SSO'],ecc3['SSO'], \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(sma['GEO'],ecc3['GEO'], \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(sma['HEO'],ecc3['HEO'], \
marker='D', markersize=6, color='r', linestyle='none')
axis.set_xlim(xmax=0.4e5)
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure1_debrisPopulation_eccentricityVsSemiMajorAxisNew.pdf", \
plt.savefig(outputPath + "/figure1_debrisPopulation_eccentricityVsSemiMajorAxis.pdf", \
dpi = figureDPI)
plt.close()

Expand All @@ -215,32 +174,21 @@
axis = figure.add_subplot(111)
plt.xlabel("$e \cos{\omega}$ [-]")
plt.ylabel("$e \sin{\omega}$ [-]")
plt.plot(ecc*np.cos(aop),ecc*np.sin(aop), marker='o', markersize=1, color='k', linestyle='none')
plt.plot(ecc3['all']*np.cos(aop3['all']),ecc3['all']*np.sin(aop3['all']), \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(ecc3['SSO']*np.cos(aop3['SSO']),ecc3['SSO']*np.sin(aop3['SSO']), \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(ecc3['GEO']*np.cos(aop3['GEO']),ecc3['GEO']*np.sin(aop3['GEO']), \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(ecc3['HEO']*np.cos(aop3['HEO']),ecc3['HEO']*np.sin(aop3['HEO']), \
marker='D', markersize=6, color='r', linestyle='none')
plt.axis('equal')
axis.set_xlim(xmin=-.21, xmax=.21)
axis.set_ylim(ymin=-.21, ymax=.21)
axis.set(xticks=[-.2,-.1,0,.1,.2])#, xticklabels=datelabels) #Same as plt.xticks
axis.set_xlim(xmin=-.82, xmax=.82)
axis.set_ylim(ymin=-.82, ymax=.82)
axis.set(xticks=[-.8,-.4,0,.4,.8])
axis.set(yticks=[-.8,-.4,0,.4,.8])
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure2_debrisPopulation_eccentricityVector.pdf", dpi = figureDPI)
# plt.savefig(outputPath + "/figure2_debrisPopulation_eccentricityVector.pdf", dpi = figureDPI)
plt.close()

figure = plt.figure()
axis = figure.add_subplot(111)
plt.xlabel("$e \cos{\omega}$ [-]")
plt.ylabel("$e \sin{\omega}$ [-]")
# plt.plot(ecc*np.cos(aop),ecc*np.sin(aop), marker='o', markersize=1, color='k', linestyle='none')
plt.plot(ecc3['all']*np.cos(aop3['all']),ecc3['all']*np.sin(aop3['all']), marker='.', markersize=1, color='k', linestyle='none')
plt.plot(ecc3['SSO']*np.cos(aop3['SSO']),ecc3['SSO']*np.sin(aop3['SSO']), marker='s', markersize=10, color='c', linestyle='none')
plt.plot(ecc3['GEO']*np.cos(aop3['GEO']),ecc3['GEO']*np.sin(aop3['GEO']), marker='^', markersize=10, color='g', linestyle='none')
plt.plot(ecc3['HEO']*np.cos(aop3['HEO']),ecc3['HEO']*np.sin(aop3['HEO']), marker='D', markersize=6, color='r', linestyle='none')
plt.axis('equal')
# axis.set_xlim(xmin=-.21, xmax=.21)
# axis.set_ylim(ymin=-.21, ymax=.21)
# axis.set(xticks=[-.2,-.1,0,.1,.2])#, xticklabels=datelabels) #Same as plt.xticks
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure2_debrisPopulation_eccentricityVectorNew.pdf", dpi = figureDPI)
# plt.savefig(outputPath + "/figure2_debrisPopulation_eccentricityVector.pdf", dpi = figureDPI)
plt.close()


Expand All @@ -250,21 +198,17 @@
plt.xlabel("Semi-major axis altitude [km]")
plt.ylabel("Inclination [deg]")
plt.ticklabel_format(style='sci', axis='x', scilimits=(0,0))
plt.plot(sma['all'],incl5['all'], \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(sma['SSO'],incl5['SSO'], \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(sma['GEO'],incl5['GEO'], \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(sma['HEO'],incl5['HEO'], \
marker='D', markersize=6, color='r', linestyle='none')
# plt.plot([convertMeanMotionToSemiMajorAxis(debrisObject.no/60.0, getgravconst('wgs72')[1]) \
# for debrisObject in debrisObjects], \
# [np.rad2deg(debrisObject.inclo) for debrisObject in debrisObjects], \
# marker='o', markersize=1, color='k', linestyle='none')
axis.set_xlim(xmax=0.5e5)
plt.plot(sma['all'],incl5['all'], \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(sma['SSO'],incl5['SSO'], \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(sma['GEO'],incl5['GEO'], \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(sma['HEO'],incl5['HEO'], \
marker='D', markersize=6, color='r', linestyle='none')
axis.set_xlim(xmax=0.4e5)
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure3_debrisPopulation_inclinationVsSemiMajorAxisNew.pdf", \
plt.savefig(outputPath + "/figure3_debrisPopulation_inclinationVsSemiMajorAxis.pdf", \
dpi = figureDPI)
plt.close()

Expand All @@ -273,37 +217,25 @@
axis = figure.add_subplot(111)
plt.xlabel("$i \cos{\Omega}$ [deg]")
plt.ylabel("$i \sin{\Omega}$ [deg]")
plt.plot(np.rad2deg(inclinations)*np.cos(raan),np.rad2deg(inclinations)*np.sin(raan), \
marker='o', markersize=1, color='k', linestyle='none')
plt.plot(np.rad2deg(inclinations3['all'])*np.cos(raan3['all']), \
np.rad2deg(inclinations3['all'])*np.sin(raan3['all']), \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(np.rad2deg(inclinations3['SSO'])*np.cos(raan3['SSO']), \
np.rad2deg(inclinations3['SSO'])*np.sin(raan3['SSO']), \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(np.rad2deg(inclinations3['GEO'])*np.cos(raan3['GEO']), \
np.rad2deg(inclinations3['GEO'])*np.sin(raan3['GEO']), \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(np.rad2deg(inclinations3['HEO'])*np.cos(raan3['HEO']), \
np.rad2deg(inclinations3['HEO'])*np.sin(raan3['HEO']), \
marker='D', markersize=6, color='r', linestyle='none')
plt.axis('equal')
axis.set_xlim(xmin=-180.0, xmax=180.0)
axis.set_ylim(ymin=-180.0, ymax=180.0)
axis.set_xlim(xmin=-110.0, xmax=110.0)
axis.set_ylim(ymin=-110.0, ymax=110.0)
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure4_debrisPopulation_inclinationVector.pdf", dpi = figureDPI)
plt.close()

# Plot components of inclination vector [deg].
figure = plt.figure()
axis = figure.add_subplot(111)
plt.xlabel("$i \cos{\Omega}$ [deg]")
plt.ylabel("$i \sin{\Omega}$ [deg]")
# plt.plot(np.rad2deg(inclinations)*np.cos(raan),np.rad2deg(inclinations)*np.sin(raan), \
# marker='o', markersize=1, color='k', linestyle='none')
plt.plot(np.rad2deg(inclinations3['all'])*np.cos(raan3['all']),np.rad2deg(inclinations3['all'])*np.sin(raan3['all']), \
marker='.', markersize=1, color='k', linestyle='none')
plt.plot(np.rad2deg(inclinations3['SSO'])*np.cos(raan3['SSO']),np.rad2deg(inclinations3['SSO'])*np.sin(raan3['SSO']), \
marker='s', markersize=10, color='c', linestyle='none')
plt.plot(np.rad2deg(inclinations3['GEO'])*np.cos(raan3['GEO']),np.rad2deg(inclinations3['GEO'])*np.sin(raan3['GEO']), \
marker='^', markersize=10, color='g', linestyle='none')
plt.plot(np.rad2deg(inclinations3['HEO'])*np.cos(raan3['HEO']),np.rad2deg(inclinations3['HEO'])*np.sin(raan3['HEO']), \
marker='D', markersize=6, color='r', linestyle='none')
plt.axis('equal')
axis.set_xlim(xmin=-180.0, xmax=180.0)
axis.set_ylim(ymin=-180.0, ymax=180.0)
figure.set_tight_layout(True)
plt.savefig(outputPath + "/figure4_debrisPopulation_inclinationVectorNew.pdf", dpi = figureDPI)
plt.close()




Expand Down

0 comments on commit 74e5aac

Please sign in to comment.