[example] add python GRegion demo

ChunelFeng · ChunelFeng · commit b62ffc94fd71 · 2025-02-25T00:19:26.000+08:00
diff --git a/python/PyCGraph.cpp b/python/PyCGraph.cpp
@@ -56,7 +56,15 @@ PYBIND11_MODULE(PyCGraph, m) {
         .def("setLoop", &GElement::setLoop);
 
     py::class_<GNode, PywGNode, GElement, std::unique_ptr<GNode, py::nodelete> >(m, "GNode")
-        .def(py::init<>());
+        .def(py::init<const std::string&>(),
+             py::arg("name"))
+        .def(py::init<const std::string&, int>(),
+             py::arg("name"),
+             py::arg("loop"))
+        .def(py::init<const GElementPtrSet&, const std::string&, int>(),
+             py::arg("depends") = GElementPtrSet{},
+             py::arg("name") = CGRAPH_EMPTY,
+             py::arg("loop") = CGRAPH_DEFAULT_LOOP_TIMES);
 
     py::class_<PyGCluster, GElement, std::unique_ptr<PyGCluster, py::nodelete> >(m, "GCluster")
         .def(py::init<>())
diff --git a/python/tutorial/T02-Cluster.py b/python/tutorial/T02-Cluster.py
@@ -12,11 +12,7 @@
 from MyPyGNode.MyPyNode2 import MyPyNode2
 
 def tutorial_cluster():
-    b1, b2, b3 = MyPyNode1(), MyPyNode1(), MyPyNode2()
-    b1.setName("nodeB1")
-    b2.setName("nodeB2")
-    b2.setLoop(3)
-    b3.setName("nodeB3")
+    b1, b2, b3 = MyPyNode1("nodeB1"), MyPyNode1("nodeB2", 3), MyPyNode2("nodeB3")
     b_cluster = GCluster()
     b_cluster.addGElements([b1, b2, b3])
 
diff --git a/python/tutorial/T03-Region.py b/python/tutorial/T03-Region.py
@@ -0,0 +1,33 @@
+"""
+@Author: Chunel
+@Contact: chunel@foxmail.com
+@File: T03-Region
+@Time: 2025/2/24 22:52
+@Desc: 
+"""
+
+from PyCGraph import GNode, GPipeline, GRegion, CStatus
+
+from MyPyGNode.MyPyNode1 import MyPyNode1
+from MyPyGNode.MyPyNode2 import MyPyNode2
+
+def tutorial_region():
+    b1 = MyPyNode1("nodeB1")
+    b2 = MyPyNode2({b1}, "nodeB2", 2)
+    b3 = MyPyNode1({b1}, "nodeB3", 1)
+    b4 = MyPyNode1({b2, b3}, "nodeB4", 1)
+    b_region = GRegion()
+    b_region.addGElements([b1, b2, b3, b4])
+
+    pipeline = GPipeline()
+    a, c, d = MyPyNode1(), MyPyNode2(), MyPyNode1()
+
+    pipeline.registerGElement(a, set(), "nodeA")
+    pipeline.registerGElement(b_region, {a}, "regionB", 2)
+    pipeline.registerGElement(c, {b_region}, "nodeC")
+
+    pipeline.process()
+
+
+if __name__ == '__main__':
+    tutorial_region()
diff --git a/python/wrapper/PywGNode.h b/python/wrapper/PywGNode.h
@@ -16,6 +16,24 @@
 namespace py = pybind11;
 
 class PywGNode : public CGraph::GNode {
+public:
+    explicit PywGNode(const std::string& name) {
+        setName(name);
+    }
+
+    explicit PywGNode(const std::string& name, int loop) {
+        setName(name);
+        setLoop(loop);
+    }
+
+    explicit PywGNode(const CGraph::GElementPtrSet& depends = CGraph::GElementPtrSet{},
+                      const std::string& name = CGraph::CGRAPH_EMPTY,
+                      int loop = CGraph::CGRAPH_DEFAULT_LOOP_TIMES) {
+        addDependGElements(depends);
+        setLoop(loop);
+        setName(name);
+    }
+
 protected:
     CStatus init() override {
         PYBIND11_OVERLOAD(CStatus, GNode, init);
