Complete algebra (+, -, scalar mult.) for combinations of tridiagonal matrices

Author: loiseaujc

src/stdlib_specialmatrices.fypp

@@ -468,11 +468,51 @@ module stdlib_specialmatrices
         !! be of the same type and kind.
         !! [Specifications](../page/specs/stdlib_specialmatrices.html#operators)
         #:for k1, t1, s1 in (KINDS_TYPES)
-        pure module function matrix_add_tridiagonal_${s1}$(A, B) result(C)
-            class(tridiagonal_${s1}$_type), intent(in) :: A
-            class(tridiagonal_${s1}$_type), intent(in) :: B
-            class(tridiagonal_${s1}$_type), allocatable :: C
+        pure module function matrix_add_tridiag_tridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A, B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_tridiag_symtridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A
+            type(symtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_symtridiag_tridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A
+            type(tridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_symtridiag_symtridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A, B
+            type(symtridiagonal_${s1}$_type) :: C
+        end function
+
+        #:if t1.startswith("complex")
+        pure module function matrix_add_tridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A
+            type(hermtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_hermtridiag_tridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A
+            type(tridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_symtridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A
+            type(hermtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
         end function
+        pure module function matrix_add_hermtridiag_symtridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A
+            type(symtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_add_hermtridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A, B
+            type(hermtridiagonal_${s1}$_type) :: C
+        end function
+        #:endif
         #:endfor
     end interface
 
@@ -481,11 +521,51 @@ module stdlib_specialmatrices
         !! be of the same type and kind.
         !! [Specifications](../page/specs/stdlib_specialmatrices.html#operators)
         #:for k1, t1, s1 in (KINDS_TYPES)
-        pure module function matrix_sub_tridiagonal_${s1}$(A, B) result(C)
-            class(tridiagonal_${s1}$_type), intent(in) :: A
-            class(tridiagonal_${s1}$_type), intent(in) :: B
-            class(tridiagonal_${s1}$_type), allocatable :: C
+        pure module function matrix_sub_tridiag_tridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A, B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_tridiag_symtridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A
+            type(symtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_symtridiag_tridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A
+            type(tridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_symtridiag_symtridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A, B
+            type(symtridiagonal_${s1}$_type) :: C
+        end function
+
+        #:if t1.startswith("complex")
+        pure module function matrix_sub_tridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(tridiagonal_${s1}$_type), intent(in) :: A
+            type(hermtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_hermtridiag_tridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A
+            type(tridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_symtridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(symtridiagonal_${s1}$_type), intent(in) :: A
+            type(hermtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
         end function
+        pure module function matrix_sub_hermtridiag_symtridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A
+            type(symtridiagonal_${s1}$_type), intent(in) :: B
+            type(tridiagonal_${s1}$_type) :: C
+        end function
+        pure module function matrix_sub_hermtridiag_hermtridiag_${s1}$(A, B) result(C)
+            type(hermtridiagonal_${s1}$_type), intent(in) :: A, B
+            type(hermtridiagonal_${s1}$_type) :: C
+        end function
+        #:endif
         #:endfor
     end interface