/****************************************************************************
 *
 *   Copyright (c) 2023 PX4 Development Team. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *	notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *	notice, this list of conditions and the following disclaimer in
 *	the documentation and/or other materials provided with the
 *	distribution.
 * 3. Neither the name PX4 nor the names of its contributors may be
 *	used to endorse or promote products derived from this software
 *	without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 ****************************************************************************/

#include "GZMixingInterfaceServo.hpp"

bool GZMixingInterfaceServo::init(const std::string &model_name)
{
	// /model/rascal_110_0/servo_2
	for (int i = 0; i < 8; i++) {
		std::string joint_name = "servo_" + std::to_string(i);
		std::string servo_topic = "/model/" + model_name + "/" + joint_name;
		//std::cout << "Servo topic: " << servo_topic << std::endl;
		_servos_pub.push_back(_node.Advertise<gz::msgs::Double>(servo_topic));

		if (!_servos_pub.back().Valid()) {
			PX4_ERR("failed to advertise %s", servo_topic.c_str());
			return false;
		}
	}

	ScheduleNow();

	return true;
}

bool GZMixingInterfaceServo::updateOutputs(bool stop_motors, uint16_t outputs[MAX_ACTUATORS], unsigned num_outputs,
		unsigned num_control_groups_updated)
{
	bool updated = false;
	// cmd.command_value = (float)outputs[i] / 500.f - 1.f; // [-1, 1]

	int i = 0;

	for (auto &servo_pub : _servos_pub) {
		if (_mixing_output.isFunctionSet(i)) {
			gz::msgs::Double servo_output;
			///TODO: Normalize output data
			double output = (outputs[i] - 500) / 500.0;
			// std::cout << "outputs[" << i << "]: " << outputs[i] << std::endl;
			// std::cout << "  output: " << output << std::endl;
			servo_output.set_data(output);

			if (servo_pub.Valid()) {
				servo_pub.Publish(servo_output);
				updated = true;
			}
		}

		i++;
	}

	return updated;
}

void GZMixingInterfaceServo::Run()
{
	pthread_mutex_lock(&_node_mutex);
	_mixing_output.update();
	_mixing_output.updateSubscriptions(false);
	pthread_mutex_unlock(&_node_mutex);
}