108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
import ternary
|
|
import matplotlib.pyplot as pp
|
|
import numpy as np
|
|
import data_ternary
|
|
import data_udiagram
|
|
import data_zdiagram
|
|
|
|
|
|
def plotTdiagram():
|
|
scale = 100
|
|
|
|
groups = [['grt_E11_a', 'grt_E11_b', 'grt_E11_c'], ['grt_E11_d', 'grt_E11_e', 'grt_E11_g'], ['grt_E11_j', 'grt_E11_k'],
|
|
['grt_Fro_b', 'grt_Fro_c', 'grt_Fro_d'], ['grt_Fro_e', 'grt_Fro_f', 'grt_Fro_g'], ['grt_Fro_n', 'grt_Fro_o', 'grt_Fro_p'], ['grt_Fro_q', 'grt_Fro_r', 'grt_Fro_s'],
|
|
['grt_Rane_a', 'grt_Rane_b'], ['grt_Rane_c', 'grt_Rane_d']]
|
|
|
|
for group in groups:
|
|
#group = [group]
|
|
|
|
figure, tax = ternary.figure(scale=scale)
|
|
|
|
# Draw Boundary and Gridlines
|
|
tax.boundary(linewidth=2.0)
|
|
tax.gridlines(color="grey", multiple=20)
|
|
|
|
# Set Axis labels and Title
|
|
fontsize = 12
|
|
offset = 0.14
|
|
#tax.set_title(group+"\n", fontsize=fontsize)
|
|
tax.right_corner_label("Sps", fontsize=fontsize)
|
|
tax.top_corner_label("Grs", fontsize=fontsize)
|
|
tax.left_corner_label("Prp", fontsize=fontsize)
|
|
#tax.left_axis_label("Left label $\\alpha^2$", fontsize=fontsize, offset=offset)
|
|
#tax.right_axis_label("Right label $\\beta^2$", fontsize=fontsize, offset=offset)
|
|
#tax.bottom_axis_label("Bottom label $\\Gamma - \\Omega$", fontsize=fontsize, offset=offset)
|
|
|
|
# Draw lines parallel to the axes
|
|
#tax.horizontal_line(16)
|
|
#tax.left_parallel_line(10, linewidth=2., color='red', linestyle="--")
|
|
#tax.right_parallel_line(20, linewidth=3., color='blue')
|
|
|
|
# Draw an arbitrary line, ternary will project the points for you
|
|
#p1 = (22, 8, 10)
|
|
#p2 = (2, 22, 16)
|
|
#tax.line(p1, p2, linewidth=3., marker='s', color='green', linestyle=":")
|
|
|
|
for item in group:
|
|
points = list(zip(*(data_ternary.data[item]['Sps'],
|
|
data_ternary.data[item]['Grs'],
|
|
data_ternary.data[item]['Prp'],
|
|
)))
|
|
#print(points)
|
|
tax.scatter(points, label=item)
|
|
|
|
#p1 = (data_ternary.data[item][val][0] for val in ['Prp', 'Sps', 'Grs'])
|
|
p1 = points[0]
|
|
#p2 = (data_ternary.data[item][val][-1] for val in ['Prp', 'Sps', 'Grs'])
|
|
p2 = points[-1]
|
|
print(p1, p2)
|
|
#tax.line(p1, p2, linewidth=3., marker='s', color='green', linestyle="-")
|
|
# line.axes.annotate('',
|
|
# xytext=(xdata[start_ind], ydata[start_ind]),
|
|
# xy=(xdata[end_ind], ydata[end_ind]),
|
|
# arrowprops=dict(arrowstyle="->", color=color),
|
|
# size=size
|
|
# )
|
|
|
|
tax.legend(frameon=False)
|
|
|
|
# edit this for ax ticks
|
|
tax.ticks(axis='lbr', multiple=20, linewidth=1, offset=0.025)
|
|
|
|
# leave this alone
|
|
tax.get_axes().axis('off')
|
|
tax.clear_matplotlib_ticks()
|
|
pp.savefig(f"ternary_{group}.png")
|
|
tax.show()
|
|
|
|
def plotUdiagram():
|
|
|
|
groups = []
|
|
|
|
for g in data_udiagram.data.keys():
|
|
|
|
dat = data_udiagram.data[g]
|
|
|
|
for i, zone in enumerate(dat['zones']):
|
|
|
|
fig, ax = pp.subplots(1, num=g+" "+zone)
|
|
ax.plot(dat['index'][i], dat['al4'][i], label=dat['zones'][i]+"Al(iv)", color="blue")
|
|
ax.plot(dat['index'][i], dat['al6'][i], label=dat['zones'][i]+"Al(vi)", color="orange")
|
|
ax.set_xlabel("No.")
|
|
ax.set_ylabel("value")
|
|
ax.set_title(f"{g}: {zone}")
|
|
pp.savefig(f"uplot_{g}_{zone}.png")
|
|
|
|
pp.show()
|
|
|
|
|
|
def plotZdiagram():
|
|
raise NotImplementedError()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pass
|
|
|
|
#plotTdiagram()
|
|
plotUdiagram()
|