@@ -29,71 +29,71 @@ class TCOPFLOW_wrapper {
29
29
void init_exago_tcopflow (pybind11::module &m) {
30
30
31
31
pybind11::class_<TCOPFLOW_wrapper>(m, " TCOPFLOW" )
32
- .def (pybind11::init ())
33
- .def (" set_network_data" ,
34
- [](TCOPFLOW_wrapper &w, std::string filename) {
35
- PetscErrorCode ierr;
36
- ierr = TCOPFLOWSetNetworkData (w.tcopf , filename.c_str ());
37
- ExaGOCheckError (ierr);
38
- })
39
- .def (" set_solver" , [](TCOPFLOW_wrapper &w, std::string solver) {
40
- PetscErrorCode ierr;
41
- ierr = TCOPFLOWSetSolver (w.tcopf , solver.c_str ());
42
- ExaGOCheckError (ierr);
43
- })
44
- .def (" set_tolerance" ,
32
+ .def (pybind11::init ())
33
+ .def (" set_network_data" ,
34
+ [](TCOPFLOW_wrapper &w, std::string filename) {
35
+ PetscErrorCode ierr;
36
+ ierr = TCOPFLOWSetNetworkData (w.tcopf , filename.c_str ());
37
+ ExaGOCheckError (ierr);
38
+ })
39
+ .def (" set_solver" ,
40
+ [](TCOPFLOW_wrapper &w, std::string solver) {
41
+ PetscErrorCode ierr;
42
+ ierr = TCOPFLOWSetSolver (w.tcopf , solver.c_str ());
43
+ ExaGOCheckError (ierr);
44
+ })
45
+ .def (" set_tolerance" ,
45
46
[](TCOPFLOW_wrapper &w, double tol) {
46
47
PetscErrorCode ierr;
47
48
ierr = TCOPFLOWSetTolerance (w.tcopf , tol);
48
49
ExaGOCheckError (ierr);
49
50
})
50
- .def (" setup" , [](TCOPFLOW_wrapper &w) {
51
- PetscErrorCode ierr;
52
- ierr = TCOPFLOWSetUp (w.tcopf );
53
- ExaGOCheckError (ierr);
54
- })
55
- .def (" solve" ,
51
+ .def (" setup" ,
52
+ [](TCOPFLOW_wrapper &w) {
53
+ PetscErrorCode ierr;
54
+ ierr = TCOPFLOWSetUp (w.tcopf );
55
+ ExaGOCheckError (ierr);
56
+ })
57
+ .def (" solve" ,
56
58
[](TCOPFLOW_wrapper &w) {
57
59
PetscErrorCode ierr;
58
60
ierr = TCOPFLOWSolve (w.tcopf );
59
61
ExaGOCheckError (ierr);
60
62
})
61
63
62
- .def (" get_convergence_status" ,
63
- [](TCOPFLOW_wrapper &w) -> bool {
64
- PetscErrorCode ierr;
65
- PetscBool flag;
66
- ierr = TCOPFLOWGetConvergenceStatus (w.tcopf , &flag);
67
- ExaGOCheckError (ierr);
68
- return flag;
69
- })
70
- .def (" get_objective" ,
71
- [](TCOPFLOW_wrapper &w) -> double {
72
- PetscErrorCode ierr;
73
- double obj;
74
- ierr = TCOPFLOWGetObjective (w.tcopf , &obj);
75
- ExaGOCheckError (ierr);
76
- return obj;
77
- })
78
- .def (" get_num_iterations" ,
79
- [](TCOPFLOW_wrapper &w) -> int {
80
- PetscErrorCode ierr;
81
- PetscInt n;
82
- ierr = TCOPFLOWGetNumIterations (w.tcopf , &n);
83
- ExaGOCheckError (ierr);
84
- return n;
85
- });
86
- /* .def("save_solution",
87
- [](TCOPFLOW_wrapper &w, OutputFormat fmt, std::string outfile) {
88
- PetscErrorCode ierr;
89
- ierr = TCOPFLOWSaveSolution(w.tcopf, fmt, outfile.c_str());
90
- ExaGOCheckError(ierr);
91
- })
92
- .def("save_solution_all",
93
- [](TCOPFLOW_wrapper &w, OutputFormat fmt, std::string outdir) {
94
- PetscErrorCode ierr;
95
- ierr = TCCOPFLOWSaveSolutionAll(w.tcopf, fmt, outdir.c_str());
96
- ExaGOCheckError(ierr);
97
- })*/
98
-
99
- }
64
+ .def (" get_convergence_status" ,
65
+ [](TCOPFLOW_wrapper &w) -> bool {
66
+ PetscErrorCode ierr;
67
+ PetscBool flag;
68
+ ierr = TCOPFLOWGetConvergenceStatus (w.tcopf , &flag);
69
+ ExaGOCheckError (ierr);
70
+ return flag;
71
+ })
72
+ .def (" get_objective" ,
73
+ [](TCOPFLOW_wrapper &w) -> double {
74
+ PetscErrorCode ierr;
75
+ double obj;
76
+ ierr = TCOPFLOWGetObjective (w.tcopf , &obj);
77
+ ExaGOCheckError (ierr);
78
+ return obj;
79
+ })
80
+ .def (" get_num_iterations" , [](TCOPFLOW_wrapper &w) -> int {
81
+ PetscErrorCode ierr;
82
+ PetscInt n;
83
+ ierr = TCOPFLOWGetNumIterations (w.tcopf , &n);
84
+ ExaGOCheckError (ierr);
85
+ return n;
86
+ });
87
+ /* .def("save_solution",
88
+ [](TCOPFLOW_wrapper &w, OutputFormat fmt, std::string outfile) {
89
+ PetscErrorCode ierr;
90
+ ierr = TCOPFLOWSaveSolution(w.tcopf, fmt, outfile.c_str());
91
+ ExaGOCheckError(ierr);
92
+ })
93
+ .def("save_solution_all",
94
+ [](TCOPFLOW_wrapper &w, OutputFormat fmt, std::string outdir) {
95
+ PetscErrorCode ierr;
96
+ ierr = TCCOPFLOWSaveSolutionAll(w.tcopf, fmt, outdir.c_str());
97
+ ExaGOCheckError(ierr);
98
+ })*/
99
+ }
0 commit comments