import numpy as np
import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator

# Define the x range
x = np.linspace(-2, 2, 1000)

# Define the wave packet: high-frequency sine modulated by a Gaussian envelope
y = np.sin(50 * x) * np.exp(-x**2)



#Plotting information 
# Create 3 subplots (3 rows, 1 column)
fig, ax = plt.subplots(figsize=(9, 7))  # Adjust figure size if needed

# Customize tick parameters for bold text
ax.tick_params(axis='both', direction='in', which='both')
ax.tick_params(which='both', width=2)
ax.tick_params(which='major', length=7, color='r')
ax.tick_params(which='minor', length=4)
ax.minorticks_on()  # Turn on minor ticks

# Set bold tick labels in Times New Roman
ax.tick_params(axis='x', labelsize=12)
ax.tick_params(axis='y', labelsize=12)
for label in ax.get_xticklabels() + ax.get_yticklabels():
    label.set_fontweight('bold')
    label.set_fontname("Times New Roman")
    label.set_fontsize(20)

# Set X-axis tick spacing (step size = 2)
ax.xaxis.set_major_locator(MultipleLocator(2))

    
# Define font settings
font_settings = {
    "fontname": "Times New Roman",
    "fontsize": 20,  # Change this for font size
    "fontweight": "bold"
}

# Labeling axes and title
ax.plot(x, y, label= 'Group Wave', color='blue')
ax.set_xlabel('Wave packet length', **font_settings)
ax.set_title('High-Frequency Group Wave Packet', **font_settings) 
ax.set_ylabel('Amplitude', **font_settings)
ax.legend(prop={'family': 'Times New Roman', 'weight': 'bold', 'size': 12})
ax.grid(True, linestyle="--", linewidth=0.5)
#plt.savefig('wave packet ')
plt.grid(True)
plt.show()