Script Attachment shader/texture changing support

Added support for getting/setting/resetting Script Attachment model shaders/textures

Author: xr-lucy

gamedata/scripts/lua_help_ex.script

@@ -659,6 +659,10 @@
         function set_ui_position();
         function set_ui_rotation(vector);
         function set_ui_rotation(number, number, number);
+		function get_shaders()
+		function get_default_shaders()
+		function set_shader(number, string, string)
+		function reset_shader(number)
     }
 
     flags:

src/xrGame/script_attachment_manager.cpp

@@ -624,4 +624,126 @@ Fvector script_attachment::GetCenter()
 		return m_model->getVisData().sphere.P;
 
 	return { 0,0,0 };
+}
+
+luabind::object script_attachment::GetShaders()
+{
+	luabind::object table = luabind::newtable(ai().script_engine().lua());
+
+	if (!m_model)
+	{
+		table["error"] = true;
+		return table;
+	}
+
+	xr_vector<IRenderVisual*>* children = m_model->get_children();
+
+	if (!children)
+	{
+		luabind::object subtable = luabind::newtable(ai().script_engine().lua());
+		subtable["shader"] = m_model->getDebugShader();
+		subtable["texture"] = m_model->getDebugTexture();
+		table[1] = subtable;
+		return table;
+	}
+
+	int i = 1;
+
+	for (auto* child : *children)
+	{
+		luabind::object subtable = luabind::newtable(ai().script_engine().lua());
+		subtable["shader"] = child->getDebugShader();
+		subtable["texture"] = child->getDebugTexture();
+		table[i] = subtable;
+		++i;
+	}
+
+	return table;
+}
+
+luabind::object script_attachment::GetDefaultShaders()
+{
+	luabind::object table = luabind::newtable(ai().script_engine().lua());
+
+	if (!m_model)
+	{
+		table["error"] = true;
+		return table;
+	}
+
+	xr_vector<IRenderVisual*>* children = m_model->get_children();
+
+	if (!children)
+	{
+		luabind::object subtable = luabind::newtable(ai().script_engine().lua());
+		subtable["shader"] = m_model->getDebugShaderDef();
+		subtable["texture"] = m_model->getDebugTextureDef();
+		table[1] = subtable;
+		return table;
+	}
+
+	int i = 1;
+
+	for (auto* child : *children)
+	{
+		luabind::object subtable = luabind::newtable(ai().script_engine().lua());
+		subtable["shader"] = child->getDebugShaderDef();
+		subtable["texture"] = child->getDebugTextureDef();
+		table[i] = subtable;
+		++i;
+	}
+
+	return table;
+}
+
+void script_attachment::SetShaderTexture(int id, LPCSTR shader, LPCSTR texture)
+{
+	if (!m_model) return;
+	xr_vector<IRenderVisual*>* children = m_model->get_children();
+
+	if (!children)
+	{
+		m_model->SetShaderTexture(shader, texture);
+		return;
+	}
+
+	if (id == -1)
+	{
+		for (auto* child : *children)
+		{
+			child->SetShaderTexture(shader, texture);
+		}
+		return;
+	}
+
+	id--;
+
+	if (id >= 0 && children->size() > id)
+		children->at(id)->SetShaderTexture(shader, texture);
+}
+
+void script_attachment::ResetShaderTexture(int id)
+{
+	if (!m_model) return;
+	xr_vector<IRenderVisual*>* children = m_model->get_children();
+
+	if (!children)
+	{
+		m_model->ResetShaderTexture();
+		return;
+	}
+
+	if (id == -1)
+	{
+		for (auto* child : *children)
+		{
+			child->ResetShaderTexture();
+		}
+		return;
+	}
+
+	id--;
+
+	if (id >= 0 && children->size() > id)
+		children->at(id)->ResetShaderTexture();
 }

src/xrGame/script_attachment_manager.h

@@ -13,16 +13,16 @@ enum script_attachment_flags
 struct script_attachment_bone_cb
 {
 	u16 m_bone_id, m_attachment_bone_id;
-	luabind::functor<Fmatrix>* m_func;
+	::luabind::functor<Fmatrix>* m_func;
 	script_attachment* m_attachment;
 	Fmatrix m_mat;
 	bool m_overwrite;
 
-	script_attachment_bone_cb(const luabind::functor<Fmatrix>& func, script_attachment* att, u16 id, bool overwrite)
+	script_attachment_bone_cb(const ::luabind::functor<Fmatrix>& func, script_attachment* att, u16 id, bool overwrite)
 	{
 		m_attachment = att;
 		m_attachment_bone_id = id;
-		m_func = xr_new<luabind::functor<Fmatrix>>(func);
+		m_func = xr_new<::luabind::functor<Fmatrix>>(func);
 		m_mat = Fidentity;
 		m_bone_id = BI_NONE;
 		m_overwrite = overwrite;
@@ -111,7 +111,7 @@ class script_attachment
 	void SetParent(script_attachment* att);
 	void SetParent(CGameObject* obj);
 	void SetParent(CScriptGameObject* obj);
-	luabind::object GetParent();
+	::luabind::object GetParent();
 
 	void SetParentBone(u16 bone_id) { m_parent_bone = bone_id; }
 	u16 GetParentBone() { return m_parent_bone; }
@@ -149,13 +149,18 @@ class script_attachment
 
 	static void _BCL ScriptAttachmentBoneCallback(CBoneInstance* B);
 	void SetBoneCallback(u16 bone_id, u16 parent_bone, bool overwrite = false);
-	void SetBoneCallback(u16 bone_id, const luabind::functor<Fmatrix>& func, bool overwrite = false);
+	void SetBoneCallback(u16 bone_id, const ::luabind::functor<Fmatrix>& func, bool overwrite = false);
 	void RemoveBoneCallback(u16 bone_id);
 
 	Fmatrix GetBoneMatrix(u16 bone_id);
 	Fmatrix GetTransform() { return m_transform; };
 	Fvector GetCenter();
 
+	::luabind::object GetShaders();
+	::luabind::object GetDefaultShaders();
+	void SetShaderTexture(int id, LPCSTR shader, LPCSTR texture);
+	void ResetShaderTexture(int id);
+
 	DECLARE_SCRIPT_REGISTER_FUNCTION
 };
 

src/xrGame/script_attachment_script.cpp

@@ -2,7 +2,7 @@
 #include "pch_script.h"
 #include "script_attachment_manager.h"
 
-using namespace luabind;
+using namespace ::luabind;
 
 #pragma optimize("s",on)
 void script_attachment::script_register(lua_State* L)
@@ -57,6 +57,10 @@ void script_attachment::script_register(lua_State* L)
 		.def("get_transform", &script_attachment::GetTransform)
 		.def("get_center", &script_attachment::GetCenter)
 		.def("bone_callback", (void (script_attachment::*)(u16,u16,bool)) &script_attachment::SetBoneCallback)
-		.def("bone_callback", (void (script_attachment::*)(u16, const luabind::functor<Fmatrix>&,bool)) &script_attachment::SetBoneCallback)
+		.def("bone_callback", (void (script_attachment::*)(u16, const ::luabind::functor<Fmatrix>&,bool)) &script_attachment::SetBoneCallback)
+		.def("get_shaders", &script_attachment::GetShaders)
+		.def("get_default_shaders", &script_attachment::GetDefaultShaders)
+		.def("set_shader", &script_attachment::SetShaderTexture)
+		.def("reset_shader", &script_attachment::ResetShaderTexture)
 	];
 }