def omni_load(dates, level, downloadonly=False, varformat=None, get_support_data=False, prefix='', suffix=''): """Loads OMNI data into pytplot variables""" file_list = omni_filename(dates, level) # 1. Download files count = 0 downloaded_files = [] for remotef, localf in file_list: count += 1 resp, err, localfile = pyspedas.download_files(remotef, localf) if resp: print(str(count) + '. File was downloaded. Location: ' + localfile) downloaded_files.append(localfile) else: print(str(count) + '. Error: Could not download file: ' + remotef) print(err) print('Downloaded ' + str(len(downloaded_files)) + ' files.') # 2. Load files into tplot downloaded_vars = [] if not downloadonly: try: downloaded_vars = pytplot.cdf_to_tplot(downloaded_files, varformat, get_support_data, prefix, suffix, False, True) except TypeError as e: msg = "cdf_to_tplot could not load all data.\nError:\n" + str(e) print(msg) print('Loaded ' + str(len(downloaded_vars)) + ' variables.') # 3. Time clip if len(downloaded_vars) > 0: pyspedas.time_clip(downloaded_vars, dates[0], dates[1], '')
def themis_load(dates, probes, instruments, level, downloadonly=False, varformat=None, get_support_data=False, prefix='', suffix=''): """Loads themis data into pytplot variables""" file_list = themis_filename(dates, probes, instruments, level) # print(file_list) count = 0 dcount = 0 for remotef, localf in file_list: count += 1 resp, err, localfile = pyspedas.download_files(remotef, localf) if resp: print(str(count) + '. File was downloaded. Location: ' + localfile) dcount += 1 if not downloadonly: try: cdfvars = pyspedas.cdf_to_tplot(localfile, varformat, get_support_data, prefix, suffix, False, True) except TypeError as e: msg = "cdf_to_tplot could not load " + localfile + "\nError:\n" + str(e) print(msg) else: print( str(count) + '. There was a problem. Could not download \ file: ' + remotef) print(err) if len(dates) == 2: pyspedas.time_clip(cdfvars, dates[0], dates[1], cdfvars) print('Downloaded ' + str(dcount) + ' files.') print('tplot variables:') print(pyspedas.tplot_names())
from pyspedas import subtract_median subtract_median('mms4_fgm_b_gsm_brst_l2') tplot([ 'mms4_fgm_b_gsm_brst_l2', 'mms4_fgm_b_gsm_brst_l2-d', 'mms4_fgm_b_gsm_brst_l2-m' ]) # time clip from pyspedas import time_clip time_clip([ 'mms4_fgm_b_gsm_brst_l2', 'mms4_fgm_b_gsm_brst_l2-d', 'mms4_fgm_b_gsm_brst_l2-m' ], '2015-10-16/13:06:45', '2015-10-16/13:07', suffix='') tplot([ 'mms4_fgm_b_gsm_brst_l2', 'mms4_fgm_b_gsm_brst_l2-d', 'mms4_fgm_b_gsm_brst_l2-m' ]) # clip the data from pyspedas import tclip tclip([ 'mms4_fgm_b_gsm_brst_l2', 'mms4_fgm_b_gsm_brst_l2-d', 'mms4_fgm_b_gsm_brst_l2-m'
def test_timeclip(self): """Test time_clip.""" time_clip('aaabbbccc', 1577308800, 1577598800) # Test non-existent tn = [ 1577112800, 1577308800, 1577598800, 1577608800, 1577998800, 1587998800 ] dn = [3., 5., 8., 15., 20., 1.] store_data('test1', data={'x': tn, 'y': dn}) time_clip('aaabbb', 1577308800, 1577598800) time_clip('test1', 1577112800, 1577608800) d = get_data('test1-tclip') dd = d[1] time_clip('test', 1577308800, 1577598800, new_names='name-clip') time_clip(['test', 'name-clip'], 1577308800, 1577598800, new_names='name1-ci') time_clip('test', 1577308800, 1577598800, overwrite=1) time_clip('test', 1577308800, 1577598800, new_names="testtest") time_clip(['test', 'test1'], 1577308800, 1577598800, new_names="testtest2") time_clip('test1', 1677112800, 1577608800) self.assertTrue((dd == [3., 5., 8., 15.]).all())