Merge pull request 20tab#365 from jiamingfeng/master

Roberto De Ioris · web-flow · commit 45e3574ddabb · 2018-08-05T14:59:08.000+02:00
Added "set static switch parameter" for material instance
diff --git a/Source/UnrealEnginePython/Private/UEPyModule.cpp b/Source/UnrealEnginePython/Private/UEPyModule.cpp
@@ -1075,6 +1075,7 @@ static PyMethodDef ue_PyUObject_methods[] = {
 	{ "set_material", (PyCFunction)py_ue_set_material, METH_VARARGS, "" },
 	{ "set_material_by_name", (PyCFunction)py_ue_set_material_by_name, METH_VARARGS, "" },
 	{ "set_material_scalar_parameter", (PyCFunction)py_ue_set_material_scalar_parameter, METH_VARARGS, "" },
+	{ "set_material_static_switch_parameter", (PyCFunction)py_ue_set_material_static_switch_parameter, METH_VARARGS, "" },
 	{ "set_material_vector_parameter", (PyCFunction)py_ue_set_material_vector_parameter, METH_VARARGS, "" },
 	{ "set_material_texture_parameter", (PyCFunction)py_ue_set_material_texture_parameter, METH_VARARGS, "" },
 	{ "get_material_scalar_parameter", (PyCFunction)py_ue_get_material_scalar_parameter, METH_VARARGS, "" },
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyMaterial.cpp b/Source/UnrealEnginePython/Private/UObject/UEPyMaterial.cpp
@@ -116,6 +116,73 @@ PyObject *py_ue_set_material_scalar_parameter(ue_PyUObject *self, PyObject * arg
 
 }
 
+PyObject *py_ue_set_material_static_switch_parameter(ue_PyUObject *self, PyObject * args)
+{
+
+	ue_py_check(self);
+
+	char *switchName = nullptr;
+	PyObject *py_bool = nullptr;
+	if (!PyArg_ParseTuple(args, "sO:set_material_static_switch_parameter", &switchName, &py_bool))
+	{
+		return NULL;
+	}
+
+	FName parameterName(UTF8_TO_TCHAR(switchName));	
+
+	bool switchValue = false;
+	if (PyObject_IsTrue(py_bool))
+	{
+		switchValue = true;
+	}
+
+	bool valid = false;
+
+#if WITH_EDITOR
+	if (self->ue_object->IsA<UMaterialInstance>())
+	{
+		UMaterialInstance *material_instance = (UMaterialInstance *)self->ue_object;
+		valid = true;
+		FStaticParameterSet staticParameterSet = material_instance->GetStaticParameters();
+
+		bool isExisting = false;
+		for (auto& parameter : staticParameterSet.StaticSwitchParameters)
+		{
+			if (parameter.bOverride && parameter.ParameterInfo.Name == parameterName)
+			{
+				parameter.Value = switchValue;
+				isExisting = true;
+				break;
+			}
+		}
+
+		if (!isExisting)
+		{
+			FStaticSwitchParameter SwitchParameter;
+			SwitchParameter.ParameterInfo.Name = parameterName;
+			SwitchParameter.Value = switchValue;
+
+			SwitchParameter.bOverride = true;
+			staticParameterSet.StaticSwitchParameters.Add(SwitchParameter);
+		}
+
+
+		material_instance->UpdateStaticPermutation(staticParameterSet);
+
+	}
+
+
+#endif
+
+	if (!valid)
+	{
+		return PyErr_Format(PyExc_Exception, "uobject is not a MaterialInstance");
+	}
+
+	Py_RETURN_NONE;
+
+}
+
 PyObject *py_ue_get_material_scalar_parameter(ue_PyUObject *self, PyObject * args)
 {
 
diff --git a/Source/UnrealEnginePython/Private/UObject/UEPyMaterial.h b/Source/UnrealEnginePython/Private/UObject/UEPyMaterial.h
@@ -5,6 +5,7 @@
 
 
 PyObject *py_ue_set_material_scalar_parameter(ue_PyUObject *, PyObject *);
+PyObject *py_ue_set_material_static_switch_parameter(ue_PyUObject *, PyObject *);
 PyObject *py_ue_set_material_vector_parameter(ue_PyUObject *, PyObject *);
 PyObject *py_ue_set_material_texture_parameter(ue_PyUObject *, PyObject *);
 
