127 lines
4.2 KiB
C
127 lines
4.2 KiB
C
/*BHEADER**********************************************************************
|
|
* (c) 1997 The Regents of the University of California
|
|
*
|
|
* See the file COPYRIGHT_and_DISCLAIMER for a complete copyright
|
|
* notice, contact person, and disclaimer.
|
|
*
|
|
* $Revision$
|
|
*********************************************************************EHEADER*/
|
|
/******************************************************************************
|
|
*
|
|
* Member functions for hypre_DistributedMatrix class for par_csr storage scheme.
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#include "./distributed_matrix.h"
|
|
|
|
#include "HYPRE_parcsr_mv.h"
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixDestroyParCSR
|
|
* Internal routine for freeing a matrix stored in Parcsr form.
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
int
|
|
hypre_DistributedMatrixDestroyParCSR( hypre_DistributedMatrix *distributed_matrix )
|
|
{
|
|
|
|
return(0);
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixInitializeParCSR
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
/* matrix must be set before calling this function*/
|
|
|
|
int
|
|
hypre_DistributedMatrixInitializeParCSR(hypre_DistributedMatrix *matrix)
|
|
{
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* Optional routines that depend on underlying storage type
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixPrintParCSR
|
|
* Internal routine for printing a matrix stored in Parcsr form.
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
int
|
|
hypre_DistributedMatrixPrintParCSR( hypre_DistributedMatrix *matrix )
|
|
{
|
|
int ierr=0;
|
|
HYPRE_ParCSRMatrix Parcsr_matrix = (HYPRE_ParCSRMatrix) hypre_DistributedMatrixLocalStorage(matrix);
|
|
|
|
HYPRE_ParCSRMatrixPrint( Parcsr_matrix, "STDOUT" );
|
|
return(ierr);
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixGetLocalRangeParCSR
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
int
|
|
hypre_DistributedMatrixGetLocalRangeParCSR( hypre_DistributedMatrix *matrix,
|
|
int *row_start,
|
|
int *row_end,
|
|
int *col_start,
|
|
int *col_end )
|
|
{
|
|
int ierr=0;
|
|
HYPRE_ParCSRMatrix Parcsr_matrix = (HYPRE_ParCSRMatrix) hypre_DistributedMatrixLocalStorage(matrix);
|
|
|
|
if (!Parcsr_matrix) return(-1);
|
|
|
|
|
|
ierr = HYPRE_ParCSRMatrixGetLocalRange( Parcsr_matrix, row_start, row_end,
|
|
col_start, col_end );
|
|
|
|
return(ierr);
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixGetRowParCSR
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
int
|
|
hypre_DistributedMatrixGetRowParCSR( hypre_DistributedMatrix *matrix,
|
|
int row,
|
|
int *size,
|
|
int **col_ind,
|
|
double **values )
|
|
{
|
|
int ierr = 0;
|
|
HYPRE_ParCSRMatrix Parcsr_matrix = (HYPRE_ParCSRMatrix) hypre_DistributedMatrixLocalStorage(matrix);
|
|
|
|
if (!Parcsr_matrix) return(-1);
|
|
|
|
ierr = HYPRE_ParCSRMatrixGetRow( Parcsr_matrix, row, size, col_ind, values);
|
|
|
|
return(ierr);
|
|
}
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_DistributedMatrixRestoreRowParCSR
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
int
|
|
hypre_DistributedMatrixRestoreRowParCSR( hypre_DistributedMatrix *matrix,
|
|
int row,
|
|
int *size,
|
|
int **col_ind,
|
|
double **values )
|
|
{
|
|
int ierr;
|
|
HYPRE_ParCSRMatrix Parcsr_matrix = (HYPRE_ParCSRMatrix) hypre_DistributedMatrixLocalStorage(matrix);
|
|
|
|
if (Parcsr_matrix == NULL) return(-1);
|
|
|
|
ierr = HYPRE_ParCSRMatrixRestoreRow( Parcsr_matrix, row, size, col_ind, values);
|
|
|
|
return(ierr);
|
|
}
|