#include <stdio.h>
#include <poll.h>
#include <sys/select.h>
#include <alsa/asoundlib.h>
#include "alsa.h"
#include "usb.h"
#include "log.h"
#include "dump.h"

#define ALSA_BUFSIZE 4096
#define MAX_PFDS 20

static snd_seq_t *snd_handle = NULL;
static int seq_port;
static snd_midi_event_t *midi_event_parser = NULL;

int npfd;
struct pollfd pfd[MAX_PFDS];

void alsa_interrupting()
{
    //Currently nothing to do.
}

int
alsa_setup(const char *name)
{
	if (snd_seq_open(&snd_handle, "default", SND_SEQ_OPEN_DUPLEX, 0) < 0) {
		perror("snd_seq_open");
		return -1;
	}
	snd_seq_nonblock(snd_handle, 1);
	snd_seq_set_client_name(snd_handle, name);
	seq_port =
	    snd_seq_create_simple_port(snd_handle, name,
				       SND_SEQ_PORT_CAP_READ | SND_SEQ_PORT_CAP_WRITE | SND_SEQ_PORT_CAP_SUBS_READ |
				       SND_SEQ_PORT_CAP_SUBS_WRITE, SND_SEQ_PORT_TYPE_MIDI_GENERIC);

	// Allocate parser object for converting to and from MIDI
	if (snd_midi_event_new(ALSA_BUFSIZE, &midi_event_parser) < 0) {
		fatal("ALSA cannot allocate MIDI event parser");
        return -1;
	} else {
        snd_midi_event_no_status(midi_event_parser,1);
        return 0;
    }
}

void
alsa_close()
{
    if (midi_event_parser) {
        snd_midi_event_free(midi_event_parser);
    }
    if (snd_handle) {
        snd_seq_close(snd_handle);
    }
}


void
alsa_fd_setup(int *nfds, fd_set *rfds, fd_set *wfds)
{
	npfd = snd_seq_poll_descriptors_count(snd_handle, POLLIN);
	int i;
	
	if (npfd > MAX_PFDS) {
		fatal("ALSA wants too many file descriptors");
	}
	
	snd_seq_poll_descriptors(snd_handle, pfd, npfd, POLLIN);
	for (i = 0; i < npfd; i += 1) {
		if (pfd[i].fd > *nfds) {
			*nfds = pfd[i].fd;
		}
		if (pfd[i].events & POLLIN) {
			FD_SET(pfd[i].fd, rfds);
		}
	}
}

void
alsa_read_ready()
{
	snd_midi_event_init(midi_event_parser);
	
	for (;;) {
		snd_seq_event_t *ev;
		unsigned char buf[ALSA_BUFSIZE];
		long converted;
		int r;
		
		r = snd_seq_event_input(snd_handle, &ev);
		
		if (r == -EAGAIN) {
			// input queue empty
			break;
		} else if (r == -ENOSPC) {
			warn("Input queue overflow");
			continue;
		} else if (r < 0) {
			warn("snd_seq_event_input() = %d", r);
			continue;
		}
        
		converted = snd_midi_event_decode(midi_event_parser, buf, ALSA_BUFSIZE, ev);
		if (converted < 0) {
            if (ev->type == SND_SEQ_EVENT_CLIENT_START ) {
                printf("Client started \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_CLIENT_EXIT) {
                printf("Client exited \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_CLIENT_CHANGE) {
                printf("Client status/info changed \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_PORT_START) {
                printf("Port created \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_PORT_EXIT) {
                printf("Port deleted \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_PORT_CHANGE) {
                printf("Port status/info changed \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_PORT_SUBSCRIBED) {
                printf("Port connected \n");
                continue;
            } else if (ev->type == SND_SEQ_EVENT_PORT_UNSUBSCRIBED) {
                printf("Port disconnected \n");
                continue;
            } else {
                warn("Can't decode MIDI event type 0x%02x", ev->type);
            }
			continue;
		}
		
		if (converted < 3) {
			int i;
			
			warn("Uh oh, weird MIDI packet with length %ld (does this make sense if prefixed with 90?)", converted);
			
			for (i = 0; i < converted; i += 1) {
				fprintf(stderr, "  %02x", buf[i]);
			}
			fprintf(stderr, "\n");
		}
		
		usb_write(buf, converted);
	}
}

void
alsa_check_fds(fd_set *rfds, fd_set *wfds)
{
	int i;
	
	for (i = 0; i < npfd; i += 1) {
		int fd = pfd[i].fd;

		if (FD_ISSET(fd, rfds) || FD_ISSET(fd, wfds)) {
			alsa_read_ready();
			return;
		}
	}
}

void
alsa_write(uint8_t *data, size_t datalen)
{
	size_t offset = 0;
	
	snd_midi_event_init(midi_event_parser);
	
	while( datalen > offset) {
		snd_seq_event_t ev;
		long encoded;
		int r;
		
		encoded = snd_midi_event_encode(midi_event_parser, data + offset, datalen - offset, &ev);
		if (encoded <= 1) {
			int i;
	
			warn("Unable to encode MIDI message (%ld < %ld)", encoded, (long int)datalen);
			fprintf(stderr, "    ");
			for (i = offset; i < (long int)datalen; i += 1) {
				fprintf(stderr, "%02x ", data[i]);
			}
			fprintf(stderr, "\n");
			
			return;
		}
		
		snd_seq_ev_set_direct(&ev);
		snd_seq_ev_set_source(&ev, seq_port);
		snd_seq_ev_set_subs(&ev);
		r = snd_seq_event_output(snd_handle, &ev);
		
		if (r < 0) {
			warn("Couldn't write an output event");
		}
		if (r > 200) {
			warn("Output buffer size %d", r);
		}
		
		offset += encoded;
	}
	snd_seq_drain_output(snd_handle);
}