#include <stdio.h>
#include <ecg/db.h>
#include <ecg/ecgmap.h>

main(argc, argv)
int argc;
char *argv[];
{
    int btype,i,j,nbeats=0,nsig,hwindow,window,v[DB_MAXSIG],vb[DB_MAXSIG];
    long stoptime = 0L, *sum[DB_MAXSIG];
    DB_Anninfo a;
    DB_Annotation annot;
    static DB_Siginfo s[DB_MAXSIG];
    void *calloc();

    if (argc < 3) {
        fprintf(stderr,
                "usage: %s annotator record [beat-type from to]\n",
                argv[0]);
        exit(1);
    }
    a.name = argv[1]; a.stat = READ;
    if ((nsig = dbinit(argv[2], &a, 1, s, DB_MAXSIG)) < 1) exit(2);
    hwindow = strtim(".05"); window = 2*hwindow + 1;
    btype = (argc > 3) ? strann(argv[3]) : NORMAL;
    if (argc > 4) iannsettime(strtim(argv[4]));
    if (argc > 5) {
        if ((stoptime = strtim(argv[5])) < 0L)
            stoptime = -stoptime;
        if (s[0].nsamp > 0L && stoptime > s[0].nsamp)
            stoptime = s[0].nsamp;
    }
    else stoptime = s[0].nsamp;
    if (stoptime > 0L) stoptime -= hwindow;
    for (i = 0; i < nsig; i++)
        if ((sum[i]=(long *)calloc(window,sizeof(long))) == NULL) {
            fprintf(stderr, "%s: insufficient memory\n", argv[0]);
            exit(3);
        }
    while (getann(0, &annot) == 0 && annot.time < hwindow)
        ;
    do {
        if (annot.anntyp != btype) continue;
        isigsettime(annot.time - hwindow - 1);
        getvec(vb);
        for (j = 0; j < window && getvec(v) > 0; j++)
            for (i = 0; i < nsig; i++)
                sum[i][j] += v[i] - vb[i];
        nbeats++;
    } while (getann(0, &annot) == 0 &&
             (stoptime == 0L || annot.time < stoptime));
    if (nbeats < 1) {
        fprintf(stderr, "%s: no `%s' beats found\n",
                argv[0], annstr(btype));
        exit(4);
    }
    printf("Average of %d `%s' beats:\n", nbeats, annstr(btype));
    for (j = 0; j < window; j++)
        for (i = 0; i < nsig; i++)
            printf("%g%c", (double)sum[i][j]/nbeats,
                   (i == nsig-1) ? '\n' : '\t');
    exit(0);
}