sort tcopflow functions

Author: Jaya Mohan

interfaces/python/exago_python_tcopflow.cpp

@@ -30,20 +30,62 @@ void init_exago_tcopflow(pybind11::module &m) {
 
   pybind11::class_<TCOPFLOW_wrapper>(m, "TCOPFLOW")
           .def(pybind11::init())
-          .def("set_tolerance",
-           [](TCOPFLOW_wrapper &w, double tol) {
+          .def("set_network_data",
+           [](TCOPFLOW_wrapper &w, std::string filename) {
              PetscErrorCode ierr;
-             ierr = TCOPFLOWSetTolerance(w.tcopf, tol);
+             ierr = TCOPFLOWSetNetworkData(w.tcopf, filename.c_str());
              ExaGOCheckError(ierr);
+           .def("set_solver",
+           [](TCOPFLOW_wrapper &w, std::string solver) {
+             PetscErrorCode ierr;
+             ierr = TCOPFLOWSetSolver(w.tcopf, solver.c_str());
+             ExaGO
+           });
+
+           .def("setup",
+           [](TCOPFLOW_wrapper &w) {
+             PetscErrorCode ierr;
+             ierr = TCOPFLOWSetUp(w.tcopf);
+             ExaGOCheckError(ierr);
+           });
+
+           .def("get_convergence_status",
+           [](TCOPFLOW_wrapper &w) -> bool {
+             PetscErrorCode ierr;
+             PetscBool flag;
+             ierr = TCOPFLOWGetConvergenceStatus(w.tcopf, &flag);
+             ExaGOCheckError(ierr);
+             return flag;
            })
-          .def("get_tolerance",
+          .def("get_objective",
            [](TCOPFLOW_wrapper &w) -> double {
              PetscErrorCode ierr;
-             PetscReal tol;
-             ierr = TCOPFLOWGetTolerance(w.tcopf, &tol);
+             double obj;
+             ierr = TCOPFLOWGetObjective(w.tcopf, &obj);
+             ExaGOCheckError(ierr);
+             return obj;
+           })
+           .def("get_num_iterations",
+           [](TCCOPFLOW_wrapper &w) -> int {
+             PetscErrorCode ierr;
+             PetscInt n;
+             ierr = TCOPFLOWGetNumIterations(w.tcopf, &n);
+             ExaGOCheckError(ierr);
+             return n;
+           }) 
+           .def("save_solution",
+           [](TCOPFLOW_wrapper &w, OutputFormat fmt, std::string outfile) {
+             PetscErrorCode ierr;
+             ierr = TCOPFLOWSaveSolution(w.tcopf, fmt, outfile.c_str());
              ExaGOCheckError(ierr);
-             return tol;
            });
+            .def("save_solution_all",
+           [](TCCOPFLOW_wrapper &w, OutputFormat fmt, std::string outdir) {
+             PetscErrorCode ierr;
+             ierr = TCCOPFLOWSaveSolutionAll(w.tcopf, fmt, outdir.c_str());
+             ExaGOCheckError(ierr);
+           });
+    
     }