add new file

src/DoubleSparseSquareMatrix.cpp

@@ -0,0 +1,175 @@
+//-------------------------------------------------------------------------------------------------------
+// The MIT License (MIT)
+//
+// Copyright (c) 2021 Yoshiya Usui
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in all
+// copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+//-------------------------------------------------------------------------------------------------------
+#include <stddef.h> // For null pointer
+#include <stdlib.h> // For exit
+#include <iostream>
+#include "DoubleSparseSquareMatrix.h"
+#include "OutputFiles.h"
+#include <assert.h>
+
+//Default Constructer
+DoubleSparseSquareMatrix::DoubleSparseSquareMatrix():
+	DoubleSparseMatrix()
+{}
+
+// Constructer
+DoubleSparseSquareMatrix::DoubleSparseSquareMatrix( const int nEq, const int nRhs ):
+	DoubleSparseMatrix( nEq, nEq, nRhs )
+{
+	//if( nEq <= 0  ){
+	//	OutputFiles::m_logFile << "Error : Total number of equation specified is less than or equals to zero. : nEq = " << nEq << std::endl;
+	//	exit(1);		
+	//}
+	//
+	//if( nRhs <= 0 ){
+	//	OutputFiles::m_logFile << "Error : Total number of right-hand side vectors is specified to be less than or equals to zero. : nRhs = " << nRhs << std::endl;
+	//	exit(1);		
+	//}
+	assert( nEq > 0 );
+	assert( nRhs > 0 );
+}
+
+// Destructer
+DoubleSparseSquareMatrix::~DoubleSparseSquareMatrix(){
+	if( m_pardisoSolver.getSolutionStage() > PARDISOSolver::MEMORY_RELEASED ){
+		m_pardisoSolver.releaseMemory();
+	}
+}
+
+// Set number of rows and columns
+void DoubleSparseSquareMatrix::setNumRowsAndColumns( const int nrows, const int ncols ){
+
+	//if( nrows != ncols ){
+	//	OutputFiles::m_logFile << "Error : Number of rows and the one of columns are different for square matrix. : nrows = " << nrows << ", ncols = " << ncols << std::endl;
+	//	exit(1);		
+	//}
+	assert( nrows == ncols );
+
+	DoubleSparseMatrix::setNumRowsAndColumns( nrows, ncols );
+
+}
+
+// Set Degree of equation
+// Note : This function must be called BEFORE the matrix is converted into CRS format
+void DoubleSparseSquareMatrix::setDegreeOfEquation( const int nEq ){
+
+	//if( nEq <= 0  ){
+	//	OutputFiles::m_logFile << "Error : Total number of equation specified is less than or equals to zero. : nEq = " << nEq << std::endl;
+	//	exit(1);		
+	//}
+	assert( nEq > 0 );
+
+	setNumRowsAndColumns( nEq, nEq );
+}
+
+//Initialize matrix and right-hand side vectors
+void DoubleSparseSquareMatrix::initializeMatrixAndRhsVectors( const int nEq, const int nRhs ){
+
+	//if( nEq <= 0  ){
+	//	OutputFiles::m_logFile << "Error : Total number of equation is specified to be less than or equals to zero. : nEq = " << nEq << std::endl;
+	//	exit(1);		
+	//}
+	//
+	//if( nRhs <= 0 ){
+	//	OutputFiles::m_logFile << "Error : Total number of right-hand side vectors is specified to be less than or equals to zero. : nRhs = " << nRhs << std::endl;
+	//	exit(1);		
+	//}
+	assert( nEq > 0 );
+	assert( nRhs > 0 );
+
+	releaseMemoryMatrixSolver();
+
+	DoubleSparseMatrix::initializeMatrixAndRhsVectors( nEq, nEq, nRhs );
+
+}
+
+//Anaysis phase of matrix solver
+void DoubleSparseSquareMatrix::analysisPhaseMatrixSolver(){
+	assert( m_hasConvertedToCRSFormat );
+	m_pardisoSolver.analysis( m_numRows, m_rowIndex, m_columns );
+}
+
+//Numerical factorization phase of matrix solver
+void DoubleSparseSquareMatrix::factorizationPhaseMatrixSolver(){
+	assert( m_hasConvertedToCRSFormat );
+	m_pardisoSolver.numericalFactorization( m_rowIndex, m_columns, m_values );
+}
+
+//Solve phase of matrix solver with a specified number of right-hand side
+void DoubleSparseSquareMatrix::solvePhaseMatrixSolver( double* solution, const int iRhsStart ,const int nRhs ){
+	assert( m_hasConvertedToCRSFormat );
+	const long long index = static_cast<long long>(m_numRows) * static_cast<long long>(iRhsStart);
+	m_pardisoSolver.solve( m_rowIndex, m_columns, m_values, nRhs, &m_rightHandSideVector[index], solution );
+}
+
+//Solve phase of matrix solver
+void DoubleSparseSquareMatrix::solvePhaseMatrixSolver( double* solution ){
+	assert( m_hasConvertedToCRSFormat );
+	m_pardisoSolver.solve( m_rowIndex, m_columns, m_values, m_numRightHandSideVectors, m_rightHandSideVector, solution );
+}
+
+//Solve phase of matrix solver
+void DoubleSparseSquareMatrix::solvePhaseMatrixSolver( const int nrhs, double* rhs, double* solution ){
+	assert( m_hasConvertedToCRSFormat );
+	m_pardisoSolver.solve( m_rowIndex, m_columns, m_values, nrhs, rhs, solution );
+}
+
+//Release memory of matrix solver
+void DoubleSparseSquareMatrix::releaseMemoryMatrixSolver(){
+	if( m_pardisoSolver.getSolutionStage() > PARDISOSolver::MEMORY_RELEASED ){
+		m_pardisoSolver.releaseMemory();
+	}
+}
+
+//Get memory required by matrix solver
+void DoubleSparseSquareMatrix::writeMemoryRequiredByMatrixSolver(){
+	m_pardisoSolver.writeMemoryRequired();
+}
+
+//Release memory
+void DoubleSparseSquareMatrix::releaseMemory(){
+
+	if( m_pardisoSolver.getSolutionStage() > PARDISOSolver::MEMORY_RELEASED ){
+		m_pardisoSolver.releaseMemory();
+	}
+	DoubleSparseMatrix::releaseMemory();
+
+}
+
+// Get Degree of equation
+int DoubleSparseSquareMatrix::getDegreeOfEquation() const{
+	return m_numRows;
+}
+
+//Copy constructer
+DoubleSparseSquareMatrix::DoubleSparseSquareMatrix(const DoubleSparseSquareMatrix &matrix ){
+	std::cerr << "Error : Copy constructer of the class DoubleSparseSquareMatrix is not implemented." << std::endl;
+	exit(1);
+}
+
+// Assignment operator
+DoubleSparseSquareMatrix& DoubleSparseSquareMatrix::operator=(const DoubleSparseSquareMatrix& rhs){
+	std::cerr << "Error : Assignment operator of the class DoubleSparseSquareMatrix is not implemented." << std::endl;
+	exit(1);
+}