代码拉取完成,页面将自动刷新
同步操作将从 刘胜胜/气象数据可视化,用matplotlib和basemap绘制micaps数据 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
# -*- coding: utf-8 -*-
# Micaps第11类数据 继承Micaps基类
# Author: Liu xianyao
# Email: flashlxy@qq.com
# Update: 2017-04-11
# Copyright: ©江西省气象台 2017
# Version: 2.0.20170411
import codecs
import math
import re
import nclcmaps
from datetime import datetime
import matplotlib.pyplot as plt
import numpy as np
from MicapsData import Micaps
class Micaps11Data(Micaps):
def __init__(self, filename):
self.filename = filename
self.U = None
self.V = None
self.linewidth = 1
self.color = 'k'
self.density = [1, 1]
self.barbsgrid = [31, 31]
self.cmap = None
self.stream = None
self.barbs = None
self.length = 5
self.scale = 700
self.colorlist = ['k', 'b', 'r', 'g']
self.wholeclip = False
self.ReadFromFile()
def ReadFromFile(self):
"""
读micaps第11类数据文件到内存
:return:
"""
try:
begin = 17
file_object = codecs.open(self.filename, mode='r', encoding='GBK')
all_the_text = file_object.read().strip()
file_object.close()
contents = re.split('[\s]+', all_the_text)
if len(contents) < begin:
return
self.dataflag = contents[0].strip()
self.style = contents[1].strip()
self.title = contents[2].strip()
self.yy = int(contents[3].strip())
self.mm = int(contents[4].strip())
self.dd = int(contents[5].strip())
self.hh = int(contents[6].strip())
self.forehh = int(contents[7].strip())
self.level = contents[8].strip()
self.deltalon = float(contents[9].strip())
self.deltalat = float(contents[10].strip())
self.beginlon = float(contents[11].strip())
self.endlon = float(contents[12].strip())
self.beginlat = float(contents[13].strip())
self.endlat = float(contents[14].strip())
self.sumlon = int(contents[15].strip())
self.sumlat = int(contents[16].strip())
self.x = np.arange(self.beginlon, self.endlon + self.deltalon, self.deltalon)
self.y = np.arange(self.beginlat, self.endlat + self.deltalat, self.deltalat)
self.X, self.Y = np.meshgrid(self.x, self.y)
if self.dataflag == 'diamond' and self.style == '11':
self.U = np.zeros((self.sumlat, self.sumlon))
self.V = np.zeros((self.sumlat, self.sumlon))
self.Z = np.zeros((self.sumlat, self.sumlon))
for i in range(self.sumlon):
for j in range(self.sumlat):
self.U[j, i] = float(contents[begin + j * self.sumlon + i])
vbegin = begin + self.sumlat * self.sumlon
for i in range(self.sumlon):
for j in range(self.sumlat):
self.V[j, i] = float(contents[vbegin + j * self.sumlon + i])
for i in range(self.sumlon):
for j in range(self.sumlat):
self.Z[j, i] = math.sqrt(self.U[j, i] ** 2 + self.V[j, i] ** 2)
if self.deltalat < 0:
self.TransposeYaxis()
except Exception as err:
print(u'【{0}】{1}-{2}'.format(self.filename, err, datetime.now()))
def TransposeYaxis(self):
self.beginlat, self.endlat = self.endlat, self.beginlat
self.deltalat = math.fabs(self.deltalat)
self.y = np.arange(self.beginlat, self.endlat + self.deltalat, self.deltalat)
self.X, self.Y = np.meshgrid(self.x, self.y)
self.U = self.U[::-1, ::]
self.V = self.V[::-1, ::]
self.Z = self.Z[::-1, ::]
def UpdateData(self, products, micapsfile):
self.UpdateExtents(products)
# micapsfile = products.micapsfiles[0]
self.min = self.Z.min()
self.max = self.Z.max()
self.distance = micapsfile.contour.step
self.min = math.floor(self.min / self.distance) * self.distance
self.max = math.ceil(self.max / self.distance) * self.distance
# 如果自定义了legend的最小、最大和步长值 则用自定义的值更新
self.UpdatePinLegendValue(micapsfile)
from Main import equal
if micapsfile.uv.onspeed and not equal(self.Z.max(), 0):
self.linewidth = 5 * self.Z / self.Z.max()
else:
self.linewidth = micapsfile.uv.linewidth
self.density = micapsfile.uv.density
self.barbsgrid = micapsfile.uv.barbsgrid
if micapsfile.uv.oncolor:
self.color = self.Z
self.cmap = nclcmaps.cmaps(micapsfile.legend.micapslegendcolor)
else:
self.color = micapsfile.uv.color
self.barbs = micapsfile.uv.barbs
self.stream = micapsfile.uv.stream
self.length = micapsfile.uv.length
self.scale = micapsfile.uv.scale
self.wholeclip = micapsfile.uv.wholecilp
self.colorlist = micapsfile.legend.legendcolor
def GetPatches(self, paths):
ps = []
for path in paths:
from matplotlib import patches
ps.append(patches.PathPatch(path, linewidth=1, facecolor='none', edgecolor='k'))
return ps
def ConvertPacth(self, ax, patch):
path = patch.get_path()
lon = []
lat = []
for points in path.vertices:
x, y = points[0], points[1]
xy_pixels = ax.transData.transform(np.vstack([x, y]).T)
xpix, ypix = xy_pixels.T
lon.append(xpix[0])
lat.append(ypix[0])
from matplotlib.path import Path
apath = Path(zip(lon, lat))
from matplotlib import patches
apatch = patches.PathPatch(apath,
linewidth=1,
facecolor='none',
edgecolor='k')
plt.gca().add_patch(apatch)
return apatch
def DrawUV(self, m, micapsfile, clipborder, patch):
if m is plt:
if self.stream:
plot = m.streamplot(self.X, self.Y, self.U, self.V,
density=self.density,
linewidth=self.linewidth,
color=self.color,
cmap=self.cmap
)
if self.barbs:
barbs = m.barbs(self.X, self.Y, self.U, self.V,
length=self.length, barb_increments=dict(half=2, full=4, flag=20),
sizes=dict(emptybarb=0))
pass
else:
# transform vectors to projection grid.
uproj, vproj, xx, yy = \
m.transform_vector(self.U, self.V, self.x, self.y, self.barbsgrid[0], self.barbsgrid[1],
returnxy=True, masked=True)
if isinstance(self.color, np.ndarray):
self.color = self.colorlist
if self.stream:
# plot = m.streamplot(xx, yy, uproj, vproj # , latlon=True
# # density=self.density,
# # linewidth=self.linewidth,
# # color=self.color
# # cmap=self.cmap
# )
# now plot.
Q = m.quiver(xx, yy, uproj, vproj, color=self.color, scale=self.scale)
# make quiver key.
speed = micapsfile.uv.markscalelength
qk = plt.quiverkey(Q, 0.1, 0.1, speed, '%.0f m/s' % speed, labelpos='W')
if self.barbs:
barbs = m.barbs(xx, yy, uproj, vproj, length=self.length,
barb_increments=dict(half=2, full=4, flag=20),
sizes=dict(emptybarb=0),
barbcolor='k', flagcolor='r',
linewidth=0.5)
if clipborder.path is not None and clipborder.using:
from matplotlib.patches import FancyArrowPatch
from matplotlib.collections import PolyCollection
from matplotlib.lines import Line2D
for artist in plt.gca().get_children():
if self.wholeclip:
# from matplotlib.patches import Polygon
if not isinstance(artist, Line2D):
artist.set_clip_path(patch)
else:
from matplotlib.collections import LineCollection
if isinstance(artist, FancyArrowPatch) or isinstance(artist, PolyCollection) or \
isinstance(artist, LineCollection):
artist.set_clip_path(patch)
# if m is plt and self.stream:
# for ax in fig.axes:
# from matplotlib.patches import FancyArrowPatch
# artists = ax.get_children()
# for artist in artists:
# if isinstance(artist, FancyArrowPatch):
# artist.set_clip_path(patch)
# plot.lines.set_clip_path(patch)
#
# if self.barbs or (m is not plt and self.stream):
# for ax in fig.axes:
# artists = ax.get_children()
# for artist in artists:
# from matplotlib.collections import PolyCollection
# if isinstance(artist, PolyCollection):
# artist.set_clip_path(patch)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。