Documentation

Author: Julius Herb

eg8_plot_localization.py

@@ -1,12 +1,13 @@
+"""
+Plot the strain localization operator E and stress localization operator S at different temperatures
+"""
 #%%
 from operator import itemgetter
 
 import numpy.linalg as la
 import matplotlib.pyplot as plt
-from interpolate_fluctuation_modes import interpolate_fluctuation_modes
 from microstructures import *
-from optimize_alpha import opt1, opt2, opt4, naive
-from utilities import read_h5, construct_stress_localization, volume_average, compute_residual_efficient
+from utilities import read_h5, construct_stress_localization
 
 np.random.seed(0)
 file_name, data_path, temp1, temp2, n_tests, sampling_alphas = itemgetter('file_name', 'data_path', 'temp1', 'temp2', 'n_tests',
@@ -24,12 +25,14 @@
 n_gp = mesh['n_integration_points']
 n_modes = ref[0]['strain_localization'].shape[-1]
 
+# Lowest temperature
 temp0 = ref[0]['temperature']
 E0 = ref[0]['strain_localization']
 C0 = ref[0]['mat_stiffness']
 eps0 = ref[0]['mat_thermal_strain']
 S0 = construct_stress_localization(E0, C0, eps0, mat_id, n_gauss, strain_dof)
 
+# First enrichment temperature
 alpha = sampling_alphas[1][1]
 a = int(alpha * n_tests)
 tempa = ref[a]['temperature']
@@ -38,6 +41,7 @@
 epsa = ref[a]['mat_thermal_strain']
 Sa = construct_stress_localization(Ea, Ca, epsa, mat_id, n_gauss, strain_dof)
 
+# Highest temperature
 temp1 = ref[-1]['temperature']
 E1 = ref[-1]['strain_localization']
 C1 = ref[-1]['mat_stiffness']
@@ -47,19 +51,29 @@
 # %%
 
 
-def plot_localization(ax, mesh, E, i=0, j=0):
+def plot_localization(ax, mesh, E, i=0):
+    """Plots the euclidean norm of the `i`-th column of
+    a localization operator `E` on the y-z-cross section at x=0.
+
+    Args:
+        ax: matplotlib axis
+        mesh: dictionary with mesh informationen
+        E (np.ndarray): localization operator with shape (nx*ny*nz*ngauss, 6, 7)
+        i (int, optional): column no. of E to be plotted. Defaults to 0.
+    """
     discr = mesh['combo_discretisation']
     n_gauss = mesh['n_gauss']
     assert E.ndim == 3
     assert E.shape[0] == n_gauss * np.prod(discr)
     assert E.shape[1] == 6
     assert E.shape[2] == 7
-    E_r = E.reshape(*discr, n_gauss, 6, 7)  # resize
+    E_r = E.reshape(*discr, n_gauss, 6, 7)  # reshape
     E_ra = np.mean(E_r, axis=3)  # average over gauss points
-    E_rai = np.linalg.norm(E_ra[:, :, :, i, :], axis=-1)  # compute norm of i-th row
+    E_rai = la.norm(E_ra[:, :, :, i, :], axis=-1)  # compute norm of i-th column
     ax.imshow(E_rai[0, :, :], interpolation='spline16')  # plot y-z-cross section at x=0
 
 
+# Plot strain localization operator E at different temperatures
 fig, ax = plt.subplots(1, 3)
 plot_localization(ax[0], mesh, E0, i=0)
 ax[0].set_title(r'$\underline{\underline{E}}\;\mathrm{at}\;\theta=' +
@@ -71,9 +85,10 @@ def plot_localization(ax, mesh, E, i=0, j=0):
 plot_localization(ax[2], mesh, E1, i=0)
 ax[2].set_title(r'$\underline{\underline{E}}\;\mathrm{at}\;\theta=' +
                 f'{temp1:.2f}' + r'\mathrm{K}$')
-plt.savefig('E.pgf', dpi=300)
+plt.savefig('output/E.pgf', dpi=300)
 plt.show()
 
+# Plot strain localization operator S at different temperatures
 fig, ax = plt.subplots(1, 3)
 plot_localization(ax[0], mesh, S0, i=0)
 ax[0].set_title(r'$\underline{\underline{S}}\;\mathrm{at}\;\theta=' +
@@ -85,5 +100,5 @@ def plot_localization(ax, mesh, E, i=0, j=0):
 plot_localization(ax[2], mesh, S1, i=0)
 ax[2].set_title(r'$\underline{\underline{S}}\;\mathrm{at}\;\theta=' +
                 f'{temp1:.2f}' + r'\mathrm{K}$')
-plt.savefig('S.pgf', dpi=300)
+plt.savefig('output/S.pgf', dpi=300)
 plt.show()