#!/usr/bin/env python

#

# Copyright (c) 2020 Stefano Franchi

#

# tcxaet.py is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License,

# or (at your option) any later version.

#

# tcxaet.py is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

# See the GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with tcxaet.py. If not, see http://www.gnu.org/licenses/.

from __future__ import print_function

from ntpath import basename

import sys, re, os

from datetime import datetime, timedelta

from argparse import ArgumentParser, SUPPRESS, REMAINDER

import lxml.etree as ET

import pandas as pd

from imath import floor

# CONSTANTS

# Defining a dictionary of Garmin's TCX format namespaces

# All non-default namespaces defined in Garmin's TCX files asof Jan 2020, for future reference

#NSMAP = {"ns5" : "http://www.garmin.com/xmlschemas/ActivityGoals/v1",

#"ns3" : "http://www.garmin.com/xmlschemas/ActivityExtension/v2",

#"ns2" : "http://www.garmin.com/xmlschemas/UserProfile/v2",

#"tcd" : "http://www.garmin.com/xmlschemas/TrainingCenterDatabase/v2",

#"xsi" : "http://www.w3.org/2001/XMLSchema-instance",

#"ns4" : "http://www.garmin.com/xmlschemas/ProfileExtension/v1"}

# Garmin's TCX format default namespace

NSMAP = {"tcd" : "http://www.garmin.com/xmlschemas/TrainingCenterDatabase/v2"}

# Other useful constant

METERS2MILES = 1609.34

MIN2SECS = 60 # for clarity in formulas

# Parsing command line arguments, using options for required arguments

# Disable default help

parser = ArgumentParser(description='Reads speed and heart rate data from (a list of) TCX files and computes overall average BPM and the cardiac drift between the lap\'s first and second half', add_help=False)

required = parser.add_argument_group('required arguments')

optional = parser.add_argument_group('optional arguments')

# Add back help

optional.add_argument('-h','--help',action='help',default=SUPPRESS,help='show this help message and exit')

# Add command line arguments

required.add_argument("file_list", nargs=REMAINDER, help="One or more TCX or FIT files containing heart rate data for one or more activities", type=str)

optional.add_argument("-d", "--details", action="store_true", default=False, help="Print more data about every lap")

optional.add_argument("-c", "--columns", action="store_true", default=False, help="Print column headers in output")

optional.add_argument("-l", "--local-time", action="store_true", default=True, help="Converts laps's UTC time to local time. Needs timezonefinder package installed ")

# the treadmill option accepts a single parameter for the dummy treadmill pace, defaults to 12 min/mi if the option is given with no value, and to False if not given

optional.add_argument("-t", "--treadmill", default=None, nargs="?", const = 12, help="Interpret data as treadmill data (set speed/pace to a program defined constant)")

args = parser.parse_args()

if args.treadmill:

args.treadmill = float(args.treadmill)

try:

from timezonefinder import TimezoneFinder

import pytz

except:

print("timezonefinder package not installed. Using UTC time and ignoring --local-time option.")

args.local_time=False

# FUNCTIONS

def UTC_datetime2local(datetime, coords):

return datetime.astimezone(pytz.timezone(TimezoneFinder().timezone_at(lng=coords[0], lat=coords[1])))

def mil_min_val_to_mil_min_string(val):

return l[-2]+":"+str(round(float(l[-1]))).zfill(2)

def meter_sec_2_min_miles(n):

return 1/miles_a_minute

def min_miles2meter_sec(n):

"""Convert decimal miles a minute pace into speed in m/s"""

secs_per_meter = n* 60 /METERS2MILES

return 1/secs_per_meter

# PARSE MULTIPLE FILES INTO COLLECTIONS OF LAPS

def read_tcx_files(filename_list):

return all_laps

def parse_tcx_lap(file_laps):

return laps

def datetime64_2_datetime(aNumpyDaytime64):

"""Convert a numpy datetime64 object into a regular pythone datetime"""

return pd.Timestamp(aNumpyDaytime64).to_pydatetime()

def get_lap_times_and_duration(lap_data):

return (beginning_time, end_time, duration)

def lap_halftime_value(lap):

"""Return the time corresponding to the half point of the lap."""

return lap['Trackpoints_series'].index.min() + ((lap['Trackpoints_series'].index.max()-lap['Trackpoints_series'].index.min())/2)

# PARSE SINGLE LAPS' DATA

def parse_laps(laps):

return pd.DataFrame(all_laps_data)

# def make_lap_header(lap):

# """Return a string with all the info about the input lap"""

# #FIXME: add formatting string and remove variable placeholders

# #FIXME: function not currently used, yet needed with -d (--details) option

# beginning_date_string = lap['Time_list'][0]

# end_date_string = lap['Time_list'][-1]

# date_format_string = "%Y-%m-%dT%H:%M:%S.%fZ"

# beginning_time=datetime.strptime(beginning_date_string,date_format_string)

# end_time = datetime.strptime(end_date_string,date_format_string)

# duration = end_time - beginning_time

# total_time_computed = duration.total_seconds()

# lap_header = "Lap {i} --> Start: {lap['Time_list'][0]} Total time: {total_time_summary} Computed time: {duration.total_seconds()} \

# Total track points: {total_trackpoints} Total Distance: {round(total_distance,2)} Average Bpm: {round(average_bpm)}\

# Average speed in m/s {round(speed_meter_sec,3)} Average pace in min/mi {pace_min_miles}".format()

# return lap_header

# OUTPUT CSV-FORMATTED DATA

def csv_output(laps_array):

return laps_array.to_csv(columns = columns_to_write,header=True, index_label=index_name)

# main loop

if __name__ == "__main__":

# parse all files into the array

parsed_array = parse_laps(parse_tcx_lap(read_tcx_files(args.file_list)))

# output data as csv with optional header

print(csv_output(parsed_array))