metradar 0.1.6__py3-none-any.whl → 0.1.8.2__py3-none-any.whl

metradar/graph/parse_pal.py

@@ -0,0 +1,157 @@
+
+# _*_ coding: utf-8 _*_
+
+'''
+专门用来解析色标文件的脚本
+'''
+
+
+
+import numpy as np
+from matplotlib.colors import LinearSegmentedColormap, Normalize
+import matplotlib.pyplot as plt
+from matplotlib.colorbar import ColorbarBase
+import re
+
+# 请将色标文件中的solidcolor字段替换为color
+def parse(file):
+	lines = []
+	vals = []
+	red = []
+	green = []
+	blue = []
+	last_color = None
+	transit_flag = False
+	with open(file, 'r',encoding='gb18030') as f:
+		for line in f:
+			if not line.lower().startswith('color:'):
+				continue
+			l = line.lower().lstrip('color: ').strip()
+			lines.append(l)
+	lines.sort(key=lambda x: float(x.split(' ')[0]))
+	color_len = len(lines)
+	for idx, l in enumerate(lines):
+		segs = [i for i in l.split(' ') if i]
+		vals.append(float(segs[0]))
+		current_color = tuple(int(i) / 255 for i in segs[1:4])
+		if color_len - idx == 2:
+			transit_flag = transit_flag if transit_flag else False
+		if not isinstance(last_color, tuple) and len(segs) == 3:
+			red.append((0, current_color[0]))
+			green.append((0, current_color[1]))
+			blue.append((0, current_color[2]))
+			last_color = current_color
+		else:
+			if len(segs) == 7 or color_len - idx == 2:
+				transit_color = tuple(int(i) / 255 for i in segs[4:7])
+				if transit_flag:
+					red.append((last_color[0], current_color[0]))
+					green.append((last_color[1], current_color[1]))
+					blue.append((last_color[2], current_color[2]))
+				else:
+					red.append((current_color[0], current_color[0]))
+					green.append((current_color[1], current_color[1]))
+					blue.append((current_color[2], current_color[2]))
+				last_color = transit_color
+				transit_flag = True
+			else:
+				red.append((current_color[0], current_color[0]))
+				green.append((current_color[1], current_color[1]))
+				blue.append((current_color[2], current_color[2]))
+				last_color = current_color
+				transit_flag = False
+	norm_array = (np.array(vals) - vals[0]) / (vals[-1] - vals[0])
+	cdict = {'red':[], 'green':[], 'blue':[]}
+	for idx in range(len(norm_array)):
+		cdict['red'].append((norm_array[idx],) + red[idx])
+		cdict['green'].append((norm_array[idx],) + green[idx])
+		cdict['blue'].append((norm_array[idx],) + blue[idx])
+	return LinearSegmentedColormap('cmap', cdict), Normalize(vals[0], vals[-1])
+
+# 可以支持alpha，朱文剑，20210831
+def parse_pro(file):
+	lines = []
+	vals = []
+	red = []
+	green = []
+	blue = []
+	alpha=[]
+	last_color = None
+	transit_flag = False
+	units=''
+	with open(file, 'r') as f:
+		for line in f:
+			if line.lower().startswith('units'):
+				units = re.split(r"[ ]+",line)[1].strip()
+			if not line.lower().startswith('color'):
+				continue
+			lines.append(line.strip())
+	lines.sort(key=lambda x: float(re.split(r"[ ]+",x)[1]))
+	color_len = len(lines)
+	for idx, l in enumerate(lines):
+		segs = [i for i in l.split(' ') if i]
+		vals.append(float(segs[1]))
+		
+		current_color = tuple(int(i) / 255 for i in segs[2::])
+		
+		if color_len - idx == 2:
+			transit_flag = transit_flag if transit_flag else False
+		
+		if len(segs) >=8 or color_len - idx == 2:
+			if len(segs) == 10:
+				transit_color = tuple(int(i) / 255 for i in segs[6:10])
+			elif len(segs)==8:
+				transit_color = tuple(int(i) / 255 for i in segs[5:8])
+			
+			if transit_flag:
+				red.append((last_color[0], current_color[0]))
+				green.append((last_color[1], current_color[1]))
+				blue.append((last_color[2], current_color[2]))
+				if segs[0]=='Color4:':
+					alpha.append((last_color[3], current_color[3]))
+				else:
+					alpha.append((1, 1))
+			else:
+				red.append((current_color[0], current_color[0]))
+				green.append((current_color[1], current_color[1]))
+				blue.append((current_color[2], current_color[2]))
+				if segs[0]=='Color4:':
+					alpha.append((current_color[3], current_color[3]))
+				else:
+					alpha.append((1.0, 1.0))
+			if len(segs) == 10 or len(segs) == 8:
+				last_color = transit_color
+				transit_flag = True
+		else:
+			red.append((current_color[0], current_color[0]))
+			green.append((current_color[1], current_color[1]))
+			blue.append((current_color[2], current_color[2]))
+			if segs[0]=='Color4:':
+				alpha.append((current_color[3], current_color[3]))
+			else:
+				alpha.append((1.0, 1.0))
+			last_color = current_color
+			transit_flag = False
+
+	norm_array = (np.array(vals) - vals[0]) / (vals[-1] - vals[0])
+	cdict = {'red':[], 'green':[], 'blue':[], 'alpha':[]}
+	
+	for idx in range(len(norm_array)):
+		cdict['red'].append((norm_array[idx],) + red[idx])
+		cdict['green'].append((norm_array[idx],) + green[idx])
+		cdict['blue'].append((norm_array[idx],) + blue[idx])
+		cdict['alpha'].append((norm_array[idx],) + alpha[idx])
+	
+	outdic=dict()
+	outdic['cmap'] = LinearSegmentedColormap('cmap', cdict)
+	outdic['norm'] = Normalize(vals[0], vals[-1])
+	outdic['units'] = units
+	return outdic
+
+if __name__ == "__main__":
+
+	outdic = parse_pro('gr2_colors/IR_dark_alpha.pal')
+	fig = plt.figure(figsize=(3, 11))
+	ax = plt.gca()
+	cbar = ColorbarBase(ax, orientation="vertical", cmap=outdic['cmap'], norm=outdic['norm'])
+	plt.show()

metradar/io/__init__.py

@@ -0,0 +1,9 @@
+"""
+METRADAR is a python package to read or retrieve radar data
+
+"""
+
+__author__ = "The R & D Center for Weather Forecasting Technology in NMC, CMA"
+
+
+__all__ = [s for s in dir() if not s.startswith('_')]

metradar/{cnrad_level2.py → io/cnrad_level2.py}

@@ -1,3 +1,6 @@
+
+# _*_ coding: utf-8 _*_
+
 """
 Functions for reading CINRAD level 2 files.
 some functions are based on pyart
@@ -770,9 +773,10 @@ class CNRADLevel2File(object):
         else:
             outdic_vol['height'] = int(self.dic_stcfg['grd_height'])
         if self.dic_stcfg['ana_height'] > 8000:
-            outdic_vol['feedhorn_height'] = int(self.dic_stcfg['ana_height']/1000)
+            # 这里馈源高度减去地基高度，才是天线相对地基的高度
+            outdic_vol['feedhorn_height'] = int(self.dic_stcfg['ana_height']/1000) - outdic_vol['height']
         else:
-            outdic_vol['feedhorn_height'] = int(self.dic_stcfg['ana_height'])
+            outdic_vol['feedhorn_height'] = int(self.dic_stcfg['ana_height']) - outdic_vol['height']
 
         outdic_vol['refl_calib'] = 1.0
         outdic_vol['power_h'] = 0.0
@@ -892,7 +896,46 @@ class CNRADLevel2File(object):
 
         return block_name, outdic,ptr
 
+    def get_beam_width_h(self):
+        """
+        Return the horizontal beam width in degrees.
+
+        Returns
+        -------
+        beam_width_h : float
+            Horizontal beam width in degrees.
+
+        """
+        return [self.dic_stcfg['beamwidth_h'],]
+
+    def get_beam_width_v(self):
+        """
+        Return the vertical beam width in degrees.
+
+        Returns
+        -------
+        beam_width_v : float
+            Vertical beam width in degrees.
+
+        """
+        return [self.dic_stcfg['beamwidth_v'],]
+    
+    def get_antenna_gain(self):
+        """
+        Return the antenna gain in dBi.
+
+        Returns
+        -------
+        antenna_gain : float
+            Antenna gain in dBi.
 
+        """
+        return [45,] # db
+    
+    def get_receiver_bandwidth(self):
+        
+        return [1.5e6,]  # Hz
+    
     def _get_msg1_from_buf(self,buf, pos, dic):
         """ Retrieve and unpack a MSG1 record from a buffer. """
         msg_header_size = _structure_size(MSG_HEADER)

metradar/{decode_fmt_pyart.py → io/decode_fmt_pyart.py}

@@ -1,7 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (c) 2022 NMC Developers.
-# Distributed under the terms of the GPL V3 License.
+
 # 将FMT格式数据直接解析为pyart的radar object格式
 # Wenjian Zhu
 # 
@@ -16,6 +15,7 @@ from .cnrad_level2 import CNRADLevel2File
 import warnings
 import numpy as np
 
+
 mapping = {
     "REF": "reflectivity",
     "VEL": "velocity",
@@ -176,8 +176,25 @@ def read_cnrad_fmt(filename, field_names=None, additional_metadata=None,
     unambiguous_range['data'] = (
         nfile.get_unambigous_range(scans).astype('float32'))
 
+    radar_beam_width_h = filemetadata('radar_beam_width_h')
+    radar_beam_width_v = filemetadata('radar_beam_width_v')
+    radar_beam_width_h['data'] = nfile.get_beam_width_h()
+    radar_beam_width_v['data'] = nfile.get_beam_width_v()
+    
+    radar_atenna_gain = filemetadata('radar_antenna_gain')
+    radar_receiver_bandwidth = filemetadata('radar_receiver_bandwidth')
+    radar_atenna_gain['data'] = nfile.get_antenna_gain()
+    radar_receiver_bandwidth['data'] = nfile.get_receiver_bandwidth()
+    
+
     instrument_parameters = {'unambiguous_range': unambiguous_range,
-                             'nyquist_velocity': nyquist_velocity, }
+                             'nyquist_velocity': nyquist_velocity, 
+                             'radar_beam_width_h': radar_beam_width_h,
+                             'radar_beam_width_v': radar_beam_width_v,
+                             'radar_antenna_gain_h': radar_atenna_gain,
+                             'radar_antenna_gain_v': radar_atenna_gain,
+                             'radar_receiver_bandwidth': radar_receiver_bandwidth,
+                             }
 
     nfile.close()
     return Radar(