txt2kml.py

import csv
import simplekml
import pandas as pd
import argparse
from pyproj import Transformer
import math

def convert(timestamp, latitude, longitude, altitude):
    from datetime import datetime

    # # Original GPS data
    # timestamp = 1732433695
    # latitude_dms = 3031.67475
    # longitude_dms = 11421.38304
    # altitude = 34.09600

    # Convert timestamp to human-readable format
    time_converted = datetime.utcfromtimestamp(timestamp).strftime('%Y-%m-%d %H:%M:%S UTC')

    # Convert latitude and longitude from DMS to decimal degrees
    latitude_degrees = int(latitude / 100) + (latitude % 100) / 60
    longitude_degrees = int(longitude / 100) + (longitude % 100) / 60

    # Prepare final result
    converted_data = {
        "UTC Time": time_converted,
        "Timestamp": timestamp,
        "Latitude": latitude_degrees,
        "Longitude": longitude_degrees,
        "Altitude": altitude
    }
    print(converted_data)
    
    return converted_data


def main(input_file, output_file):
    utm_transformer = Transformer.from_crs("EPSG:4326", "EPSG:32650", always_xy=True)  # UTM 50N
    
    columns = [
        "Timestamp", "Latitude", "Longitude", "Altitude",
        "SatNum", "Xfactor", "Yfactor", "Zfactor"
    ]
    
    df_csv = pd.read_csv(input_file, delim_whitespace=True, names=columns)
    
    if df_csv.empty:
        raise ValueError("The input file is empty or not properly formatted.")

    # 创建一个 simplekml 对象
    kml = simplekml.Kml()

    freq = 1
    if freq < 1:
        raise ValueError("Sampling frequency (freq) must be a positive integer.")

    total_distance = 0.0
    path_coords = []
    first_ts = -1
    total_time = 0.0
    for ind in df_csv.index[::freq]:
        try:
            lon = df_csv.at[ind, "Longitude"]
            lat = df_csv.at[ind, "Latitude"]
            alt = df_csv.at[ind, "Altitude"]
            ts = df_csv.at[ind, "Timestamp"]
            
            if first_ts < 0:
                first_ts = ts
            
            if lon == 0 or lat == 0:
                continue
            converted_data = convert(ts, lat, lon, alt)
            name = str(ts)

            point = kml.newpoint(name="", coords=[(converted_data["Longitude"], converted_data["Latitude"])])
            point.style.iconstyle.scale = 0.5
            
            last_one = path_coords[-1] if len(path_coords) > 0 else None
            if last_one is not None:
                last_lon, last_lat, last_alt = last_one
                curr_lon, curr_lat, curr_alt = converted_data["Longitude"], converted_data["Latitude"], converted_data["Altitude"]
                x1, y1 = utm_transformer.transform(last_lon, last_lat)
                x2, y2 = utm_transformer.transform(curr_lon, curr_lat)

                # 计算欧几里得距离
                distance = math.sqrt((x2 - x1)**2 + (y2 - y1)**2)
                total_distance += distance

            ambstatus = 1
            if ambstatus == 1:
                fix_color = simplekml.Color.yellow
                point.style.labelstyle.color = fix_color
                point.style.iconstyle.icon.href = (
                    "http://maps.google.com/mapfiles/kml/shapes/placemark_circle.png"
                ) 
                point.style.iconstyle.color = fix_color
            
            path_coords.append((converted_data["Longitude"], converted_data["Latitude"], converted_data["Altitude"]))
            total_time = ts - first_ts
        except Exception as e:
            print(f"Error processing row {ind}: {e}")  
            
    if len(path_coords) > 1:  # 至少两个点才能绘制连线
        line = kml.newlinestring(name="Path")
        line.coords = path_coords
        line.style.linestyle.width = 2  # 线宽
        line.style.linestyle.color = simplekml.Color.red

    # 保存KML文件
    kml.save(output_file)
    
    print(f"KML file saved to {output_file}")
    print(f"Total distance: {total_distance:.2f} meters")
    print(f"Total time(min): {total_time/60:.2f}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Convert CSV GPS data to KML format")
    parser.add_argument("input_file", help="Input CSV file path")
    parser.add_argument("output_file", help="Output KML file path")
    args = parser.parse_args()

    main(args.input_file, args.output_file)