Included a method to create a diagonal matrix with constant diagonal …

…value
labmec · Dec 13, 2021 · 0d1e87f · 0d1e87f
1 parent 9276d61
commit 0d1e87f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/Matrix/pzmatrix.cpp b/Matrix/pzmatrix.cpp
@@ -186,6 +186,19 @@ void TPZMatrix<TVar>::Identity() {
     }
 }
 
+template<class TVar>
+void TPZMatrix<TVar>::Diagonal(TVar val) {
+
+    if ( Cols() != Rows() ) {
+        Error( "Diagonal (TPZMatrix<>*) <TPZMatrix<>must be square>" );
+    }
+    for ( int64_t row = 0; row < Rows(); row++) {
+        for ( int64_t col = 0; col < Cols(); col++ ) {
+            (row == col)? PutVal(row,col,val):PutVal(row,col,0.);
+        }
+    }
+}
+
 
 template<class TVar>
 void TPZMatrix<TVar>::Input(std::istream& in )

diff --git a/Matrix/pzmatrix.h b/Matrix/pzmatrix.h
@@ -209,6 +209,8 @@ class TPZMatrix: public TPZBaseMatrix
 
 	/** @brief Converts the matrix in an identity matrix*/
 	virtual void Identity();
+    /** @brief Converts the matrix in a diagonal matrix*/
+    virtual void Diagonal(TVar val);
 	/** @brief It makes *T the transpose of current matrix. */
 	virtual void Transpose(TPZMatrix<TVar>*const T) const;