path: root/apps/position_estimator_inav/position_estimator_inav_main.c

/****************************************************************************
 *
 *   Copyright (C) 2008-2012 PX4 Development Team. All rights reserved.
 *   Author: 	Damian Aregger	<daregger@student.ethz.ch>
 *   			Tobias Naegeli <naegelit@student.ethz.ch>
 * 				Lorenz Meier <lm@inf.ethz.ch>
 *
 * 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.
 *
 ****************************************************************************/

/**
 * @file position_estimator_inav_main.c
 * Model-identification based position estimator for multirotors
 */

#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <stdbool.h>
#include <fcntl.h>
#include <float.h>
#include <string.h>
#include <nuttx/config.h>
#include <nuttx/sched.h>
#include <sys/prctl.h>
#include <termios.h>
#include <errno.h>
#include <limits.h>
#include <math.h>
#include <uORB/uORB.h>
#include <uORB/topics/sensor_combined.h>
#include <uORB/topics/parameter_update.h>
#include <uORB/topics/vehicle_attitude.h>
#include <uORB/topics/vehicle_status.h>
#include <uORB/topics/vehicle_local_position.h>
#include <uORB/topics/vehicle_gps_position.h>
#include <mavlink/mavlink_log.h>
#include <poll.h>
#include <systemlib/geo/geo.h>
#include <systemlib/systemlib.h>
#include <drivers/drv_hrt.h>

#include "position_estimator_inav_params.h"
//#include <uORB/topics/debug_key_value.h>
#include "sounds.h"
#include <drivers/drv_tone_alarm.h>
#include "kalman_filter_inertial.h"

static bool thread_should_exit = false; /**< Deamon exit flag */
static bool thread_running = false; /**< Deamon status flag */
static int position_estimator_inav_task; /**< Handle of deamon task / thread */

__EXPORT int position_estimator_inav_main(int argc, char *argv[]);

int position_estimator_inav_thread_main(int argc, char *argv[]);
/**
 * Print the correct usage.
 */
static void usage(const char *reason);

static void usage(const char *reason) {
	if (reason)
		fprintf(stderr, "%s\n", reason);
		fprintf(stderr,"usage: position_estimator_inav {start|stop|status} [-p <additional params>]\n\n");
	exit(1);
}

		/**
		 * The position_estimator_inav_thread only briefly exists to start
		 * the background job. The stack size assigned in the
		 * Makefile does only apply to this management task.
		 *
		 * The actual stack size should be set in the call
		 * to task_create().
		 */
int position_estimator_inav_main(int argc, char *argv[]) {
	if (argc < 1)
		usage("missing command");

	if (!strcmp(argv[1], "start")) {

		if (thread_running) {
			printf("position_estimator_inav already running\n");
			/* this is not an error */
			exit(0);
		}

		thread_should_exit = false;
		position_estimator_inav_task = task_spawn("position_estimator_inav",
				SCHED_RR, SCHED_PRIORITY_MAX - 5, 4096,
				position_estimator_inav_thread_main,
				(argv) ? (const char **) &argv[2] : (const char **) NULL );
		exit(0);
	}
	if (!strcmp(argv[1], "stop")) {
		thread_should_exit = true;
		exit(0);
	}

	if (!strcmp(argv[1], "status")) {
		if (thread_running) {
			printf("\tposition_estimator_inav is running\n");
		} else {
			printf("\tposition_estimator_inav not started\n");
		}
		exit(0);
	}

	usage("unrecognized command");
	exit(1);
}

/****************************************************************************
 * main
 ****************************************************************************/
int position_estimator_inav_thread_main(int argc, char *argv[]) {
	/* welcome user */
	printf("[position_estimator_inav] started\n");
	static int mavlink_fd;
	mavlink_fd = open(MAVLINK_LOG_DEVICE, 0);
	mavlink_log_info(mavlink_fd, "[position_estimator_inav] started");

	/* initialize values */
	static float k[3][2] = { { 0.0f, 0.0f }, { 0.0f, 0.0f }, { 0.0f, 0.0f } };
	static float x_est[3] = { 0.0f, 0.0f, 0.0f };
	static float y_est[3] = { 0.0f, 0.0f, 0.0f };
	static float z_est[3] = { 0.0f, 0.0f, 0.0f };
	const static float dT_const_120 = 1.0f / 120.0f;
	const static float dT2_const_120 = 1.0f / 120.0f / 120.0f / 2.0f;

	int baro_loop_cnt = 0;
	int baro_loop_end = 70; /* measurement for 1 second */
	float baro_alt0 = 0.0f; /* to determin while start up */
	float rho0 = 1.293f; /* standard pressure */
	const static float const_earth_gravity = 9.81f;

	static double lat_current = 0.0d; //[°]] --> 47.0
	static double lon_current = 0.0d; //[°]] -->8.5
	static double alt_current = 0.0d; //[m] above MSL

	/* declare and safely initialize all structs */
	struct vehicle_status_s vehicle_status;
	memset(&vehicle_status, 0, sizeof(vehicle_status)); /* make sure that baroINITdone = false */
	struct sensor_combined_s sensor;
	memset(&sensor, 0, sizeof(sensor));
	struct vehicle_gps_position_s gps;
	memset(&gps, 0, sizeof(gps));
	struct vehicle_attitude_s att;
	memset(&att, 0, sizeof(att));
	struct vehicle_local_position_s local_pos_est;
	memset(&local_pos_est, 0, sizeof(local_pos_est));

	/* subscribe */
	int parameter_update_sub = orb_subscribe(ORB_ID(parameter_update));
	int vehicle_status_sub = orb_subscribe(ORB_ID(vehicle_status));
	int sensor_combined_sub = orb_subscribe(ORB_ID(sensor_combined));
	int vehicle_attitude_sub = orb_subscribe(ORB_ID(vehicle_attitude));
	int vehicle_gps_position_sub = orb_subscribe(ORB_ID(vehicle_gps_position));

	/* advertise */
	orb_advert_t local_pos_est_pub = orb_advertise(
			ORB_ID(vehicle_local_position), &local_pos_est);

	struct position_estimator_inav_params pos_inav_params;
	struct position_estimator_inav_param_handles pos_inav_param_handles;
	/* initialize parameter handles */
	parameters_init(&pos_inav_param_handles);

	bool local_flag_baroINITdone = false; /* in any case disable baroINITdone */
	/* FIRST PARAMETER READ at START UP*/
	struct parameter_update_s param_update;
	orb_copy(ORB_ID(parameter_update), parameter_update_sub, &param_update); /* read from param to clear updated flag */
	/* FIRST PARAMETER UPDATE */
	parameters_update(&pos_inav_param_handles, &pos_inav_params);
	/* END FIRST PARAMETER UPDATE */
	/* wait until gps signal turns valid, only then can we initialize the projection */
	struct pollfd fds_init[1] = { { .fd = vehicle_gps_position_sub, .events =
			POLLIN } };

	while (gps.fix_type < 3) {

		/* wait for GPS updates, BUT READ VEHICLE STATUS (!)
		 * this choice is critical, since the vehicle status might not
		 * actually change, if this app is started after GPS lock was
		 * aquired.
		 */
		if (poll(fds_init, 1, 5000)) { /* poll only two first subscriptions */
			if (fds_init[0].revents & POLLIN) {
				/* Wait for the GPS update to propagate (we have some time) */
				usleep(5000);
				orb_copy(ORB_ID(vehicle_gps_position), vehicle_gps_position_sub,
						&gps);
			}
		}
		static int printcounter = 0;
		if (printcounter == 100) {
			printcounter = 0;
			printf("[pos_est1D] wait for GPS fix type 3\n");
		}
		printcounter++;
	}

	/* get gps value for first initialization */
	orb_copy(ORB_ID(vehicle_gps_position), vehicle_gps_position_sub, &gps);
	lat_current = ((double) (gps.lat)) * 1e-7;
	lon_current = ((double) (gps.lon)) * 1e-7;
	alt_current = gps.alt * 1e-3;
	/* initialize coordinates */
	map_projection_init(lat_current, lon_current);
	/* publish global position messages only after first GPS message */
	printf("[pos_est1D] initialized projection with: lat: %.10f,  lon:%.10f\n",
			lat_current, lon_current);
	uint64_t last_time = 0;
	thread_running = true;

	/** main loop */
	struct pollfd fds[5] = { { .fd = parameter_update_sub, .events = POLLIN }, {
			.fd = vehicle_status_sub, .events = POLLIN }, { .fd =
			vehicle_attitude_sub, .events = POLLIN }, { .fd =
			sensor_combined_sub, .events = POLLIN }, { .fd =
			vehicle_gps_position_sub, .events = POLLIN }, };
	while (!thread_should_exit) {
		int ret = poll(fds, 5, 20); //wait maximal this 20 ms = 50 Hz minimum rate
		if (ret < 0) {
			/* poll error */
		} else {
			/* parameter update */
			if (fds[0].revents & POLLIN) {
				/* read from param to clear updated flag */
				struct parameter_update_s update;
				orb_copy(ORB_ID(parameter_update), parameter_update_sub,
						&update);
				/* update parameters */
				parameters_update(&pos_inav_param_handles, &pos_inav_params);
				//r = pos_inav_params.r;
			}
			/* vehicle status */
			if (fds[1].revents & POLLIN) {
				orb_copy(ORB_ID(vehicle_status), vehicle_status_sub, &vehicle_status);
			}
			/* vehicle attitude */
			if (fds[2].revents & POLLIN) {
				orb_copy(ORB_ID(vehicle_attitude), vehicle_attitude_sub, &att);
			}
			/* sensor combined */
			if (fds[3].revents & POLLIN) {
				orb_copy(ORB_ID(sensor_combined), sensor_combined_sub,
						&sensor);
			}
			/* vehicle GPS position */
			if (fds[4].revents & POLLIN) {
				/* new GPS value */
				orb_copy(ORB_ID(vehicle_gps_position), vehicle_gps_position_sub,
						&gps);
				static float local_pos_gps[3] = { 0.0f, 0.0f, 0.0f }; /* output variables from tangent plane mapping */
				/* Project gps lat lon (Geographic coordinate system) to plane */
				map_projection_project(((double) (gps.lat)) * 1e-7,
						((double) (gps.lon)) * 1e-7, &(local_pos_gps[0]),
						&(local_pos_gps[1]));
				local_pos_gps[2] = (float) (gps.alt * 1e-3);
			}
			// barometric pressure estimation at start up
			if (!local_flag_baroINITdone) {
				// mean calculation over several measurements
				if (baro_loop_cnt < baro_loop_end) {
					baro_alt0 += sensor.baro_alt_meter;
					baro_loop_cnt++;
				} else {
					baro_alt0 /= (float) (baro_loop_cnt);
					local_flag_baroINITdone = true;
					char *baro_m_start = "barometer initialized with alt0 = ";
					char p0_char[15];
					sprintf(p0_char, "%8.2f", baro_alt0 / 100);
					char *baro_m_end = " m";
					char str[80];
					strcpy(str, baro_m_start);
					strcat(str, p0_char);
					strcat(str, baro_m_end);
					mavlink_log_info(mavlink_fd, str);
				}
			}
			/* TODO convert accelerations from UAV frame to NED frame */
			float accel_NED[3];
			accel_NED[2] = sensor.accelerometer_m_s2[2] + const_earth_gravity;
			/* prediction */
			kalman_filter_inertial_predict(dT_const_120, z_est);
			/* prepare vectors for kalman filter correction */
			float z_meas[2];	// pos, accel
			bool use_z[2] = { false, true };
			if (local_flag_baroINITdone) {
				z_meas[0] = sensor.baro_alt_meter - baro_alt0;
				use_z[0] = true;
			}
			z_meas[1] = accel_NED[2];
			/* correction */
			kalman_filter_inertial_update(z_est, z_meas, k, use_z);
			printf("[pos_est_inav] att = %.3f, %.3f, %.3f, %.3f\n", att.q[0], att.q[1], att.q[2], att.q[3]);
			printf("[pos_est_inav] z = %.2f, vz = %.2f, az = %.2f\n", z_est[0], z_est[1], z_est[2]);

			local_pos_est.x = 0.0;
			local_pos_est.vx = 0.0;
			local_pos_est.y = 0.0;
			local_pos_est.vy = 0.0;
			local_pos_est.z = z_est[0];
			local_pos_est.vz = z_est[1];
			local_pos_est.timestamp = hrt_absolute_time();
			if ((isfinite(local_pos_est.x)) && (isfinite(local_pos_est.vx))
					&& (isfinite(local_pos_est.y))
					&& (isfinite(local_pos_est.vy))
					&& (isfinite(local_pos_est.z))
					&& (isfinite(local_pos_est.vz))) {
				orb_publish(ORB_ID(
						vehicle_local_position), local_pos_est_pub,
						&local_pos_est);
			}
		} /* end of poll return value check */
	}

	printf("[pos_est_inav] exiting.\n");
	mavlink_log_info(mavlink_fd, "[pos_est_inav] exiting");
	thread_running = false;
	return 0;
}