nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

refactor(src): make compute_stft and process_damage_case to be pure function that explicitly need STFT arguments to be passed

Browse Source
This commit is contained in:
nuluh
2025-07-01 14:32:52 +07:00
parent a93adc8af3
commit 5ba628b678
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
code/src/process_stft.py
View File

@@ -28,7 +28,7 @@ Fs = 1024
num_damage_cases = 0 # Change to 30 if you have 30 damage cases num_damage_cases = 0 # Change to 30 if you have 30 damage cases
# Function to perform STFT and return magnitude # Function to perform STFT and return magnitude
def compute_stft(vibration_data): def compute_stft(vibration_data, Fs=Fs, window_size=window_size, hop_size=hop_size):
frequencies, times, Zxx = stft( frequencies, times, Zxx = stft(
vibration_data, vibration_data,
fs=Fs, fs=Fs,
@@ -39,7 +39,7 @@ def compute_stft(vibration_data):
stft_magnitude = np.abs(Zxx) stft_magnitude = np.abs(Zxx)
return stft_magnitude.T # Transpose to have frequencies as columns return stft_magnitude.T # Transpose to have frequencies as columns
def process_damage_case(damage_num): def process_damage_case(damage_num, Fs=Fs, window_size=window_size, hop_size=hop_size, output_dirs=output_dirs):
damage_folder = os.path.join(damage_base_path, f'DAMAGE_{damage_num}') damage_folder = os.path.join(damage_base_path, f'DAMAGE_{damage_num}')
if damage_num == 0: if damage_num == 0:
# Number of test runs per damage case # Number of test runs per damage case
@@ -83,7 +83,7 @@ def process_damage_case(damage_num):
vibration_data = df.iloc[:, 1].values vibration_data = df.iloc[:, 1].values
# Perform STFT # Perform STFT
stft_magnitude = compute_stft(vibration_data) stft_magnitude = compute_stft(vibration_data, Fs=Fs, window_size=window_size, hop_size=hop_size)
# Convert STFT result to DataFrame # Convert STFT result to DataFrame
df_stft = pd.DataFrame( df_stft = pd.DataFrame(