godot/core/method_ptrcall.h

#ifndef METHOD_PTRCALL_H
#define METHOD_PTRCALL_H

#include "typedefs.h"
#include "math_2d.h"
#include "variant.h"

#ifdef PTRCALL_ENABLED

template<class T>
struct PtrToArg {

};

#define MAKE_PTRARG(m_type) \
template<>\
struct PtrToArg<m_type> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		return *reinterpret_cast<const m_type*>(p_ptr);\
	}\
	_FORCE_INLINE_ static void encode(m_type p_val, void* p_ptr) {\
		*((m_type*)p_ptr)=p_val;\
	}\
};\
template<>\
struct PtrToArg<const m_type&> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		return *reinterpret_cast<const m_type*>(p_ptr);\
	}\
	_FORCE_INLINE_ static void encode(m_type p_val, void* p_ptr) {\
		*((m_type*)p_ptr)=p_val;\
	}\
}


#define MAKE_PTRARGR(m_type,m_ret) \
template<>\
struct PtrToArg<m_type> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		return *reinterpret_cast<const m_type*>(p_ptr);\
	}\
	_FORCE_INLINE_ static void encode(m_type p_val, void* p_ptr) {\
		*((m_ret*)p_ptr)=p_val;\
	}\
};\
template<>\
struct PtrToArg<const m_type&> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		return *reinterpret_cast<const m_type*>(p_ptr);\
	}\
	_FORCE_INLINE_ static void encode(m_type p_val, void* p_ptr) {\
		*((m_ret*)p_ptr)=p_val;\
	}\
}



MAKE_PTRARG(bool);
MAKE_PTRARGR(uint8_t,int);
MAKE_PTRARGR(int8_t,int);
MAKE_PTRARGR(uint16_t,int);
MAKE_PTRARGR(int16_t,int);
MAKE_PTRARGR(uint32_t,int);
MAKE_PTRARGR(int32_t,int);
MAKE_PTRARGR(int64_t,int);
MAKE_PTRARGR(uint64_t,int);
MAKE_PTRARG(float);
MAKE_PTRARGR(double,float);

MAKE_PTRARG(String);
MAKE_PTRARG(Vector2);
MAKE_PTRARG(Rect2);
MAKE_PTRARG(Vector3);
MAKE_PTRARG(Transform2D);
MAKE_PTRARG(Plane);
MAKE_PTRARG(Quat);
MAKE_PTRARG(AABB);
MAKE_PTRARG(Basis);
MAKE_PTRARG(Transform);
MAKE_PTRARG(Color);
MAKE_PTRARG(Image);
MAKE_PTRARG(NodePath);
MAKE_PTRARG(RID);
MAKE_PTRARG(InputEvent);
MAKE_PTRARG(Dictionary);
MAKE_PTRARG(Array);
MAKE_PTRARG(ByteArray);
MAKE_PTRARG(IntArray);
MAKE_PTRARG(RealArray);
MAKE_PTRARG(StringArray);
MAKE_PTRARG(Vector2Array);
MAKE_PTRARG(Vector3Array);
MAKE_PTRARG(ColorArray);
MAKE_PTRARG(Variant);


//this is for Object

template<class T>
struct PtrToArg< T* > {

	_FORCE_INLINE_ static T* convert(const void* p_ptr) {

		return const_cast<T*>(reinterpret_cast<const T*>(p_ptr));
	}

	_FORCE_INLINE_ static void encode(T* p_var, void* p_ptr) {

		*((T**)p_ptr)=p_var;
	}

};

template<class T>
struct PtrToArg< const T* > {

	_FORCE_INLINE_ static const T* convert(const void* p_ptr) {

		return reinterpret_cast<const T*>(p_ptr);
	}

	_FORCE_INLINE_ static void encode(T* p_var, void* p_ptr) {

		*((T**)p_ptr)=p_var;
	}

};


//this is for the special cases used by Variant

#define MAKE_VECARG(m_type) \
template<>\
struct PtrToArg<Vector<m_type> > {\
	_FORCE_INLINE_ static Vector<m_type> convert(const void* p_ptr) {\
		const PoolVector<m_type> *dvs = reinterpret_cast<const PoolVector<m_type> *>(p_ptr);\
		Vector<m_type> ret;\
		int len = dvs->size();\
		ret.resize(len);\
		{\
			PoolVector<m_type>::Read r=dvs->read();\
			for(int i=0;i<len;i++) {\
				ret[i]=r[i];\
			}\
		}		\
		return ret;\
	}\
	_FORCE_INLINE_ static void encode(Vector<m_type> p_vec, void* p_ptr) {\
		PoolVector<m_type> *dv = reinterpret_cast<PoolVector<m_type> *>(p_ptr);\
		int len=p_vec.size();\
		dv->resize(len);\
		{\
			PoolVector<m_type>::Write w=dv->write();\
			for(int i=0;i<len;i++) {\
				w[i]=p_vec[i];\
			}\
		}	\
	}\
};\
template<>\
struct PtrToArg<const Vector<m_type>& > {\
	_FORCE_INLINE_ static Vector<m_type> convert(const void* p_ptr) {\
		const PoolVector<m_type> *dvs = reinterpret_cast<const PoolVector<m_type> *>(p_ptr);\
		Vector<m_type> ret;\
		int len = dvs->size();\
		ret.resize(len);\
		{\
			PoolVector<m_type>::Read r=dvs->read();\
			for(int i=0;i<len;i++) {\
				ret[i]=r[i];\
			}\
		}		\
		return ret;\
	}\
}

MAKE_VECARG(String);
MAKE_VECARG(uint8_t);
MAKE_VECARG(int);
MAKE_VECARG(float);
MAKE_VECARG(Vector2);
MAKE_VECARG(Vector3);
MAKE_VECARG(Color);

//for stuff that gets converted to Array vectors
#define MAKE_VECARR(m_type) \
template<>\
struct PtrToArg<Vector<m_type> > {\
	_FORCE_INLINE_ static Vector<m_type> convert(const void* p_ptr) {\
		const Array *arr = reinterpret_cast<const Array *>(p_ptr);\
		Vector<m_type> ret;\
		int len = arr->size();\
		ret.resize(len);\
		for(int i=0;i<len;i++) {\
			ret[i]=(*arr)[i];\
		}\
		return ret;\
	}\
	_FORCE_INLINE_ static void encode(Vector<m_type> p_vec, void* p_ptr) {\
		Array *arr = reinterpret_cast<Array *>(p_ptr);\
		int len = p_vec.size();\
		arr->resize(len);\
		for(int i=0;i<len;i++) {\
			(*arr)[i]=p_vec[i];\
		}\
	}	\
};\
template<>\
struct PtrToArg<const Vector<m_type>& > {\
	_FORCE_INLINE_ static Vector<m_type> convert(const void* p_ptr) {\
		const Array *arr = reinterpret_cast<const Array *>(p_ptr);\
		Vector<m_type> ret;\
		int len = arr->size();\
		ret.resize(len);\
		for(int i=0;i<len;i++) {\
			ret[i]=(*arr)[i];\
		}\
		return ret;\
	}\
}


MAKE_VECARR(Variant);
MAKE_VECARR(RID);
MAKE_VECARR(Plane);

#define MAKE_DVECARR(m_type) \
template<>\
struct PtrToArg<PoolVector<m_type> > {\
	_FORCE_INLINE_ static PoolVector<m_type> convert(const void* p_ptr) {\
		const Array *arr = reinterpret_cast<const Array *>(p_ptr);\
		PoolVector<m_type> ret;\
		int len = arr->size();\
		ret.resize(len);\
		{\
			PoolVector<m_type>::Write w=ret.write();\
			for(int i=0;i<len;i++) {\
				w[i]=(*arr)[i];\
			}\
		}\
		return ret;\
	}\
	_FORCE_INLINE_ static void encode(PoolVector<m_type> p_vec, void* p_ptr) {\
		Array *arr = reinterpret_cast<Array *>(p_ptr);\
		int len = p_vec.size();\
		arr->resize(len);\
		{\
			PoolVector<m_type>::Read r=p_vec.read();\
			for(int i=0;i<len;i++) {\
				(*arr)[i]=r[i];\
			}\
		}\
	}	\
};\
template<>\
struct PtrToArg<const PoolVector<m_type>& > {\
	_FORCE_INLINE_ static PoolVector<m_type> convert(const void* p_ptr) {\
	const Array *arr = reinterpret_cast<const Array *>(p_ptr);\
	PoolVector<m_type> ret;\
	int len = arr->size();\
	ret.resize(len);\
	{\
		PoolVector<m_type>::Write w=ret.write();\
		for(int i=0;i<len;i++) {\
			w[i]=(*arr)[i];\
		}\
	}\
	return ret;\
	}\
}

MAKE_DVECARR(Plane);
//for special case StringName

#define MAKE_STRINGCONV(m_type) \
template<>\
struct PtrToArg<m_type> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		m_type s = *reinterpret_cast<const String*>(p_ptr);\
		return s;\
	}\
	_FORCE_INLINE_ static void encode(m_type p_vec, void* p_ptr) {\
		String *arr = reinterpret_cast<String *>(p_ptr);\
		*arr=p_vec;\
	}\
};\
\
template<>\
struct PtrToArg<const m_type&> {\
	_FORCE_INLINE_ static m_type convert(const void* p_ptr) {\
		m_type s = *reinterpret_cast<const String*>(p_ptr);\
		return s;\
	}\
}

MAKE_STRINGCONV(StringName);
MAKE_STRINGCONV(IP_Address);

template<>
struct PtrToArg<PoolVector<Face3> > {
	_FORCE_INLINE_ static PoolVector<Face3> convert(const void* p_ptr) {
		const PoolVector<Vector3> *dvs = reinterpret_cast<const PoolVector<Vector3> *>(p_ptr);
		PoolVector<Face3> ret;
		int len = dvs->size()/3;
		ret.resize(len);
		{
			PoolVector<Vector3>::Read r=dvs->read();
			PoolVector<Face3>::Write w=ret.write();
			for(int i=0;i<len;i++) {
				w[i].vertex[0]=r[i*3+0];
				w[i].vertex[1]=r[i*3+1];
				w[i].vertex[2]=r[i*3+2];
			}
		}
		return ret;
	}
	_FORCE_INLINE_ static void encode(PoolVector<Face3> p_vec, void* p_ptr) {\
		PoolVector<Vector3> *arr = reinterpret_cast<PoolVector<Vector3> *>(p_ptr);\
		int len = p_vec.size();\
		arr->resize(len*3);\
		{\
			PoolVector<Face3>::Read r=p_vec.read();\
			PoolVector<Vector3>::Write w=arr->write();\
			for(int i=0;i<len;i++) {\
				w[i*3+0]=r[i].vertex[0];\
				w[i*3+1]=r[i].vertex[1];\
				w[i*3+2]=r[i].vertex[2];\
			}\
		}\
	}	\
};
template<>
struct PtrToArg<const PoolVector<Face3>& > {
	_FORCE_INLINE_ static PoolVector<Face3> convert(const void* p_ptr) {
		const PoolVector<Vector3> *dvs = reinterpret_cast<const PoolVector<Vector3> *>(p_ptr);
		PoolVector<Face3> ret;
		int len = dvs->size()/3;
		ret.resize(len);
		{
			PoolVector<Vector3>::Read r=dvs->read();
			PoolVector<Face3>::Write w=ret.write();
			for(int i=0;i<len;i++) {
				w[i].vertex[0]=r[i*3+0];
				w[i].vertex[1]=r[i*3+1];
				w[i].vertex[2]=r[i*3+2];
			}
		}
		return ret;
	}
};


#endif // METHOD_PTRCALL_H
#endif