/*
    Copyright (C) 2013 Waves Audio Ltd.

    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., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <iostream>
#include "waves_midi_device.h"
#include "waves_midi_event.h"

// use non-zero latency because we want output to be timestapmed
#define LATENCY 0

#define QUEUE_LENGTH 1024

using namespace ARDOUR;

WavesMidiDevice::WavesMidiDevice (const std::string& device_name)
    : _pm_input_id (pmNoDevice)
    , _pm_output_id (pmNoDevice)
    , _name (device_name)
    , _input_queue (NULL)
    , _output_queue (NULL)
    , _input_pm_stream (NULL)
    , _output_pm_stream (NULL)
    , _incomplete_waves_midi_event (NULL)
{
    validate ();
}

WavesMidiDevice::~WavesMidiDevice ()
{
    // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::~WavesMidiDevice ():" << name () << std::endl;
    close ();
}

void
WavesMidiDevice::validate ()
{
    _pm_input_id = 
    _pm_output_id = pmNoDevice;
    int count = Pm_CountDevices ();

    for (int i = 0; i < count; i++) {

        const PmDeviceInfo* pm_device_info = Pm_GetDeviceInfo (i);

        if (pm_device_info == NULL) {
            continue;
        }
        if (name () == pm_device_info->name) {
            if (pm_device_info->input){
                _pm_input_id = i;
            }
            if (pm_device_info->output){
                _pm_output_id = i;
            }
        }
    }
}

int
WavesMidiDevice::open (PmTimeProcPtr time_proc, void* time_info)
{
    // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::open ():" << name () << std::endl;
    
    if (is_input () && !_input_pm_stream) {
        if (pmNoError != Pm_OpenInput (&_input_pm_stream, 
                                      _pm_input_id,
                                      NULL,
                                      1024,
                                      time_proc,
                                      time_info)) {
                std::cerr << "WavesMidiDevice::open (): Pm_OpenInput () failed for " << _pm_input_id << "-[" << name () <<  "]!" << std::endl;
                _input_pm_stream = NULL;
                _pm_input_id = pmNoDevice;
                return -1;
        }
        _input_queue = Pm_QueueCreate (QUEUE_LENGTH, sizeof (const WavesMidiEvent*));
        if (NULL == _input_queue) {
            std::cerr << "WavesMidiDevice::open (): _input_queue = Pm_QueueCreate () failed for " << _pm_input_id << "-[" << name () <<  "]!" << std::endl;
            close ();
            return -1;
        }
    }

    if (is_output () && !_output_pm_stream) {
        if (pmNoError != Pm_OpenOutput (&_output_pm_stream, 
                                       _pm_output_id, 
                                       NULL,
                                       1024,
                                       time_proc,
                                       time_info,
                                       LATENCY)) {
                std::cerr << "WavesMidiDevice::open (): Pm_OpenOutput () failed for " << _pm_output_id << "-[" << name () <<  "]!" << std::endl;
                _output_pm_stream = NULL;
                _pm_output_id = pmNoDevice;
                return -1;
        }
        _output_queue = Pm_QueueCreate (QUEUE_LENGTH, sizeof (const WavesMidiEvent*));
        if (NULL == _output_queue) {
            std::cerr << "WavesMidiDevice::open (): _output_queue = Pm_QueueCreate () failed for " << _pm_output_id << "-[" << name () <<  "]!" << std::endl;
            close ();
            return -1;
        }
    }
    return 0;
}


void
WavesMidiDevice::close ()
{
        DEBUG_TRACE (DEBUG::WavesMIDI, string_compose ("WavesMidiDevice::close (): %1\n", name ()));
        WavesMidiEvent *waves_midi_event;
        
        if (_input_pm_stream) {
                PmError err = Pm_Close (_input_pm_stream);

                if (err != pmNoError) {
	                std::cerr << "WavesMidiDevice::close (): Pm_Close () failed for " << _pm_input_id << "-[" << name () <<  "]!" << std::endl;
	                char* err_msg = new char[256];
	                Pm_GetHostErrorText(err_msg, 256);
	                std::cerr << "Error: " << err_msg << std::endl;
	                std::cerr << "Aborting!" << std::endl;
	                Pm_Abort (_input_pm_stream);
                }

                while (1 == Pm_Dequeue (_input_queue, &waves_midi_event)) {
                        delete waves_midi_event; // XXX possible dup free in ~WavesMidiBuffer() (?)
                }

                Pm_QueueDestroy (_input_queue);
                _input_queue = NULL;
                _input_pm_stream = NULL;
                _pm_input_id = pmNoDevice;
        }
        
        
        if ( _output_pm_stream ) {
                PmError err = Pm_Close (_output_pm_stream);

                if (err != pmNoError) {
	                std::cerr << "WavesMidiDevice::close (): Pm_Close () failed for " << _pm_output_id << "-[" << name () <<  "]!" << std::endl;
	                char* err_msg = new char[256];
	                Pm_GetHostErrorText(err_msg, 256);
	                std::cerr << "Error: " << err_msg << std::endl;
	                std::cerr << "Aborting!" << std::endl;
	                Pm_Abort (_output_pm_stream);
                }
                
                while (1 == Pm_Dequeue (_output_queue, &waves_midi_event)) {
                        delete waves_midi_event; // XXX possible dup free in ~WavesMidiBuffer() (?)
                }

                Pm_QueueDestroy (_output_queue);
                _output_queue = NULL;
                _output_pm_stream = NULL;
                _pm_output_id = pmNoDevice;
        }
}

void
WavesMidiDevice::do_io ()
{
    read_midi ();
    write_midi ();
}

void
WavesMidiDevice::read_midi ()
{
    if (NULL == _input_pm_stream) {
        return;
    }

    // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "]" << std::endl;

    while (Pm_Poll (_input_pm_stream) > 0) {
        PmEvent pm_event; // just one message at a time
        int result = Pm_Read (_input_pm_stream, &pm_event, 1);
        if (result < 0) {
            std::cerr << "WavesMidiDevice::_read_midi (): Pm_Read () failed (" << result << ") for [" << name () << "]!" << std::endl;
            break;
        }
        // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "] evt-tm:" << pm_event.timestamp << std::endl;
        if (_incomplete_waves_midi_event == NULL ) {
            // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "] : new _incomplete_waves_midi_event" << std::endl;
            _incomplete_waves_midi_event = new WavesMidiEvent (pm_event.timestamp);
        }
        
        WavesMidiEvent *nested_pm_event = _incomplete_waves_midi_event->append_data (pm_event);
        if (nested_pm_event) {
            Pm_Enqueue (_input_queue, &nested_pm_event);
            // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "] : Pm_Enqueue (_input_queue, nested_pm_event)" << std::endl;
        }
        switch ( _incomplete_waves_midi_event->state ()) {
            case WavesMidiEvent::BROKEN:
                delete _incomplete_waves_midi_event;
                _incomplete_waves_midi_event = NULL;
                // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "] : case WavesMidiEvent::BROKEN:" << std::endl;
            break;
            case WavesMidiEvent::COMPLETE:
                // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_read_midi (): " << _pm_device_id << "-[" << name () << "] : Pm_Enqueue (_input_queue, _incomplete_waves_midi_event); " << std::hex << (void*)_incomplete_waves_midi_event << std::dec << std::endl;
                Pm_Enqueue (_input_queue, &_incomplete_waves_midi_event);
                _incomplete_waves_midi_event = NULL;
            break;
            default:
            break;
        }
    }
}


void
WavesMidiDevice::write_midi ()
{
    if (NULL == _output_pm_stream) {
        return;
    }
    // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_write_midi (): " << _pm_device_id << "-[" << name () << "]" << std::endl;

    PmError err;
    WavesMidiEvent *waves_midi_event;

    while (1 == Pm_Dequeue (_output_queue, &waves_midi_event)) {
        if (waves_midi_event->sysex ()) {
            // LATENCY compensation
            err = Pm_WriteSysEx (_output_pm_stream, waves_midi_event->timestamp () - LATENCY, waves_midi_event->data ());
            if (0 > err) {
                std::cout << "WavesMidiDevice::write_event_to_device (): [" << name () << "] Pm_WriteSysEx () failed (" << err << ")!" << std::endl;
            };
            // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_write_midi (): SYSEX used, ev->tm:" << waves_midi_event->timestamp () - LATENCY << std::endl;
        }
        else
        {
            err = Pm_WriteShort (_output_pm_stream, waves_midi_event->timestamp () - LATENCY, * (PmMessage*)waves_midi_event->data ());
            if (0 > err) {
                std::cout << "WavesMidiDevice::write_event_to_device (): [" << name () << "] Pm_WriteShort () failed (" << err << ")!" << std::endl;
            }
            // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::_write_midi (): SHORTMSG used, ev->tm:" << waves_midi_event->timestamp () - LATENCY <<  std::endl;
        }
    }
    return;
}

int
WavesMidiDevice::enqueue_output_waves_midi_event (const WavesMidiEvent* waves_midi_event)
{
    // COMMENTED DBG LOGS */ std::cout << "WavesMidiDevice::enqueue_output_waves_midi_event (): " << _pm_device_id << "-[" << name () << "]" << std::endl;

    if (waves_midi_event == NULL) {
        std::cerr << "WavesMidiDevice::put_event_to_callback (): 'waves_midi_event' is NULL!" << std::endl;
        return -1;
    }

    PmError err = Pm_Enqueue (_output_queue, &waves_midi_event);

    if (0 > err) {
        std::cerr << "WavesMidiDevice::put_event_to_callback (): Pm_Enqueue () failed (" << err << ")!" << std::endl;
        return -1;
    };

    return 0;
}

WavesMidiEvent* 
WavesMidiDevice::dequeue_input_waves_midi_event ()
{
    WavesMidiEvent* waves_midi_event;
    if (Pm_Dequeue (_input_queue, &waves_midi_event) == 1) {
        return waves_midi_event;
    }
    return NULL;
}