/*
 * Copyright (C) 2020 Luciano Iam <lucianito@gmail.com>
 *
 * 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.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include "ardour/meter.h"
#include "ardour/plugin_insert.h"
#include "ardour/session.h"
#include "ardour/tempo.h"

#include "feedback.h"
#include "globals.h"
#include "server.h"
#include "state.h"
#include "strips.h"

using namespace ARDOUR;

struct TempoObserver {
	void operator() (ArdourFeedback* p)
	{
		p->update_all (Node::tempo, p->globals ().tempo ());
	}
};

struct StripGainObserver {
	void operator() (ArdourFeedback* p, uint32_t strip_n)
	{
		// fires multiple times (4x as of ardour 6.0)
		p->update_all (Node::strip_gain, strip_n, p->strips ().strip_gain (strip_n));
	}
};

struct StripPanObserver {
	void operator() (ArdourFeedback* p, uint32_t strip_n)
	{
		p->update_all (Node::strip_pan, strip_n, p->strips ().strip_pan (strip_n));
	}
};

struct StripMuteObserver {
	void operator() (ArdourFeedback* p, uint32_t strip_n)
	{
		p->update_all (Node::strip_mute, strip_n, p->strips ().strip_mute (strip_n));
	}
};

struct PluginBypassObserver {
	void operator() (ArdourFeedback* p, uint32_t strip_n, uint32_t plugin_n)
	{
		p->update_all (Node::strip_plugin_enable, strip_n, plugin_n,
		               p->strips ().strip_plugin_enabled (strip_n, plugin_n));
	}
};

struct PluginParamValueObserver {
	void operator() (ArdourFeedback* p, uint32_t strip_n, uint32_t plugin_n,
	                 uint32_t param_n, boost::weak_ptr<AutomationControl> ctrl)
	{
		boost::shared_ptr<AutomationControl> control = ctrl.lock ();
		if (!control) {
			return;
		}
		p->update_all (Node::strip_plugin_param_value, strip_n, plugin_n, param_n,
		               ArdourStrips::plugin_param_value (control));
	}
};

int
ArdourFeedback::start ()
{
	observe_globals ();
	observe_strips ();

	// some things need polling like the strip meters
	Glib::RefPtr<Glib::TimeoutSource> periodic_timeout = Glib::TimeoutSource::create (100); // ms
	_periodic_connection                               = periodic_timeout->connect (sigc::mem_fun (*this,
                                                                         &ArdourFeedback::poll));
	periodic_timeout->attach (main_loop ()->get_context ());

	return 0;
}

int
ArdourFeedback::stop ()
{
	_periodic_connection.disconnect ();
	_signal_connections.drop_connections ();
	return 0;
}

void
ArdourFeedback::update_all (std::string node, TypedValue value) const
{
	update_all (node, ADDR_NONE, ADDR_NONE, ADDR_NONE, value);
}

void
ArdourFeedback::update_all (std::string node, uint32_t strip_n, TypedValue value) const
{
	update_all (node, strip_n, ADDR_NONE, ADDR_NONE, value);
}

void
ArdourFeedback::update_all (std::string node, uint32_t strip_n, uint32_t plugin_n,
                            TypedValue value) const
{
	update_all (node, strip_n, plugin_n, ADDR_NONE, value);
}

void
ArdourFeedback::update_all (std::string node, uint32_t strip_n, uint32_t plugin_n, uint32_t param_n,
                            TypedValue value) const
{
	AddressVector addr = AddressVector ();

	if (strip_n != ADDR_NONE) {
		addr.push_back (strip_n);
	}

	if (plugin_n != ADDR_NONE) {
		addr.push_back (plugin_n);
	}

	if (param_n != ADDR_NONE) {
		addr.push_back (param_n);
	}

	ValueVector val = ValueVector ();
	val.push_back (value);

	server ().update_all_clients (NodeState (node, addr, val), false);
}

bool
ArdourFeedback::poll () const
{
	for (uint32_t strip_n = 0; strip_n < strips ().strip_count (); ++strip_n) {
		// meters
		boost::shared_ptr<Stripable> strip = strips ().nth_strip (strip_n);
		boost::shared_ptr<PeakMeter> meter = strip->peak_meter ();
		float                        db    = meter ? meter->meter_level (0, MeterMCP) : -193;
		update_all (Node::strip_meter, strip_n, static_cast<double> (db));
	}

	return true;
}

void
ArdourFeedback::observe_globals ()
{
	session ().tempo_map ().PropertyChanged.connect (_signal_connections, MISSING_INVALIDATOR,
	                                                 boost::bind<void> (TempoObserver (), this), event_loop ());
}

void
ArdourFeedback::observe_strips ()
{
	for (uint32_t strip_n = 0; strip_n < strips ().strip_count (); ++strip_n) {
		boost::shared_ptr<Stripable> strip = strips ().nth_strip (strip_n);

		strip->gain_control ()->Changed.connect (_signal_connections, MISSING_INVALIDATOR,
		                                         boost::bind<void> (StripGainObserver (), this, strip_n), event_loop ());

		if (strip->pan_azimuth_control ()) {
			strip->pan_azimuth_control ()->Changed.connect (_signal_connections, MISSING_INVALIDATOR,
			                                                boost::bind<void> (StripPanObserver (), this, strip_n), event_loop ());
		}

		strip->mute_control ()->Changed.connect (_signal_connections, MISSING_INVALIDATOR,
		                                         boost::bind<void> (StripMuteObserver (), this, strip_n), event_loop ());

		observe_strip_plugins (strip_n, strip);
	}
}

void
ArdourFeedback::observe_strip_plugins (uint32_t strip_n, boost::shared_ptr<ARDOUR::Stripable> strip)
{
	for (uint32_t plugin_n = 0;; ++plugin_n) {
		boost::shared_ptr<PluginInsert> insert = strips ().strip_plugin_insert (strip_n, plugin_n);
		if (!insert) {
			break;
		}

		uint32_t                             bypass  = insert->plugin ()->designated_bypass_port ();
		Evoral::Parameter                    param   = Evoral::Parameter (PluginAutomation, 0, bypass);
		boost::shared_ptr<AutomationControl> control = insert->automation_control (param);

		if (control) {
			control->Changed.connect (_signal_connections, MISSING_INVALIDATOR,
			                          boost::bind<void> (PluginBypassObserver (), this, strip_n, plugin_n), event_loop ());
		}

		observe_strip_plugin_param_values (strip_n, plugin_n, insert);
	}
}

void
ArdourFeedback::observe_strip_plugin_param_values (uint32_t strip_n,
                                                   uint32_t plugin_n, boost::shared_ptr<ARDOUR::PluginInsert> insert)
{
	boost::shared_ptr<Plugin> plugin = insert->plugin ();

	for (uint32_t param_n = 0; param_n < plugin->parameter_count (); ++param_n) {
		boost::shared_ptr<AutomationControl> control = strips ().strip_plugin_param_control (
		    strip_n, plugin_n, param_n);

		if (!control) {
			continue;
		}

		control->Changed.connect (_signal_connections, MISSING_INVALIDATOR,
		                          boost::bind<void> (PluginParamValueObserver (), this, strip_n, plugin_n, param_n,
		                                             boost::weak_ptr<AutomationControl>(control)),
		                          event_loop ());
	}
}