moose_data_generation/plot_res.py at main · conlain-k/moose_data_generation · GitHub

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
import numpy as np
import matplotlib.pyplot as plt
from helpers.plot_cube import plot_cube

from h5py import File

PLOT_SLICE = 10

E_BAR = 0.001

FILENAME_MOOSE = "poly_strain_out_0001.csv"
FILENAME_MOOSE = "outputs/2phase_strain_out_0001.csv"

# exx_ab = np.load("crystal_pred.npy")[0, 0] / E_BAR


ab_file = File("00000.h5")

print(ab_file.keys())
exx_ab = ab_file["strain"][0, 0] / E_BAR

eps_file = np.genfromtxt(
    FILENAME_MOOSE,
    names=True,
    delimiter=",",
)

# sig_file = np.genfromtxt(
#     "outputs/2phase_stress_out_0001.csv",
#     names=True,
#     delimiter=",",
# )


# sigxx = sig_file["stress_xx"][:]
# sigxx = sigxx.reshape(exx_ab.shape, order="F")


exx = eps_file["strain_xx"][:] / E_BAR
exx = exx.reshape(exx_ab.shape, order="F")

m = np.load("structure.npy")


# print(sigxx.shape, exx.shape, exx_ab.shape)

# exx = exx.transpose(-3, -2, -1)


def rr(field):
    field = np.roll(field, -25, 0)
    field = np.roll(field, -33, 1)
    field = np.roll(field, -17, 2)

    return field


m = rr(m)
exx = rr(exx)
# sigxx = rr(sigxx)
exx_ab = rr(exx_ab)


plot_cube(m, title="m", savedir="m.png", add_cb=False, cmap="viridis")

plot_cube(exx, title="exx", savedir="exx.png")

# plot_cube(sigxx, title="sigxx", savedir="sigxx.png")

plot_cube(exx_ab, title="exx_ab", savedir="exx_ab.png")

plot_cube(exx_ab - exx, title="diff", savedir="diff.png")


plt.show()