/* This file is part of Ingen.
 * Copyright (C) 2007 Dave Robillard <http://drobilla.net>
 * 
 * Ingen 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.
 * 
 * Ingen 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 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.,
 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
 */

#include <cassert>
#include <sys/mman.h>
#include <iostream>
#include <unistd.h>
#include <raul/Deletable.hpp>
#include <raul/Maid.hpp>
#include "Engine.hpp"	
#include "../../../../config/config.h"
#include "tuning.hpp"
#include "Event.hpp"
#include "JackAudioDriver.hpp"
#include "NodeFactory.hpp"
#include "ClientBroadcaster.hpp"
#include "Patch.hpp"
#include "ObjectStore.hpp"
#include "MidiDriver.hpp"
#include "OSCDriver.hpp"
#include "QueuedEventSource.hpp"
#include "PostProcessor.hpp"
#include "CreatePatchEvent.hpp"
#include "EnablePatchEvent.hpp"
#include "OSCEngineReceiver.hpp"
#ifdef HAVE_JACK_MIDI
#include "JackMidiDriver.hpp"
#endif
/*#ifdef HAVE_LASH
#include "LashDriver.hpp"
#endif*/
using namespace std;

namespace Ingen {


Engine::Engine(Ingen::Shared::World* world)
: _world(world),
  _midi_driver(NULL),
  _osc_driver(NULL),
  _maid(new Raul::Maid(maid_queue_size)),
  _post_processor(new PostProcessor(*_maid, post_processor_queue_size)),
  _broadcaster(new ClientBroadcaster()),
  _object_store(new ObjectStore()),
  _node_factory(new NodeFactory(world)),
/*#ifdef HAVE_LASH
  _lash_driver(new LashDriver()),
#else */
  _lash_driver(NULL),
//#endif
  _quit_flag(false),
  _activated(false)
{
}


Engine::~Engine()
{
	deactivate();

	for (ObjectStore::Objects::const_iterator i = _object_store->objects().begin();
			i != _object_store->objects().end(); ++i) {
		if ( ! i->second->parent())
			delete i->second;
	}
	
	delete _object_store;
	delete _broadcaster;
	delete _node_factory;
	delete _midi_driver;
	delete _osc_driver;
	
	delete _maid;

	munlockall();
}


Driver*
Engine::driver(DataType type)
{
	if (type == DataType::FLOAT)
		return _audio_driver.get();
	else if (type == DataType::MIDI)
		return _midi_driver;
	else if (type == DataType::OSC)
		return _osc_driver;
	else
		return NULL;
}


int
Engine::main()
{
	// Loop until quit flag is set (by OSCReceiver)
	while ( ! _quit_flag) {
		nanosleep(&main_rate, NULL);
		main_iteration();
	}
	cout << "[Main] Done main loop." << endl;
	
	_event_source->deactivate();
	
	if (_activated)
		deactivate();

	return 0;
}


/** Run one iteration of the main loop.
 *
 * NOT realtime safe (this is where deletion actually occurs)
 */
bool
Engine::main_iteration()
{
/*#ifdef HAVE_LASH
	// Process any pending LASH events
	if (lash_driver->enabled())
		lash_driver->process_events();
#endif*/
	// Run the maid (garbage collector)
	_maid->cleanup();
	
	return !_quit_flag;
}


void
Engine::start_jack_driver()
{
	if (_audio_driver)
		cerr << "[Engine] Warning:  replaced audio driver" << endl;
	
	_audio_driver = SharedPtr<AudioDriver>(new JackAudioDriver(*this));
}


void
Engine::start_osc_driver(int port)
{
	if (_event_source)
		cerr << "[Engine] Warning: replaced event source (engine interface)" << endl;

	_event_source = SharedPtr<EventSource>(new OSCEngineReceiver(
			*this, pre_processor_queue_size, port));
	
	//_osc_driver = _event_source;
}
	

SharedPtr<QueuedEngineInterface>
Engine::new_queued_interface()
{
	assert(!_event_source);

	SharedPtr<QueuedEngineInterface> result(new QueuedEngineInterface(
			*this, Ingen::event_queue_size, Ingen::event_queue_size));
	
	_event_source = result;

	return result;
}

/*
void
Engine::set_event_source(SharedPtr<EventSource> source)
{
	if (_event_source)
		cerr << "Warning:  Dropped event source (engine interface)" << endl;

	_event_source = source;
}
*/


bool
Engine::activate()
{
	if (_activated)
		return false;

	assert(_audio_driver);
	assert(_event_source);

#ifdef HAVE_JACK_MIDI
	_midi_driver = new JackMidiDriver(((JackAudioDriver*)_audio_driver.get())->jack_client());
#else
	_midi_driver = new DummyMidiDriver();
#endif
	
	_event_source->activate();

	// Create root patch

	Patch* root_patch = new Patch("", 1, NULL,
			_audio_driver->sample_rate(), _audio_driver->buffer_size(), 1);
	root_patch->activate();
	_object_store->add(root_patch);
	root_patch->process_order(root_patch->build_process_order());
	root_patch->enable();

	assert(_audio_driver->root_patch() == NULL);
	_audio_driver->set_root_patch(root_patch);

	_audio_driver->activate();
	
	_post_processor->start();

	_activated = true;
	
	return true;
}


void
Engine::deactivate()
{
	if (!_activated)
		return;

	/*for (Tree<GraphObject*>::iterator i = _object_store->objects().begin();
			i != _object_store->objects().end(); ++i)
		if ((*i)->as_node() != NULL && (*i)->as_node()->parent() == NULL)
			(*i)->as_node()->deactivate();*/
	
	if (_midi_driver != NULL)
		_midi_driver->deactivate();
	
	_audio_driver->deactivate();

	_audio_driver->root_patch()->deactivate();

	// Finalize any lingering events (unlikely)
	_post_processor->whip();
	_post_processor->stop();

	_audio_driver.reset();

	_event_source.reset();
	
	_activated = false;
}


} // namespace Ingen