added support for python generators in component tick

Roberto De Ioris · Roberto De Ioris · commit 6a3f0700b979 · 2018-04-30T10:23:15.000+02:00
diff --git a/Source/UnrealEnginePython/Private/PythonComponent.cpp b/Source/UnrealEnginePython/Private/PythonComponent.cpp
@@ -13,8 +13,11 @@ UPythonComponent::UPythonComponent()
 
 	PythonTickForceDisabled = false;
 	PythonDisableAutoBinding = false;
+	PythonTickEnableGenerator = false;
 
 	bWantsInitializeComponent = true;
+
+	py_generator = nullptr;
 }
 
 void UPythonComponent::InitializePythonComponent()
@@ -163,6 +166,23 @@ void UPythonComponent::TickComponent(float DeltaTime, ELevelTick TickType, FActo
 
 	FScopePythonGIL gil;
 
+	if (PythonTickEnableGenerator && py_generator)
+	{
+		PyObject *ret = PyIter_Next(py_generator);
+		if (!ret)
+		{
+			if (PyErr_Occurred())
+			{
+				unreal_engine_py_log_error();
+			}
+			Py_DECREF(py_generator);
+			py_generator = nullptr;
+			return;
+		}
+		Py_DECREF(ret);
+		return;
+	}
+
 	// no need to check for method availability, we did it in component initialization
 
 	PyObject *ret = PyObject_CallMethod(py_component_instance, (char *)"tick", (char *)"f", DeltaTime);
@@ -171,6 +191,15 @@ void UPythonComponent::TickComponent(float DeltaTime, ELevelTick TickType, FActo
 		unreal_engine_py_log_error();
 		return;
 	}
+
+	if (PythonTickEnableGenerator)
+	{
+		py_generator = PyObject_GetIter(ret);
+		if (!py_generator)
+		{
+			UE_LOG(LogPython, Error, TEXT("tick is not a python generator"));
+		}
+	}
 	Py_DECREF(ret);
 
 }
diff --git a/Source/UnrealEnginePython/Public/PythonComponent.h b/Source/UnrealEnginePython/Public/PythonComponent.h
@@ -37,6 +37,9 @@ class UNREALENGINEPYTHON_API UPythonComponent : public UActorComponent
 	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Python")
 		bool PythonDisableAutoBinding;
 
+	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Python")
+		bool PythonTickEnableGenerator;
+
 	UFUNCTION(BlueprintCallable, Category = "Python")
 		void InitializePythonComponent();
 
@@ -89,5 +92,7 @@ class UNREALENGINEPYTHON_API UPythonComponent : public UActorComponent
 	PyObject * py_component_instance;
 	// mapped uobject, required for debug and advanced reflection
 	ue_PyUObject *py_uobject;
+
+	PyObject *py_generator;
 };
 
