iPOLARRIS_v1.0.1

.gitignore

@@ -1,9 +1,14 @@
 *.ipynb
-*.nc
 *.cdf
 *.png
 *.tar.gz
 *.tar
 *.pyc
 *.p
 *.npy
+*.log
+output/*
+other_samples/*
+outputfig/*
+configtxt/*
+nexrad*

GeneralFunctions.py

@@ -35,7 +35,8 @@
 
 #import analysis_tools as AT
 #import lightning_tools as LT
-from CSU_RadarTools.csu_radartools import csu_fhc
+#from CSU_RadarTools.csu_radartools import csu_fhc
+import csu_fhc
 import general_tools as gentools
 import RadarConfig
 from matplotlib.colors import from_levels_and_colors

README

@@ -7,5 +7,11 @@ Colorado State University, Dept. Atmos. Sci.
 bdolan@atmos.colostate.edu
 May 2017
 
+Contributions by Anthony Di Stefano
+The University of British Columbia, Dept. EOAS
+adistefa@eoas.ubc.ca
+June 2021
+
 based on code from Brody Fuchs
 
+Last updated on June 23, 2021

RadarConfig.py

@@ -12,7 +12,8 @@ class RadarConfig(object):
 
     def __init__(self, dz='DZ', zdr='DR', kdp='KD', ldr='LH', rho='RH', hid = 'HID',conv='Con',
             temp='T', x='x', y='y', z='z', u='U', v='V',rr='RR', w='Wvar',vr='VR',mphys='None',exper = 'Case',
-            band = 'C',lat_0 = 0,lon_0=90.0,lat_r=None,lon_r=None,lat=None,lon=None,tm = None,radar_name = None):
+            band = 'C',lat_0 = 0,lon_0=90.0,lat_r=None,lon_r=None,lat=None,lon=None,tm = None,radar_name = None,
+            color_blind = False):
         # ******** first the polarimetric stuff *************
         self.dz_name = dz
         self.zdr_name = zdr
@@ -56,21 +57,23 @@ def __init__(self, dz='DZ', zdr='DR', kdp='KD', ldr='LH', rho='RH', hid = 'HID',
 
 
         self.species = np.array(['DZ','RN','CR','AG','WS','VI','LDG','HDG','HA','BD'])
-        self.hid_colors = ['White','LightBlue','MediumBlue','Darkorange','LightPink','Cyan','DarkGray',\
-            'Lime','Yellow','Red','Fuchsia']
+        #self.hid_colors = ['White','LightBlue','MediumBlue','Darkorange','LightPink','Cyan','DarkGray',\
+        #    'Lime','Yellow','Red','Fuchsia']
+        self.hid_colors = ['LightBlue','MediumBlue','Darkorange','LightPink','Cyan','DarkGray',\
+            'Lime','Yellow','Red','Fuchsia']       
         self.pol_vars = np.array([self.dz_name, self.zdr_name, self.kdp_name, self.ldr_name, self.rho_name, self.hid_name])
 
         self.cs_colors = ['#FFFFFF', 'DodgerBlue', 'Red', 'Khaki']
         self.cs_labels = ['', 'Strat', 'Conv', 'Mixed']
 
-        self.set_dbz_colorbar()
+        self.set_dbz_colorbar(color_blind=color_blind)
         self.set_hid_colorbar()
         self.set_cs_colorbar()
 
         # Now just set some defaults
         self.lims = {dz: [0,80], zdr: [-1, 3], kdp: [-0.5, 3], ldr: [-35, -20], rho: [0.95, 1.00], hid: [0, len(self.species)+1],w:[-25,25],vr:[-25,25],self.cs_name:[0,4],self.rr_name:[0.01,150]}
         self.delta = {dz: 10, zdr: 1, kdp: 1, ldr: 5, rho: 0.005, hid: 1,w:5,vr:5,self.cs_name:1,self.rr_name:10}
-        self.units = {dz: '(dBZ)', zdr: '(dB)', kdp: '($^{\circ}$/km)', ldr: '(dB)', rho: '', hid: '',w:'m s$^{-1}$',vr:'m s$^{-1}$',self.cs_name:'',self.rr_name:'mm hr$^{-1}$'}
+        self.units = {dz: '(dBZ)', zdr: '(dB)', kdp: '($^{\circ}$/km)', ldr: '(dB)', rho: '', hid: '',w:'(m s$^{-1}$)',vr:'(m s$^{-1}$)',self.cs_name:'',self.rr_name:'(mm hr$^{-1}$)'}
         self.names = {dz: 'Z', zdr: 'Z$_{DR}$', kdp: 'K$_{dp}$', ldr: 'LDR', rho: r'$\rho_{hv}$', hid: '',w:'',vr:'V$_r$',self.cs_name:'',self.rr_name:'RR'}
         self.longnames = {dz: 'Reflectivity', zdr: 'Differntial reflectivity', kdp: 'Specific differential phase',\
                 ldr: 'Linear depolarization ratio', rho: 'Correlation coefficient', hid: 'Hydrometeor identification',w:'Vertical Velocity',vr:'Radial Velocity',\
@@ -126,13 +129,19 @@ def sav_title(self,tm = None):
         return extra
 #############################################################################################################
 
-    def set_dbz_colorbar(self, color_list=None):
+    def set_dbz_colorbar(self, color_list=None, color_blind=False):
         if color_list is None:
             # just use the default here
-            radarcbar = ['PeachPuff','Aqua','DodgerBlue','MediumBlue','Lime', \
-                'LimeGreen','Green','Yellow','Orange','OrangeRed', \
-                'Red', 'Crimson','Fuchsia','Indigo','DarkCyan','White']
-        else:
+            if color_blind is not True:
+                radarcbar = ['PeachPuff','Aqua','DodgerBlue','Blue','Lime', \
+                    'LimeGreen','Green','Yellow','Orange','DarkOrange','Red', \
+                    'Crimson','Fuchsia','Purple','Indigo','MidnightBlue'] 
+            #radarcbar = ['PeachPuff','Aqua','DodgerBlue','MediumBlue','Lime', \
+            #    'LimeGreen','Green','Yellow','Orange','OrangeRed','Red', \
+            #    'Crimson','Fuchsia','Indigo','DarkCyan','White'] 
+            else:
+                radarcbar = ['Lavender', 'Thistle', 'Plum', 'MediumPurple', 'CornFlowerBlue', 'SkyBlue', 'PaleTurquoise', 'LightCyan', 'Yellow', 'Gold', 'Orange', 'DarkOrange', 'Chocolate', 'IndianRed', 'FireBrick', 'Maroon']
+        else: 
             radarcbar = deepcopy(color_list)
 
         temp_cmap = colors.ListedColormap(radarcbar)
@@ -151,7 +160,8 @@ def set_hid_colorbar(self, color_list=None):
             hidcbar = deepcopy(color_list)
         self.hid_cmap = colors.ListedColormap(hidcbar)
 
-        self.boundshid = np.arange(0,12)
+        #self.boundshid = np.arange(0,12)
+        self.boundshid = np.arange(self.hid_cmap.N+1)
         self.normhid = colors.BoundaryNorm(self.boundshid, self.hid_cmap.N)
 #############################################################################################################
     def set_cs_colorbar(self, color_list=None):