/*
    Copyright (C) 2000 Paul Davis 

    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.

    $Id: diskstream.h 579 2006-06-12 19:56:37Z essej $
*/

#ifndef __ardour_midi_diskstream_h__
#define __ardour_midi_diskstream_h__

#include <sigc++/signal.h>

#include <cmath>
#include <cassert>
#include <string>
#include <queue>
#include <map>
#include <vector>

#include <time.h>

#include <pbd/fastlog.h>
#include <pbd/ringbufferNPT.h>

#include <ardour/ardour.h>
#include <ardour/configuration.h>
#include <ardour/diskstream.h>
#include <ardour/midi_playlist.h>
#include <ardour/midi_ring_buffer.h>
#include <ardour/midi_state_tracker.h>
#include <ardour/utils.h>

struct tm;

namespace ARDOUR {

class IO;
class MidiEngine;
class MidiPort;
class MidiRingbuffer;
class SMFSource;
class Send;
class Session;

class MidiDiskstream : public Diskstream
{	
  public:
	MidiDiskstream (Session &, const string& name, Diskstream::Flag f = Recordable);
	MidiDiskstream (Session &, const XMLNode&);
	~MidiDiskstream();

	float playback_buffer_load() const;
	float capture_buffer_load() const;
	
	void get_playback(MidiBuffer& dst, nframes_t start, nframes_t end, nframes_t offset);

	void set_record_enabled (bool yn);

	boost::shared_ptr<MidiPlaylist> midi_playlist () { return boost::dynamic_pointer_cast<MidiPlaylist>(_playlist); }

	int use_playlist (boost::shared_ptr<Playlist>);
	int use_new_playlist ();
	int use_copy_playlist ();

	/* stateful */
	XMLNode& get_state(void);
	int set_state(const XMLNode& node);

	void monitor_input (bool);

	boost::shared_ptr<SMFSource> write_source () { return _write_source; }
	
	int set_destructive (bool yn); // doom!
	
	void set_note_mode (NoteMode m);
	
	uint16_t get_channel_mask() { 
		uint16_t playback_mask = _playback_buf->get_channel_mask();
#ifndef NDEBUG
		uint16_t capture_mask  = _capture_buf->get_channel_mask();
		assert(playback_mask == capture_mask);
#endif
		return playback_mask;
	}

	void set_channel_mode(ChannelMode mode, uint16_t mask) {
		_playback_buf->set_channel_mode(mode, mask); 
		_capture_buf->set_channel_mode(mode, mask); 
	}
	
	ChannelMode get_channel_mode() {
		ChannelMode playback_mode = _playback_buf->get_channel_mode();
#ifndef NDEBUG
		ChannelMode capture_mode  = _capture_buf->get_channel_mode();
		assert(playback_mode == capture_mode);
#endif
		return playback_mode;
	}

  protected:
	friend class Session;

	/* the Session is the only point of access for these
	   because they require that the Session is "inactive"
	   while they are called.
	*/

	void set_pending_overwrite(bool);
	int  overwrite_existing_buffers ();
	void set_block_size (nframes_t);
	int  internal_playback_seek (nframes_t distance);
	int  can_internal_playback_seek (nframes_t distance);
	int  rename_write_sources ();
	void reset_write_sources (bool, bool force = false);
	void non_realtime_input_change ();
	void non_realtime_locate (nframes_t location);

	static void set_readahed_frames( nframes_t frames_ahead ) { midi_readahead = frames_ahead; }

  protected:
	int seek (nframes_t which_sample, bool complete_refill = false);

  protected:
	friend class MidiTrack;

	int  process (nframes_t transport_frame, nframes_t nframes, nframes_t offset, bool can_record, bool rec_monitors_input);
	bool commit  (nframes_t nframes);
	static nframes_t midi_readahead;

  private:

	/* The two central butler operations */
	int do_flush (RunContext context, bool force = false);
	int do_refill ();
	
	int do_refill_with_alloc();

	int read (nframes_t& start, nframes_t cnt, bool reversed);

	void finish_capture (bool rec_monitors_input);
	void transport_stopped (struct tm&, time_t, bool abort);
	void transport_looped (nframes_t transport_frame);

	void init (Diskstream::Flag);

	int use_new_write_source (uint32_t n=0);

	int find_and_use_playlist (const string&);

	void allocate_temporary_buffers ();

	int use_pending_capture_data (XMLNode& node);

	void get_input_sources ();
	void check_record_status (nframes_t transport_frame, nframes_t nframes, bool can_record);
	void set_align_style_from_io();
	
	void engage_record_enable ();
	void disengage_record_enable ();

	MidiRingBuffer<nframes_t>*   _playback_buf;
	MidiRingBuffer<nframes_t>*   _capture_buf;
	MidiPort*                    _source_port;
	boost::shared_ptr<SMFSource> _write_source;
	nframes_t                    _last_flush_frame;
	NoteMode                     _note_mode;  
	MidiStateTracker             _midi_state_tracker;
	volatile gint                _frames_written_to_ringbuffer;
	volatile gint                _frames_read_from_ringbuffer;
};

}; /* namespace ARDOUR */

#endif /* __ardour_midi_diskstream_h__ */