GPR2PCD.py

# -*- coding: utf-8 -*-
"""
Created on Mon Jul  8 12:01:38 2024

@author: user
Merge each slice into one CSV file so that PCD software could recognize as a 3D file.
"""

import os
import pandas as pd
import re
from tqdm import tqdm

# 定义slices文件夹的路径
folder_path = 'E:\\OneDrive - The University of Hong Kong - Connect\\sci_research\\2025.GPRAR_heritage\\[1209]Muiwo-watchtower-detection-sources\\GPR-survey\\csv_Survey_MWWT_Ground1_Floor3_251209'

# 获取slices文件夹下所有的CSV文件并按文件名中的顺序排序
csv_files = sorted(
    [f for f in os.listdir(folder_path) if f.startswith('Slice') and f.endswith('.csv')],
    key=lambda x: int(re.search(r'Slice (\d+) -', x).group(1))
)

# 初始化一个空的DataFrame，用于存储合并后的数据
merged_df = pd.DataFrame()

# 使用tqdm包装文件列表，显示处理进度
for csv_file in tqdm(csv_files, desc='Processing files', unit='file'):
    # 构造文件的完整路径
    file_path = os.path.join(folder_path, csv_file)
    
    # 读取CSV文件
    df = pd.read_csv(file_path)
    
    # 使用正则表达式从文件名中提取数字
    match = re.search(r'Slice \d+ - ([0-9]+(?:\.[0-9]+)?) m\.csv', csv_file)
    if match:
        z_value = -float(match.group(1))  # 提取的数字作为Z列的值
        # 在Y列之后Amplitude列之前添加Z列
        df.insert(loc=df.columns.get_loc('Amplitude'), column='Z', value=z_value)
    else:
        continue  # 如果文件名不符合预期格式，则跳过
    
    # 将当前文件的数据追加到合并的DataFrame中
    merged_df = pd.concat([merged_df, df], ignore_index=True)

# 将合并后的DataFrame写入CSV文件
output_file_path = os.path.join(folder_path + '.csv')
print('Writing into the file...')
merged_df.to_csv(output_file_path, index=False)
print(f'Merged CSV file has been created at {output_file_path}')