/* This file is part of Machina.
 * Copyright 2007-2011 David Robillard <http://drobilla.net>
 *
 * Machina 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 3 of the License, or
 * (at your option) any later version.
 *
 * Machina 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 Machina.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef MACHINA_MACHINE_HPP
#define MACHINA_MACHINE_HPP

#include <vector>
#include <set>
#include <boost/utility.hpp>

#include "machina/types.hpp"
#include "raul/Atom.hpp"
#include "raul/RingBuffer.hpp"
#include "raul/TimeSlice.hpp"
#include "sord/sordmm.hpp"

#include "types.hpp"
#include "Node.hpp"

namespace machina {

class Context;
class LearnRequest;

/** A (Finite State) Machine.
 */
class Machine
	: public Stateful
{
public:
	Machine(TimeUnit unit);
	Machine(const Machine& copy);

	Machine& operator=(const Machine& other);

	// Kluge to appease Eugene
	bool operator==(const Machine& other) { return false; }

	// Main context
	void activate()   { _is_activated = true; }
	void deactivate() { _is_activated = false; }

	bool is_empty()     { return _nodes.empty(); }
	bool is_finished()  { return _is_finished; }
	bool is_activated() { return _is_activated; }

	void add_node(SPtr<Node> node);
	void remove_node(SPtr<Node> node);
	void learn(SPtr<Raul::Maid> maid, SPtr<Node> node);

	void write_state(Sord::Model& model);

	// Audio context
	void     reset(MIDISink* sink, Raul::TimeStamp time);
	uint32_t run(Context& context, SPtr<Raul::RingBuffer> updates);

	// Any context
	inline Raul::TimeStamp time() const { return _time; }

	SPtr<LearnRequest> pending_learn()       { return _pending_learn; }
	void               clear_pending_learn() { _pending_learn.reset(); }

	typedef std::set< SPtr<Node> > Nodes;
	Nodes&       nodes()       { return _nodes; }
	const Nodes& nodes() const { return _nodes; }

	SPtr<Node> random_node();
	SPtr<Edge> random_edge();

private:
	// Audio context
	SPtr<Node> earliest_node() const;

	bool enter_node(Context&               context,
	                SPtr<Node>             node,
	                SPtr<Raul::RingBuffer> updates);

	void exit_node(Context&               context,
	               SPtr<Node>             node,
	               SPtr<Raul::RingBuffer> updates);

	static const size_t MAX_ACTIVE_NODES = 128;

	std::vector< SPtr<Node> > _active_nodes;

	SPtr<LearnRequest> _pending_learn;
	Nodes              _nodes;
	Raul::TimeStamp    _time;
	bool               _is_activated;
	bool               _is_finished;
};

} // namespace machina

#endif // MACHINA_MACHINE_HPP