frequence_response.py
f855cb73c1c5main

frequence_response.py
View Options

			# Plot frequency response of the 1D wave propagation with DDM

			import numpy as np
			import matplotlib.pyplot as plt
			from ddm_wave_prop import *

			path = r'C:/Travail/EPFL/Cours/M2/Projet CSE/LAMMPS/Dataset/'

			# PARAMETERS
			Ttime = 1.0
			dt = 0.001
			L = 0.05
			A = L**2
			E = 2e6 # fix value to get right strain energy
			rho = 2000 # 2500*0.8

			# SINUS PARAMETERS
			nb_periods = 500 # about 30 minutes for 500 periods
			periods = np.logspace(-2,0,num=nb_periods)
			ampl = 0.01

			# DATASET
			dataset = load_dataset(path)

			ampl_ratio = []
			for k, t0 in enumerate(periods):
			Ttime = 10*t0 # 10 PERIODS
			dt = t0/50 # 50 timestep per period
			displacement = lambda t: amplnp.sin(2np.pi*t/t0)
			coords, all_displacements, KE, SE, TE = wave_propagation_pressure(dataset, E, A, L, displacement, Ttime, dt, rho)
			disp_nodes = np.array(all_displacements).T # transpose displacements to index it by nodes
			max_ampl = np.abs(np.max(disp_nodes[-1]))
			ampl_ratio.append(max_ampl/ampl)
			if k%10 == 0: print('Progression: {}%'.format(100*k/nb_periods))

			plt.figure(figsize=(10, 6))
			plt.semilogx(periods, ampl_ratio)
			plt.xlabel('Period [s]')
			plt.ylabel('Amplification ratio')
			plt.savefig('Frequency_response.png')
			plt.show()