diff --git a/data/QUGS/convert.py b/data/QUGS/convert.py index 865b633..9dc22c9 100644 --- a/data/QUGS/convert.py +++ b/data/QUGS/convert.py @@ -14,10 +14,12 @@ def create_damage_files(base_path, output_base): # Define the damage scenarios and the corresponding original file indices damage_scenarios = { - 1: range(6, 11), # Damage 1 files from zzzAD6.csv to zzzAD10.csv - 2: range(11, 16), # Damage 2 files from zzzAD11.csv to zzzAD15.csvs - 3: range(16, 21), # Damage 3 files from zzzAD16.csv to zzzAD20.csv - 4: range(21, 26) # Damage 4 files from zzzAD21.csv to zzzAD25.csv + 1: range(1, 6), # Damage 1 files from zzzAD1.csv to zzzAD5.csv + 2: range(6, 11), # Damage 2 files from zzzAD6.csv to zzzAD10.csv + 3: range(11, 16), # Damage 3 files from zzzAD11.csv to zzzAD15.csvs + 4: range(16, 21), # Damage 4 files from zzzAD16.csv to zzzAD20.csv + 5: range(21, 26), # Damage 5 files from zzzAD21.csv to zzzAD25.csv + 6: range(26, 31) # Damage 6 files from zzzAD26.csv to zzzAD30.csv } damage_pad = len(str(len(damage_scenarios))) test_pad = len(str(30)) @@ -30,7 +32,7 @@ def create_damage_files(base_path, output_base): top_sensor = columns[i-1] print(top_sensor, type(top_sensor)) - output_file_1 = os.path.join(output_base, f'DAMAGE_{damage}', f'D{damage:0{damage_pad}}_TEST{i:0{test_pad}}_01.csv') + output_file_1 = os.path.join(output_base, f'DAMAGE_{damage}', f'DAMAGE_{damage}_TEST{i}_01.csv') print(f"Creating {output_file_1} from taking zzzAD{file_index}.TXT") print("Taking datetime column on index 0...") print(f"Taking `{top_sensor}`...") @@ -38,7 +40,7 @@ def create_damage_files(base_path, output_base): print(Fore.GREEN + "Done") bottom_sensor = sensor_end_map[i] - output_file_2 = os.path.join(output_base, f'DAMAGE_{damage}', f'D{damage}_TEST{i}_02.csv') + output_file_2 = os.path.join(output_base, f'DAMAGE_{damage}', f'DAMAGE_{damage}_TEST{i}_02.csv') print(f"Creating {output_file_2} from taking zzzAD{file_index}.TXT") print("Taking datetime column on index 0...") print(f"Taking `{bottom_sensor}`...")