From 14b2ead96efe1d899aa92c32c3c6ca2efd1f95a2 Mon Sep 17 00:00:00 2001
From: dcluo <dc.luo@outlook.com>
Date: Thu, 13 Jul 2023 18:26:09 -0500
Subject: [PATCH] updated docs

---
 doc/soupy.solver.rst          |  8 +++++++
 soupy/solver/PETScLUSolver.py | 42 ++++++++++++++++++++++++++---------
 2 files changed, 40 insertions(+), 10 deletions(-)

diff --git a/doc/soupy.solver.rst b/doc/soupy.solver.rst
index a03f32f..fb66e87 100644
--- a/doc/soupy.solver.rst
+++ b/doc/soupy.solver.rst
@@ -16,3 +16,11 @@ soupy.solver.newtonBacktrackSolver
     :members:
     :undoc-members:
     :show-inheritance:
+
+soupy.solver.PETScLUSolver 
+-----------------------------------------------
+
+.. automodule:: soupy.solver.PETScLUSolver
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/soupy/solver/PETScLUSolver.py b/soupy/solver/PETScLUSolver.py
index 809633a..71a49fe 100644
--- a/soupy/solver/PETScLUSolver.py
+++ b/soupy/solver/PETScLUSolver.py
@@ -5,16 +5,34 @@
 
 class PETScLUSolver:
     """
-    Wrapper for `dolfin.PETScLUSolver` with methods
-    :code:`solve_transpose` and :code:`set_operator`
+    LU solver for linear systems :math:`Ax = b` and :math:`A^Tx = b`. \
+        It is a wrapper for :py:class:`dolfin.PETScLUSolver` \
+        providing custom implementations for methods :code:`solve_transpose` \
+        and :code:`set_operator` if they are unavailable in the \
+        :code:`dolfin` version being used.
     """
     def __init__(self, mpi_comm=MPI.COMM_WORLD, method="mumps"):
+        """
+        Constructor:
+
+        :param mpi_comm: MPI Communicator for the linear system
+        :type mpi_comm: :py:class:`MPI.Comm`
+        :param method: LU method 
+        :type method: str
+        """
         self.mpi_comm = mpi_comm 
         self.method = method 
         self.solver = dl.PETScLUSolver(mpi_comm, method)
         self.ksp = self.solver.ksp()
 
     def set_operator(self, A):
+        """
+        Set the linear operator 
+
+        :param A: Matrix for the solves
+        :type A: :py:class:`dolfin.Matrix` or :py:class:`dolfin.PETScMatrix`
+        """
+
         if hasattr(A, 'mat'):
             self.ksp.setOperators(A.mat())
         else:
@@ -22,22 +40,26 @@ def set_operator(self, A):
 
     def solve(self, x, b):
         """
-        Solve the linear system 
+        Solve the linear system :math:`Ax = b` \
+            and stores result to :code:`x`
 
-        .. math:: `Ax = b`
-
-        and stores result to :code:`x`
+        :param x: Solution vector
+        :type x: :py:class:`dolfin.Vector` or :py:class:`dolfin.PETScVector`
+        :param b: Right hand side vector
+        :type b: :py:class:`dolfin.Vector` or :py:class:`dolfin.PETScVector`
         """
 
         self.solver.solve(x, b)
 
     def solve_transpose(self, x, b):
         """
-        Solve the linear system 
-
-        .. math:: `A^T x = b`
+        Solve the linear system :math:`A^T x = b` \
+            and stores result to :code:`x`
 
-        and stores result to :code:`x`
+        :param x: Solution vector
+        :type x: :py:class:`dolfin.Vector` or :py:class:`dolfin.PETScVector`
+        :param b: Right hand side vector
+        :type b: :py:class:`dolfin.Vector` or :py:class:`dolfin.PETScVector`
         """
         if hasattr(self.solver, 'solve_transpose'):
             self.solver.solveTranspose(x, b)