Heliophysics-Python-Programs/cdfprogram.py at master · rjaylwar/Heliophysics-Python-Programs · 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
# -*- coding: utf-8 -*-
"""
Spyder Editor

This temporary script file is located here:
/Users/rjaylward/.spyder2/.temp.py
"""
def getcdfmfiho():
    import os
    import urllib2
    from spacepy import pycdf
    import math
    import matplotlib.pyplot as plt
    # import spacepy.toolbox as tb

    year = input('Year: ')
    month = input('Month: ')
    day = input('Day: ')
    hour = input('Hour: ')
    minute = input('Minute: ')
    seconds = input('Seconds: ')


    timepercent = ((hour*3600 + minute*60 + seconds)/(24.*60*60))*884

    timevar = int(math.floor(timepercent))
    print timevar

    mfiho_url = 'http://cdaweb.sci.gsfc.nasa.gov/sp_phys/data/wind/mfi/mfi_h0/' + str(year) + '/' + 'wi_h0_mfi_' + str(year) + str(month) + str(day) + '_v05.cdf'

    file_ho = 'wi_h0_mfi' + str(year) + str(month) + str(day) + '.cdf'

    have_cdfho = 0

    try:
        cdftest = pycdf.CDF(file_ho)
    except:
        print "Preparing to download..."
    else:
        print "you already have this file"
        have_cdfho = 1
        cdftest.close()

    if have_cdfho == 0:
        f = urllib2.urlopen(mfiho_url)

        f_cdf = open(file_ho, 'w')

        f_cdf.write(f.read())

        f_cdf.close()

    with pycdf.CDF(file_ho) as cdffile1:
        #print cdffile
        array1 = cdffile1['P1GSM']
        print 'this is the ho file'
        print array1[:,0]
        sat_position = array1[hour,0]
        print ' '
        print sat_position
        print ' '
        #tb.dictree(cdffile)
#
#
#    print sat_position
    file_ko = 'wi_k0_swe' + str(year) + str(month) + str(day) + '.cdf'

    have_cdf = 0

    try:
        cdftest = pycdf.CDF(file_ko)
    except:
        print "Preparing to download..."
    else:
        print "you already have this file"
        have_cdf = 1
        cdftest.close()


    sweko_url = 'http://cdaweb.sci.gsfc.nasa.gov/sp_phys/data/wind/swe/swe_k0/' + str(year) + '/' + 'wi_k0_swe_' + str(year) + str(month) + str(day) + '_v01.cdf'
    # note, sometimes it is not v01, could be v02, or v05
    if have_cdf == 0:
        f = urllib2.urlopen(sweko_url)

        f_cdf = open(file_ko, 'w')

        f_cdf.write(f.read())

        f_cdf.close()

    dontquit = 1
    endearly = 1
    timevar_copy = timevar

    with pycdf.CDF(file_ko) as cdffile2:
        # print cdffile2
        array2 = cdffile2['V_GSM']
        swvelocity = array2[timevar,0]
        if swvelocity > -200 or swvelocity < -900 and swvelocity > -1000:
            print "Solar wind speed is: " + str(swvelocity)
            advance = raw_input('Try the Solar Wind speed 1.5 min later? (y/n) ')

            if advance == 'Y' or advance == 'y' or advance == 'Yes' or advance == 'YES' or advance == 'yes':
                timevar += 3
                swvelocity = array2[timevar, 0]
                while swvelocity < -900 and endearly:
                    timevar += 3
                    swvelocity = array2[timevar, 0]
                    if timevar - timevar_copy > 30:
                        endearly = 0
                if endearly == 0:
                    print ('Solar wind velocity data unavilible.')

            elif advance == 'N' or advance == 'n' or advance == 'No' or advance == 'NO' or advance == 'no':
                print "Warning, check the plot to verify this data"
            else:
                advance = raw_input('Input not recognized. Try Solar Wind speed 3 min later? (y/n) ')

        elif swvelocity < -1000:
            ### could also add a loop here to check...
            print "Solar wind velocity data unavalible for this time period"
            advance = raw_input('Use Solar Wind speed 3 min later? (y/n) ')

            if advance == 'Y' or advance == 'y' or advance == 'Yes' or advance == 'YES' or advance == 'yes':
                timevar += 3
                swvelocity = array2[timevar, 0]
                while swvelocity < -900 and endearly:
                    timevar += 3
                    swvelocity = array2[timevar, 0]
                    if timevar - timevar_copy > 15:
                        endearly = 0
                if endearly == 0:
                    print ('Solar wind velocity data unavilible.')
            elif advance == 'N' or advance == 'n' or advance == 'No' or advance == 'NO' or advance == 'no':
                dontquit = 0
            else:
                advance = raw_input('Input not recognized. Try Solar Wind speed 1 min later? (y/n) ')

        #sat_position = array1[hour,0]
        #tb.dictree(cdffile)
    time_adj = (timevar - timevar_copy)/884.0*(24*60.0)
    print "The velocity is "
    print swvelocity
    print 'The time adjustment is ' + str(time_adj) + ' min'
    print str(timevar) + ' min'
    print endearly

    with pycdf.CDF(file_ko) as cdffile2:
        # print cdffile2
        array2 = cdffile2['V_GSM']
        print array2
        b = array2[:,0]

    a = list(range (0, 885))

   # plt.plot(a,b)
   # plt.show()