import os
import matplotlib.pyplot as plt
import hyperspy.api as hs

path = "b009_diffraction_plots"
os.makedirs(path, exist_ok=True)

##########################################
filename = "b006_011_A_LFO_diff.hspy"
s_diff = hs.load(filename)
s_diff.change_dtype('float64')
s_diff.data /= s_diff.data.max()
fig, ax = plt.subplots(figsize=(10, 10))
ax.imshow(s_diff.data, vmin=0.0, vmax=0.1)
fig.subplots_adjust(0, 0, 1, 1)
filename_output = os.path.join(path, filename.replace(".hspy", ".jpg"))
fig.savefig(filename_output)

##########################################
filename = "b006_011_B_LFO_diff.hspy"
s_diff = hs.load(filename)
s_diff.change_dtype('float64')
s_diff.data /= s_diff.data.max()
fig, ax = plt.subplots(figsize=(10, 10))
ax.imshow(s_diff.data, vmin=0.0, vmax=0.1)
fig.subplots_adjust(0, 0, 1, 1)
filename_output = os.path.join(path, filename.replace(".hspy", ".jpg"))
fig.savefig(filename_output)

##########################################
filename = "b006_011_O_LFO_diff.hspy"
s_diff = hs.load(filename)
s_diff.change_dtype('float64')
s_diff.data /= s_diff.data.max()
fig, ax = plt.subplots(figsize=(10, 10))
ax.imshow(s_diff.data, vmin=0.0, vmax=0.1)
fig.subplots_adjust(0, 0, 1, 1)
filename_output = os.path.join(path, filename.replace(".hspy", ".jpg"))
fig.savefig(filename_output)

##########################################
filename = "b006_011_A_STO_diff.hspy"
s_diff = hs.load(filename)
s_diff.change_dtype('float64')
s_diff.data /= s_diff.data.max()
fig, ax = plt.subplots(figsize=(10, 10))
ax.imshow(s_diff.data, vmin=0.0, vmax=0.1)
fig.subplots_adjust(0, 0, 1, 1)
filename_output = os.path.join(path, filename.replace(".hspy", ".jpg"))
fig.savefig(filename_output)

##########################################
filename = "b006_011_A_LMO_diff.hspy"
s_diff = hs.load(filename)
s_diff.change_dtype('float64')
s_diff.data /= s_diff.data.max()
fig, ax = plt.subplots(figsize=(10, 10))
ax.imshow(s_diff.data, vmin=0.0, vmax=0.1)
fig.subplots_adjust(0, 0, 1, 1)
filename_output = os.path.join(path, filename.replace(".hspy", ".jpg"))
fig.savefig(filename_output)