d9fcb77306
dimension data (see [issue[926]). Also added initial implementations of the new loops and named them with the prefix 'zypre_'. These new loops will be optimized further before swapping them in permanently.
104 lines
4.8 KiB
C
104 lines
4.8 KiB
C
/*BHEADER**********************************************************************
|
|
* Copyright (c) 2008, Lawrence Livermore National Security, LLC.
|
|
* Produced at the Lawrence Livermore National Laboratory.
|
|
* This file is part of HYPRE. See file COPYRIGHT for details.
|
|
*
|
|
* HYPRE is free software; you can redistribute it and/or modify it under the
|
|
* terms of the GNU Lesser General Public License (as published by the Free
|
|
* Software Foundation) version 2.1 dated February 1999.
|
|
*
|
|
* $Revision$
|
|
***********************************************************************EHEADER*/
|
|
|
|
/******************************************************************************
|
|
*
|
|
* Header info for the hypre_StructMatrix structures
|
|
*
|
|
*****************************************************************************/
|
|
|
|
#ifndef hypre_STRUCT_MATRIX_HEADER
|
|
#define hypre_STRUCT_MATRIX_HEADER
|
|
|
|
#include <assert.h>
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* hypre_StructMatrix:
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
typedef struct hypre_StructMatrix_struct
|
|
{
|
|
MPI_Comm comm;
|
|
|
|
hypre_StructGrid *grid;
|
|
hypre_StructStencil *user_stencil;
|
|
hypre_StructStencil *stencil;
|
|
HYPRE_Int num_values; /* Number of "stored" coefficients */
|
|
|
|
hypre_BoxArray *data_space;
|
|
|
|
double *data; /* Pointer to matrix data */
|
|
HYPRE_Int data_alloced; /* Boolean used for freeing data */
|
|
HYPRE_Int data_size; /* Size of matrix data */
|
|
HYPRE_Int **data_indices; /* num-boxes by stencil-size array
|
|
of indices into the data array.
|
|
data_indices[b][s] is the starting
|
|
index of matrix data corresponding
|
|
to box b and stencil coefficient s */
|
|
HYPRE_Int constant_coefficient; /* normally 0; set to 1 for
|
|
constant coefficient matrices
|
|
or 2 for constant coefficient
|
|
with variable diagonal */
|
|
|
|
HYPRE_Int symmetric; /* Is the matrix symmetric */
|
|
HYPRE_Int *symm_elements;/* Which elements are "symmetric" */
|
|
HYPRE_Int num_ghost[6]; /* Num ghost layers in each direction */
|
|
|
|
HYPRE_Int global_size; /* Total number of nonzero coeffs */
|
|
|
|
hypre_CommPkg *comm_pkg; /* Info on how to update ghost data */
|
|
|
|
HYPRE_Int ref_count;
|
|
|
|
} hypre_StructMatrix;
|
|
|
|
/*--------------------------------------------------------------------------
|
|
* Accessor macros: hypre_StructMatrix
|
|
*--------------------------------------------------------------------------*/
|
|
|
|
#define hypre_StructMatrixComm(matrix) ((matrix) -> comm)
|
|
#define hypre_StructMatrixGrid(matrix) ((matrix) -> grid)
|
|
#define hypre_StructMatrixUserStencil(matrix) ((matrix) -> user_stencil)
|
|
#define hypre_StructMatrixStencil(matrix) ((matrix) -> stencil)
|
|
#define hypre_StructMatrixNumValues(matrix) ((matrix) -> num_values)
|
|
#define hypre_StructMatrixDataSpace(matrix) ((matrix) -> data_space)
|
|
#define hypre_StructMatrixData(matrix) ((matrix) -> data)
|
|
#define hypre_StructMatrixDataAlloced(matrix) ((matrix) -> data_alloced)
|
|
#define hypre_StructMatrixDataSize(matrix) ((matrix) -> data_size)
|
|
#define hypre_StructMatrixDataIndices(matrix) ((matrix) -> data_indices)
|
|
#define hypre_StructMatrixConstantCoefficient(matrix) ((matrix) -> constant_coefficient)
|
|
#define hypre_StructMatrixSymmetric(matrix) ((matrix) -> symmetric)
|
|
#define hypre_StructMatrixSymmElements(matrix) ((matrix) -> symm_elements)
|
|
#define hypre_StructMatrixNumGhost(matrix) ((matrix) -> num_ghost)
|
|
#define hypre_StructMatrixGlobalSize(matrix) ((matrix) -> global_size)
|
|
#define hypre_StructMatrixCommPkg(matrix) ((matrix) -> comm_pkg)
|
|
#define hypre_StructMatrixRefCount(matrix) ((matrix) -> ref_count)
|
|
|
|
#define hypre_StructMatrixDim(matrix) \
|
|
hypre_StructGridDim(hypre_StructMatrixGrid(matrix))
|
|
|
|
#define hypre_StructMatrixBox(matrix, b) \
|
|
hypre_BoxArrayBox(hypre_StructMatrixDataSpace(matrix), b)
|
|
|
|
#define hypre_StructMatrixBoxData(matrix, b, s) \
|
|
(hypre_StructMatrixData(matrix) + hypre_StructMatrixDataIndices(matrix)[b][s])
|
|
|
|
#define hypre_StructMatrixBoxDataValue(matrix, b, s, index) \
|
|
(hypre_StructMatrixBoxData(matrix, b, s) + \
|
|
hypre_BoxIndexRank(hypre_StructMatrixBox(matrix, b), index))
|
|
|
|
#define hypre_CCStructMatrixBoxDataValue(matrix, b, s, index) \
|
|
(hypre_StructMatrixBoxData(matrix, b, s) + \
|
|
hypre_CCBoxIndexRank(hypre_StructMatrixBox(matrix, b), index))
|
|
|
|
#endif
|