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

feat(convert): Update damage scenarios and output file naming conventions

Browse Source
This commit is contained in:
nuluh
2024-12-08 18:08:59 +07:00
parent 832b6c49db
commit a1fbe8bd93
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
data/QUGS/convert.py
View File

@@ -14,10 +14,12 @@ def create_damage_files(base_path, output_base):
# Define the damage scenarios and the corresponding original file indices # Define the damage scenarios and the corresponding original file indices
damage_scenarios = { damage_scenarios = {
1: range(6, 11), # Damage 1 files from zzzAD6.csv to zzzAD10.csv 1: range(1, 6), # Damage 1 files from zzzAD1.csv to zzzAD5.csv
2: range(11, 16), # Damage 2 files from zzzAD11.csv to zzzAD15.csvs 2: range(6, 11), # Damage 2 files from zzzAD6.csv to zzzAD10.csv
3: range(16, 21), # Damage 3 files from zzzAD16.csv to zzzAD20.csv 3: range(11, 16), # Damage 3 files from zzzAD11.csv to zzzAD15.csvs
4: range(21, 26) # Damage 4 files from zzzAD21.csv to zzzAD25.csv 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))) damage_pad = len(str(len(damage_scenarios)))
test_pad = len(str(30)) test_pad = len(str(30))
@@ -30,7 +32,7 @@ def create_damage_files(base_path, output_base):
top_sensor = columns[i-1] top_sensor = columns[i-1]
print(top_sensor, type(top_sensor)) 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(f"Creating {output_file_1} from taking zzzAD{file_index}.TXT")
print("Taking datetime column on index 0...") print("Taking datetime column on index 0...")
print(f"Taking `{top_sensor}`...") print(f"Taking `{top_sensor}`...")
@@ -38,7 +40,7 @@ def create_damage_files(base_path, output_base):
print(Fore.GREEN + "Done") print(Fore.GREEN + "Done")
bottom_sensor = sensor_end_map[i] 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(f"Creating {output_file_2} from taking zzzAD{file_index}.TXT")
print("Taking datetime column on index 0...") print("Taking datetime column on index 0...")
print(f"Taking `{bottom_sensor}`...") print(f"Taking `{bottom_sensor}`...")