/*
    Read and write cotoGPS files.

    Copyright (C) 2005 Tobias Minich,
    
    Based on the Cetus I/O Filter,
    Copyright (C) 2002 Robert Lipe, robertlipe@usa.net

    This program 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 2 of the License, or
    (at your option) any later version.

    This program 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 this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111 USA
    
 */


#include "defs.h"
#if PDBFMTS_ENABLED
#include "csv_util.h"
#include "pdbfile.h"
#include "grtcirc.h"

#define MYNAME "cotoGPS"

#define MYTYPETRACK	0x5452434b  	/* TRCK */
#define MYTYPEWPT	0x44415441  	/* DATA */
#define MYCREATOR	0x636f4750	/* coGP */

#define NOTESZ 4096
#define DESCSZ 4096

#define MAX_MARKER_NAME_LENGTH 20
#define CATEGORY_NAME_LENGTH 16

typedef enum {
	cotofixNone = 0,	/* No Fix or Warning */
	cotofixReserved = 1,	/* Shouldn't occur*/
	cotofix2D = 2,		/* retrieved from a GPS with a 2D fix */
	cotofix3D = 3,		/* retrieved from a GPS with a 3D fix  */
	cotofixDGPS = 4,	/* retrieved from a GPS with a DGPS signal */
} fix_quality;

struct record_track {

	pdb_double latitude;	/* radians, s=negative */
	pdb_double longitude;	/* same as lat; e=negative */
	pdb_double distance;	/* Distance to thel last point; discarded since it's calculated by gpsbabel on write */
	pdb_double arc;		/* Course, unknown dimension */
	pdb_double x,y;		/* Internal virtual coordinates used for drawing the track on the Palm */

	gbuint16 alt;		/* Altitude */

	/* accuracy and precision information for use where applicable */
	gbuint16 hdop; /* _dop * 10 */
	gbuint16 vdop;
	gbuint16 pdop;
	gbuint8 sat_tracked;
	gbuint8 fix_quality;

	gbuint16 speed; /* *10 */
	gbuint32 time; /* Palm Time */
};

struct record_wpt {
     char lon[8];
     char lat[8];
     char name[MAX_MARKER_NAME_LENGTH];
     char notes[1];
};


// We need the pdb AppInfo for waypoint categories

typedef char appinfo_category[16];

typedef struct appinfo {
	gbuint8 U0;
	gbuint8 renamedCategories;
	appinfo_category categories[CATEGORY_NAME_LENGTH];
	gbuint8 ids[16];
	gbuint8 maxid;
} appinfo_t;

#define APPINFO_SIZE sizeof(appinfo_t)

static pdbfile *file_in, *file_out;
static const char *out_fname;
static const char *in_fname; /* We might need that for naming tracks */
static short_handle  mkshort_wr_handle;
static int ct;

static char *zerocat = NULL;
static char *internals = NULL;

static
arglist_t coto_args[] = {
	{"zerocat", &zerocat, "Name of the 'unassigned' category", NULL, 
		ARGTYPE_STRING, ARG_NOMINMAX },
	{"internals", &internals, "Export some internal stuff to notes", NULL, 
		ARGTYPE_STRING | ARGTYPE_HIDDEN, ARG_NOMINMAX },
	ARG_TERMINATOR
};

static void
rd_init(const char *fname)
{
	file_in = pdb_open(fname, MYNAME);
	in_fname = fname;
}

static void
rd_deinit(void)
{
	pdb_close(file_in);
}

static void
wr_init(const char *fname)
{
	file_out = pdb_create(fname, MYNAME);
	out_fname = fname;
	ct = 0;
}

static void
wr_deinit(void)
{
	pdb_close(file_out);
}

/* helpers */

static char *
coto_get_icon_descr(int category, const appinfo_t *app)
{
	char buff[CATEGORY_NAME_LENGTH + 1] = "Not Assigned";
	if ((category >= 0) && (category < 16))
	{
		if ((category > 0) && (app->categories[category][0] == '\0'))
			category = 0;
			
		strncpy(buff, app->categories[category], sizeof(buff) - 1);
		if (buff[0] == '\0')
			return NULL;
	}
	return xstrdup(buff);
}

static void
coto_track_read(void)
{
	struct record_track *rec;
	pdbrec_t *pdb_rec;
	route_head *trk_head;
	char *track_name;
	
	if (strncmp(file_in->name, "cotoGPS TrackDB", PDB_DBNAMELEN) != 0)
		// Use database name if not default
		track_name = xstrndup(file_in->name, PDB_DBNAMELEN);
	else {
		// Use filename for new track title
		const char *fnametmp = strrchr(in_fname, '/');
		if (fnametmp == NULL)
			fnametmp = strrchr(in_fname, '\\');
		if (fnametmp)
			fnametmp++;
		else
			fnametmp = in_fname;
		if (strrchr(fnametmp, '.') != NULL)
			track_name = xstrndup(fnametmp, strrchr(fnametmp,'.') - fnametmp);
		else
			track_name = xstrdup(fnametmp);
	}
	
	trk_head = route_head_alloc();
	track_add_head(trk_head);
		
	trk_head->rte_name = track_name;
	
	for (pdb_rec = file_in->rec_list; pdb_rec; pdb_rec = pdb_rec->next)
	{
		waypoint *wpt_tmp;

		wpt_tmp = waypt_new();

		rec = (struct record_track *) pdb_rec->data;
		
		wpt_tmp->longitude = DEG(-pdb_read_double(&rec->longitude));
		wpt_tmp->latitude = DEG(pdb_read_double(&rec->latitude));

		// It's not the course, so leave it out for now
		// WAYPT_SET(wpt_tmp, course, pdb_read_double(&rec->arc));
		wpt_tmp->altitude = be_read16(&rec->alt);
		
		if (internals)
		{
			// Parse the option as xcsv delimiter
			const char *inter = xcsv_get_char_from_constant_table(internals);
			char temp[256];
			snprintf(temp, sizeof(temp), "%.20f%s%.20f%s%.20f%s%.20f", pdb_read_double(&rec->distance), inter, 
				pdb_read_double(&rec->arc), inter, pdb_read_double(&rec->x), inter, pdb_read_double(&rec->y));
			wpt_tmp->notes = xstrdup(temp);
		}
		
		wpt_tmp->pdop = be_read16(&rec->pdop)/10.0;
		wpt_tmp->hdop = be_read16(&rec->hdop)/10.0;
		wpt_tmp->vdop = be_read16(&rec->vdop)/10.0;
		wpt_tmp->sat = rec->sat_tracked;
		switch (rec->fix_quality)
		{
			case cotofixNone:
				wpt_tmp->fix = fix_none; 
				break;
			case cotofixReserved:  
				wpt_tmp->fix = fix_unknown; 
				break;
			case cotofix2D: 
				wpt_tmp->fix = fix_2d; 
				break;
			case cotofix3D: 
				wpt_tmp->fix = fix_3d; 
				break;
			case cotofixDGPS: 
				wpt_tmp->fix = fix_dgps; 
				break;
		}
		WAYPT_SET(wpt_tmp, speed, be_read16(&rec->speed)/10.0);
		rec->time = be_read32(&rec->time);
		if (rec->time != 0)
		{
			rec->time -= 2082844800U;
			wpt_tmp->creation_time = rec->time;
		}
		track_add_wpt(trk_head, wpt_tmp);
	} 
}

static void
coto_wpt_read(void)
{
	struct record_wpt *rec;
	pdbrec_t *pdb_rec;
	appinfo_t *app;
	app = (struct appinfo *) file_in->appinfo;
	
	for(pdb_rec = file_in->rec_list; pdb_rec; pdb_rec = pdb_rec->next)
	{
		waypoint *wpt_tmp;
		char *c;
		
		wpt_tmp = waypt_new();

		rec = (struct record_wpt *) pdb_rec->data;
			
		wpt_tmp->longitude = DEG(-pdb_read_double(&rec->lon));
		wpt_tmp->latitude = DEG(pdb_read_double(&rec->lat));
		
		wpt_tmp->shortname = xstrndup(rec->name, sizeof(rec->name));
		
		wpt_tmp->icon_descr = coto_get_icon_descr(pdb_rec->category, app);
		if (wpt_tmp->icon_descr)
			wpt_tmp->wpt_flags.icon_descr_is_dynamic = 1; 
	
		if ((c = strstr(rec->notes, "\nNotes:\n"))) {	/* remove our contruct */
			wpt_tmp->notes = xstrdup(c + 8);
			if (c != rec->notes) {
				wpt_tmp->description = xstrndup(rec->notes, c - rec->notes);
			}
		} else {
			wpt_tmp->notes = xstrdup(rec->notes);
		}
		
		waypt_add(wpt_tmp);
	}
}

static void
data_read(void)
{
	if ((file_in->creator != MYCREATOR) || ((file_in->type != MYTYPETRACK) && (file_in->type != MYTYPEWPT))) {
		warning("Creator %x Type %x Version %d\n", (int) file_in->creator, (int) file_in->type, (int) file_in->version);
		fatal(MYNAME ": Not a cotoGPS file.\n");
	}
	
        is_fatal((file_in->version > 0), 
		MYNAME ": This file is from an unsupported newer version of cotoGPS.  It may be supported in a newer version of GPSBabel.\n");
	
	switch(file_in->type)
	{
		case MYTYPETRACK:
			coto_track_read();
			break;
		case MYTYPEWPT:
			coto_wpt_read();
			break;
	}
}

static void
coto_prepare_wpt_write(void)
{
	struct appinfo *ai;

	file_out->name[PDB_DBNAMELEN-1] = 0;
	file_out->attr = PDB_FLAG_BACKUP;
	file_out->type = MYTYPEWPT;  
	file_out->creator = MYCREATOR; 
	file_out->version = 0;
	
	strncpy(file_out->name, "cotoGPS MarkerDB", PDB_DBNAMELEN);
	
	file_out->appinfo_len = APPINFO_SIZE;
	file_out->appinfo = calloc(APPINFO_SIZE,1);
	
	ai = (struct appinfo *) file_out->appinfo;
	be_write16(&ai->renamedCategories, 31); // Don't ask me why...
	if (zerocat)
		strncpy(ai->categories[0], zerocat, 16);
	else
		strncpy(ai->categories[0], "Not Assigned", 16); // FIXME: Replace by default English Palm 'Not Assigned' category
	
}

static void
coto_wpt_write(const waypoint *wpt)
{
	struct record_wpt *rec;
	struct appinfo *ai = (struct appinfo *) file_out->appinfo;
	char *notes = NULL;
	char *shortname = NULL;
	int size;
	gbuint8 cat = 0;
	int i;
	
	mkshort_wr_handle = mkshort_new_handle();
	setshort_length(mkshort_wr_handle, MAX_MARKER_NAME_LENGTH);
	setshort_whitespace_ok(mkshort_wr_handle, 1);
	
	if ((global_opts.synthesize_shortnames && wpt->description) || (wpt->shortname == NULL))
		shortname = mkshort_from_wpt(mkshort_wr_handle, wpt);
	else
		shortname = xstrdup(wpt->shortname);
	
	if ((wpt->description) && ((strlen(wpt->description) > MAX_MARKER_NAME_LENGTH) || (strcmp(wpt->description, wpt->shortname))))
	{
		if ((wpt->notes) && (strcmp(wpt->description, wpt->notes) != 0)) 
		{
			notes = xcalloc(strlen(wpt->description) + strlen(wpt->notes) + 9, 1);
			sprintf(notes, "%s\nNotes:\n%s", wpt->description, wpt->notes);
		} else {
			notes = xstrdup(wpt->description);
		}
	} 
	else if (wpt->notes != NULL)
	{
		notes = xstrdup(wpt->notes);
	}

	size = sizeof(*rec);
	if (notes != NULL)
		size += strlen(notes);
	rec = xcalloc(size, 1);
	
	pdb_write_double(&rec->lon, RAD(-wpt->longitude));
	pdb_write_double(&rec->lat, RAD(wpt->latitude));
	strncpy(rec->name, shortname, MAX_MARKER_NAME_LENGTH);
	
	if (notes)
	{
		strcpy(rec->notes, notes);
		xfree(notes);
	}
	
	if (wpt->icon_descr)
	{
		for(i = 1; i < 16; i++)
			if (!strncmp(wpt->icon_descr, ai->categories[i], 16)) {cat=i; break;}
		if (!cat) {
			// We have a new one
			if (ai->maxid<15) {
				i = ++ai->maxid;
				snprintf(ai->categories[i], 16, "%s", wpt->icon_descr);
				cat = ai->ids[i] = i;
			} else {
				// We're full!
				warning(MYNAME ": Categories full. Category '%s' written as %s.\n", wpt->icon_descr, zerocat?zerocat:"Not Assigned");
			}
		}
	}
	
	pdb_write_rec(file_out, 0, cat, ct++, (const gbuint8 *)rec, size);
	
	xfree(shortname);
	xfree(rec);
	
	mkshort_del_handle(&mkshort_wr_handle);
}

static void
data_write(void)
{
	coto_prepare_wpt_write();
	waypt_disp_all(coto_wpt_write);
}


ff_vecs_t coto_vecs = {
	ff_type_file,
	{ff_cap_read|ff_cap_write, ff_cap_read, ff_cap_none},
	rd_init,
	wr_init,
	rd_deinit,
	wr_deinit,
	data_read,
	data_write,
	NULL,
	coto_args,
	CET_CHARSET_ASCII, 0	/* CET-REVIEW */
};
#endif
