// 2009 © Václav Šmilauer <eudoxos@arcig.cz>
#include <core/State.hpp>

namespace yade { // Cannot have #include directive inside.

CREATE_LOGGER(State);
CREATE_LOGGER(ThermalState);

void State::setDOFfromVector3r(Vector3r disp, Vector3r rot)
{
	blockedDOFs = ((disp[0] == 1.0) ? DOF_X : 0) | ((disp[1] == 1.0) ? DOF_Y : 0) | ((disp[2] == 1.0) ? DOF_Z : 0) | ((rot[0] == 1.0) ? DOF_RX : 0)
	        | ((rot[1] == 1.0) ? DOF_RY : 0) | ((rot[2] == 1.0) ? DOF_RZ : 0);
}

std::string State::blockedDOFs_vec_get() const
{
	std::string ret;
#define _SET_DOF(DOF_ANY, ch)                                                                                                                                  \
	if ((blockedDOFs & State::DOF_ANY) != 0) ret.push_back(ch);
	_SET_DOF(DOF_X, 'x');
	_SET_DOF(DOF_Y, 'y');
	_SET_DOF(DOF_Z, 'z');

	_SET_DOF(DOF_RX, 'X');
	_SET_DOF(DOF_RY, 'Y');
	_SET_DOF(DOF_RZ, 'Z');
#undef _SET_DOF
	return ret;
}

void State::blockedDOFs_vec_set(const std::string& dofs)
{
	blockedDOFs = 0;
	for (char c : dofs) {
#define _GET_DOF(DOF_ANY, ch)                                                                                                                                  \
	if (c == ch) {                                                                                                                                         \
		blockedDOFs |= State::DOF_ANY;                                                                                                                 \
		continue;                                                                                                                                      \
	}
		_GET_DOF(DOF_X, 'x');
		_GET_DOF(DOF_Y, 'y');
		_GET_DOF(DOF_Z, 'z');
		_GET_DOF(DOF_RX, 'X');
		_GET_DOF(DOF_RY, 'Y');
		_GET_DOF(DOF_RZ, 'Z');
#undef _GET_DOF
	}
}

} // namespace yade
