worldspawn/plugins/entity/rotation.h

/*
   Copyright (C) 2001-2006, William Joseph.
   All Rights Reserved.

   This file is part of GtkRadiant.

   GtkRadiant 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.

   GtkRadiant 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 GtkRadiant; if not, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#if !defined( INCLUDED_ROTATION_H )
#define INCLUDED_ROTATION_H

#include "ientity.h"

#include "stream/stringstream.h"
#include "math/quaternion.h"
#include "generic/callback.h"
#include "stringio.h"

#include "angle.h"

typedef float Float9[9];

inline void default_rotation(Float9 rotation)
{
	rotation[0] = 1;
	rotation[1] = 0;
	rotation[2] = 0;
	rotation[3] = 0;
	rotation[4] = 1;
	rotation[5] = 0;
	rotation[6] = 0;
	rotation[7] = 0;
	rotation[8] = 1;
}

inline void write_rotation(const Float9 rotation, Entity *entity, const char *key = "rotation")
{
	if (rotation[0] == 1
	    && rotation[1] == 0
	    && rotation[2] == 0
	    && rotation[3] == 0
	    && rotation[4] == 1
	    && rotation[5] == 0
	    && rotation[6] == 0
	    && rotation[7] == 0
	    && rotation[8] == 1) {
		entity->setKeyValue(key, "");
	} else {
		StringOutputStream value(256);
		value << rotation[0] << ' '
		      << rotation[1] << ' '
		      << rotation[2] << ' '
		      << rotation[3] << ' '
		      << rotation[4] << ' '
		      << rotation[5] << ' '
		      << rotation[6] << ' '
		      << rotation[7] << ' '
		      << rotation[8];
		entity->setKeyValue(key, value.c_str());
	}
}

inline void read_rotation(Float9 rotation, const char *value)
{
	if (!string_parse_vector(value, rotation, rotation + 9)) {
		default_rotation(rotation);
	}
}

inline Matrix4 rotation_toMatrix(const Float9 rotation)
{
	return Matrix4(
		rotation[0],
		rotation[1],
		rotation[2],
		0,
		rotation[3],
		rotation[4],
		rotation[5],
		0,
		rotation[6],
		rotation[7],
		rotation[8],
		0,
		0,
		0,
		0,
		1
		);
}

inline void rotation_fromMatrix(Float9 rotation, const Matrix4 &matrix)
{
	rotation[0] = matrix.xx();
	rotation[1] = matrix.xy();
	rotation[2] = matrix.xz();
	rotation[3] = matrix.yx();
	rotation[4] = matrix.yy();
	rotation[5] = matrix.yz();
	rotation[6] = matrix.zx();
	rotation[7] = matrix.zy();
	rotation[8] = matrix.zz();
}

inline void rotation_assign(Float9 rotation, const Float9 other)
{
	rotation[0] = other[0];
	rotation[1] = other[1];
	rotation[2] = other[2];
	rotation[3] = other[3];
	rotation[4] = other[4];
	rotation[5] = other[5];
	rotation[6] = other[6];
	rotation[7] = other[7];
	rotation[8] = other[8];
}

inline void rotation_rotate(Float9 rotation, const Quaternion &rotate)
{
	rotation_fromMatrix(rotation,
	                    matrix4_multiplied_by_matrix4(
				    rotation_toMatrix(rotation),
				    matrix4_rotation_for_quaternion_quantised(rotate)
				    )
	                    );
}

inline void read_angle(Float9 rotation, const char *value)
{
	float angle;
	if (!string_parse_float(value, angle)) {
		default_rotation(rotation);
	} else {
		rotation_fromMatrix(rotation, matrix4_rotation_for_z_degrees(angle));
	}
}

class RotationKey {
Callback<void()> m_rotationChanged;
public:
Float9 m_rotation;


RotationKey(const Callback<void()> &rotationChanged)
	: m_rotationChanged(rotationChanged)
{
	default_rotation(m_rotation);
}

void angleChanged(const char *value)
{
	read_angle(m_rotation, value);
	m_rotationChanged();
}

typedef MemberCaller<RotationKey, void (const char *), &RotationKey::angleChanged> AngleChangedCaller;

void rotationChanged(const char *value)
{
	read_rotation(m_rotation, value);
	m_rotationChanged();
}

typedef MemberCaller<RotationKey, void (const char *), &RotationKey::rotationChanged> RotationChangedCaller;

void write(Entity *entity) const
{
	Vector3 euler = matrix4_get_rotation_euler_xyz_degrees(rotation_toMatrix(m_rotation));
	if (euler[0] == 0 && euler[1] == 0) {
		entity->setKeyValue("rotation", "");
		write_angle(euler[2], entity);
	} else {
		entity->setKeyValue("angle", "");
		write_rotation(m_rotation, entity);
	}
}
};

#endif