matcreateseqadj.c
Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 static char const Id[] = "$Id: matcreateseqadj.c 2962 2010-02-04 19:50:44Z scholz $\n\n";
00025 static char const Td[] = "$Today: " __FILE__ " " __DATE__ " " __TIME__ " $\n\n";
00026
00027 #define PREALLOC_DG 20
00028 #include "util.h"
00029
00030 int matcreateseqadj(int nvert,int nele,int *elevert,Mat *matout)
00031 {
00032 MagparFunctionInfoBegin;
00033
00034 int *ia,*ja;
00035 ierr = Mesh2Nodal(nele,nvert,elevert,&ia,&ja);CHKERRQ(ierr);
00036
00037
00038 for (int i=0;i<nvert;i++) {
00039 ia[i]=ia[i+1]-ia[i]+1;
00040 }
00041
00042
00043 Mat mat;
00044 ierr = MatCreateSeqAIJ(
00045 PETSC_COMM_SELF,
00046 nvert,nvert,
00047 0,ia,
00048
00049
00050
00051 &mat
00052 );CHKERRQ(ierr);
00053 ierr = MatSetFromOptions(mat);CHKERRQ(ierr);
00054
00055 ierr = PetscFree(ia);CHKERRQ(ierr);
00056 ierr = PetscFree(ja);CHKERRQ(ierr);
00057
00058 for (int i=0; i<nele; i++) {
00059 for (int l=0;l<NV;l++) {
00060 for (int j=0;j<NV;j++) {
00061
00062 ierr = MatSetValue(
00063 mat,
00064 elevert[i*NV+j],
00065 elevert[i*NV+l],
00066 1.0,
00067 ADD_VALUES
00068 );CHKERRQ(ierr);
00069 }
00070 }
00071 }
00072
00073 PetscInfo(0,"mat matrix assembly complete\n");
00074 ierr = MatAssemblyBegin(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
00075 ierr = MatAssemblyEnd(mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
00076
00077 *matout=mat;
00078
00079 MagparFunctionInfoReturn(0);
00080 }
00081