/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            jackclient.cc
 *
 *  Sun Jul 20 21:48:44 CEST 2008
 *  Copyright 2008 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of DrumGizmo.
 *
 *  DrumGizmo 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.
 *
 *  DrumGizmo 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 DrumGizmo; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "jackclient.h"

extern "C"
{
  void _wrap_jack_shutdown(void *arg){
    ((JackClient*)arg)->shutdown();}
  int _wrap_jack_process(jack_nframes_t nframes, void *arg){
    return ((JackClient*)arg)->process(nframes);}
  void _wrap_jack_thread_init(void *arg){
    ((JackClient*)arg)->thread_init();}
  void _wrap_jack_freewheel_mode(int freewheel_mode, void *arg){
    ((JackClient*)arg)->freewheel_mode(freewheel_mode);}
  int _wrap_jack_buffer_size(jack_nframes_t nframes, void *arg){
    return ((JackClient*)arg)->buffer_size(nframes);}
  int _wrap_jack_sample_rate(jack_nframes_t nframes, void *arg){
    return ((JackClient*)arg)->sample_rate(nframes);}
  void _wrap_jack_port_registration(jack_port_id_t port, int i, void *arg){
    ((JackClient*)arg)->port_registration(port, i);}
  int _wrap_jack_graph_order(void *arg){
    return ((JackClient*)arg)->graph_order();}
  int _wrap_jack_xrun(void *arg){
    return ((JackClient*)arg)->xrun();}
}  // extern "C"

JackClient::JackClient(DrumKit *drumkit)
  : midimapper(drumkit)
{
  this->drumkit = drumkit;

	jack_status_t status;

	jack_client = jack_client_open("DrumGizmo", JackNullOption, &status);

	// Setup input port
	midi_port = jack_port_register(jack_client,
                                 "midi_in",
                                 JACK_DEFAULT_MIDI_TYPE,
                                 JackPortIsInput,// | JackPortIsTerminal,
                                 0);

	// Setup input ports
  Instruments::iterator ii = drumkit->instruments.begin();
	while(ii != drumkit->instruments.end()) {
    Instrument *instrument = ii->second;
		instrument->port = jack_port_register(jack_client,
                                          instrument->name.c_str(),
                                          JACK_DEFAULT_AUDIO_TYPE,
                                          JackPortIsInput | JackPortIsTerminal,
                                          0);

    BeatMapper *beatmapper = new BeatMapper(instrument);
    beatmappers.push_back(beatmapper);

    //    input_ports.push_back(instrument->port);
    ii++;
	}

	// Setup output ports
  Channels::iterator ci = drumkit->channels.begin();
  while(ci != drumkit->channels.end()) {
    Channel *channel = ci->second;
		channel->port = jack_port_register(jack_client,
                                       channel->name.c_str(),
                                       JACK_DEFAULT_AUDIO_TYPE,
                                       JackPortIsOutput | JackPortIsTerminal,
                                       0);
    output_ports.push_back(channel->port);
    ci++;
	}

	//jack_on_shutdown(jack_client, _wrap_jack_shutdown, this);
  jack_set_process_callback(jack_client, _wrap_jack_process, this);
	//jack_set_thread_init_callback(jack_client, _wrap_jack_thread_init, this);
	//jack_set_freewheel_callback(jack_client, _wrap_jack_freewheel_mode, this);
	//jack_set_buffer_size_callback(jack_client, _wrap_jack_buffer_size, this);
	//jack_set_sample_rate_callback(jack_client, _wrap_jack_sample_rate, this);
	//jack_set_port_registration_callback(jack_client, _wrap_jack_port_registration, this);
	//jack_set_graph_order_callback(jack_client, _wrap_jack_graph_order, this);
	//jack_set_xrun_callback(jack_client, _wrap_jack_xrun, this);
}

JackClient::~JackClient()
{
	jack_client_close(jack_client);
}

void JackClient::activate()
{
	jack_activate(jack_client);
}

void JackClient::shutdown()
{
}

int JackClient::process(jack_nframes_t nframes)
{
  //
  // Look for midi input
  //
	void *midibuffer = jack_port_get_buffer(midi_port, nframes);
	jack_nframes_t midievents = jack_midi_get_event_count(midibuffer);
	for(jack_nframes_t i = 0; i < midievents; i++) {
		jack_midi_event_t midi_event;
		jack_midi_event_get(&midi_event, midibuffer, i);
    
    Sample *sample = midimapper.map(midi_event);
    if(!sample) continue;

    AudioFiles::iterator ai = sample->audiofiles.begin();
    while(ai != sample->audiofiles.end()) {
      AudioFile *audiofile = ai->second;
      audiofile->load();

      if(drumkit->channels.find(audiofile->channel) != drumkit->channels.end()) {
        Channel *channel = drumkit->channels[audiofile->channel];
        Event event(channel->port, audiofile, midi_event.time);
        events.insert(event);
      }
      ai++;
    }

	}
  jack_midi_clear_buffer(midibuffer);

  //
  // Look for audio trigger input
  //
  //  std::vector< BeatMapper >::iterator bi = beatmappers.begin();
  //  while(bi != beatmappers.end()) {
  for(size_t bi = 0; bi < beatmappers.size(); bi++) {
    BeatMapper *beatmapper = beatmappers[bi];//*bi;

    Sample *sample = beatmapper->map(nframes);
    if(!sample) continue;

    AudioFiles::iterator ai = sample->audiofiles.begin();
    while(ai != sample->audiofiles.end()) {
      AudioFile *audiofile = ai->second;
      audiofile->load();

      if(drumkit->channels.find(audiofile->channel) != drumkit->channels.end()) {
        Channel *channel = drumkit->channels[audiofile->channel];
        Event event(channel->port, audiofile, 0);
        events.insert(event);
      }
      ai++;
    }

    //bi++;
  }

  //
  // Reset ports 
  //
  Ports::iterator pi = output_ports.begin();
  while(pi != output_ports.end()) {
    
    jack_default_audio_sample_t *buffer;
    buffer = (jack_default_audio_sample_t *)jack_port_get_buffer(*pi, nframes);
    
    for(size_t j = 0; j < nframes; j++) {
      buffer[j] = 0;
    }
    
    pi++;
  }

  Events nextevents;

  //
  // Handle events
  //
  Events::iterator ei = events.begin();
  while(ei != events.end()) {

    Event event = *ei;

    jack_default_audio_sample_t *buffer;
    buffer = (jack_default_audio_sample_t *)jack_port_get_buffer(event.port, nframes);
    
    size_t dtime = nframes - event.time; // how much buffer is left?
    size_t size = event.sample->size - event.duration; // how much audio is left?
    if(size > dtime) size = dtime;
    
    for(size_t j = event.time; j < event.time + size; j++) {
      buffer[j] += event.sample->data[event.duration + j - event.time];
    }
    
    if(event.duration + size < event.sample->size) {
      Event e(event.port, event.sample, 0, event.duration + size);
      nextevents.insert(e);
    }
    
    ei++;
  }

  // Remove all dead events
  events = nextevents;

	return 0;
}

void JackClient::thread_init()
{
}

void JackClient::freewheel_mode(int freewheel_mode)
{
}

int JackClient::buffer_size(jack_nframes_t nframes)
{
  return 0;
}

int JackClient::sample_rate(jack_nframes_t nframes)
{
  return 0;
}

void JackClient::port_registration(jack_port_id_t port, int i)
{
}

int JackClient::graph_order()
{
  return 0;
}

int JackClient::xrun()
{
  return 0;
}