/* file: ahaecg2mit.c		G. Moody		7 May 2008
				Last revised:	     11 December 2017
Convert *.ecg or *.txt files from an AHA Database DVD to WFDB-compatible format
*/

#include <stdio.h>
#include <wfdb/wfdb.h>
#include <wfdb/ecgmap.h>

int format;
FILE *ifile;
WFDB_Annotation a;
WFDB_Sample v[2];

void process(char *r);
int readbindata(void)
{
    char data[5];

    if (fread(data, 1, 5, ifile) != 5)
	return (0);	/* EOF */
    v[0] = (data[0] & 0xff) | ((data[1] << 8) & 0xff00);
    v[1] = (data[2] & 0xff) | ((data[3] << 8) & 0xff00);
    if (data[4] == '.')
	return (1);	/* samples, no annotation */
    a.anntyp = ammap(data[4]);
    a.subtyp = (data[4] == 'U' ? -1 : 0);
    return (2);		/* samples and annotation */
}

int readtxtdata(void)
{
    char buf[16], c;
    int i, n;

    for (i = 0; i < sizeof(buf); i++) {
	c = getc(ifile);
	if (c == '\r' || c == EOF || c == '\n') break;
	buf[i] = c;
    }
    if (c == EOF) return (0);
    n = sscanf(buf, "%d,%d,\"%c", v, v+1, &c);
    if (n < 3) {
	fprintf(stderr, "Error: improperly formatted input!\n");
	fprintf(stderr,
		" n = %d, buf = |%s|, v[0] = %d, v[1] = %d, c = |%c|\n",
		n, buf, v[0], v[1], c);
	return (0);
    }
    if (c == '"')
	return (1);	/* samples, no annotation */
    a.anntyp = ammap(c);
    a.subtyp = (c == 'U' ? -1 : 0);
    return (2);		/* samples and annotation */
}
	
main(int argc, char **argv)
{
    char *p, *record;
    int i, sflag = 0;

    if (argc < 2 || strcmp(argv[1], "-h") == 0) {
	fprintf(stderr, "usage: %s [-s] RECORD.XXX [RECORD.XXX]...\n",
		argv[0]);
	fprintf(stderr, 
		" where RECORD.XXX is the name of the AHA DVD input file to"
		" be converted\n"
		" ('RECORD' is one of the 4-digit AHA DB record names, and"
		" 'XXX' may be either\n 'ecg' or 'txt', depending on the"
		" version of the AHA DVD.)\n"
		" Output files are RECORD.hea (text), RECORD.atr (binary),"
		" RECORD.dat (binary).\n"
		" Two or more input files may be converted in a single run.\n"
		" Use -s to make short-format (35-minute) records from"
		" long-format (2.5 hour)\n input files.  When using -s, the"
		" second digit in the output record name is\n"
		" incremented by 2.\n");
	exit(1);
    }
    if (strcmp(argv[i = 1], "-s") == 0) {
	sflag = 1;	/* produce short-format (35-minute) records */
	i = 2;
    }
    for ( ; i < argc; i++) {
	p = argv[i] + strlen(argv[i]) - 4;    /* pointer to '.ecg' or '.txt' */
	if (strcmp(".ecg", p) == 0 || strcmp(".ECG", p) == 0)
	    format = 1;
	else if (strcmp(".txt", p) == 0 || strcmp(".TXT", p) == 0)
	    format = 2;
	else {
	    fprintf(stderr, "%s: ignoring '%s'\n", argv[0], argv[i]);
	    continue;	/* not an .ecg or .txt file */
	}
	if ((ifile = fopen(argv[i], "rb")) == NULL) {
	    fprintf(stderr, "%s: can't open '%s'\n", argv[0], argv[1]);
	    continue;
	}
	*p = '\0';	/* strip off extension */
	record = p - 4;	/* AHA record names are 4 (ASCII) digits */
	if (sflag) {/* skip first 145 minutes if making a short-format record */
	    if (format == 1)
		fseek(ifile, 145L*60L*250L*5L, SEEK_SET);
	    else if (format == 2) {
		long t;

		for (t = 0; t < 145L*60L*250L; t++)
		    (void)readtxtdata();
	    }		
	    record[1] += 2;	/* fix record name (n0nn->n2nn, n1nn->n3nn) */
	}
	process(record);
	fclose(ifile);
    }
    exit(0);
}

void process(char *record)
{
    char ofname[10];
    int stat;
    WFDB_Time t = -1;
    static WFDB_Siginfo si[2];
    static WFDB_Anninfo ai;

    setsampfreq(250.0);	/* AHA DB is sampled at 250 Hz for each signal */
    sprintf(ofname, "%s.dat", record);
    si[0].fname = ofname;
    si[0].desc = "ECG0";
    si[0].units = "mV";
    si[0].gain = 400;
    si[0].fmt = 212;
    si[0].adcres = 12;
    si[1] = si[0];
    si[1].desc = "ECG1";
    ai.name = "atr";
    ai.stat = WFDB_WRITE;
    if (osigfopen(si, 2) != 2 || annopen(record, &ai, 1) < 0) {
	wfdbquit();
	return;
    }
    if (format == 1) {
	while (stat = readbindata()) {
	    (void)putvec(v);
	    if (stat > 1) {
		a.time = t;
		(void)putann(0, &a);
	    }
	    t++;
	}
    }
    else {
	while (stat = readtxtdata()) {
	    t++;
	    (void)putvec(v);
	    if (stat > 1) {
		a.time = t;
		(void)putann(0, &a);
	    }
	}
    }
    (void)newheader(record);
    wfdbquit();
    fprintf(stderr, "wrote %s.atr, %s.dat, and %s.hea\n", record,record,record);
    return;
}