netradiant-custom/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
 */

#pragma once

#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
	{
		const auto value = StringStream(
			rotation[0], ' ',
			rotation[1], ' ',
			rotation[2], ' ',
			rotation[3], ' ',
			rotation[4], ' ',
			rotation[5], ' ',
			rotation[6], ' ',
			rotation[7], ' ',
			rotation[8] );
		entity->setKeyValue( key, value );
	}
}
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 m_rotationChanged;
public:
	Float9 m_rotation;


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

	void angleChanged( const char* value ){
		read_angle( m_rotation, value );
		m_rotationChanged();
	}
	typedef MemberCaller1<RotationKey, const char*, &RotationKey::angleChanged> AngleChangedCaller;

	void rotationChanged( const char* value ){
		read_rotation( m_rotation, value );
		m_rotationChanged();
	}
	typedef MemberCaller1<RotationKey, 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 );
		}
	}
};