forked from flashlxy/PyMICAPS
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Legend.py
123 lines (100 loc) · 4.63 KB
/
Legend.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# -*- coding: utf-8 -*-
#
# Author: Liu xianyao
# Email: [email protected]
# Update: 2017-04-11
# Copyright: ©江西省气象台 2017
# Version: 2.0.20170411
from __future__ import print_function
from Projection import Projection
class Legend:
def __init__(self, root):
p = root.find("Legend")
if p is None:
print("没有图例节点")
return
# 图例图片文件路径，图例叠加到产品图上的位置，为空表示自行生成图例
legend_pic = p.find("LegendPic").text
self.islegendpic = False
self.legendpic = ""
self.legendpos = [0, 0]
self.legendopacity = 1
if not (legend_pic is None or legend_pic.strip() == ""):
self.islegendpic = True
legend_pics = legend_pic.strip().split(",")
if 1 == len(legend_pics):
self.legendpic = legend_pics[0].strip()
elif 4 == len(legend_pics):
self.legendpic = legend_pics[0].strip()
self.legendpos = [
float(legend_pics[1].strip()),
float(legend_pics[2].strip()),
]
self.legendopacity = float(legend_pics[3].strip())
else:
self.legendpic = legend_pics[0].strip()
# 图例的延展类型:neither,min,max default is neither
self.extend = Projection.leaf_to_string(p, "Extend", "neither")
# 是否取MICAPS数据本身的图例值
self.micapslegendvalue = Projection.leaf_to_bool(
p, "MicapsLegendValue", True, "TRUE"
)
# When micapslegendvalue is true, the setting is working.
# if pin the legend values, [begin value, stop value, step] is working.default is None
# else use the data self legend values
self.pinlegendvalue = Projection.leaf_to_list(p, "PinLegendValue", None)
self.valid(self.pinlegendvalue)
# NCL colorbar 的别名
self.micapslegendcolor = Projection.leaf_to_string(
p, "MicapsLegendColor", "ncl_default"
)
# 图例填充样式数组，一般和自定义的legend结合使用，如不够LegendValue数组的长度，则会用最后一个样式自动补齐
self.hatches = Projection.leaf_to_list(p, "Hatches", [""])
self.validhatches(self.hatches)
# 图例等级值
self.legendvalue = Projection.leaf_to_list(p, "LegendValue", None)
# 实际显示在图例上的值
self.legendvaluealias = Projection.leaf_to_list(p, "LegendValueAlias", None)
# 图例颜色值
self.legendcolor = Projection.leaf_to_list(p, "LegendColor", None)
# 图例抽稀间隔
self.thinning = Projection.leaf_to_int(p, "Thinning", 1)
# 图例字体
self.font = Projection.leaf_to_dict(p, "Font", None)
# 图例标题
self.title = Projection.leaf_to_string(p, "Title", "")
# 图例标题字体
self.titlefont = Projection.leaf_to_dict(p, "TitleFont", None)
# 图例标题位置
self.titlepos = Projection.leaf_to_dict(
p,
"TitlePos",
{"rotation": 90, "va": "top", "ha": "center", "ypercent": 0.5},
)
# ---------- 无投影时的图例配置 start --------
# 图例放置方式
self.orientation = Projection.leaf_to_string(p, "Orientation", "vertical")
# 图例离边框位置
self.anchor = Projection.leaf_to_list(p, "Anchor", [0, 0])
# 图例收缩系数
self.shrink = Projection.leaf_to_float(p, "Shrink", 1)
self.fraction = Projection.leaf_to_float(p, "Fraction", 0.15)
# ---------- 有投影时的图例配置 start --------
# 图例收缩系数
self.size = Projection.leaf_to_string(leaf=p, code="Size", defvalue="5%")
# 图例离边框位置
self.pad = Projection.leaf_to_string(leaf=p, code="Pad", defvalue="2%")
# 图例放置位置
self.location = Projection.leaf_to_string(
leaf=p, code="Location", defvalue="right"
)
def valid(self, pin):
if pin is not None and len(pin) == 3:
if pin[1] < pin[0] or pin[2] < 0:
self.pinlegendvalue = None
else:
self.pinlegendvalue = None
def validhatches(self, hatches):
hatches_list = ["", "/", "\\", "|", "-", "+", "x", "o", "O", ".", "*"]
if hatches is not None:
self.hatches = [a if a in hatches_list else "" for a in hatches]