@@ -23,16 +23,30 @@ PYBIND11_MODULE(pyCGraph, m) {
23
23
py::arg (" runTimes" ) = 1 )
24
24
.def (" destroy" , &GPipelinePy::destroy)
25
25
.def (" registerGElement" , &GPipelinePy::registerGElement,
26
- py::arg (" element" ),
27
- py::arg (" depends" ) = GElementPtrSet{},
28
- py::arg (" name" ) = CGRAPH_EMPTY,
29
- py::arg (" loop" ) = CGRAPH_DEFAULT_LOOP_TIMES,
30
- " register a GElement with dependencies, name, and loop count." );
26
+ py::arg (" element" ),
27
+ py::arg (" depends" ) = GElementPtrSet{},
28
+ py::arg (" name" ) = CGRAPH_EMPTY,
29
+ py::arg (" loop" ) = CGRAPH_DEFAULT_LOOP_TIMES,
30
+ " register a GElement with dependencies, name, and loop count." );
31
31
32
32
py::class_<GElement, GElementPyw, std::unique_ptr<GElement, py::nodelete> >(m, " GElement" )
33
- .def (py::init<>());
33
+ .def (py::init<>())
34
+ .def (" getName" , &GElement::getName)
35
+ .def (" setName" , &GElement::setName)
36
+ .def (" addDependGElements" , &GElement::addDependGElements,
37
+ py::arg (" elements" ))
38
+ .def (" setLoop" , &GElement::setLoop);
34
39
35
40
py::class_<GNode, GNodePyw, GElement, std::unique_ptr<GNode, py::nodelete> >(m, " GNode" )
41
+ .def (py::init<>());
42
+
43
+ py::class_<GRegionPy, GElement, std::unique_ptr<GRegionPy, py::nodelete> >(m, " GRegion" )
44
+ .def (py::init<>())
45
+ .def (" addGElement" , &GRegionPy::addGElement,
46
+ py::arg (" element" ));
47
+
48
+ py::class_<GConditionPyw, GElement, std::unique_ptr<GConditionPyw, py::nodelete> >(m, " GCondition" )
36
49
.def (py::init<>())
37
- .def (" getName" , &GNode::getName);
50
+ .def (" addGElement" , &GConditionPyw::addGElement,
51
+ py::arg (" element" ));
38
52
}
0 commit comments