Skip to content

Commit

Permalink
rename libs name and modif import path
Browse files Browse the repository at this point in the history
  • Loading branch information
zy committed Sep 24, 2019
1 parent 257d319 commit 7efdc6a
Show file tree
Hide file tree
Showing 21 changed files with 59 additions and 80 deletions.
2 changes: 2 additions & 0 deletions NuistRadar/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from . import core, graph, io
__all__ = ["core", "graph", "io"]
2 changes: 1 addition & 1 deletion NuistRadar/configure/default_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
注意:距离单位统一到m
"""

FILL_VALUE = -999.0
FILL_VALUE = -9999.0
_LIGHT_SPEED = 2.99792458e8

DEFAULT_METADATA = {
Expand Down
2 changes: 1 addition & 1 deletion NuistRadar/configure/location_config.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# -*- coding: utf-8 -*-
import pandas as pd
from cfgs import Radar_info_Path
from ..cfgs import Radar_info_Path

radar_info = pd.read_json(Radar_info_Path)
2 changes: 1 addition & 1 deletion NuistRadar/core/NRadar.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
"""
import numpy as np
import xarray as xr
from configure.default_config import DEFAULT_METADATA, FILL_VALUE, CINRAD_field_mapping
from ..configure.default_config import DEFAULT_METADATA, FILL_VALUE, CINRAD_field_mapping

class NuistRadar(object):
"""
Expand Down
8 changes: 5 additions & 3 deletions NuistRadar/core/PyartRadar.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,13 +21,15 @@
"""
# the code for Radar Object in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart
from __future__ import print_function

import numpy as np
import sys
from configure.pyart_config import get_metadata
from configure.pyart_lazydict import LazyLoadDict
from libs.core.transforms import antenna_vectors_to_cartesian, cartesian_to_geographic
from ..configure.pyart_config import get_metadata
from ..configure.pyart_lazydict import LazyLoadDict
from .transforms import antenna_vectors_to_cartesian, cartesian_to_geographic


class Radar(object):
Expand Down
2 changes: 1 addition & 1 deletion NuistRadar/core/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from libs.core import NRadar, PyartRadar, transforms
from . import NRadar, PyartRadar, transforms

__all__ = ["NRadar", "PyartRadar", "transforms"]
3 changes: 3 additions & 0 deletions NuistRadar/core/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,9 @@
"""

# the code for transforms in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart

import warnings

import numpy as np
Expand Down
2 changes: 2 additions & 0 deletions NuistRadar/graph/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from . import colormap, RadarPlotModule
__all__ = ["colormap", "RadarPlotModule"]
2 changes: 2 additions & 0 deletions NuistRadar/graph/colormap/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from . import cm, cm_colorblind
__all__ = ["cm", "cm_colorblind"]
2 changes: 2 additions & 0 deletions NuistRadar/graph/colormap/_cm.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
Data for radar related colormaps.
"""
# the data for colormaps in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart

# The data for the colormaps in this file were adapted from those found in
# ctables.py of the Deprecated_Metpy package by Patrick Marsh and Ryan May.
Expand Down
3 changes: 3 additions & 0 deletions NuistRadar/graph/colormap/_cm_colorblind.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,9 @@
Data for colorblind friendly radar colormaps
"""
# the code for colormaps in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart

import numpy as np
import os

Expand Down
5 changes: 4 additions & 1 deletion NuistRadar/graph/colormap/cm.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,8 @@
* LangRainbow12
"""
# the code for colormaps in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart

# This file was adapted from the cm.py file of the matplotlib project,
# http:https://matplotlib.org/.
Expand All @@ -66,6 +68,7 @@
import matplotlib as mpl
import matplotlib.colors as colors
from ._cm import datad
import matplotlib.cm

cmap_d = dict()

Expand Down Expand Up @@ -147,4 +150,4 @@ def _generate_cmap(name, lutsize):
# register the colormaps so that can be accessed with the names pyart_XXX
for name, cmap in cmap_d.items():
full_name = 'pyart_' + name
mpl.cm.register_cmap(name=full_name, cmap=cmap)
matplotlib.cm.register_cmap(name=full_name, cmap=cmap)
2 changes: 2 additions & 0 deletions NuistRadar/graph/colormap/cm_colorblind.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,8 @@
* HomeyerRainbow
"""
# the code for colormaps in this file were adapted from pyart by Helmus, J.J. & Collis, S.M.
# https://github.com/ARM-DOE/pyart

import warnings

Expand Down
2 changes: 2 additions & 0 deletions NuistRadar/io/BaseDataProtocol/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from . import CCProtocol, SABProtocol, SCProtocol, WSR98DProtocol
__all__ = ["CCProtocol", "SABProtocol", "SCProtocol", "WSR98DProtocol"]
21 changes: 6 additions & 15 deletions NuistRadar/io/CCFile.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,13 @@
# -*- coding: utf-8 -*-
import numpy as np
from libs.io.BaseDataProtocol.CCProtocol import dtype_cc
from libs.io.util import _prepare_for_read, _unpack_from_buf, get_radar_info, make_time_unit_str, get_radar_sitename
import time
from .BaseDataProtocol.CCProtocol import dtype_cc
from .util import _prepare_for_read, _unpack_from_buf, get_radar_info, make_time_unit_str, get_radar_sitename
import datetime
import pandas as pd
from libs.core.NRadar import NuistRadar
from configure.pyart_config import get_metadata, get_fillvalue
from configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from libs.core.PyartRadar import Radar
from ..core.NRadar import NuistRadar
from ..configure.pyart_config import get_metadata, get_fillvalue
from ..configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from ..core.PyartRadar import Radar
from netCDF4 import date2num

class CCBaseData(object):
Expand Down Expand Up @@ -406,11 +405,3 @@ def _get_instrument_parameters(self):
nyquist_velocity['data'] = self.get_nyquist_velocity()
instrument_parameters['nyquist_velocity'] = nyquist_velocity
return instrument_parameters


if __name__ == "__main__":
start = time.time()
test = CCBaseData(r"E:\RadarBaseData\CINRAD-CC\2016070818.00V")
CC = CC2NRadar(test)
end = time.time()
print(end - start)
22 changes: 6 additions & 16 deletions NuistRadar/io/SABFile.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,14 @@
# -*- coding: utf-8 -*-
import numpy as np
from scipy import interpolate
from libs.io.BaseDataProtocol.SABProtocol import dtype_sab
from libs.io.util import _prepare_for_read, _unpack_from_buf, julian2date, \
from .BaseDataProtocol.SABProtocol import dtype_sab
from .util import _prepare_for_read, _unpack_from_buf, julian2date, \
get_radar_info, make_time_unit_str, get_radar_sitename
import time
from netCDF4 import date2num
from libs.core.NRadar import NuistRadar
from configure.pyart_config import get_metadata, get_fillvalue
from configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from libs.core.PyartRadar import Radar
from ..core.NRadar import NuistRadar
from ..configure.pyart_config import get_metadata, get_fillvalue
from ..configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from ..core.PyartRadar import Radar


class SABBaseData(object):
Expand Down Expand Up @@ -479,12 +478,3 @@ def _get_instrument_parameters(self):
nyquist_velocity['data'] = self.get_nyquist_velocity()
instrument_parameters['nyquist_velocity'] = nyquist_velocity
return instrument_parameters


if __name__ == "__main__":
start = time.time()
# test = SABBaseData("/home/zy/data/code_data/ERIC/Radar/2010081202.13A")
test = SABBaseData(r"E:\RadarBaseData\CINRAD-SA\z9250\BASE150427\Z_RADR_I_Z9250_20150427111500_O_DOR_SA_CAP.bin")
SAB = SAB2NRadar(test)
end = time.time()
print(end - start)
21 changes: 6 additions & 15 deletions NuistRadar/io/SCFile.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,13 @@
# -*- coding: utf-8 -*-
import numpy as np
from libs.io.BaseDataProtocol.SCProtocol import dtype_sc
from libs.io.util import _prepare_for_read, _unpack_from_buf, get_radar_info, make_time_unit_str, get_radar_sitename
import time
from .BaseDataProtocol.SCProtocol import dtype_sc
from .util import _prepare_for_read, _unpack_from_buf, make_time_unit_str, get_radar_sitename
import pandas as pd
import datetime
from libs.core.NRadar import NuistRadar
from configure.pyart_config import get_metadata, get_fillvalue
from configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from libs.core.PyartRadar import Radar
from ..core.NRadar import NuistRadar
from ..configure.pyart_config import get_metadata, get_fillvalue
from ..configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from ..core.PyartRadar import Radar
from netCDF4 import date2num

class SCBaseData(object):
Expand Down Expand Up @@ -413,11 +412,3 @@ def _get_instrument_parameters(self):
nyquist_velocity['data'] = self.get_nyquist_velocity()
instrument_parameters['nyquist_velocity'] = nyquist_velocity
return instrument_parameters


if __name__ == "__main__":
start = time.time()
test = SCBaseData(r"E:\RadarBaseData\CINRAD_SC_old\Z_RADR_I_Z9280_20180209132700_O_DOR_SC_CAP.bin")
SC = SC2NRadar(test)
end = time.time()
print(end-start)
23 changes: 6 additions & 17 deletions NuistRadar/io/WSR98DFile.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
# -*- coding: utf-8 -*-
import numpy as np
from libs.io.BaseDataProtocol.WSR98DProtocol import dtype_98D
from libs.io.util import _prepare_for_read, _unpack_from_buf, julian2date_SEC, make_time_unit_str
import time
from libs.core.NRadar import NuistRadar
from configure.pyart_config import get_metadata, get_fillvalue
from configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from libs.core.PyartRadar import Radar
from .BaseDataProtocol.WSR98DProtocol import dtype_98D
from .util import _prepare_for_read, _unpack_from_buf, julian2date_SEC, make_time_unit_str
from ..core.NRadar import NuistRadar
from ..configure.pyart_config import get_metadata, get_fillvalue
from ..configure.default_config import CINRAD_field_mapping, _LIGHT_SPEED
from ..core.PyartRadar import Radar
from netCDF4 import date2num

class WSR98DBaseData(object):
Expand Down Expand Up @@ -477,13 +476,3 @@ def _get_instrument_parameters(self):
nyquist_velocity['data'] = self.get_nyquist_velocity()
instrument_parameters['nyquist_velocity'] = nyquist_velocity
return instrument_parameters

if __name__ == "__main__":
start = time.time()
# test = WSR98DBaseData(r"E:\RadarBaseData\郑玉\利奇马\LQM_Z9513_NT\Z_RADR_I_Z9513_20190811234540_O_DOR_SAD_CAP_FMT.bin.bz2")
test = WSR98DBaseData(r"E:\RadarBaseData\StandardFormat\厦门\Z9592.20160728.111443.AR2.bz2")
# test = WSR98DBaseData(r'E:\RadarBaseData\StandardFormat\NUIST\NUIST.20170408.153319.AR2')
wsr98d = WSR98D2NRadar(test)
NRadar = wsr98d.ToNuistRadar()
end = time.time()
print(end - start)
4 changes: 1 addition & 3 deletions NuistRadar/io/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from libs.io import auto_io
from libs.io import CCFile
from libs.io import SCFile, WSR98DFile, SABFile
from . import SCFile, WSR98DFile, SABFile, auto_io, CCFile

__all__ = ["auto_io", "CCFile", "SCFile", "WSR98DFile", "SABFile"]
7 changes: 2 additions & 5 deletions NuistRadar/io/auto_io.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,6 @@
# -*- coding: utf-8 -*-
from libs.io.util import radar_format
from libs.io import CCFile
from libs.io import SABFile
from libs.io import SCFile
from libs.io import WSR98DFile
from .util import radar_format
from . import CCFile, SABFile, SCFile, WSR98DFile

class radar_io(object):
def __init__(self, filename):
Expand Down
2 changes: 1 addition & 1 deletion NuistRadar/io/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import gzip
import datetime
import os
from configure.location_config import radar_info
from ..configure.location_config import radar_info

def _structure_size(structure):
"""计算structure的字节大小"""
Expand Down

0 comments on commit 7efdc6a

Please sign in to comment.