/*
    Copyright (C) 2005 Paul Davis 
    Written by Taybin Rutkin

    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 <map>
#include <cerrno>

#include <gtkmm/box.h>
#include <gtkmm/stock.h>

#include <pbd/convert.h>

#include <gtkmm2ext/utils.h>

#include <ardour/audio_library.h>
#include <ardour/audioregion.h>
#include <ardour/audiofilesource.h>

#include "ardour_ui.h"
#include "gui_thread.h"
#include "prompter.h"
#include "sfdb_ui.h"
#include "utils.h"

#include "i18n.h"

using namespace ARDOUR;
using namespace PBD;
using namespace std;

SoundFileBox::SoundFileBox ()
	:
	_session(0),
	current_pid(0),
	fields(Gtk::ListStore::create(label_columns)),
	main_box (false, 3),
	top_box (true, 4),
	bottom_box (true, 4),
	play_btn(_("Play")),
	stop_btn(_("Stop")),
	add_field_btn(_("Add Field...")),
	remove_field_btn(_("Remove Field"))
{
	set_name (X_("SoundFileBox"));
	border_frame.set_label (_("Soundfile Info"));
	border_frame.add (main_box);

	pack_start (border_frame);
	set_border_width (4);

	main_box.set_border_width (4);

	main_box.pack_start(length, false, false);
	main_box.pack_start(format, false, false);
	main_box.pack_start(channels, false, false);
	main_box.pack_start(samplerate, false, false);
	main_box.pack_start(field_view, true, true);
	main_box.pack_start(top_box, false, false);
	main_box.pack_start(bottom_box, false, false);

	field_view.set_model (fields);
	field_view.set_size_request(200, 150);
	field_view.append_column (_("Field"), label_columns.field);
	field_view.append_column_editable (_("Value"), label_columns.data);

	top_box.set_homogeneous(true);
	top_box.pack_start(add_field_btn);
	top_box.pack_start(remove_field_btn);

	remove_field_btn.set_sensitive(false);

	bottom_box.set_homogeneous(true);
	bottom_box.pack_start(play_btn);
	bottom_box.pack_start(stop_btn);

	play_btn.signal_clicked().connect (mem_fun (*this, &SoundFileBox::play_btn_clicked));
	stop_btn.signal_clicked().connect (mem_fun (*this, &SoundFileBox::stop_btn_clicked));

	add_field_btn.signal_clicked().connect
	                (mem_fun (*this, &SoundFileBox::add_field_clicked));
	remove_field_btn.signal_clicked().connect
	                (mem_fun (*this, &SoundFileBox::remove_field_clicked));
	
	Gtk::CellRendererText* cell(static_cast<Gtk::CellRendererText*>(field_view.get_column_cell_renderer(1)));
	cell->signal_edited().connect (mem_fun (*this, &SoundFileBox::field_edited));

	field_view.get_selection()->signal_changed().connect (mem_fun (*this, &SoundFileBox::field_selected));
	Library->fields_changed.connect (mem_fun (*this, &SoundFileBox::setup_fields));

	show_all();
	stop_btn.hide();
}

void
SoundFileBox::set_session(Session* s)
{
	_session = s;

	if (!_session) {
		play_btn.set_sensitive(false);
	} else {
		_session->AuditionActive.connect(mem_fun (*this, &SoundFileBox::audition_status_changed));
	}
}

bool
SoundFileBox::setup_labels (string filename) 
{
	path = filename;

	string error_msg;
	if(!AudioFileSource::get_soundfile_info (filename, sf_info, error_msg)) {
		return false;
	}

	length.set_alignment (0.0f, 0.0f);
	length.set_text (string_compose("Length: %1", PBD::length2string(sf_info.length, sf_info.samplerate)));

	format.set_alignment (0.0f, 0.0f);
	format.set_text (sf_info.format_name);

	channels.set_alignment (0.0f, 0.0f);
	channels.set_text (string_compose("Channels: %1", sf_info.channels));

	samplerate.set_alignment (0.0f, 0.0f);
	samplerate.set_text (string_compose("Samplerate: %1", sf_info.samplerate));

	setup_fields ();

	return true;
}

void
SoundFileBox::setup_fields ()
{
	ENSURE_GUI_THREAD(mem_fun (*this, &SoundFileBox::setup_fields));

	fields->clear ();

	vector<string> field_list;
	Library->get_fields(field_list);

	vector<string>::iterator i;
	Gtk::TreeModel::iterator iter;
	Gtk::TreeModel::Row row;
	for (i = field_list.begin(); i != field_list.end(); ++i) {
		if (!(*i == _("channels") || *i == _("samplerate") ||
			*i == _("resolution") || *i == _("format"))) {
			iter = fields->append();
			row = *iter;

			string value = Library->get_field(path, *i);
			row[label_columns.field] = *i;
			row[label_columns.data]  = value;
		}
	}
}

void
SoundFileBox::play_btn_clicked ()
{
	if (!_session) {
		return;
	}

	_session->cancel_audition();

	if (access(path.c_str(), R_OK)) {
		warning << string_compose(_("Could not read file: %1 (%2)."), path, strerror(errno)) << endmsg;
		return;
	}

	static std::map<string, AudioRegion*> region_cache;

	if (region_cache.find (path) == region_cache.end()) {
		AudioRegion::SourceList srclist;
		AudioFileSource* afs;

		for (int n = 0; n < sf_info.channels; ++n) {
			try {
				afs = AudioFileSource::create (path+":"+string_compose("%1", n));
				srclist.push_back(afs);

			} catch (failed_constructor& err) {
				error << _("Could not access soundfile: ") << path << endmsg;
				return;
			}
		}

		if (srclist.empty()) {
			return;
		}

		string result;
		_session->region_name (result, Glib::path_get_basename(srclist[0]->name()), false);
		AudioRegion* a_region = new AudioRegion(srclist, 0, srclist[0]->length(), result, 0, Region::DefaultFlags, false);
		region_cache[path] = a_region;
	}

	play_btn.hide();
	stop_btn.show();

	_session->audition_region(*region_cache[path]);
}

void
SoundFileBox::stop_btn_clicked ()
{
	if (_session) {
		_session->cancel_audition();
		play_btn.show();
		stop_btn.hide();
	}
}

void
SoundFileBox::add_field_clicked ()
{
    ArdourPrompter prompter (true);
    string name;

    prompter.set_prompt (_("Name for Field"));
    prompter.add_button (Gtk::Stock::ADD, Gtk::RESPONSE_ACCEPT);
    prompter.set_response_sensitive (Gtk::RESPONSE_ACCEPT, false);

    switch (prompter.run ()) {
		case Gtk::RESPONSE_ACCEPT:
	        prompter.get_result (name);
			if (name.length()) {
				Library->add_field (name);
				Library->save_changes ();
			}
	        break;

	    default:
	        break;
    }
}

void
SoundFileBox::remove_field_clicked ()
{
	field_view.get_selection()->selected_foreach_iter(mem_fun(*this, &SoundFileBox::delete_row));

	Library->save_changes ();
}

void
SoundFileBox::field_edited (const Glib::ustring& str1, const Glib::ustring& str2)
{
	cout << "field_edited" << endl;
	Library->save_changes ();
}

void
SoundFileBox::delete_row (const Gtk::TreeModel::iterator& iter)
{
	Gtk::TreeModel::Row row = *iter;

	Library->remove_field(row[label_columns.field]);
}

void
SoundFileBox::audition_status_changed (bool active)
{
	ENSURE_GUI_THREAD(bind (mem_fun (*this, &SoundFileBox::audition_status_changed), active));
	
	if (!active) {
		stop_btn_clicked ();
	}
}

void
SoundFileBox::field_selected ()
{
	if (field_view.get_selection()->count_selected_rows()) {
		remove_field_btn.set_sensitive(true);
	} else {
		remove_field_btn.set_sensitive(false);
	}
}

SoundFileBrowser::SoundFileBrowser (string title, ARDOUR::Session* s)
	: ArdourDialog (title, false),
	  chooser (Gtk::FILE_CHOOSER_ACTION_OPEN)
{
	get_vbox()->pack_start(chooser);
	chooser.set_preview_widget(preview);
	chooser.set_select_multiple (true);

	chooser.signal_update_preview().connect(mem_fun(*this, &SoundFileBrowser::update_preview));

	set_session (s);
}

void
SoundFileBrowser::set_session (Session* s)
{
	preview.set_session(s);
}

void
SoundFileBrowser::update_preview ()
{
	chooser.set_preview_widget_active(preview.setup_labels(chooser.get_filename()));
}

SoundFileChooser::SoundFileChooser (string title, ARDOUR::Session* s)
	:
	SoundFileBrowser(title, s)
{
	add_button (Gtk::Stock::OPEN, Gtk::RESPONSE_OK);
	add_button (Gtk::Stock::CANCEL, Gtk::RESPONSE_CANCEL);
	show_all ();
}

static const char *import_mode_strings[] = {
	X_("Add to Region list"),
	X_("Add as new Track(s)"),
	X_("Add to selected Track(s)"),
	0
};

vector<string> SoundFileOmega::mode_strings;

SoundFileOmega::SoundFileOmega (string title, ARDOUR::Session* s)
	: SoundFileBrowser (title, s),
	  split_check (_("Split Channels"))
{
	if (mode_strings.empty()) {
		mode_strings = PBD::internationalize (import_mode_strings);
	}

	ARDOUR_UI::instance()->tooltips().set_tip(split_check, 
			_("Create a region for each channel"));

	Gtk::Button* btn = add_button (_("Embed"), ResponseEmbed);
	ARDOUR_UI::instance()->tooltips().set_tip(*btn, 
			_("Link to an external file"));

	btn = add_button (_("Import"), ResponseImport);
	ARDOUR_UI::instance()->tooltips().set_tip(*btn, 
			_("Copy a file to the session folder"));

	add_button (Gtk::Stock::CLOSE, Gtk::RESPONSE_CLOSE);

	Gtk::HBox *box = manage (new Gtk::HBox());

	Gtkmm2ext::set_popdown_strings (mode_combo, mode_strings);

	set_mode (Editing::ImportAsRegion);

	box->pack_start (split_check);
	box->pack_start (mode_combo);

	mode_combo.signal_changed().connect (mem_fun (*this, &SoundFileOmega::mode_changed));

	chooser.set_extra_widget (*box);
	
	show_all ();
}

bool
SoundFileOmega::get_split ()
{
	return split_check.get_active();
}

vector<Glib::ustring>
SoundFileOmega::get_paths ()
{
	return chooser.get_filenames();
}

void
SoundFileOmega::set_mode (Editing::ImportMode mode)
{
	mode_combo.set_active_text (mode_strings[(int)mode]);

	switch (mode) {
	case Editing::ImportAsRegion:
		split_check.set_sensitive (true);
		break;
	case Editing::ImportAsTrack:
		split_check.set_sensitive (true);
		break;
	case Editing::ImportToTrack:
		split_check.set_sensitive (false);
		break;
	}
}

Editing::ImportMode
SoundFileOmega::get_mode ()
{
	vector<string>::iterator i;
	uint32_t n;
	string str = mode_combo.get_active_text ();

	for (n = 0, i = mode_strings.begin (); i != mode_strings.end(); ++i, ++n) {
		if (str == (*i)) {
			break;
		}
	}

	if (i == mode_strings.end()) {
		fatal << string_compose (_("programming error: %1"), X_("unknown import mode string")) << endmsg;
		/*NOTREACHED*/
	}

	return (Editing::ImportMode) (n);
}

void
SoundFileOmega::mode_changed ()
{
	Editing::ImportMode mode = get_mode();

	switch (mode) {
	case Editing::ImportAsRegion:
		split_check.set_sensitive (true);
		break;
	case Editing::ImportAsTrack:
		split_check.set_sensitive (true);
		break;
	case Editing::ImportToTrack:
		split_check.set_sensitive (false);
		break;
	}
}