geospacelab 0.10.0__py3-none-any.whl → 0.10.2__py3-none-any.whl

geospacelab/__init__.py

@@ -6,7 +6,7 @@ __author__ = "Lei Cai"
 __copyright__ = "Copyright 2021, GeospaceLAB"
 __credits__ = ["Lei Cai"]
 __license__ = "BSD-3-Clause License"
-__version__ = "0.10.0"
+__version__ = "0.10.2"
 __maintainer__ = "Lei Cai"
 __email__ = "lei.cai@oulu.fi"
 __status__ = "Developing"

geospacelab/datahub/sources/madrigal/isr/eiscat/__init__.py

@@ -72,7 +72,8 @@ class Dataset(datahub.DatasetSourced):
         self.data_file_type = kwargs.pop('data_file_type', '')
         self.affiliation = kwargs.pop('affiliation', '')
         self.allow_download = kwargs.pop('allow_download', True)
-        self.metadata = None
+        self.gate_num = kwargs.pop('gate_num', None)
+        self.metadata = {}
         self.add_AACGM = kwargs.pop('add_AACGM', True)
         self.add_APEX = kwargs.pop('add_APEX', False)
 
@@ -116,7 +117,9 @@ class Dataset(datahub.DatasetSourced):
         self.check_data_files(**kwargs)
 
         for file_path in self.data_file_paths:
-            load_obj = self.loader(file_path, file_type=self.data_file_type)
+            load_obj = self.loader(file_path, file_type=self.data_file_type, gate_num=self.gate_num)
+            if self.gate_num is None:
+                self.gate_num = load_obj.gate_num
 
             for var_name in self._variables.keys():
                 self._variables[var_name].join(load_obj.variables[var_name])
@@ -130,6 +133,12 @@ class Dataset(datahub.DatasetSourced):
             self.experiment = rawdata_path.split('/')[-1].split('@')[0]
             self.affiliation = load_obj.metadata['affiliation']
             self.metadata = load_obj.metadata
+
+        inds_cmb = np.argsort(self['DATETIME'].flatten())
+        if any(np.diff(np.array(inds_cmb))<0):
+            for var_name in self.keys():
+                self[var_name].value = self[var_name].value[inds_cmb, :]
+
         if self.add_AACGM or self.add_APEX:
             self.calc_lat_lon()
             # self.select_beams(field_aligned=True)
@@ -357,21 +366,23 @@ class Dataset(datahub.DatasetSourced):
             fps_sub = []
             for ii in inds_id:
                 fp = file_paths[ii]
-                rc = re.compile(r".*_([\d]{8}T[\d]{6}).*_([\d]{8}T[\d]{6}).*[\d]{4}\-[\d]{2}\-[\d]{2}_([\w]+)@.*")
+                rc = re.compile(r".*_([\d]{8}T[\d]{6}).*_([\d]{8}T[\d]{6}).*[\d]{4}\-[\d]{2}\-[\d]{2}_([\w.]+)@.*")
                 res = rc.search(str(fp))
                 dt_0 = datetime.datetime.strptime(res.groups()[0], '%Y%m%dT%H%M%S')
                 dt_1 = datetime.datetime.strptime(res.groups()[1], '%Y%m%dT%H%M%S')
                 if (dt_0 >= self.dt_to) or (dt_1<=self.dt_fr):
                     continue
                 if str(self.pulse_code):
-                    if self.pulse_code not in res.groups()[2]:
+                    if self.pulse_code not in res.groups()[2].lower():
                         continue
                 if str(self.modulation):
-                    if self.modulation not in res.groups()[2]:
+                    if self.modulation not in res.groups()[2].lower():
                         continue
+                if '_v' in res.groups()[2].lower():
+                    continue
                 fps_sub.extend([fp])
             if len(fps_sub) > 1:
-                mylog.StreamLogger.warning("Multiple data files for a single experiment detected! Only the first one is selected.")
+                mylog.StreamLogger.warning("Multiple data files for a single experiment detected!")
                 # for fp in fps_sub:
                 #     mylog.simpleinfo.info(str(fp))
                 # fps_sub = fps_sub[0]

geospacelab/datahub/sources/madrigal/isr/eiscat/downloader.py

@@ -207,8 +207,8 @@ class Downloader(DownloaderBase):
             dt_to_exp = datetime.datetime(
                 exp.endyear, exp.endmonth, exp.endday, exp.endhour, exp.endmin, exp.endsec
             )
-            if (dt_fr_exp >= self.dt_to) or (dt_to_exp <= self.dt_fr):
-                continue
+            # if (dt_fr_exp >= self.dt_to) or (dt_to_exp <= self.dt_fr):
+            #     continue
             for file in list(exp.files):
 
                 file_path_remote = pathlib.Path(file.name)

geospacelab/datahub/sources/madrigal/isr/eiscat/loader.py

@@ -13,6 +13,7 @@ import h5py
 import numpy as np
 import pathlib
 import re
+import scipy.interpolate as si
 
 from geospacelab.config import pref as prf
 
@@ -32,12 +33,12 @@ default_variable_names = [
 
 
 class Loader:
-    def __init__(self, file_path, file_type="eiscat-hdf5"):
+    def __init__(self, file_path, file_type="eiscat-hdf5", gate_num=None):
         self.variables = {}
         self.metadata = {}
         self.file_path = file_path
         self.file_type = file_type
-
+        self.gate_num = gate_num
         self.load_data()
 
     def load_data(self):
@@ -131,8 +132,20 @@ class Loader:
                         if num_col != nrec[0]:
                             print("Note: the number of range gates doesn't match nrec!")
                         var = var.reshape(num_row, num_col)
+                        if self.gate_num is None:
+                            num_gates=self.gate_num = num_col
+                        else:
+                            num_gates = self.gate_num
+                        var_array = np.empty((num_row, num_gates))
+                        var_array[::] = np.nan
+                        for i in range(num_row):
+                            var_array[i, 0:num_col] = var[i, :]
+                        var = var_array
                     elif nrec_group == 'par1d':
-                        num_gates = int(np.max(nrec))
+                        if self.gate_num is None:
+                            self.gate_num = num_gates = int(np.max(nrec))
+                        else:
+                            num_gates = self.gate_num
                         var_array = np.empty((num_row, num_gates))
                         var_array[:, :] = np.nan
                         rec_ind_1 = 0
@@ -201,21 +214,30 @@ class Loader:
         # check height and range
         vars['HEIGHT'] = vars['HEIGHT'] / 1000.
         vars['RANGE'] = vars['RANGE'] / 1000.
-        inds = np.where(np.isnan(vars['HEIGHT']))
-        for i in range(len(inds[0])):
-            ind_0 = inds[0][i]
-            ind_1 = inds[1][i]
-            x0 = np.arange(vars['HEIGHT'].shape[0])
-            y0 = vars['HEIGHT'][:, ind_1]
-            xp = x0[np.where(np.isfinite(y0))[0]]
-            yp = y0[np.isfinite(y0)]
-            vars['HEIGHT'][ind_0, ind_1] = np.interp(x0[ind_0], xp, yp)
-
-            x0 = np.arange(vars['RANGE'].shape[0])
-            y0 = vars['RANGE'][:, ind_1]
-            xp = x0[np.isfinite(y0)]
-            yp = y0[np.isfinite(y0)]
-            vars['RANGE'][ind_0, ind_1] = np.interp(x0[ind_0], xp, yp)
+
+        inds_nan = np.where(np.isnan(vars['HEIGHT']))
+        if list(inds_nan):
+            m, n = vars['HEIGHT'].shape
+            for i in range(m):
+                yy = vars['HEIGHT'][i, :].flatten()
+                iii = np.where(~np.isfinite(yy))
+                if not list(iii):
+                    continue
+                iii = np.where(np.isfinite(yy))[0]
+                xx = np.arange(0, n)
+                f = si.interp1d(xx[iii], yy[iii], kind='linear', bounds_error=False, fill_value='extrapolate')
+                yy_new = f(xx)
+                vars['HEIGHT'][i, :] = yy_new
+
+                yy = vars['RANGE'][i, :].flatten()
+                iii = np.where(~np.isfinite(yy))
+                if not list(iii):
+                    continue
+                iii = np.where(np.isfinite(yy))[0]
+                xx = np.arange(0, n)
+                f = si.interp1d(xx[iii], yy[iii], kind='linear', bounds_error=False, fill_value='extrapolate')
+                yy_new = f(xx)
+                vars['RANGE'][i, :] = yy_new
 
         if np.isscalar(vars['AZ']):
             az = np.empty((vars['DATETIME_1'].shape[0], 1))
@@ -336,8 +358,12 @@ class Loader:
             inds_ran_max = np.append(inds_ran_max, len(ran)-1)
             inds_ran_max.sort()
 
-            ngates_max = np.max(np.diff(inds_ran_max))
-            data_array = np.empty((nvar_h5, inds_ran_min.shape[0], ngates_max))
+            if self.gate_num is None:
+                self.gate_num = ngates_max = np.max(np.diff(inds_ran_max))
+            else:
+                ngates_max = self.gate_num
+            num_row = inds_ran_min.shape[0]
+            data_array = np.empty((nvar_h5, num_row, ngates_max))
             data_array[::] = np.nan
             for ip in range(nvar_h5):
                 var_tmp = np.array(data[ip])
@@ -345,7 +371,7 @@ class Loader:
                     ind1 = inds_ran_min[i]
                     ind2 = inds_ran_max[i]
                     data_array[ip, i, 0: ind2-ind1+1] = var_tmp[ind1: ind2+1]
-            num_row = inds_ran_min.shape[0]
+
             vars_h5 = {}
             for ip in range(nvar_h5):
 
@@ -382,21 +408,31 @@ class Loader:
             vars['T_e_err'] = vars['T_e'] * np.sqrt((vars['T_i_err'] / vars['T_i']) ** 2
                                                     + (vars['T_r_err'] / vars['T_r']) ** 2)
             vars['AZ'] = vars['AZ'] % 360.
-            inds = np.where(np.isnan(vars['HEIGHT']))
-            for i in range(len(inds[0])):
-                ind_0 = inds[0][i]
-                ind_1 = inds[1][i]
-                x0 = np.arange(vars['HEIGHT'].shape[0])
-                y0 = vars['HEIGHT'][:, ind_1]
-                xp = x0[np.where(np.isfinite(y0))[0]]
-                yp = y0[np.isfinite(y0)]
-                vars['HEIGHT'][ind_0, ind_1] = np.interp(x0[ind_0], xp, yp)
-
-                x0 = np.arange(vars['RANGE'].shape[0])
-                y0 = vars['RANGE'][:, ind_1]
-                xp = x0[np.isfinite(y0)]
-                yp = y0[np.isfinite(y0)]
-                vars['RANGE'][ind_0, ind_1] = np.interp(x0[ind_0], xp, yp)
+
+            inds_nan = np.where(np.isnan(vars['HEIGHT']))
+            if list(inds_nan):
+                m, n = vars['HEIGHT'].shape
+                for i in range(m):
+                    yy = vars['HEIGHT'][i, :].flatten()
+                    iii = np.where(~np.isfinite(yy))
+                    if not list(iii):
+                        continue
+                    iii = np.where(np.isfinite(yy))[0]
+                    xx = np.arange(0, n)
+                    f = si.interp1d(xx[iii], yy[iii], kind='linear', bounds_error=False, fill_value='extrapolate')
+                    yy_new = f(xx)
+                    vars['HEIGHT'][i, :] = yy_new
+
+                    yy = vars['RANGE'][i, :].flatten()
+                    iii = np.where(~np.isfinite(yy))
+                    if not list(iii):
+                        continue
+                    iii = np.where(np.isfinite(yy))[0]
+                    xx = np.arange(0, n)
+                    f = si.interp1d(xx[iii], yy[iii], kind='linear', bounds_error=False, fill_value='extrapolate')
+                    yy_new = f(xx)
+                    vars['RANGE'][i, :] = yy_new
+
 
             self.variables = vars
             self.metadata = metadata

geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/__init__.py

@@ -46,7 +46,7 @@ default_variable_names = [
     'DATETIME', 'AZ', 'EL', 'PULSE_LENGTH',
     'P_Tx', 'n_e', 'n_e_err', 'T_i', 'T_i_err', 'T_e', 'T_e_err',
     'v_i_los', 'v_i_los_err', 'comp_mix', 'comp_mix_err',
-    'HEIGHT', 'RANGE', 'CGM_LAT', 'CGM_LON'
+    'HEIGHT', 'RANGE', 'CGM_LAT', 'CGM_LON', 'BEAM_ID', 'CHISQ'
 ]
 
 # default_data_search_recursive = True
@@ -55,7 +55,7 @@ default_attrs_required = []
 
 pulse_code_dict = {
     'alternating code': 'AC',
-    'long pulse': 'PL',
+    'long pulse': 'LP',
 }
 
 
@@ -138,8 +138,8 @@ class Dataset(datahub.DatasetSourced):
         self.beam_az = load_obj.beam_az
         self.beam_el = load_obj.beam_el
 
-        if self.add_APEX or self.add_AACGM:
-            self.calc_lat_lon()
+        # if self.add_APEX or self.add_AACGM:
+        self.calc_lat_lon()
 
         if self['HEIGHT'].value is None:
             self['HEIGHT'] = self['GEO_ALT']
@@ -353,7 +353,7 @@ class Dataset(datahub.DatasetSourced):
             pulse_code=self.pulse_code,
             dry_run=dry_run,
             data_file_root_dir=self.data_root_dir,
-            include_exp_ids=self.exp_ids,
+            include_exp_ids=self.experiment_ids,
             include_exp_name_patterns=include_exp_name_patterns)
         return download_obj.data_file_paths
 

geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/downloader.py

@@ -15,7 +15,7 @@ EXCLUDE_FILE_TYPE_PATTERNS = [['from power'], ['velocity'], ['uncorrected']]
 
 pulse_code_dict = {
     'alternating code': 'AC',
-    'long pulse': 'PL',
+    'long pulse': 'LP',
 }
 
 

geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/loader.py

@@ -37,6 +37,8 @@ var_name_dict = {
     'CGM_LAT': 'cgm_lat',
     'CGM_LON': 'cgm_long',
     'HEIGHT': 'gdalt',
+    'BEAM_ID': 'beamid',
+    'CHISQ': 'chisq',
 }
 
 

geospacelab/datahub/sources/madrigal/satellites/dmsp/s1/__init__.py

@@ -205,8 +205,8 @@ class Dataset(datahub.DatasetSourced):
         sin_glon_1 = np.sin(glon_1 * factor)
         sin_glon_2 = np.sin(glon_2 * factor)
         cos_glon_2 = np.cos(glon_2 * factor)
-        itpf_sin = interp1d(sectime_2, sin_glon_2, kind='cubic', bounds_error=False, fill_value='extrapolate')
-        itpf_cos = interp1d(sectime_2, cos_glon_2, kind='cubic', bounds_error=False, fill_value='extrapolate')
+        itpf_sin = interp1d(sectime_2, sin_glon_2, kind='linear', bounds_error=False, fill_value='extrapolate')
+        itpf_cos = interp1d(sectime_2, cos_glon_2, kind='linear', bounds_error=False, fill_value='extrapolate')
         sin_glon_2_i = itpf_sin(sectime_1)
         sin_glon_2_i = np.where(sin_glon_2_i > 1., 1., sin_glon_2_i)
         sin_glon_2_i = np.where(sin_glon_2_i < -1., -1., sin_glon_2_i)

geospacelab/datahub/sources/madrigal/satellites/dmsp/s4/__init__.py

@@ -177,8 +177,8 @@ class Dataset(datahub.DatasetSourced):
         sin_glon_1 = np.sin(glon_1 * factor)
         sin_glon_2 = np.sin(glon_2 * factor)
         cos_glon_2 = np.cos(glon_2 * factor)
-        itpf_sin = interp1d(sectime_2, sin_glon_2, kind='cubic', bounds_error=False, fill_value='extrapolate')
-        itpf_cos = interp1d(sectime_2, cos_glon_2, kind='cubic', bounds_error=False, fill_value='extrapolate')
+        itpf_sin = interp1d(sectime_2, sin_glon_2, kind='linear', bounds_error=False, fill_value='extrapolate')
+        itpf_cos = interp1d(sectime_2, cos_glon_2, kind='linear', bounds_error=False, fill_value='extrapolate')
         sin_glon_2_i = itpf_sin(sectime_1)
         cos_glon_2_i = itpf_cos(sectime_1)
         rad = np.sign(sin_glon_2_i) * (np.pi / 2 - np.arcsin(cos_glon_2_i))

geospacelab/toolbox/utilities/numpyarray.py

@@ -10,9 +10,11 @@ __docformat__ = "reStructureText"
 
 
 import numpy
-from scipy.interpolate import interp1d
+import numpy as np
+from scipy.interpolate import interp1d, griddata
 import geospacelab.toolbox.utilities.pydatetime as dttool
 import geospacelab.toolbox.utilities.pybasic as basic
+import geospacelab.toolbox.utilities.pylogging as mylog
 import datetime
 
 
@@ -133,6 +135,140 @@ def data_resample(
 
     return xnew, ynew
 
+# def resample_2d(
+#         x_data, y_data, z_data,
+#         x_data_type=None,
+#         x_data_res=None,
+#         x_data_res_scale=1.,
+#         x_grid_res=None,
+#         x_grid_res_scale=1,
+#         x_grid_min=None,
+#         x_grid_max=None,
+#         along_x_interp=True,
+#         along_x_interp_method='nearest',
+#         along_x_binning=False,
+#         y_data_res=None,
+#         y_data_res_scale=1.,
+#         y_grid_res=None,
+#         y_grid_res_scale=1.,
+#         y_grid_min=None,
+#         y_grid_max=None,
+#         along_y_interp=True,
+#         along_y_interp_method='nearest',
+#         along_y_binning=False,
+# ):
+#     if along_x_interp==True and along_x_binning==True:
+#         mylog.StreamLogger.error('The keywords "along_x_interp" and "along_x_binning" cannot be True at the same time!')
+#         raise ValueError
+#     if along_y_interp==True and along_y_binning==True:
+#         mylog.StreamLogger.error('The keywords "along_y_interp" and "along_y_binning" cannot be True at the same time!')
+#         raise ValueError
+#
+#     num_x, num_y = z_data.shape
+#     if x_data_type == 'datetime':
+#         dt0 = dttool.get_start_of_the_day(numpy.nanmin(x_data.flatten()))
+#         sectime, dt0 = dttool.convert_datetime_to_sectime(x_data, dt0=dt0)
+#         x_data = sectime
+#
+#     # check x dim:
+#     if len(x_data.shape) == 1:
+#         x_dim = 1
+#     elif len(x_data.shape) == 2:
+#         if x_data.shape[1] == num_y:
+#             x_dim = 2
+#         else:
+#             x_dim = 1
+#     else:
+#         raise ValueError
+#     if x_dim == 1:
+#         xd = np.tile(x_data.flatten(), (1, num_y))
+#     else:
+#         xd = x_data
+#     min_x = np.nanmin(xd.flatten())
+#     max_x = np.nanmax(xd.flatten())
+#     # check y dim:
+#     if len(y_data.shape) == 1:
+#         y_dim = 1
+#     elif len(y_data.shape) == 2:
+#         if y_data.shape[1] == num_y:
+#             y_dim = 2
+#         else:
+#             y_dim = 1
+#     else:
+#         raise ValueError
+#     if y_dim == 1:
+#         yd = np.tile(x_data.flatten(), (num_x, 1))
+#     else:
+#         yd = y_data
+#     min_y = np.nanmin(yd.flatten())
+#     max_y = np.nanmax(yd.flatten())
+#
+#     if x_data_res is None:
+#         x_data_res_ = np.median(np.diff(xd[:, 0].flatten()))
+#     else:
+#         x_data_res_ = x_data_res
+#     if y_data_res is None:
+#         y_data_res_ = np.median(np.diff(yd[0, :].flatten()))
+#     else:
+#         y_data_res_ = y_data_res
+#
+#     if x_grid_res is None:
+#         xx = xd[:, 0].flatten()
+#         # along_x_interp=False
+#         # along_x_binning=False
+#         # x_grid_res=x_data_res
+#     else:
+#         if x_grid_min is None:
+#             x_grid_min = np.floor((min_x / x_grid_res)) * x_grid_res
+#         if x_grid_max is None:
+#             x_grid_max = np.ceil((max_x / x_grid_res)) * x_grid_res
+#         xx = np.arange(x_grid_min, x_grid_max+x_grid_res, x_grid_res)
+#     if y_grid_res is None:
+#         yy = yd[0, :].flatten()
+#         # along_y_interp=False
+#         # along_y_binning=False
+#         # y_grid_res=y_data_res
+#     else:
+#         if y_grid_min is None:
+#             y_grid_min = np.floor((min_y / y_grid_res)) * y_grid_res
+#         if y_grid_max is None:
+#             y_grid_max = np.ceil((max_y / y_grid_res)) * y_grid_res
+#         yy = np.arange(y_grid_min, y_grid_max+y_grid_res, y_grid_res)
+#
+#     grid_x, grid_y = numpy.meshgrid(xx,yy)
+#     grid_z = np.empty_like(grid_x)
+#     grid_z[::] = np.nan
+#
+#     if along_y_interp:
+#         grid_x_1 = np.empty_like((num_x, grid_z.shape[1])) * np.nan
+#         grid_y_1 = np.empty_like((num_x, grid_z.shape[1])) * np.nan
+#         grid_z_1 = np.empty_like((num_x, grid_z.shape[1])) * np.nan
+#         for i in range(num_x):
+#             x1 = xd[i, :].flatten()
+#             y1 = yd[i, :].flatten()
+#             z1 = z_data[i, :].flatten()
+#
+#             inds_finite = np.where(np.isfinite(x1) & np.isfinite(y1) & np.isfinite(z1))[0]
+#             if not list(inds_finite):
+#                 continue
+#
+#             f = interp1d(y1[inds_finite], x1[inds_finite],
+#                 kind='nearest', bounds_error=False, fill_value=np.nan)
+#             x_i = f(grid_y[i, :].flatten())
+#             grid_x_1[i, :] = x_i
+#
+#             f = interp1d(y1[inds_finite], y1[inds_finite],
+#                 kind='nearest', bounds_error=False, fill_value=np.nan)
+#             y_i = f(grid_y[i, :].flatten())
+#             grid_y_1[i, :] = y_i
+#
+#             f = interp1d(
+#                 y1[inds_finite], z1[inds_finite],
+#                 kind=along_y_interp_method, bounds_error=False, fill_value=np.nan)
+#             z_i = f(grid_y[i, :].flatten())
+#
+#             grid_z_1[i, :] = z_i
+
 
 def regridding_2d_xgaps(
     x, y, z,

geospacelab/visualization/mpl/panels.py

@@ -736,6 +736,7 @@ class TSPanel(Panel):
         time_gap = var.visual.axis[0].mask_gap
         if time_gap is None:
             time_gap = self.time_gap
+        y_data = self._check_ydata(y_data)
         if time_gap:
             # x, y, z = arraytool.data_resample_2d(
             #     x=x_data, y=y_data, z=z_data, xtype='datetime', xres=x_data_res, method='Null', axis=0)
@@ -748,6 +749,21 @@ class TSPanel(Panel):
         data = {'x': x, 'y': y, 'z': z}
         return data
 
+    def _check_ydata(self, ydata):
+        inds_infinite = np.where(~np.isfinite(ydata))
+        if list(inds_infinite):
+            m, n = ydata.shape
+            for i in range(m):
+                yy = ydata[i, :].flatten()
+                iii = np.where(~np.isfinite(yy))
+                if not list(iii):
+                    continue
+                iii = np.where(np.isfinite(yy))[0]
+                xx = np.arange(0, n)
+                f = interp1d(xx[iii], yy[iii], kind='linear', bounds_error=False, fill_value='extrapolate')
+                yy_new = f(xx)
+                ydata[i, :] = yy_new
+        return ydata
     @staticmethod
     def generate_label(label: str, unit: str='', style: str='double'):
         label = label

{geospacelab-0.10.0.dist-info → geospacelab-0.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: geospacelab
-Version: 0.10.0
+Version: 0.10.2
 Summary: Collect, manage, and visualize geospace data.
 Home-page: https://github.com/JouleCai/geospacelab
 Author: Lei Cai
@@ -378,14 +378,9 @@ Output:
 From then on, the data source has been switched from JHUAPL to CDAWeb.
  
 ```python
-import datetime
-import matplotlib.pyplot as plt
-
-import geospacelab.visualization.mpl.geomap.geodashboards as geomap
-
 dt_fr = datetime.datetime(2015, 9, 8, 8)
 dt_to = datetime.datetime(2015, 9, 8, 23, 59)
-time1 = datetime.datetime(2015, 9, 8, 20, 21)
+time_c = datetime.datetime(2015, 9, 8, 20, 21)
 pole = 'N'
 sat_id = 'f16'
 band = 'LBHS'
@@ -396,49 +391,51 @@ dashboard = geomap.GeoDashboard(dt_fr=dt_fr, dt_to=dt_to, figure_config={'figsiz
 # If the orbit_id is specified, only one file will be downloaded. This option saves the downloading time.
 # dashboard.dock(datasource_contents=['jhuapl', 'dmsp', 'ssusi', 'edraur'], pole='N', sat_id='f17', orbit_id='46863')
 # If not specified, the data during the whole day will be downloaded.
-dashboard.dock(
-    datasource_contents=dashboard.dock(datasource_contents=['cdaweb', 'dmsp', 'ssusi', 'edr_aur'], 
-    pole=pole, sat_id=sat_id, orbit_id=None
-)
+ds_ssusi = dashboard.dock(datasource_contents=['cdaweb', 'dmsp', 'ssusi', 'edr_aur'], pole=pole, sat_id=sat_id, orbit_id=None)
 ds_s1 = dashboard.dock(
     datasource_contents=['madrigal', 'satellites', 'dmsp', 's1'],
-    dt_fr=time1 - datetime.timedelta(minutes=45),
-    dt_to=time1 + datetime.timedelta(minutes=45),
-    sat_id=sat_id
-)
+    dt_fr=time_c - datetime.timedelta(minutes=45),
+    dt_to=time_c + datetime.timedelta(minutes=45),
+    sat_id=sat_id, replace_orbit=True)
 
 dashboard.set_layout(1, 1)
 
 # Get the variables: LBHS emission intensiy, corresponding times and locations
-lbhs = dashboard.assign_variable('GRID_AUR_' + band, dataset_index=0)
-dts = dashboard.assign_variable('DATETIME', dataset_index=0).value.flatten()
-mlat = dashboard.assign_variable('GRID_MLAT', dataset_index=0).value
-mlon = dashboard.assign_variable('GRID_MLON', dataset_index=0).value
-mlt = dashboard.assign_variable(('GRID_MLT'), dataset_index=0).value
+lbhs = ds_ssusi['GRID_AUR_' + band]
+dts = ds_ssusi['DATETIME'].flatten()
+mlat = ds_ssusi['GRID_MLAT']
+mlon = ds_ssusi['GRID_MLON']
+mlt = ds_ssusi['GRID_MLT']
 
 # Search the index for the time to plot, used as an input to the following polar map
-ind_t = dashboard.datasets[0].get_time_ind(ut=time1)
+ind_t = dashboard.datasets[0].get_time_ind(ut=time_c)
+if (dts[ind_t] - time_c).total_seconds()/60 > 60:     # in minutes
+    raise ValueError("The time does not match any SSUSI data!")
 lbhs_ = lbhs.value[ind_t]
-mlat_ = mlat[ind_t]
-mlon_ = mlon[ind_t]
-mlt_ = mlt[ind_t]
+mlat_ = mlat.value[ind_t]
+mlon_ = mlon.value[ind_t]
+mlt_ = mlt.value[ind_t]
 # Add a polar map panel to the dashboard. Currently the style is the fixed MLT at mlt_c=0. See the keywords below:
-panel1 = dashboard.add_polar_map(row_ind=0, col_ind=0, style='mlt-fixed', cs='AACGM', mlt_c=0., pole=pole, ut=time1, boundary_lat=65., mirror_south=True)
+panel = dashboard.add_polar_map(
+    row_ind=0, col_ind=0, style='mlt-fixed', cs='AACGM',
+    mlt_c=0., pole=pole, ut=time_c, boundary_lat=55., mirror_south=True
+)
 
 # Some settings for plotting.
 pcolormesh_config = lbhs.visual.plot_config.pcolormesh
 # Overlay the SSUSI image in the map.
-ipm = panel1.overlay_pcolormesh(data=lbhs_, coords={'lat': mlat_, 'lon': mlon_, 'mlt': mlt_}, cs='AACGM',
-                                regridding=True, **pcolormesh_config)
+ipc = panel.overlay_pcolormesh(
+    data=lbhs_, coords={'lat': mlat_, 'lon': mlon_, 'mlt': mlt_}, cs='AACGM', 
+    regridding=False, **pcolormesh_config)
 # Add a color bar
-panel1.add_colorbar(ipm, c_label=band + " (R)", c_scale=pcolormesh_config['c_scale'], left=1.1, bottom=0.1,
+panel.add_colorbar(ipc, c_label=band + " (R)", c_scale=pcolormesh_config['c_scale'], left=1.1, bottom=0.1,
                     width=0.05, height=0.7)
 
 # Overlay the gridlines
-panel1.overlay_gridlines(lat_res=5, lon_label_separator=5)
+panel.overlay_gridlines(lat_res=5, lon_label_separator=5)
 
 # Overlay the coastlines in the AACGM coordinate
-panel1.overlay_coastlines()
+panel.overlay_coastlines()
 
 # Overlay cross-track velocity along satellite trajectory
 sc_dt = ds_s1['SC_DATETIME'].value.flatten()
@@ -448,14 +445,22 @@ sc_alt = ds_s1['SC_GEO_ALT'].value.flatten()
 sc_coords = {'lat': sc_lat, 'lon': sc_lon, 'height': sc_alt}
 
 v_H = ds_s1['v_i_H'].value.flatten()
-panel1.overlay_cross_track_vector(vector=v_H, unit_vector=1000, alpha=0.5, color='r', sc_coords=sc_coords, sc_ut=sc_dt, cs='GEOC')
+panel.overlay_cross_track_vector(
+    vector=v_H, unit_vector=1000, vector_unit='m/s', alpha=0.3, color='red',
+    sc_coords=sc_coords, sc_ut=sc_dt, cs='GEO',
+)
 # Overlay the satellite trajectory with ticks
-panel1.overlay_sc_trajectory(sc_ut=sc_dt, sc_coords=sc_coords, cs='GEOC')
+panel.overlay_sc_trajectory(sc_ut=sc_dt, sc_coords=sc_coords, cs='GEO')
+
+# Overlay sites
+panel.overlay_sites(
+    site_ids=['TRO', 'ESR'], coords={'lat': [69.58, 78.15], 'lon': [19.23, 16.02], 'height': 0.}, 
+    cs='GEO', marker='^', markersize=2)
 
 # Add the title and save the figure
 polestr = 'North' if pole == 'N' else 'South'
-panel1.add_title(title='DMSP/SSUSI, ' + band + ', ' + sat_id.upper() + ', ' + polestr + ', ' + time1.strftime('%Y-%m-%d %H%M UT'))
-plt.savefig('DMSP_SSUSI_' + time1.strftime('%Y%m%d-%H%M') + '_' + band + '_' + sat_id.upper() + '_' + pole, dpi=300)
+panel.add_title(title='DMSP/SSUSI, ' + band + ', ' + sat_id.upper() + ', ' + polestr + ', ' + time_c.strftime('%Y-%m-%d %H%M UT'))
+plt.savefig('DMSP_SSUSI_' + time_c.strftime('%Y%m%d-%H%M') + '_' + band + '_' + sat_id.upper() + '_' + pole, dpi=300)
 
 # show the figure
 plt.show()

{geospacelab-0.10.0.dist-info → geospacelab-0.10.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-geospacelab/__init__.py,sha256=f4iwnBOYn05XseQui8vNYJ279570jPZKwy0eTnws92Y,801
+geospacelab/__init__.py,sha256=ERG2N0CIY6iNLEbC9RPSn3k4KdxrXlMj4BI9U9Tzsg0,801
 geospacelab/config/__init__.py,sha256=D5A0ORTubSaLEXGqPmg-mLH_KNqINicOeeNFqkGpXrk,641
 geospacelab/config/_preferences.py,sha256=DakPjKJQ0VRe2Mgc8bakw585u4N8qVqYYvqnoLdyvH4,4726
 geospacelab/coords/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -154,9 +154,9 @@ geospacelab/datahub/sources/madrigal/gnss/tecmap/downloader.py,sha256=9Z4zZm-xNS
 geospacelab/datahub/sources/madrigal/gnss/tecmap/loader.py,sha256=JytJZfzUp2ijn0w1ADoA5-aRP1c674T4wz0mRSFSvPg,2895
 geospacelab/datahub/sources/madrigal/gnss/tecmap/variable_config.py,sha256=d-1XeMMZOyYFQAOYjPwIS2USBY9B8YjRfIww1TowVqE,1517
 geospacelab/datahub/sources/madrigal/isr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-geospacelab/datahub/sources/madrigal/isr/eiscat/__init__.py,sha256=XsRlb-89n4SoP0lELoYeHBFpKLLFmTnPwkTEXsCb1H0,19963
-geospacelab/datahub/sources/madrigal/isr/eiscat/downloader.py,sha256=UDIVMuFmNsC97F-nTYib5FyU3ag41wUHnDDmGZ0Bcec,26140
-geospacelab/datahub/sources/madrigal/isr/eiscat/loader.py,sha256=YbpbFtVyk15ejqPq9O1VeUZKyK95Epmv2LsMwsQ3bf0,16337
+geospacelab/datahub/sources/madrigal/isr/eiscat/__init__.py,sha256=sxHcvgMy_Kjnl9TppU-jVw7yPoPtUZwxsieuCOBr2aY,20413
+geospacelab/datahub/sources/madrigal/isr/eiscat/downloader.py,sha256=d-oUtCh8NHbkmEKfbe1oGb0-DzZsQH0CQXAedncMfTU,26144
+geospacelab/datahub/sources/madrigal/isr/eiscat/loader.py,sha256=vxip8UcPqA0S4H3aFSVoT8BgMi6J9Q3noaMIjbIuNy4,17953
 geospacelab/datahub/sources/madrigal/isr/eiscat/utilities.py,sha256=TImaJSVDyqvXezSCTAzMgAQ7CmrgD5YjyUiRl-K_dWk,2514
 geospacelab/datahub/sources/madrigal/isr/eiscat/variable_config.py,sha256=Lny8eVbtE7wxcYtRcfUrVDY6vkzfxD5hb29mc3ytENI,9153
 geospacelab/datahub/sources/madrigal/isr/eiscat/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -174,9 +174,9 @@ geospacelab/datahub/sources/madrigal/isr/millstonehill/vi/loader.py,sha256=kgV5c
 geospacelab/datahub/sources/madrigal/isr/millstonehill/vi/variable_config.py,sha256=tz650AGv4XedOEnB-DoN6VgIk6SNkEjnBOweACuXbwc,8951
 geospacelab/datahub/sources/madrigal/isr/pfisr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 geospacelab/datahub/sources/madrigal/isr/pfisr/downloader.py,sha256=uCrJR-MJ_LJ2JyQ971VkiN56_Kl2C9y8RhdvG9B0vX8,6521
-geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/__init__.py,sha256=v5Wx2NfFX3GsOfNNJ26sRjDBT-8T_gG1Sne_NR3hXiE,16177
-geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/downloader.py,sha256=oO8QpDGw7XFSbJjtNtBES5vA4g1HGFUMkYteX6h17xE,2763
-geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/loader.py,sha256=jN95YU11l0PuJZ7k08lh5-4h2-Mhn_p-2VgCghnd_mQ,8495
+geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/__init__.py,sha256=twxLco9U0-qZrZspe2LBokIkmG-8bHDlu_mZVMrPHY0,16202
+geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/downloader.py,sha256=-2y1BRdQOmNjnXoWHIHh1PCRbeuKZh0Bn3LIMYw0yV4,2763
+geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/loader.py,sha256=JsZwNSsEly__FxT1Px57Te5pKbteymE3j7B0oLBgKWc,8542
 geospacelab/datahub/sources/madrigal/isr/pfisr/fitted/variable_config.py,sha256=D4xdYFk--ll0ATqNtjoX9pwrsCryVV3TuPSbGR92drI,8167
 geospacelab/datahub/sources/madrigal/isr/pfisr/vi/__init__.py,sha256=rHx79V0gFwC4ZsSt79QkKaCE6XtdWWoQUaFaXVq49LY,12719
 geospacelab/datahub/sources/madrigal/isr/pfisr/vi/downloader.py,sha256=az1OxIa6-F4pvTAJGTqjuc6rfza1e3qyAiRf6sEje7c,2413
@@ -188,10 +188,10 @@ geospacelab/datahub/sources/madrigal/satellites/dmsp/downloader.py,sha256=jnA8Xp
 geospacelab/datahub/sources/madrigal/satellites/dmsp/e/__init__.py,sha256=2LmuPOzOH9lhMqXkrb4gj9Wxsmdq2mXSA0phTlrK0eU,10309
 geospacelab/datahub/sources/madrigal/satellites/dmsp/e/loader.py,sha256=kZ9AeI0LsNWS26Kwfp6azvDl2IutcWKBPDd6o0tKASc,4902
 geospacelab/datahub/sources/madrigal/satellites/dmsp/e/variable_config.py,sha256=exp1bYVmDBVlIbraeNgAuVEnjxg5IC2ED1cZIEiPj1U,10280
-geospacelab/datahub/sources/madrigal/satellites/dmsp/s1/__init__.py,sha256=FcSeOfcymCcqDukxSvHTEpdxa-r2ZpZXC-dbb0t7M7I,11534
+geospacelab/datahub/sources/madrigal/satellites/dmsp/s1/__init__.py,sha256=u7Z0SFUaLNzPV7aIqcaEUTyIIYPHHJF3rXMWCDYbSUY,11536
 geospacelab/datahub/sources/madrigal/satellites/dmsp/s1/loader.py,sha256=o5zEkosq7LyNlBIjlefKUpYpePCDbyL_yd3HrrUrk5Q,3360
 geospacelab/datahub/sources/madrigal/satellites/dmsp/s1/variable_config.py,sha256=Amp90wHogRQOESLH2b4D0jfeBozIKvvOnwx-_waU0us,6745
-geospacelab/datahub/sources/madrigal/satellites/dmsp/s4/__init__.py,sha256=YIVUE3rrOs67KCBUTfExoHz-EoVx2OoJqEfPGClp8Tk,10196
+geospacelab/datahub/sources/madrigal/satellites/dmsp/s4/__init__.py,sha256=0Qkn18pmVBmc0y0Kk6TRJBynoU_ysYMsXLzAAXB37M8,10198
 geospacelab/datahub/sources/madrigal/satellites/dmsp/s4/loader.py,sha256=LL8IAMwmgQX06lgVECh6Rx6TGXEoHmAkFDmouJ6kQOo,3229
 geospacelab/datahub/sources/madrigal/satellites/dmsp/s4/variable_config.py,sha256=4DhvQxFonoYiU_3f9ObE2OkJOXDkHfw0RUMHa3FLTv4,4054
 geospacelab/datahub/sources/ncei/__init__.py,sha256=MHpNJ-PJBjywheG5CZDgj89fsxVG5qSxi_vkTsxibd8,658
@@ -320,7 +320,7 @@ geospacelab/toolbox/io/__init__.py,sha256=YcFkmOGWXNiOXWsMFxFPhn32UnbxmeSrluTp6O
 geospacelab/toolbox/io/dialog.py,sha256=y87PKCXQS5_axbrNuVrcYN2uIfr-ZHICr7yMQ0yH1M4,1067
 geospacelab/toolbox/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 geospacelab/toolbox/utilities/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-geospacelab/toolbox/utilities/numpyarray.py,sha256=Xa9YBYOgIu9gUHnmyB-j1TmmPyt4KSukVmA4Y1dwxDA,7063
+geospacelab/toolbox/utilities/numpyarray.py,sha256=olclT06DRPFSYaEdiD1uqwHvM0vHNDFjkLstck6_clE,11862
 geospacelab/toolbox/utilities/numpymath.py,sha256=xr-u9vsN5UMZ6pjomJ2O3A1n00Eg3yH5fN4lcY7rR_Q,3810
 geospacelab/toolbox/utilities/pybasic.py,sha256=-wLWoQMTroJ49nn8H_-hsFxDSiHXMiea86U6u0W8_cM,3310
 geospacelab/toolbox/utilities/pyclass.py,sha256=DnnzLH1ovlxArtqCpywZUfDoNzYi7HmMW2s17Xd30LI,2813
@@ -341,7 +341,7 @@ geospacelab/visualization/mpl/axis_ticks.py,sha256=vkS7eqSop5MNsxZ8l3EqdJEW7obF7
 geospacelab/visualization/mpl/colormaps.py,sha256=o4NfZPZK29Lefc4Jw-9OAN5VveqLuEEHKbP3JjUFSFI,17685
 geospacelab/visualization/mpl/dashboards.py,sha256=jk8W_j1SzCfUUYeTm5e7uA3xjlXpUIINdJPmBc53t-E,17232
 geospacelab/visualization/mpl/figure.py,sha256=stHgAG_qH-b87-rIXBfFnG9p3cyD2oDnx3MA6kDgX3A,3335
-geospacelab/visualization/mpl/panels.py,sha256=3vAIjP0iE08Xa7fxmwbxu0vO4sokKGZryypzcXo1BUs,35151
+geospacelab/visualization/mpl/panels.py,sha256=uYC18Nxk_WjAJmv827D89znY9sqSSsDRAE23wG8c6xc,35817
 geospacelab/visualization/mpl/ts_viewer.py,sha256=4sOwXeWkdn5Z0BA6c-zxj4dAYxHP9tbuFwsbuJz4_ag,67416
 geospacelab/visualization/mpl/geomap/__base__.py,sha256=v3lVV5NVQMi0csNxq1ioORyizULqGN92M0DrlGjjBRQ,2940
 geospacelab/visualization/mpl/geomap/__init__.py,sha256=lsCyNIg5d5ZiMyRvE1zIuCnIhL47x8aYbOKZUX5E8fE,327
@@ -371,7 +371,7 @@ geospacelab/wrapper/geopack/geopack/t89.py,sha256=zDVNPrmtK1NnNHgohQEPqOOJDsm2Z-
 geospacelab/wrapper/geopack/geopack/t96.py,sha256=ktcoo1R7Z3NtkWHENuseu48ub4-JfQGqFV0ZOtd0zH8,65292
 geospacelab/wrapper/geopack/geopack/test_geopack1.md,sha256=dMUY0O1BgZsKpmJ6BLSQ80B6p6DZcB7OceFeyPOlFK0,15324
 geospacelab/wrapper/geopack/geopack/test_geopack1.py,sha256=qjLz6O3BAk3H58IpmxXyftwZTkh3vPGp49C-al4hjf0,6669
-geospacelab-0.10.0.dist-info/licenses/LICENSE,sha256=2yRlwLt4o5Z6OZAGcyvBj-zfFX1Uw7E6CzqODg7khqs,1515
+geospacelab-0.10.2.dist-info/licenses/LICENSE,sha256=2yRlwLt4o5Z6OZAGcyvBj-zfFX1Uw7E6CzqODg7khqs,1515
 test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 test/test_ampere.py,sha256=0-HZURubpv1mBK3bJ_qTqx39L1jezgRoU5neXMPYgZQ,2968
 test/test_dmsp_s1.py,sha256=5m_7mjdDGja8ovshNPV3LKW_6q6mIwT9XKqoyRiH79A,3588
@@ -381,7 +381,7 @@ test/test_omni.py,sha256=Zk1LZozPiY5V0aSRmK6GTQuB01hHn_j2j3Brm6Ea_po,1632
 test/test_superdarn.py,sha256=uP55muvXryPzNGHinWkiGv2PxvRs4f9M9h1WIBEBW7k,2846
 test/test_swarm.py,sha256=PDDE9nUshhQpXZbV_ZwcsjbMhI73fRaojTZv9rtRzZE,15568
 test/test_swarm_new.py,sha256=mzhMAx-M9W3Ue5noTyfBx4c3Vtc3b_ZUEvGgL9v8UE4,853
-geospacelab-0.10.0.dist-info/METADATA,sha256=HylyGl0glST5fjah47v0EQxqqmga1iI8iM2JKZO0DvQ,24337
-geospacelab-0.10.0.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-geospacelab-0.10.0.dist-info/top_level.txt,sha256=98eDwrSNgyQFAtSA06QMP71gw9BzgIj0uvkTudpGly4,12
-geospacelab-0.10.0.dist-info/RECORD,,
+geospacelab-0.10.2.dist-info/METADATA,sha256=qZZNzZ0LIB9gD7DVcJA0gbis5euI8GLsseAEGSgFl6I,24351
+geospacelab-0.10.2.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+geospacelab-0.10.2.dist-info/top_level.txt,sha256=98eDwrSNgyQFAtSA06QMP71gw9BzgIj0uvkTudpGly4,12
+geospacelab-0.10.2.dist-info/RECORD,,