-
Notifications
You must be signed in to change notification settings - Fork 4
/
chart.py
242 lines (215 loc) · 15.9 KB
/
chart.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
# -*- coding: utf-8 -*-
"""
Created on Thu Nov 2 16:46:02 2023
@author: kentang
"""
import drawsvg as draw
import math
import re
#第一層中間, 第二層八門
def gen_chart(first_layer, second_layer, sixth_layer):
# ... [rest of your setup code remains the same]
# Create an SVG drawing canvas
d = draw.Drawing(390, 450, origin="center")
# Set the donut's radii and number of divisions for each layer
inner_radius = 13
layer_gap = 45 # Gap between layers
num_divisions = [1, 8, 16, 16]
# Define the data for each layer
#general = dict(zip(list("貴蛇雀合勾龍空虎常玄陰后"),re.findall('..', '貴人螣蛇朱雀六合勾陳青龍天空白虎常侍玄武太陰太后')))
#skygeneral = [general.get(i) for i in skygeneral]
data = [
[first_layer],
second_layer,
[['巳','大神','楚'], ['午','大威','荊州'], ['未','天道','秦'], ['坤','大武','梁州'], ['申','武德','晉'], ['酉','太簇','趙雍'], ['戌','陰主','魯'], ['乾','陰德','冀州'], ['亥','大義','衛'], ['子','地主','齊兗'], ['丑','陽德','吳'], ['艮','和德','青州'], ['寅','呂申','燕'], ['卯','高叢','徐州'], ['辰','太陽','鄭'], ['巽','大炅','揚州']],
#[['巳','大神'], ['午','大威'], ['未','天道'], ['坤','大武'], ['申','武德'], ['酉','太簇'], ['戌','陰主'], ['乾','陰德'], ['亥','大義'], ['子','地主'], ['丑','陽德'], ['艮','和德'], ['寅','呂申'], ['卯','高叢'], ['辰','太陽'], ['巽','大炅']],
#['楚', '荊州', '秦', '梁州', '晉', '趙雍', '魯', '冀州', '衛', '齊兗', '吳', '青州', '燕', '徐州', '鄭', '揚州'],
sixth_layer,
]
rotation_angle = 248
for layer_index, divisions in enumerate(num_divisions):
for division in range(divisions):
start_angle = (360 / divisions) * division + rotation_angle
end_angle = (360 / divisions) * (division + 1) + rotation_angle
label = data[layer_index][division]
inner = inner_radius + layer_index * layer_gap
outer = inner_radius + (layer_index + 1) * layer_gap
# Calculate start and end points for both inner and outer arcs
start_outer_x, start_outer_y = outer * math.cos(math.radians(start_angle)), outer * math.sin(math.radians(start_angle))
end_outer_x, end_outer_y = outer * math.cos(math.radians(end_angle)), outer * math.sin(math.radians(end_angle))
start_inner_x, start_inner_y = inner * math.cos(math.radians(start_angle)), inner * math.sin(math.radians(start_angle))
end_inner_x, end_inner_y = inner * math.cos(math.radians(end_angle)), inner * math.sin(math.radians(end_angle))
path = draw.Path(stroke='white', stroke_width=1.8, fill='black')
path.M(start_inner_x, start_inner_y) # Move to the start point on the inner radius
path.L(start_outer_x, start_outer_y) # Line to the start point on the outer radius
path.A(outer, outer, 0, 0, 1, end_outer_x, end_outer_y) # Outer arc
path.L(end_inner_x, end_inner_y) # Line to the end point on the inner radius
path.A(inner, inner, 0, 0, 0, start_inner_x, start_inner_y) # Inner arc
path.Z() # Close the path
d.append(path)
# Add labels to the pie slices
label_x = (inner + outer) / 2 * math.cos(math.radians((start_angle + end_angle) / 2))
label_y = (inner + outer) / 2 * math.sin(math.radians((start_angle + end_angle) / 2))
#if divisions == 1:
# label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='black')
#else:
label_text = draw.Text(label, 9, label_x, label_y, center=1, fill='white')
d.append(label_text)
# ... [rest of your code remains the same]
return d.as_svg().replace('''<path d="M-4.86988571440686,-12.053390109368236 L-21.72718241812291,-53.776663564873665 A58,58,0,0,1,-21.727182418122876,-53.77666356487368 L-4.869885714406852,-12.053390109368237 A13,13,0,0,0,-4.86988571440686,-12.053390109368236 Z" stroke="white" stroke-width="1.8" fill="black" />''', "")
#太乙命法盤
def gen_chart_life(second_layer, twelve, sixth_layer):
# ... [rest of your setup code remains the same]
# Create an SVG drawing canvas
d = draw.Drawing(390, 450, origin="center")
# Set the donut's radii and number of divisions for each layer
inner_radius = 12
layer_gap = 35 # Gap between layers
num_divisions = [1, 12, 12, 12]
# Define the data for each layer
#general = dict(zip(list("貴蛇雀合勾龍空虎常玄陰后"),re.findall('..', '貴人螣蛇朱雀六合勾陳青龍天空白虎常侍玄武太陰太后')))
#skygeneral = [general.get(i) for i in skygeneral]
data = [
[second_layer],
twelve,
['巳', '午', '未', '申', '酉', '戌', '亥', '子', '丑', '寅','卯', '辰'],
#[['巳','大神'], ['午','大威'], ['未','天道'], ['坤','大武'], ['申','武德'], ['酉','太簇'], ['戌','陰主'], ['乾','陰德'], ['亥','大義'], ['子','地主'], ['丑','陽德'], ['艮','和德'], ['寅','呂申'], ['卯','高叢'], ['辰','太陽'], ['巽','大炅']],
#['楚', '荊州', '秦', '梁州', '晉', '趙雍', '魯', '冀州', '衛', '齊兗', '吳', '青州', '燕', '徐州', '鄭', '揚州'],
sixth_layer
]
rotation_angle = 248
for layer_index, divisions in enumerate(num_divisions):
for division in range(divisions):
start_angle = (360 / divisions) * division + rotation_angle
end_angle = (360 / divisions) * (division + 1) + rotation_angle
label = data[layer_index][division]
inner = inner_radius + layer_index * layer_gap
outer = inner_radius + (layer_index + 1) * layer_gap
# Calculate start and end points for both inner and outer arcs
start_outer_x, start_outer_y = outer * math.cos(math.radians(start_angle)), outer * math.sin(math.radians(start_angle))
end_outer_x, end_outer_y = outer * math.cos(math.radians(end_angle)), outer * math.sin(math.radians(end_angle))
start_inner_x, start_inner_y = inner * math.cos(math.radians(start_angle)), inner * math.sin(math.radians(start_angle))
end_inner_x, end_inner_y = inner * math.cos(math.radians(end_angle)), inner * math.sin(math.radians(end_angle))
path = draw.Path(stroke='white', stroke_width=1.88, fill='black')
path.M(start_inner_x, start_inner_y) # Move to the start point on the inner radius
path.L(start_outer_x, start_outer_y) # Line to the start point on the outer radius
path.A(outer, outer, 0, 0, 1, end_outer_x, end_outer_y) # Outer arc
path.L(end_inner_x, end_inner_y) # Line to the end point on the inner radius
path.A(inner, inner, 0, 0, 0, start_inner_x, start_inner_y) # Inner arc
path.Z() # Close the path
d.append(path)
# Add labels to the pie slices
label_x = (inner + outer) / 2 * math.cos(math.radians((start_angle + end_angle) / 2))
label_y = (inner + outer) / 2 * math.sin(math.radians((start_angle + end_angle) / 2))
#if divisions == 1:
# label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='black')
#else:
label_text = draw.Text(label, 9, label_x, label_y, center=1, fill='white')
d.append(label_text)
# ... [rest of your code remains the same]
return d.as_svg().replace('''<path d="M-4.495279120990947,-11.126206254801447 L-17.606509890547876,-43.577641164639005 A47,47,0,0,1,-17.606509890547848,-43.57764116463901 L-4.49527912099094,-11.12620625480145 A12,12,0,0,0,-4.495279120990947,-11.126206254801447 Z" stroke="white" stroke-width="1.88" fill="black" />''', "")
#日家太乙盤
#第一層中間, 第二層八門
def gen_chart_day(first_layer, second_layer, golden, sixth_layer):
# ... [rest of your setup code remains the same]
# Create an SVG drawing canvas
d = draw.Drawing(390, 450, origin="center")
# Set the donut's radii and number of divisions for each layer
inner_radius = 3
layer_gap = 31.5 # Gap between layers
num_divisions = [1, 8, 8, 16, 16]
# Define the data for each layer
#general = dict(zip(list("貴蛇雀合勾龍空虎常玄陰后"),re.findall('..', '貴人螣蛇朱雀六合勾陳青龍天空白虎常侍玄武太陰太后')))
#skygeneral = [general.get(i) for i in skygeneral]
data = [
[first_layer],
second_layer,
golden,
[['巳','大神','楚'], ['午','大威','荊州'], ['未','天道','秦'], ['坤','大武','梁州'], ['申','武德','晉'], ['酉','太簇','趙雍'], ['戌','陰主','魯'], ['乾','陰德','冀州'], ['亥','大義','衛'], ['子','地主','齊兗'], ['丑','陽德','吳'], ['艮','和德','青州'], ['寅','呂申','燕'], ['卯','高叢','徐州'], ['辰','太陽','鄭'], ['巽','大炅','揚州']],
#[['巳','大神'], ['午','大威'], ['未','天道'], ['坤','大武'], ['申','武德'], ['酉','太簇'], ['戌','陰主'], ['乾','陰德'], ['亥','大義'], ['子','地主'], ['丑','陽德'], ['艮','和德'], ['寅','呂申'], ['卯','高叢'], ['辰','太陽'], ['巽','大炅']],
#['楚', '荊州', '秦', '梁州', '晉', '趙雍', '魯', '冀州', '衛', '齊兗', '吳', '青州', '燕', '徐州', '鄭', '揚州'],
sixth_layer,
]
rotation_angle = 248
for layer_index, divisions in enumerate(num_divisions):
for division in range(divisions):
start_angle = (360 / divisions) * division + rotation_angle
end_angle = (360 / divisions) * (division + 1) + rotation_angle
label = data[layer_index][division]
inner = inner_radius + layer_index * layer_gap
outer = inner_radius + (layer_index + 1) * layer_gap
# Calculate start and end points for both inner and outer arcs
start_outer_x, start_outer_y = outer * math.cos(math.radians(start_angle)), outer * math.sin(math.radians(start_angle))
end_outer_x, end_outer_y = outer * math.cos(math.radians(end_angle)), outer * math.sin(math.radians(end_angle))
start_inner_x, start_inner_y = inner * math.cos(math.radians(start_angle)), inner * math.sin(math.radians(start_angle))
end_inner_x, end_inner_y = inner * math.cos(math.radians(end_angle)), inner * math.sin(math.radians(end_angle))
path = draw.Path(stroke='white', stroke_width=1.8, fill='black')
path.M(start_inner_x, start_inner_y) # Move to the start point on the inner radius
path.L(start_outer_x, start_outer_y) # Line to the start point on the outer radius
path.A(outer, outer, 0, 0, 1, end_outer_x, end_outer_y) # Outer arc
path.L(end_inner_x, end_inner_y) # Line to the end point on the inner radius
path.A(inner, inner, 0, 0, 0, start_inner_x, start_inner_y) # Inner arc
path.Z() # Close the path
d.append(path)
# Add labels to the pie slices
label_x = (inner + outer) / 2 * math.cos(math.radians((start_angle + end_angle) / 2))
label_y = (inner + outer) / 2 * math.sin(math.radians((start_angle + end_angle) / 2))
#if divisions == 1:
# label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='black')
#else:
label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='white')
d.append(label_text)
return d.as_svg().replace('''<path d="M-1.1238197802477368,-2.781551563700362 L-12.923927472848973,-31.987842982554163 A34.5,34.5,0,0,1,-12.923927472848954,-31.98784298255417 L-1.123819780247735,-2.7815515637003627 A3.0,3.0,0,0,0,-1.1238197802477368,-2.781551563700362 Z" stroke="white" stroke-width="1.8" fill="black" />''', "")
#第一層中間, 第二層八門
def gen_chart_hour(first_layer, second_layer, skygeneral, sixth_layer, twentyeight):
# ... [rest of your setup code remains the same]
# Create an SVG drawing canvas
d = draw.Drawing(390, 450, origin="center")
# Set the donut's radii and number of divisions for each layer
inner_radius = 3
layer_gap = 31.5 # Gap between layers
num_divisions = [1, 8, 16, 16, 16,28]
# Define the data for each layer
#general = dict(zip(list("貴蛇雀合勾龍空虎常玄陰后"),re.findall('..', '貴人螣蛇朱雀六合勾陳青龍天空白虎常侍玄武太陰太后')))
#skygeneral = [general.get(i) for i in skygeneral]
data = [
[first_layer],
second_layer,
skygeneral,
[['巳','大神','楚'], ['午','大威','荊州'], ['未','天道','秦'], ['坤','大武','梁州'], ['申','武德','晉'], ['酉','太簇','趙雍'], ['戌','陰主','魯'], ['乾','陰德','冀州'], ['亥','大義','衛'], ['子','地主','齊兗'], ['丑','陽德','吳'], ['艮','和德','青州'], ['寅','呂申','燕'], ['卯','高叢','徐州'], ['辰','太陽','鄭'], ['巽','大炅','揚州']],
#[['巳','大神'], ['午','大威'], ['未','天道'], ['坤','大武'], ['申','武德'], ['酉','太簇'], ['戌','陰主'], ['乾','陰德'], ['亥','大義'], ['子','地主'], ['丑','陽德'], ['艮','和德'], ['寅','呂申'], ['卯','高叢'], ['辰','太陽'], ['巽','大炅']],
#['楚', '荊州', '秦', '梁州', '晉', '趙雍', '魯', '冀州', '衛', '齊兗', '吳', '青州', '燕', '徐州', '鄭', '揚州'],
sixth_layer,
twentyeight
]
rotation_angle = 248
for layer_index, divisions in enumerate(num_divisions):
for division in range(divisions):
start_angle = (360 / divisions) * division + rotation_angle
end_angle = (360 / divisions) * (division + 1) + rotation_angle
label = data[layer_index][division]
inner = inner_radius + layer_index * layer_gap
outer = inner_radius + (layer_index + 1) * layer_gap
# Calculate start and end points for both inner and outer arcs
start_outer_x, start_outer_y = outer * math.cos(math.radians(start_angle)), outer * math.sin(math.radians(start_angle))
end_outer_x, end_outer_y = outer * math.cos(math.radians(end_angle)), outer * math.sin(math.radians(end_angle))
start_inner_x, start_inner_y = inner * math.cos(math.radians(start_angle)), inner * math.sin(math.radians(start_angle))
end_inner_x, end_inner_y = inner * math.cos(math.radians(end_angle)), inner * math.sin(math.radians(end_angle))
path = draw.Path(stroke='white', stroke_width=1.8, fill='black')
path.M(start_inner_x, start_inner_y) # Move to the start point on the inner radius
path.L(start_outer_x, start_outer_y) # Line to the start point on the outer radius
path.A(outer, outer, 0, 0, 1, end_outer_x, end_outer_y) # Outer arc
path.L(end_inner_x, end_inner_y) # Line to the end point on the inner radius
path.A(inner, inner, 0, 0, 0, start_inner_x, start_inner_y) # Inner arc
path.Z() # Close the path
d.append(path)
# Add labels to the pie slices
label_x = (inner + outer) / 2 * math.cos(math.radians((start_angle + end_angle) / 2))
label_y = (inner + outer) / 2 * math.sin(math.radians((start_angle + end_angle) / 2))
#if divisions == 1:
# label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='black')
#else:
label_text = draw.Text(label, 8, label_x, label_y, center=1, fill='white')
d.append(label_text)
# ... [rest of your code remains the same]
return d.as_svg().replace('''<path d="M-1.1238197802477368,-2.781551563700362 L-12.923927472848973,-31.987842982554163 A34.5,34.5,0,0,1,-12.923927472848954,-31.98784298255417 L-1.123819780247735,-2.7815515637003627 A3.0,3.0,0,0,0,-1.1238197802477368,-2.781551563700362 Z" stroke="white" stroke-width="1.8" fill="black" />''', "")