Remove DM and add DMDA and PetscDM types

Author: Jeremy E Kozdon

src/dm.jl

@@ -1,13 +1,14 @@
 const CDM = Ptr{Cvoid}
 
 abstract type AbstractDM{PetscLib} end
-mutable struct DM{PetscLib} <: AbstractDM{PetscLib}
+
+# Mainly for DM we do not know the type of, namely ones returned by PETSc
+# functions such as `KSPGetDM`
+mutable struct PetscDM{PetscLib} <: AbstractDM{PetscLib}
     ptr::CDM
-    opts::Options{PetscLib}
-    DM{PetscLib}(ptr, opts = Options(PetscLib)) where {PetscLib} =
-        new{PetscLib}(ptr, opts)
 end
 
+
 """
     DMSetUp!(da::DM, opts=da.opts)
 
@@ -17,7 +18,7 @@ $(_doc_external("DM/DMSetUp"))
 """
 function DMSetUp! end
 
-@for_petsc function DMSetUp!(da::DM{$PetscLib}, opts::Options=da.opts)
+@for_petsc function DMSetUp!(da::AbstractDM{$PetscLib}, opts::Options=da.opts)
     with(opts) do
         @chk ccall(
             (:DMSetFromOptions, $petsc_library),
@@ -31,7 +32,7 @@ function DMSetUp! end
 end
 
 @for_petsc begin
-    function destroy(da::DM{$PetscLib})
+    function destroy(da::AbstractDM{$PetscLib})
         finalized($PetscScalar) || @chk ccall(
             (:DMDestroy, $petsc_library),
             PetscErrorCode,
@@ -44,7 +45,7 @@ end
 end
 
 """
-    DMGetDimension(dm::DM)
+    DMGetDimension(dm::AbstractDM)
 
 Return the topological dimension of the `dm`
 
@@ -53,7 +54,7 @@ $(_doc_external("DM/DMGetDimension"))
 """
 function DMGetDimension end
 
-@for_petsc function DMGetDimension(dm::DM{$PetscLib})
+@for_petsc function DMGetDimension(dm::AbstractDM{$PetscLib})
     dim = Ref{$PetscInt}()
 
     @chk ccall(
@@ -68,16 +69,16 @@ function DMGetDimension end
 end
 
 """
-    gettype(dm::DM)
+    gettype(dm::AbstractDM)
 
 Gets type name of the `dm`
 
 # External Links
 $(_doc_external("DM/DMGetType"))
 """
-function gettype(::DM) end
+function gettype(::AbstractDM) end
 
-@for_petsc function gettype(dm::DM{$PetscLib})
+@for_petsc function gettype(dm::AbstractDM{$PetscLib})
     t_r = Ref{Cstring}()
     @chk ccall(
         (:DMGetType, $petsc_library),
@@ -90,17 +91,17 @@ function gettype(::DM) end
 end
 
 """
-    view(dm::DM, viewer::Viewer=ViewerStdout(petsclib, getcomm(dm)))
+    view(dm::AbstractDM, viewer::Viewer=ViewerStdout(petsclib, getcomm(dm)))
 
 view a `dm` with `viewer`
 
 # External Links
 $(_doc_external("DM/DMView"))
 """
-function view(::DM) end
+function view(::AbstractDM) end
 
 @for_petsc function view(
-    dm::DM{$PetscLib},
+    dm::AbstractDM{$PetscLib},
     viewer::AbstractViewer{$PetscLib} = ViewerStdout($petsclib, getcomm(dm)),
 )
     @chk ccall(
@@ -114,7 +115,7 @@ function view(::DM) end
 end
 
 """
-    DMCreateMatrix(dm::DM)
+    DMCreateMatrix(dm::AbstractDM)
 
 Generates a matrix from the `dm` object.
 
@@ -123,7 +124,7 @@ $(_doc_external("DM/DMCreateMatrix"))
 """
 function DMCreateMatrix end
 
-@for_petsc function DMCreateMatrix(dm::DM{$PetscLib})
+@for_petsc function DMCreateMatrix(dm::AbstractDM{$PetscLib})
     mat = Mat{$PetscScalar}(C_NULL)
 
     @chk ccall(
@@ -138,7 +139,7 @@ function DMCreateMatrix end
 end
 
 """
-    DMCreateLocalVector(dm::DM)
+    DMCreateLocalVector(dm::AbstractDM)
 
 returns a local vector from the `dm` object.
 
@@ -147,7 +148,7 @@ $(_doc_external("DM/DMCreateLocalVector"))
 """
 function DMCreateLocalVector end
 
-@for_petsc function DMCreateLocalVector(dm::DM{$PetscLib})
+@for_petsc function DMCreateLocalVector(dm::AbstractDM{$PetscLib})
     vec = Vec{$PetscScalar}(C_NULL)
 
     @chk ccall(
@@ -171,7 +172,7 @@ $(_doc_external("DM/DMCreateGlobalVector"))
 """
 function DMCreateGlobalVector end
 
-@for_petsc function DMCreateGlobalVector(dm::DM{$PetscLib})
+@for_petsc function DMCreateGlobalVector(dm::AbstractDM{$PetscLib})
     vec = Vec{$PetscScalar}(C_NULL)
 
     @chk ccall(
@@ -187,7 +188,7 @@ end
 
 """
     DMLocalToGlobal!(
-        dm::DM
+        dm::AbstractDM
         local_vec::AbstractVec,
         mode::InsertMode,
         global_vec::AbstractVec,
@@ -201,7 +202,7 @@ $(_doc_external("DM/DMLocalToGlobal"))
 function DMLocalToGlobal! end
 
 @for_petsc function DMLocalToGlobal!(
-    dm::DM{$PetscLib},
+    dm::AbstractDM{$PetscLib},
     local_vec::AbstractVec,
     mode::InsertMode,
     global_vec::AbstractVec,
@@ -235,7 +236,7 @@ $(_doc_external("DM/DMGlobalToLocal"))
 function DMGlobalToLocal! end
 
 @for_petsc function DMGlobalToLocal!(
-    dm::DM{$PetscLib},
+    dm::AbstractDM{$PetscLib},
     global_vec::AbstractVec,
     mode::InsertMode,
     local_vec::AbstractVec,
@@ -255,34 +256,35 @@ end
 
 """
     DMGetCoordinateDM(
-        dm::DM
+        dm::AbstractDM
     )
 
-Create a `DM` for the coordinates of `dm`.
+Create a `coord_dm` for the coordinates of `dm`.
 
 # External Links
 $(_doc_external("DM/DMGetCoordinateDM"))
 """
 function DMGetCoordinateDM end
 
 @for_petsc function DMGetCoordinateDM(
-    dm::DM{$PetscLib},
+    dm::AbstractDM{$PetscLib},
 )
-    coord_dm = DM{$PetscLib}(C_NULL)
+    coord_dm = empty(dm)
     @chk ccall(
         (:DMGetCoordinateDM, $petsc_library),
         PetscErrorCode,
         (CDM, Ptr{CDM}),
         dm,
         coord_dm,
     )
+    @assert gettype(dm) == gettype(coord_dm)
 
     return coord_dm
 end
 
 """
     DMGetCoordinatesLocal(
-        dm::DM
+        dm::AbstractDM
     )
 
 Gets a local vector with the coordinates associated with `dm`.
@@ -293,7 +295,7 @@ $(_doc_external("DM/DMGetCoordinatesLocal"))
 function DMGetCoordinatesLocal end
 
 @for_petsc function DMGetCoordinatesLocal(
-    dm::DM{$PetscLib},
+    dm::AbstractDM{$PetscLib},
 )
     coord_vec = Vec{$PetscScalar}(C_NULL)
     @chk ccall(

src/dmda.jl

@@ -13,6 +13,20 @@ mutable struct DMDALocalInfo{IT}
     DMDALocalInfo{IT}() where {IT} = new{IT}()
 end
 
+mutable struct DMDA{PetscLib} <: AbstractDM{PetscLib}
+    ptr::CDM
+    opts::Options{PetscLib}
+    DMDA{PetscLib}(ptr, opts = Options(PetscLib)) where {PetscLib} =
+        new{PetscLib}(ptr, opts)
+end
+
+"""
+    empty(da::DMDA)
+
+return an uninitialized `DMDA` struct.
+"""
+Base.empty(::DMDA{PetscLib}) where PetscLib = DMDA{PetscLib}(C_NULL)
+
 """
     DMDACreate1d(
         ::PetscLib
@@ -54,7 +68,7 @@ function DMDACreate1d end
         @assert length(points_per_proc) == MPI.Comm_size(comm)
         points_per_proc
     end
-    da = DM{$PetscLib}(C_NULL, opts)
+    da = DMDA{$PetscLib}(C_NULL, opts)
     with(da.opts) do
         @chk ccall(
             (:DMDACreate1d, $petsc_library),
@@ -142,7 +156,7 @@ function DMDACreate2d end
         @assert length(points_per_proc_y) == procs_y
         points_per_proc_y
     end
-    da = DM{$PetscLib}(C_NULL, opts)
+    da = DMDA{$PetscLib}(C_NULL, opts)
     with(da.opts) do
         @chk ccall(
             (:DMDACreate2d, $petsc_library),
@@ -257,7 +271,7 @@ function DMDACreate3d end
         @assert length(points_per_proc_z) == procs_z
         points_per_proc_z
     end
-    da = DM{$PetscLib}(C_NULL, opts)
+    da = DMDA{$PetscLib}(C_NULL, opts)
     with(da.opts) do
         @chk ccall(
             (:DMDACreate3d, $petsc_library),

src/ksp.jl

@@ -173,7 +173,7 @@ struct Fn_KSPComputeOperators{T} end
             ksp,
             t_dm,
         )
-        dm = DM{$PetscLib}(t_dm[])
+        dm = PetscDM{$PetscLib}(t_dm[])
         return dm
     end