1b377110cSBarry Smith 2d5d45c9bSBarry Smith /* 33369ce9aSBarry Smith Defines the basic matrix operations for the AIJ (compressed row) 4d5d45c9bSBarry Smith matrix storage format. 5d5d45c9bSBarry Smith */ 63369ce9aSBarry Smith 77c4f633dSBarry Smith 8c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h> /*I "petscmat.h" I*/ 9c6db04a5SJed Brown #include <petscblaslapack.h> 10c6db04a5SJed Brown #include <petscbt.h> 11af0996ceSBarry Smith #include <petsc/private/kernels/blocktranspose.h> 120716a85fSBarry Smith 130716a85fSBarry Smith PetscErrorCode MatGetColumnNorms_SeqAIJ(Mat A,NormType type,PetscReal *norms) 140716a85fSBarry Smith { 150716a85fSBarry Smith PetscErrorCode ierr; 160716a85fSBarry Smith PetscInt i,m,n; 170716a85fSBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)A->data; 180716a85fSBarry Smith 190716a85fSBarry Smith PetscFunctionBegin; 200716a85fSBarry Smith ierr = MatGetSize(A,&m,&n);CHKERRQ(ierr); 210716a85fSBarry Smith ierr = PetscMemzero(norms,n*sizeof(PetscReal));CHKERRQ(ierr); 220716a85fSBarry Smith if (type == NORM_2) { 230716a85fSBarry Smith for (i=0; i<aij->i[m]; i++) { 240716a85fSBarry Smith norms[aij->j[i]] += PetscAbsScalar(aij->a[i]*aij->a[i]); 250716a85fSBarry Smith } 260716a85fSBarry Smith } else if (type == NORM_1) { 270716a85fSBarry Smith for (i=0; i<aij->i[m]; i++) { 280716a85fSBarry Smith norms[aij->j[i]] += PetscAbsScalar(aij->a[i]); 290716a85fSBarry Smith } 300716a85fSBarry Smith } else if (type == NORM_INFINITY) { 310716a85fSBarry Smith for (i=0; i<aij->i[m]; i++) { 320716a85fSBarry Smith norms[aij->j[i]] = PetscMax(PetscAbsScalar(aij->a[i]),norms[aij->j[i]]); 330716a85fSBarry Smith } 340716a85fSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Unknown NormType"); 350716a85fSBarry Smith 360716a85fSBarry Smith if (type == NORM_2) { 378f1a2a5eSBarry Smith for (i=0; i<n; i++) norms[i] = PetscSqrtReal(norms[i]); 380716a85fSBarry Smith } 390716a85fSBarry Smith PetscFunctionReturn(0); 400716a85fSBarry Smith } 410716a85fSBarry Smith 423a062f41SBarry Smith PetscErrorCode MatFindOffBlockDiagonalEntries_SeqAIJ(Mat A,IS *is) 433a062f41SBarry Smith { 443a062f41SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 453a062f41SBarry Smith PetscInt i,m=A->rmap->n,cnt = 0, bs = A->rmap->bs; 463a062f41SBarry Smith const PetscInt *jj = a->j,*ii = a->i; 473a062f41SBarry Smith PetscInt *rows; 483a062f41SBarry Smith PetscErrorCode ierr; 493a062f41SBarry Smith 503a062f41SBarry Smith PetscFunctionBegin; 513a062f41SBarry Smith for (i=0; i<m; i++) { 523a062f41SBarry Smith if ((ii[i] != ii[i+1]) && ((jj[ii[i]] < bs*(i/bs)) || (jj[ii[i+1]-1] > bs*((i+bs)/bs)-1))) { 533a062f41SBarry Smith cnt++; 543a062f41SBarry Smith } 553a062f41SBarry Smith } 563a062f41SBarry Smith ierr = PetscMalloc1(cnt,&rows);CHKERRQ(ierr); 573a062f41SBarry Smith cnt = 0; 583a062f41SBarry Smith for (i=0; i<m; i++) { 593a062f41SBarry Smith if ((ii[i] != ii[i+1]) && ((jj[ii[i]] < bs*(i/bs)) || (jj[ii[i+1]-1] > bs*((i+bs)/bs)-1))) { 603a062f41SBarry Smith rows[cnt] = i; 613a062f41SBarry Smith cnt++; 623a062f41SBarry Smith } 633a062f41SBarry Smith } 643a062f41SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,cnt,rows,PETSC_OWN_POINTER,is);CHKERRQ(ierr); 653a062f41SBarry Smith PetscFunctionReturn(0); 663a062f41SBarry Smith } 673a062f41SBarry Smith 68f1f41ecbSJed Brown PetscErrorCode MatFindZeroDiagonals_SeqAIJ_Private(Mat A,PetscInt *nrows,PetscInt **zrows) 696ce1633cSBarry Smith { 706ce1633cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 716ce1633cSBarry Smith const MatScalar *aa = a->a; 726ce1633cSBarry Smith PetscInt i,m=A->rmap->n,cnt = 0; 73b2db7409Sstefano_zampini const PetscInt *ii = a->i,*jj = a->j,*diag; 746ce1633cSBarry Smith PetscInt *rows; 756ce1633cSBarry Smith PetscErrorCode ierr; 766ce1633cSBarry Smith 776ce1633cSBarry Smith PetscFunctionBegin; 786ce1633cSBarry Smith ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 796ce1633cSBarry Smith diag = a->diag; 806ce1633cSBarry Smith for (i=0; i<m; i++) { 81b2db7409Sstefano_zampini if ((diag[i] >= ii[i+1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) { 826ce1633cSBarry Smith cnt++; 836ce1633cSBarry Smith } 846ce1633cSBarry Smith } 85785e854fSJed Brown ierr = PetscMalloc1(cnt,&rows);CHKERRQ(ierr); 866ce1633cSBarry Smith cnt = 0; 876ce1633cSBarry Smith for (i=0; i<m; i++) { 88b2db7409Sstefano_zampini if ((diag[i] >= ii[i+1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) { 896ce1633cSBarry Smith rows[cnt++] = i; 906ce1633cSBarry Smith } 916ce1633cSBarry Smith } 92f1f41ecbSJed Brown *nrows = cnt; 93f1f41ecbSJed Brown *zrows = rows; 94f1f41ecbSJed Brown PetscFunctionReturn(0); 95f1f41ecbSJed Brown } 96f1f41ecbSJed Brown 97f1f41ecbSJed Brown PetscErrorCode MatFindZeroDiagonals_SeqAIJ(Mat A,IS *zrows) 98f1f41ecbSJed Brown { 99f1f41ecbSJed Brown PetscInt nrows,*rows; 100f1f41ecbSJed Brown PetscErrorCode ierr; 101f1f41ecbSJed Brown 102f1f41ecbSJed Brown PetscFunctionBegin; 1030298fd71SBarry Smith *zrows = NULL; 104f1f41ecbSJed Brown ierr = MatFindZeroDiagonals_SeqAIJ_Private(A,&nrows,&rows);CHKERRQ(ierr); 105ce94432eSBarry Smith ierr = ISCreateGeneral(PetscObjectComm((PetscObject)A),nrows,rows,PETSC_OWN_POINTER,zrows);CHKERRQ(ierr); 1066ce1633cSBarry Smith PetscFunctionReturn(0); 1076ce1633cSBarry Smith } 1086ce1633cSBarry Smith 109b3a44c85SBarry Smith PetscErrorCode MatFindNonzeroRows_SeqAIJ(Mat A,IS *keptrows) 110b3a44c85SBarry Smith { 111b3a44c85SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 112b3a44c85SBarry Smith const MatScalar *aa; 113b3a44c85SBarry Smith PetscInt m=A->rmap->n,cnt = 0; 114b3a44c85SBarry Smith const PetscInt *ii; 115b3a44c85SBarry Smith PetscInt n,i,j,*rows; 116b3a44c85SBarry Smith PetscErrorCode ierr; 117b3a44c85SBarry Smith 118b3a44c85SBarry Smith PetscFunctionBegin; 119b3a44c85SBarry Smith *keptrows = 0; 120b3a44c85SBarry Smith ii = a->i; 121b3a44c85SBarry Smith for (i=0; i<m; i++) { 122b3a44c85SBarry Smith n = ii[i+1] - ii[i]; 123b3a44c85SBarry Smith if (!n) { 124b3a44c85SBarry Smith cnt++; 125b3a44c85SBarry Smith goto ok1; 126b3a44c85SBarry Smith } 127b3a44c85SBarry Smith aa = a->a + ii[i]; 128b3a44c85SBarry Smith for (j=0; j<n; j++) { 129b3a44c85SBarry Smith if (aa[j] != 0.0) goto ok1; 130b3a44c85SBarry Smith } 131b3a44c85SBarry Smith cnt++; 132b3a44c85SBarry Smith ok1:; 133b3a44c85SBarry Smith } 134b3a44c85SBarry Smith if (!cnt) PetscFunctionReturn(0); 135854ce69bSBarry Smith ierr = PetscMalloc1(A->rmap->n-cnt,&rows);CHKERRQ(ierr); 136b3a44c85SBarry Smith cnt = 0; 137b3a44c85SBarry Smith for (i=0; i<m; i++) { 138b3a44c85SBarry Smith n = ii[i+1] - ii[i]; 139b3a44c85SBarry Smith if (!n) continue; 140b3a44c85SBarry Smith aa = a->a + ii[i]; 141b3a44c85SBarry Smith for (j=0; j<n; j++) { 142b3a44c85SBarry Smith if (aa[j] != 0.0) { 143b3a44c85SBarry Smith rows[cnt++] = i; 144b3a44c85SBarry Smith break; 145b3a44c85SBarry Smith } 146b3a44c85SBarry Smith } 147b3a44c85SBarry Smith } 148b3a44c85SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,cnt,rows,PETSC_OWN_POINTER,keptrows);CHKERRQ(ierr); 149b3a44c85SBarry Smith PetscFunctionReturn(0); 150b3a44c85SBarry Smith } 151b3a44c85SBarry Smith 1527087cfbeSBarry Smith PetscErrorCode MatDiagonalSet_SeqAIJ(Mat Y,Vec D,InsertMode is) 15379299369SBarry Smith { 15479299369SBarry Smith PetscErrorCode ierr; 15579299369SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*) Y->data; 15699e65526SBarry Smith PetscInt i,m = Y->rmap->n; 15799e65526SBarry Smith const PetscInt *diag; 15854f21887SBarry Smith MatScalar *aa = aij->a; 15999e65526SBarry Smith const PetscScalar *v; 160ace3abfcSBarry Smith PetscBool missing; 16179299369SBarry Smith 16279299369SBarry Smith PetscFunctionBegin; 16309f38230SBarry Smith if (Y->assembled) { 1640298fd71SBarry Smith ierr = MatMissingDiagonal_SeqAIJ(Y,&missing,NULL);CHKERRQ(ierr); 16509f38230SBarry Smith if (!missing) { 16679299369SBarry Smith diag = aij->diag; 16799e65526SBarry Smith ierr = VecGetArrayRead(D,&v);CHKERRQ(ierr); 16879299369SBarry Smith if (is == INSERT_VALUES) { 16979299369SBarry Smith for (i=0; i<m; i++) { 17079299369SBarry Smith aa[diag[i]] = v[i]; 17179299369SBarry Smith } 17279299369SBarry Smith } else { 17379299369SBarry Smith for (i=0; i<m; i++) { 17479299369SBarry Smith aa[diag[i]] += v[i]; 17579299369SBarry Smith } 17679299369SBarry Smith } 17799e65526SBarry Smith ierr = VecRestoreArrayRead(D,&v);CHKERRQ(ierr); 17879299369SBarry Smith PetscFunctionReturn(0); 17979299369SBarry Smith } 180acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(Y);CHKERRQ(ierr); 18109f38230SBarry Smith } 18209f38230SBarry Smith ierr = MatDiagonalSet_Default(Y,D,is);CHKERRQ(ierr); 18309f38230SBarry Smith PetscFunctionReturn(0); 18409f38230SBarry Smith } 18579299369SBarry Smith 1861a83f524SJed Brown PetscErrorCode MatGetRowIJ_SeqAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *m,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 18717ab2063SBarry Smith { 188416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 189dfbe8321SBarry Smith PetscErrorCode ierr; 19097f1f81fSBarry Smith PetscInt i,ishift; 19117ab2063SBarry Smith 1923a40ed3dSBarry Smith PetscFunctionBegin; 193d0f46423SBarry Smith *m = A->rmap->n; 1943a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 195bfeeae90SHong Zhang ishift = 0; 19653e63a63SBarry Smith if (symmetric && !A->structurally_symmetric) { 1972462f5fdSStefano Zampini ierr = MatToSymmetricIJ_SeqAIJ(A->rmap->n,a->i,a->j,PETSC_TRUE,ishift,oshift,(PetscInt**)ia,(PetscInt**)ja);CHKERRQ(ierr); 198bfeeae90SHong Zhang } else if (oshift == 1) { 1991a83f524SJed Brown PetscInt *tia; 200d0f46423SBarry Smith PetscInt nz = a->i[A->rmap->n]; 2013b2fbd54SBarry Smith /* malloc space and add 1 to i and j indices */ 202854ce69bSBarry Smith ierr = PetscMalloc1(A->rmap->n+1,&tia);CHKERRQ(ierr); 2031a83f524SJed Brown for (i=0; i<A->rmap->n+1; i++) tia[i] = a->i[i] + 1; 2041a83f524SJed Brown *ia = tia; 205ecc77c7aSBarry Smith if (ja) { 2061a83f524SJed Brown PetscInt *tja; 207854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&tja);CHKERRQ(ierr); 2081a83f524SJed Brown for (i=0; i<nz; i++) tja[i] = a->j[i] + 1; 2091a83f524SJed Brown *ja = tja; 210ecc77c7aSBarry Smith } 2116945ee14SBarry Smith } else { 212ecc77c7aSBarry Smith *ia = a->i; 213ecc77c7aSBarry Smith if (ja) *ja = a->j; 214a2ce50c7SBarry Smith } 2153a40ed3dSBarry Smith PetscFunctionReturn(0); 216a2744918SBarry Smith } 217a2744918SBarry Smith 2181a83f524SJed Brown PetscErrorCode MatRestoreRowIJ_SeqAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2196945ee14SBarry Smith { 220dfbe8321SBarry Smith PetscErrorCode ierr; 2216945ee14SBarry Smith 2223a40ed3dSBarry Smith PetscFunctionBegin; 2233a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 224bfeeae90SHong Zhang if ((symmetric && !A->structurally_symmetric) || oshift == 1) { 225606d414cSSatish Balay ierr = PetscFree(*ia);CHKERRQ(ierr); 226ecc77c7aSBarry Smith if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);} 227bcd2baecSBarry Smith } 2283a40ed3dSBarry Smith PetscFunctionReturn(0); 22917ab2063SBarry Smith } 23017ab2063SBarry Smith 2311a83f524SJed Brown PetscErrorCode MatGetColumnIJ_SeqAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2323b2fbd54SBarry Smith { 2333b2fbd54SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 234dfbe8321SBarry Smith PetscErrorCode ierr; 235d0f46423SBarry Smith PetscInt i,*collengths,*cia,*cja,n = A->cmap->n,m = A->rmap->n; 23697f1f81fSBarry Smith PetscInt nz = a->i[m],row,*jj,mr,col; 2373b2fbd54SBarry Smith 2383a40ed3dSBarry Smith PetscFunctionBegin; 239899cda47SBarry Smith *nn = n; 2403a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 2413b2fbd54SBarry Smith if (symmetric) { 2422462f5fdSStefano Zampini ierr = MatToSymmetricIJ_SeqAIJ(A->rmap->n,a->i,a->j,PETSC_TRUE,0,oshift,(PetscInt**)ia,(PetscInt**)ja);CHKERRQ(ierr); 2433b2fbd54SBarry Smith } else { 2441795a4d1SJed Brown ierr = PetscCalloc1(n+1,&collengths);CHKERRQ(ierr); 245854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&cia);CHKERRQ(ierr); 246854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&cja);CHKERRQ(ierr); 2473b2fbd54SBarry Smith jj = a->j; 2483b2fbd54SBarry Smith for (i=0; i<nz; i++) { 249bfeeae90SHong Zhang collengths[jj[i]]++; 2503b2fbd54SBarry Smith } 2513b2fbd54SBarry Smith cia[0] = oshift; 2523b2fbd54SBarry Smith for (i=0; i<n; i++) { 2533b2fbd54SBarry Smith cia[i+1] = cia[i] + collengths[i]; 2543b2fbd54SBarry Smith } 25597f1f81fSBarry Smith ierr = PetscMemzero(collengths,n*sizeof(PetscInt));CHKERRQ(ierr); 2563b2fbd54SBarry Smith jj = a->j; 257a93ec695SBarry Smith for (row=0; row<m; row++) { 258a93ec695SBarry Smith mr = a->i[row+1] - a->i[row]; 259a93ec695SBarry Smith for (i=0; i<mr; i++) { 260bfeeae90SHong Zhang col = *jj++; 2612205254eSKarl Rupp 2623b2fbd54SBarry Smith cja[cia[col] + collengths[col]++ - oshift] = row + oshift; 2633b2fbd54SBarry Smith } 2643b2fbd54SBarry Smith } 265606d414cSSatish Balay ierr = PetscFree(collengths);CHKERRQ(ierr); 2663b2fbd54SBarry Smith *ia = cia; *ja = cja; 2673b2fbd54SBarry Smith } 2683a40ed3dSBarry Smith PetscFunctionReturn(0); 2693b2fbd54SBarry Smith } 2703b2fbd54SBarry Smith 2711a83f524SJed Brown PetscErrorCode MatRestoreColumnIJ_SeqAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2723b2fbd54SBarry Smith { 273dfbe8321SBarry Smith PetscErrorCode ierr; 274606d414cSSatish Balay 2753a40ed3dSBarry Smith PetscFunctionBegin; 2763a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 2773b2fbd54SBarry Smith 278606d414cSSatish Balay ierr = PetscFree(*ia);CHKERRQ(ierr); 279606d414cSSatish Balay ierr = PetscFree(*ja);CHKERRQ(ierr); 2803a40ed3dSBarry Smith PetscFunctionReturn(0); 2813b2fbd54SBarry Smith } 2823b2fbd54SBarry Smith 2837cee066cSHong Zhang /* 2847cee066cSHong Zhang MatGetColumnIJ_SeqAIJ_Color() and MatRestoreColumnIJ_SeqAIJ_Color() are customized from 2857cee066cSHong Zhang MatGetColumnIJ_SeqAIJ() and MatRestoreColumnIJ_SeqAIJ() by adding an output 286040ebd07SHong Zhang spidx[], index of a->a, to be used in MatTransposeColoringCreate_SeqAIJ() and MatFDColoringCreate_SeqXAIJ() 2877cee066cSHong Zhang */ 2887cee066cSHong Zhang PetscErrorCode MatGetColumnIJ_SeqAIJ_Color(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscInt *spidx[],PetscBool *done) 2897cee066cSHong Zhang { 2907cee066cSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2917cee066cSHong Zhang PetscErrorCode ierr; 2927cee066cSHong Zhang PetscInt i,*collengths,*cia,*cja,n = A->cmap->n,m = A->rmap->n; 2937cee066cSHong Zhang PetscInt nz = a->i[m],row,*jj,mr,col; 2947cee066cSHong Zhang PetscInt *cspidx; 2957cee066cSHong Zhang 2967cee066cSHong Zhang PetscFunctionBegin; 2977cee066cSHong Zhang *nn = n; 2987cee066cSHong Zhang if (!ia) PetscFunctionReturn(0); 299625f6d37SHong Zhang 3001795a4d1SJed Brown ierr = PetscCalloc1(n+1,&collengths);CHKERRQ(ierr); 301854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&cia);CHKERRQ(ierr); 302854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&cja);CHKERRQ(ierr); 303854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&cspidx);CHKERRQ(ierr); 3047cee066cSHong Zhang jj = a->j; 3057cee066cSHong Zhang for (i=0; i<nz; i++) { 3067cee066cSHong Zhang collengths[jj[i]]++; 3077cee066cSHong Zhang } 3087cee066cSHong Zhang cia[0] = oshift; 3097cee066cSHong Zhang for (i=0; i<n; i++) { 3107cee066cSHong Zhang cia[i+1] = cia[i] + collengths[i]; 3117cee066cSHong Zhang } 3127cee066cSHong Zhang ierr = PetscMemzero(collengths,n*sizeof(PetscInt));CHKERRQ(ierr); 3137cee066cSHong Zhang jj = a->j; 3147cee066cSHong Zhang for (row=0; row<m; row++) { 3157cee066cSHong Zhang mr = a->i[row+1] - a->i[row]; 3167cee066cSHong Zhang for (i=0; i<mr; i++) { 3177cee066cSHong Zhang col = *jj++; 3187cee066cSHong Zhang cspidx[cia[col] + collengths[col] - oshift] = a->i[row] + i; /* index of a->j */ 3197cee066cSHong Zhang cja[cia[col] + collengths[col]++ - oshift] = row + oshift; 3207cee066cSHong Zhang } 3217cee066cSHong Zhang } 3227cee066cSHong Zhang ierr = PetscFree(collengths);CHKERRQ(ierr); 3237cee066cSHong Zhang *ia = cia; *ja = cja; 3247cee066cSHong Zhang *spidx = cspidx; 3257cee066cSHong Zhang PetscFunctionReturn(0); 3267cee066cSHong Zhang } 3277cee066cSHong Zhang 3287cee066cSHong Zhang PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Color(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool inodecompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscInt *spidx[],PetscBool *done) 3297cee066cSHong Zhang { 3307cee066cSHong Zhang PetscErrorCode ierr; 3317cee066cSHong Zhang 3327cee066cSHong Zhang PetscFunctionBegin; 3335243ef75SHong Zhang ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,inodecompressed,n,ia,ja,done);CHKERRQ(ierr); 3347cee066cSHong Zhang ierr = PetscFree(*spidx);CHKERRQ(ierr); 3357cee066cSHong Zhang PetscFunctionReturn(0); 3367cee066cSHong Zhang } 3377cee066cSHong Zhang 33887d4246cSBarry Smith PetscErrorCode MatSetValuesRow_SeqAIJ(Mat A,PetscInt row,const PetscScalar v[]) 33987d4246cSBarry Smith { 34087d4246cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 34187d4246cSBarry Smith PetscInt *ai = a->i; 34287d4246cSBarry Smith PetscErrorCode ierr; 34387d4246cSBarry Smith 34487d4246cSBarry Smith PetscFunctionBegin; 34587d4246cSBarry Smith ierr = PetscMemcpy(a->a+ai[row],v,(ai[row+1]-ai[row])*sizeof(PetscScalar));CHKERRQ(ierr); 34687d4246cSBarry Smith PetscFunctionReturn(0); 34787d4246cSBarry Smith } 34887d4246cSBarry Smith 349bd04181cSBarry Smith /* 350bd04181cSBarry Smith MatSeqAIJSetValuesLocalFast - An optimized version of MatSetValuesLocal() for SeqAIJ matrices with several assumptions 351bd04181cSBarry Smith 352bd04181cSBarry Smith - a single row of values is set with each call 353bd04181cSBarry Smith - no row or column indices are negative or (in error) larger than the number of rows or columns 354bd04181cSBarry Smith - the values are always added to the matrix, not set 355bd04181cSBarry Smith - no new locations are introduced in the nonzero structure of the matrix 356bd04181cSBarry Smith 3571f763a69SBarry Smith This does NOT assume the global column indices are sorted 358bd04181cSBarry Smith 3591f763a69SBarry Smith */ 360bd04181cSBarry Smith 361af0996ceSBarry Smith #include <petsc/private/isimpl.h> 362189e4007SBarry Smith PetscErrorCode MatSeqAIJSetValuesLocalFast(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 363189e4007SBarry Smith { 364189e4007SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3651f763a69SBarry Smith PetscInt low,high,t,row,nrow,i,col,l; 3661f763a69SBarry Smith const PetscInt *rp,*ai = a->i,*ailen = a->ilen,*aj = a->j; 3671f763a69SBarry Smith PetscInt lastcol = -1; 368189e4007SBarry Smith MatScalar *ap,value,*aa = a->a; 369189e4007SBarry Smith const PetscInt *ridx = A->rmap->mapping->indices,*cidx = A->cmap->mapping->indices; 370189e4007SBarry Smith 371f38dd0b8SBarry Smith row = ridx[im[0]]; 3721f763a69SBarry Smith rp = aj + ai[row]; 3731f763a69SBarry Smith ap = aa + ai[row]; 3741f763a69SBarry Smith nrow = ailen[row]; 375189e4007SBarry Smith low = 0; 376189e4007SBarry Smith high = nrow; 377189e4007SBarry Smith for (l=0; l<n; l++) { /* loop over added columns */ 378189e4007SBarry Smith col = cidx[in[l]]; 379f38dd0b8SBarry Smith value = v[l]; 380189e4007SBarry Smith 381189e4007SBarry Smith if (col <= lastcol) low = 0; 382189e4007SBarry Smith else high = nrow; 383189e4007SBarry Smith lastcol = col; 384189e4007SBarry Smith while (high-low > 5) { 385189e4007SBarry Smith t = (low+high)/2; 386189e4007SBarry Smith if (rp[t] > col) high = t; 387189e4007SBarry Smith else low = t; 388189e4007SBarry Smith } 389189e4007SBarry Smith for (i=low; i<high; i++) { 390189e4007SBarry Smith if (rp[i] == col) { 3911f763a69SBarry Smith ap[i] += value; 392189e4007SBarry Smith low = i + 1; 3931f763a69SBarry Smith break; 394189e4007SBarry Smith } 395189e4007SBarry Smith } 396189e4007SBarry Smith } 397f38dd0b8SBarry Smith return 0; 398189e4007SBarry Smith } 399189e4007SBarry Smith 40097f1f81fSBarry Smith PetscErrorCode MatSetValues_SeqAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 40117ab2063SBarry Smith { 402416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 403e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N; 40497f1f81fSBarry Smith PetscInt *imax = a->imax,*ai = a->i,*ailen = a->ilen; 4056849ba73SBarry Smith PetscErrorCode ierr; 406e2ee6c50SBarry Smith PetscInt *aj = a->j,nonew = a->nonew,lastcol = -1; 40754f21887SBarry Smith MatScalar *ap,value,*aa = a->a; 408ace3abfcSBarry Smith PetscBool ignorezeroentries = a->ignorezeroentries; 409ace3abfcSBarry Smith PetscBool roworiented = a->roworiented; 41017ab2063SBarry Smith 4113a40ed3dSBarry Smith PetscFunctionBegin; 41217ab2063SBarry Smith for (k=0; k<m; k++) { /* loop over added rows */ 413416022c9SBarry Smith row = im[k]; 4145ef9f2a5SBarry Smith if (row < 0) continue; 4152515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 416e32f2f54SBarry Smith if (row >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->n-1); 4173b2fbd54SBarry Smith #endif 418bfeeae90SHong Zhang rp = aj + ai[row]; ap = aa + ai[row]; 41917ab2063SBarry Smith rmax = imax[row]; nrow = ailen[row]; 420416022c9SBarry Smith low = 0; 421c71e6ed7SBarry Smith high = nrow; 42217ab2063SBarry Smith for (l=0; l<n; l++) { /* loop over added columns */ 4235ef9f2a5SBarry Smith if (in[l] < 0) continue; 4242515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 425e32f2f54SBarry Smith if (in[l] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap->n-1); 4263b2fbd54SBarry Smith #endif 427bfeeae90SHong Zhang col = in[l]; 4284b0e389bSBarry Smith if (roworiented) { 4295ef9f2a5SBarry Smith value = v[l + k*n]; 430bef8e0ddSBarry Smith } else { 4314b0e389bSBarry Smith value = v[k + l*m]; 4324b0e389bSBarry Smith } 43333b2b78bSBarry Smith if ((value == 0.0 && ignorezeroentries) && (is == ADD_VALUES)) continue; 43436db0b34SBarry Smith 4352205254eSKarl Rupp if (col <= lastcol) low = 0; 4362205254eSKarl Rupp else high = nrow; 437e2ee6c50SBarry Smith lastcol = col; 438416022c9SBarry Smith while (high-low > 5) { 439416022c9SBarry Smith t = (low+high)/2; 440416022c9SBarry Smith if (rp[t] > col) high = t; 441416022c9SBarry Smith else low = t; 44217ab2063SBarry Smith } 443416022c9SBarry Smith for (i=low; i<high; i++) { 44417ab2063SBarry Smith if (rp[i] > col) break; 44517ab2063SBarry Smith if (rp[i] == col) { 446416022c9SBarry Smith if (is == ADD_VALUES) ap[i] += value; 44717ab2063SBarry Smith else ap[i] = value; 448e44c0bd4SBarry Smith low = i + 1; 44917ab2063SBarry Smith goto noinsert; 45017ab2063SBarry Smith } 45117ab2063SBarry Smith } 452abc0a331SBarry Smith if (value == 0.0 && ignorezeroentries) goto noinsert; 453c2653b3dSLois Curfman McInnes if (nonew == 1) goto noinsert; 454e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero at (%D,%D) in the matrix",row,col); 455fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,A->rmap->n,1,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 456c03d1d03SSatish Balay N = nrow++ - 1; a->nz++; high++; 457416022c9SBarry Smith /* shift up all the later entries in this row */ 458416022c9SBarry Smith for (ii=N; ii>=i; ii--) { 45917ab2063SBarry Smith rp[ii+1] = rp[ii]; 46017ab2063SBarry Smith ap[ii+1] = ap[ii]; 46117ab2063SBarry Smith } 46217ab2063SBarry Smith rp[i] = col; 46317ab2063SBarry Smith ap[i] = value; 464416022c9SBarry Smith low = i + 1; 465e56f5c9eSBarry Smith A->nonzerostate++; 466e44c0bd4SBarry Smith noinsert:; 46717ab2063SBarry Smith } 46817ab2063SBarry Smith ailen[row] = nrow; 46917ab2063SBarry Smith } 4703a40ed3dSBarry Smith PetscFunctionReturn(0); 47117ab2063SBarry Smith } 47217ab2063SBarry Smith 47381824310SBarry Smith 474a77337e4SBarry Smith PetscErrorCode MatGetValues_SeqAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 4757eb43aa7SLois Curfman McInnes { 4767eb43aa7SLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 47797f1f81fSBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 47897f1f81fSBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 47954f21887SBarry Smith MatScalar *ap,*aa = a->a; 4807eb43aa7SLois Curfman McInnes 4813a40ed3dSBarry Smith PetscFunctionBegin; 4827eb43aa7SLois Curfman McInnes for (k=0; k<m; k++) { /* loop over rows */ 4837eb43aa7SLois Curfman McInnes row = im[k]; 484e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 485e32f2f54SBarry Smith if (row >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->n-1); 486bfeeae90SHong Zhang rp = aj + ai[row]; ap = aa + ai[row]; 4877eb43aa7SLois Curfman McInnes nrow = ailen[row]; 4887eb43aa7SLois Curfman McInnes for (l=0; l<n; l++) { /* loop over columns */ 489e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 490e32f2f54SBarry Smith if (in[l] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap->n-1); 491bfeeae90SHong Zhang col = in[l]; 4927eb43aa7SLois Curfman McInnes high = nrow; low = 0; /* assume unsorted */ 4937eb43aa7SLois Curfman McInnes while (high-low > 5) { 4947eb43aa7SLois Curfman McInnes t = (low+high)/2; 4957eb43aa7SLois Curfman McInnes if (rp[t] > col) high = t; 4967eb43aa7SLois Curfman McInnes else low = t; 4977eb43aa7SLois Curfman McInnes } 4987eb43aa7SLois Curfman McInnes for (i=low; i<high; i++) { 4997eb43aa7SLois Curfman McInnes if (rp[i] > col) break; 5007eb43aa7SLois Curfman McInnes if (rp[i] == col) { 501b49de8d1SLois Curfman McInnes *v++ = ap[i]; 5027eb43aa7SLois Curfman McInnes goto finished; 5037eb43aa7SLois Curfman McInnes } 5047eb43aa7SLois Curfman McInnes } 50597e567efSBarry Smith *v++ = 0.0; 5067eb43aa7SLois Curfman McInnes finished:; 5077eb43aa7SLois Curfman McInnes } 5087eb43aa7SLois Curfman McInnes } 5093a40ed3dSBarry Smith PetscFunctionReturn(0); 5107eb43aa7SLois Curfman McInnes } 5117eb43aa7SLois Curfman McInnes 51217ab2063SBarry Smith 513dfbe8321SBarry Smith PetscErrorCode MatView_SeqAIJ_Binary(Mat A,PetscViewer viewer) 51417ab2063SBarry Smith { 515416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 5166849ba73SBarry Smith PetscErrorCode ierr; 5176f69ff64SBarry Smith PetscInt i,*col_lens; 5186f69ff64SBarry Smith int fd; 519b37d52dbSMark F. Adams FILE *file; 52017ab2063SBarry Smith 5213a40ed3dSBarry Smith PetscFunctionBegin; 522b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 523854ce69bSBarry Smith ierr = PetscMalloc1(4+A->rmap->n,&col_lens);CHKERRQ(ierr); 5242205254eSKarl Rupp 5250700a824SBarry Smith col_lens[0] = MAT_FILE_CLASSID; 526d0f46423SBarry Smith col_lens[1] = A->rmap->n; 527d0f46423SBarry Smith col_lens[2] = A->cmap->n; 528416022c9SBarry Smith col_lens[3] = a->nz; 529416022c9SBarry Smith 530416022c9SBarry Smith /* store lengths of each row and write (including header) to file */ 531d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 532416022c9SBarry Smith col_lens[4+i] = a->i[i+1] - a->i[i]; 53317ab2063SBarry Smith } 534d0f46423SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+A->rmap->n,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 535606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 536416022c9SBarry Smith 537416022c9SBarry Smith /* store column indices (zero start index) */ 5386f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,a->j,a->nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 539416022c9SBarry Smith 540416022c9SBarry Smith /* store nonzero values */ 5416f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,a->a,a->nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 542b37d52dbSMark F. Adams 543b37d52dbSMark F. Adams ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 544b37d52dbSMark F. Adams if (file) { 54533d57670SJed Brown fprintf(file,"-matload_block_size %d\n",(int)PetscAbs(A->rmap->bs)); 546b37d52dbSMark F. Adams } 5473a40ed3dSBarry Smith PetscFunctionReturn(0); 54817ab2063SBarry Smith } 549416022c9SBarry Smith 55009573ac7SBarry Smith extern PetscErrorCode MatSeqAIJFactorInfo_Matlab(Mat,PetscViewer); 551cd155464SBarry Smith 552dfbe8321SBarry Smith PetscErrorCode MatView_SeqAIJ_ASCII(Mat A,PetscViewer viewer) 553416022c9SBarry Smith { 554416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 555dfbe8321SBarry Smith PetscErrorCode ierr; 55660e0710aSBarry Smith PetscInt i,j,m = A->rmap->n; 557e060cb09SBarry Smith const char *name; 558f3ef73ceSBarry Smith PetscViewerFormat format; 55917ab2063SBarry Smith 5603a40ed3dSBarry Smith PetscFunctionBegin; 561b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 56271c2f376SKris Buschelman if (format == PETSC_VIEWER_ASCII_MATLAB) { 56397f1f81fSBarry Smith PetscInt nofinalvalue = 0; 56460e0710aSBarry Smith if (m && ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != A->cmap->n-1))) { 565c337ccceSJed Brown /* Need a dummy value to ensure the dimension of the matrix. */ 566d00d2cf4SBarry Smith nofinalvalue = 1; 567d00d2cf4SBarry Smith } 568d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 569d0f46423SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",m,A->cmap->n);CHKERRQ(ierr); 57077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%% Nonzeros = %D \n",a->nz);CHKERRQ(ierr); 571fbfe6fa7SJed Brown #if defined(PETSC_USE_COMPLEX) 572fbfe6fa7SJed Brown ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%D,4);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 573fbfe6fa7SJed Brown #else 57477431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"zzz = zeros(%D,3);\n",a->nz+nofinalvalue);CHKERRQ(ierr); 575fbfe6fa7SJed Brown #endif 576b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"zzz = [\n");CHKERRQ(ierr); 57717ab2063SBarry Smith 57817ab2063SBarry Smith for (i=0; i<m; i++) { 57960e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 580aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 581a9bf72d8SJed Brown ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e %18.16e\n",i+1,a->j[j]+1,(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 58217ab2063SBarry Smith #else 58360e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e\n",i+1,a->j[j]+1,(double)a->a[j]);CHKERRQ(ierr); 58417ab2063SBarry Smith #endif 58517ab2063SBarry Smith } 58617ab2063SBarry Smith } 587d00d2cf4SBarry Smith if (nofinalvalue) { 588c337ccceSJed Brown #if defined(PETSC_USE_COMPLEX) 589c337ccceSJed Brown ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e %18.16e\n",m,A->cmap->n,0.,0.);CHKERRQ(ierr); 590c337ccceSJed Brown #else 591d0f46423SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%D %D %18.16e\n",m,A->cmap->n,0.0);CHKERRQ(ierr); 592c337ccceSJed Brown #endif 593d00d2cf4SBarry Smith } 594317d6ea6SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 595fb9695e5SSatish Balay ierr = PetscViewerASCIIPrintf(viewer,"];\n %s = spconvert(zzz);\n",name);CHKERRQ(ierr); 596d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 59768369a75SKris Buschelman } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO || format == PETSC_VIEWER_ASCII_INFO) { 598cd155464SBarry Smith PetscFunctionReturn(0); 599fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 600d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 60144cd7ae7SLois Curfman McInnes for (i=0; i<m; i++) { 60277431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 60360e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 604aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 60536db0b34SBarry Smith if (PetscImaginaryPart(a->a[j]) > 0.0 && PetscRealPart(a->a[j]) != 0.0) { 60660e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 60736db0b34SBarry Smith } else if (PetscImaginaryPart(a->a[j]) < 0.0 && PetscRealPart(a->a[j]) != 0.0) { 60860e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)-PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 60936db0b34SBarry Smith } else if (PetscRealPart(a->a[j]) != 0.0) { 61060e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)PetscRealPart(a->a[j]));CHKERRQ(ierr); 6116831982aSBarry Smith } 61244cd7ae7SLois Curfman McInnes #else 61360e0710aSBarry Smith if (a->a[j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)a->a[j]);CHKERRQ(ierr);} 61444cd7ae7SLois Curfman McInnes #endif 61544cd7ae7SLois Curfman McInnes } 616b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 61744cd7ae7SLois Curfman McInnes } 618d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 619fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_SYMMODU) { 62097f1f81fSBarry Smith PetscInt nzd=0,fshift=1,*sptr; 621d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 622854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&sptr);CHKERRQ(ierr); 623496be53dSLois Curfman McInnes for (i=0; i<m; i++) { 624496be53dSLois Curfman McInnes sptr[i] = nzd+1; 62560e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 626496be53dSLois Curfman McInnes if (a->j[j] >= i) { 627aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 62836db0b34SBarry Smith if (PetscImaginaryPart(a->a[j]) != 0.0 || PetscRealPart(a->a[j]) != 0.0) nzd++; 629496be53dSLois Curfman McInnes #else 630496be53dSLois Curfman McInnes if (a->a[j] != 0.0) nzd++; 631496be53dSLois Curfman McInnes #endif 632496be53dSLois Curfman McInnes } 633496be53dSLois Curfman McInnes } 634496be53dSLois Curfman McInnes } 6352e44a96cSLois Curfman McInnes sptr[m] = nzd+1; 63677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %D %D\n\n",m,nzd);CHKERRQ(ierr); 6372e44a96cSLois Curfman McInnes for (i=0; i<m+1; i+=6) { 6382205254eSKarl Rupp if (i+4<m) { 6392205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D %D %D %D %D %D\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4],sptr[i+5]);CHKERRQ(ierr); 6402205254eSKarl Rupp } else if (i+3<m) { 6412205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D %D %D %D %D\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4]);CHKERRQ(ierr); 6422205254eSKarl Rupp } else if (i+2<m) { 6432205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D %D %D %D\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3]);CHKERRQ(ierr); 6442205254eSKarl Rupp } else if (i+1<m) { 6452205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D %D %D\n",sptr[i],sptr[i+1],sptr[i+2]);CHKERRQ(ierr); 6462205254eSKarl Rupp } else if (i<m) { 6472205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D %D\n",sptr[i],sptr[i+1]);CHKERRQ(ierr); 6482205254eSKarl Rupp } else { 6492205254eSKarl Rupp ierr = PetscViewerASCIIPrintf(viewer," %D\n",sptr[i]);CHKERRQ(ierr); 6502205254eSKarl Rupp } 651496be53dSLois Curfman McInnes } 652b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 653606d414cSSatish Balay ierr = PetscFree(sptr);CHKERRQ(ierr); 654496be53dSLois Curfman McInnes for (i=0; i<m; i++) { 65560e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 65677431f27SBarry Smith if (a->j[j] >= i) {ierr = PetscViewerASCIIPrintf(viewer," %D ",a->j[j]+fshift);CHKERRQ(ierr);} 657496be53dSLois Curfman McInnes } 658b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 659496be53dSLois Curfman McInnes } 660b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 661496be53dSLois Curfman McInnes for (i=0; i<m; i++) { 66260e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 663496be53dSLois Curfman McInnes if (a->j[j] >= i) { 664aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 66536db0b34SBarry Smith if (PetscImaginaryPart(a->a[j]) != 0.0 || PetscRealPart(a->a[j]) != 0.0) { 66660e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %18.16e %18.16e ",(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 6676831982aSBarry Smith } 668496be53dSLois Curfman McInnes #else 66960e0710aSBarry Smith if (a->a[j] != 0.0) {ierr = PetscViewerASCIIPrintf(viewer," %18.16e ",(double)a->a[j]);CHKERRQ(ierr);} 670496be53dSLois Curfman McInnes #endif 671496be53dSLois Curfman McInnes } 672496be53dSLois Curfman McInnes } 673b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 674496be53dSLois Curfman McInnes } 675d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 676fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_DENSE) { 67797f1f81fSBarry Smith PetscInt cnt = 0,jcnt; 67887828ca2SBarry Smith PetscScalar value; 67968f1ed48SBarry Smith #if defined(PETSC_USE_COMPLEX) 68068f1ed48SBarry Smith PetscBool realonly = PETSC_TRUE; 68168f1ed48SBarry Smith 68268f1ed48SBarry Smith for (i=0; i<a->i[m]; i++) { 68368f1ed48SBarry Smith if (PetscImaginaryPart(a->a[i]) != 0.0) { 68468f1ed48SBarry Smith realonly = PETSC_FALSE; 68568f1ed48SBarry Smith break; 68668f1ed48SBarry Smith } 68768f1ed48SBarry Smith } 68868f1ed48SBarry Smith #endif 68902594712SBarry Smith 690d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 69102594712SBarry Smith for (i=0; i<m; i++) { 69202594712SBarry Smith jcnt = 0; 693d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 694e24b481bSBarry Smith if (jcnt < a->i[i+1]-a->i[i] && j == a->j[cnt]) { 69502594712SBarry Smith value = a->a[cnt++]; 696e24b481bSBarry Smith jcnt++; 69702594712SBarry Smith } else { 69802594712SBarry Smith value = 0.0; 69902594712SBarry Smith } 700aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 70168f1ed48SBarry Smith if (realonly) { 70260e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)PetscRealPart(value));CHKERRQ(ierr); 70368f1ed48SBarry Smith } else { 70460e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %7.5e+%7.5e i ",(double)PetscRealPart(value),(double)PetscImaginaryPart(value));CHKERRQ(ierr); 70568f1ed48SBarry Smith } 70602594712SBarry Smith #else 70760e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %7.5e ",(double)value);CHKERRQ(ierr); 70802594712SBarry Smith #endif 70902594712SBarry Smith } 710b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 71102594712SBarry Smith } 712d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 7133c215bfdSMatthew Knepley } else if (format == PETSC_VIEWER_ASCII_MATRIXMARKET) { 714150b93efSMatthew G. Knepley PetscInt fshift=1; 715d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 7163c215bfdSMatthew Knepley #if defined(PETSC_USE_COMPLEX) 71719303e72SJonathan Guyer ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate complex general\n");CHKERRQ(ierr); 7183c215bfdSMatthew Knepley #else 71919303e72SJonathan Guyer ierr = PetscViewerASCIIPrintf(viewer,"%%%%MatrixMarket matrix coordinate real general\n");CHKERRQ(ierr); 7203c215bfdSMatthew Knepley #endif 721d0f46423SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%D %D %D\n", m, A->cmap->n, a->nz);CHKERRQ(ierr); 7223c215bfdSMatthew Knepley for (i=0; i<m; i++) { 72360e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 7243c215bfdSMatthew Knepley #if defined(PETSC_USE_COMPLEX) 725a9a0e077SKarl Rupp ierr = PetscViewerASCIIPrintf(viewer,"%D %D %g %g\n", i+fshift,a->j[j]+fshift,(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 7263c215bfdSMatthew Knepley #else 727150b93efSMatthew G. Knepley ierr = PetscViewerASCIIPrintf(viewer,"%D %D %g\n", i+fshift, a->j[j]+fshift, (double)a->a[j]);CHKERRQ(ierr); 7283c215bfdSMatthew Knepley #endif 7293c215bfdSMatthew Knepley } 7303c215bfdSMatthew Knepley } 731d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 7323a40ed3dSBarry Smith } else { 733d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 734d5f3da31SBarry Smith if (A->factortype) { 73516cd7e1dSShri Abhyankar for (i=0; i<m; i++) { 73616cd7e1dSShri Abhyankar ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 73716cd7e1dSShri Abhyankar /* L part */ 73860e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 73916cd7e1dSShri Abhyankar #if defined(PETSC_USE_COMPLEX) 74016cd7e1dSShri Abhyankar if (PetscImaginaryPart(a->a[j]) > 0.0) { 74160e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 74216cd7e1dSShri Abhyankar } else if (PetscImaginaryPart(a->a[j]) < 0.0) { 7436712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)(-PetscImaginaryPart(a->a[j])));CHKERRQ(ierr); 74416cd7e1dSShri Abhyankar } else { 74560e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)PetscRealPart(a->a[j]));CHKERRQ(ierr); 74616cd7e1dSShri Abhyankar } 74716cd7e1dSShri Abhyankar #else 74860e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)a->a[j]);CHKERRQ(ierr); 74916cd7e1dSShri Abhyankar #endif 75016cd7e1dSShri Abhyankar } 75116cd7e1dSShri Abhyankar /* diagonal */ 75216cd7e1dSShri Abhyankar j = a->diag[i]; 75316cd7e1dSShri Abhyankar #if defined(PETSC_USE_COMPLEX) 75416cd7e1dSShri Abhyankar if (PetscImaginaryPart(a->a[j]) > 0.0) { 75560e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->j[j],(double)PetscRealPart(1.0/a->a[j]),(double)PetscImaginaryPart(1.0/a->a[j]));CHKERRQ(ierr); 75616cd7e1dSShri Abhyankar } else if (PetscImaginaryPart(a->a[j]) < 0.0) { 7576712e2f1SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->j[j],(double)PetscRealPart(1.0/a->a[j]),(double)(-PetscImaginaryPart(1.0/a->a[j])));CHKERRQ(ierr); 75816cd7e1dSShri Abhyankar } else { 75960e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)PetscRealPart(1.0/a->a[j]));CHKERRQ(ierr); 76016cd7e1dSShri Abhyankar } 76116cd7e1dSShri Abhyankar #else 76260e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)(1.0/a->a[j]));CHKERRQ(ierr); 76316cd7e1dSShri Abhyankar #endif 76416cd7e1dSShri Abhyankar 76516cd7e1dSShri Abhyankar /* U part */ 76660e0710aSBarry Smith for (j=a->diag[i+1]+1; j<a->diag[i]; j++) { 76716cd7e1dSShri Abhyankar #if defined(PETSC_USE_COMPLEX) 76816cd7e1dSShri Abhyankar if (PetscImaginaryPart(a->a[j]) > 0.0) { 76960e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 77016cd7e1dSShri Abhyankar } else if (PetscImaginaryPart(a->a[j]) < 0.0) { 77122ab088eSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)(-PetscImaginaryPart(a->a[j])));CHKERRQ(ierr); 77216cd7e1dSShri Abhyankar } else { 77360e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)PetscRealPart(a->a[j]));CHKERRQ(ierr); 77416cd7e1dSShri Abhyankar } 77516cd7e1dSShri Abhyankar #else 77660e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)a->a[j]);CHKERRQ(ierr); 77716cd7e1dSShri Abhyankar #endif 77816cd7e1dSShri Abhyankar } 77916cd7e1dSShri Abhyankar ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 78016cd7e1dSShri Abhyankar } 78116cd7e1dSShri Abhyankar } else { 78217ab2063SBarry Smith for (i=0; i<m; i++) { 78377431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 78460e0710aSBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 785aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 78636db0b34SBarry Smith if (PetscImaginaryPart(a->a[j]) > 0.0) { 78760e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 78836db0b34SBarry Smith } else if (PetscImaginaryPart(a->a[j]) < 0.0) { 78960e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i)",a->j[j],(double)PetscRealPart(a->a[j]),(double)-PetscImaginaryPart(a->a[j]));CHKERRQ(ierr); 7903a40ed3dSBarry Smith } else { 79160e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)PetscRealPart(a->a[j]));CHKERRQ(ierr); 79217ab2063SBarry Smith } 79317ab2063SBarry Smith #else 79460e0710aSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[j],(double)a->a[j]);CHKERRQ(ierr); 79517ab2063SBarry Smith #endif 79617ab2063SBarry Smith } 797b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 79817ab2063SBarry Smith } 79916cd7e1dSShri Abhyankar } 800d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 80117ab2063SBarry Smith } 802b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 8033a40ed3dSBarry Smith PetscFunctionReturn(0); 804416022c9SBarry Smith } 805416022c9SBarry Smith 8069804daf3SBarry Smith #include <petscdraw.h> 807dfbe8321SBarry Smith PetscErrorCode MatView_SeqAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 808416022c9SBarry Smith { 809480ef9eaSBarry Smith Mat A = (Mat) Aa; 810416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 811dfbe8321SBarry Smith PetscErrorCode ierr; 812383922c3SLisandro Dalcin PetscInt i,j,m = A->rmap->n; 813383922c3SLisandro Dalcin int color; 814b05fc000SLisandro Dalcin PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 815b0a32e0cSBarry Smith PetscViewer viewer; 816f3ef73ceSBarry Smith PetscViewerFormat format; 817cddf8d76SBarry Smith 8183a40ed3dSBarry Smith PetscFunctionBegin; 819480ef9eaSBarry Smith ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 820b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 821b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 822383922c3SLisandro Dalcin 823416022c9SBarry Smith /* loop over matrix elements drawing boxes */ 8240513a670SBarry Smith 825fb9695e5SSatish Balay if (format != PETSC_VIEWER_DRAW_CONTOUR) { 826383922c3SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 8270513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 828b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 829416022c9SBarry Smith for (i=0; i<m; i++) { 830cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 831bfeeae90SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 832bfeeae90SHong Zhang x_l = a->j[j]; x_r = x_l + 1.0; 83336db0b34SBarry Smith if (PetscRealPart(a->a[j]) >= 0.) continue; 834b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 835cddf8d76SBarry Smith } 836cddf8d76SBarry Smith } 837b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 838cddf8d76SBarry Smith for (i=0; i<m; i++) { 839cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 840bfeeae90SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 841bfeeae90SHong Zhang x_l = a->j[j]; x_r = x_l + 1.0; 842cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 843b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 844cddf8d76SBarry Smith } 845cddf8d76SBarry Smith } 846b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 847cddf8d76SBarry Smith for (i=0; i<m; i++) { 848cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 849bfeeae90SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 850bfeeae90SHong Zhang x_l = a->j[j]; x_r = x_l + 1.0; 85136db0b34SBarry Smith if (PetscRealPart(a->a[j]) <= 0.) continue; 852b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 853416022c9SBarry Smith } 854416022c9SBarry Smith } 855383922c3SLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 8560513a670SBarry Smith } else { 8570513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 8580513a670SBarry Smith /* first determine max of all nonzero values */ 859b05fc000SLisandro Dalcin PetscReal minv = 0.0, maxv = 0.0; 860383922c3SLisandro Dalcin PetscInt nz = a->nz, count = 0; 861b0a32e0cSBarry Smith PetscDraw popup; 8620513a670SBarry Smith 8630513a670SBarry Smith for (i=0; i<nz; i++) { 8640513a670SBarry Smith if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]); 8650513a670SBarry Smith } 866383922c3SLisandro Dalcin if (minv >= maxv) maxv = minv + PETSC_SMALL; 867b0a32e0cSBarry Smith ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 86845f3bb6eSLisandro Dalcin ierr = PetscDrawScalePopup(popup,minv,maxv);CHKERRQ(ierr); 869383922c3SLisandro Dalcin 870383922c3SLisandro Dalcin ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr); 8710513a670SBarry Smith for (i=0; i<m; i++) { 872383922c3SLisandro Dalcin y_l = m - i - 1.0; 873383922c3SLisandro Dalcin y_r = y_l + 1.0; 874bfeeae90SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 875383922c3SLisandro Dalcin x_l = a->j[j]; 876383922c3SLisandro Dalcin x_r = x_l + 1.0; 877b05fc000SLisandro Dalcin color = PetscDrawRealToColor(PetscAbsScalar(a->a[count]),minv,maxv); 878b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 8790513a670SBarry Smith count++; 8800513a670SBarry Smith } 8810513a670SBarry Smith } 882383922c3SLisandro Dalcin ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr); 8830513a670SBarry Smith } 884480ef9eaSBarry Smith PetscFunctionReturn(0); 885480ef9eaSBarry Smith } 886cddf8d76SBarry Smith 8879804daf3SBarry Smith #include <petscdraw.h> 888dfbe8321SBarry Smith PetscErrorCode MatView_SeqAIJ_Draw(Mat A,PetscViewer viewer) 889480ef9eaSBarry Smith { 890dfbe8321SBarry Smith PetscErrorCode ierr; 891b0a32e0cSBarry Smith PetscDraw draw; 89236db0b34SBarry Smith PetscReal xr,yr,xl,yl,h,w; 893ace3abfcSBarry Smith PetscBool isnull; 894480ef9eaSBarry Smith 895480ef9eaSBarry Smith PetscFunctionBegin; 896b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 897b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 898480ef9eaSBarry Smith if (isnull) PetscFunctionReturn(0); 899480ef9eaSBarry Smith 900d0f46423SBarry Smith xr = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0; 901480ef9eaSBarry Smith xr += w; yr += h; xl = -w; yl = -h; 902b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 903832b7cebSLisandro Dalcin ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 904b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqAIJ_Draw_Zoom,A);CHKERRQ(ierr); 9050298fd71SBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr); 906832b7cebSLisandro Dalcin ierr = PetscDrawSave(draw);CHKERRQ(ierr); 9073a40ed3dSBarry Smith PetscFunctionReturn(0); 908416022c9SBarry Smith } 909416022c9SBarry Smith 910dfbe8321SBarry Smith PetscErrorCode MatView_SeqAIJ(Mat A,PetscViewer viewer) 911416022c9SBarry Smith { 912dfbe8321SBarry Smith PetscErrorCode ierr; 913ace3abfcSBarry Smith PetscBool iascii,isbinary,isdraw; 914416022c9SBarry Smith 9153a40ed3dSBarry Smith PetscFunctionBegin; 916251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 917251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 918251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 919c45a1595SBarry Smith if (iascii) { 9203a40ed3dSBarry Smith ierr = MatView_SeqAIJ_ASCII(A,viewer);CHKERRQ(ierr); 9210f5bd95cSBarry Smith } else if (isbinary) { 9223a40ed3dSBarry Smith ierr = MatView_SeqAIJ_Binary(A,viewer);CHKERRQ(ierr); 9230f5bd95cSBarry Smith } else if (isdraw) { 9243a40ed3dSBarry Smith ierr = MatView_SeqAIJ_Draw(A,viewer);CHKERRQ(ierr); 92511aeaf0aSBarry Smith } 9264108e4d5SBarry Smith ierr = MatView_SeqAIJ_Inode(A,viewer);CHKERRQ(ierr); 9273a40ed3dSBarry Smith PetscFunctionReturn(0); 92817ab2063SBarry Smith } 92919bcc07fSBarry Smith 930dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode) 93117ab2063SBarry Smith { 932416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 9336849ba73SBarry Smith PetscErrorCode ierr; 93497f1f81fSBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 935d0f46423SBarry Smith PetscInt m = A->rmap->n,*ip,N,*ailen = a->ilen,rmax = 0; 93654f21887SBarry Smith MatScalar *aa = a->a,*ap; 9373447b6efSHong Zhang PetscReal ratio = 0.6; 93817ab2063SBarry Smith 9393a40ed3dSBarry Smith PetscFunctionBegin; 9403a40ed3dSBarry Smith if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 94117ab2063SBarry Smith 94243ee02c3SBarry Smith if (m) rmax = ailen[0]; /* determine row with most nonzeros */ 94317ab2063SBarry Smith for (i=1; i<m; i++) { 944416022c9SBarry Smith /* move each row back by the amount of empty slots (fshift) before it*/ 94517ab2063SBarry Smith fshift += imax[i-1] - ailen[i-1]; 94694a9d846SBarry Smith rmax = PetscMax(rmax,ailen[i]); 94717ab2063SBarry Smith if (fshift) { 948bfeeae90SHong Zhang ip = aj + ai[i]; 949bfeeae90SHong Zhang ap = aa + ai[i]; 95017ab2063SBarry Smith N = ailen[i]; 95117ab2063SBarry Smith for (j=0; j<N; j++) { 95217ab2063SBarry Smith ip[j-fshift] = ip[j]; 95317ab2063SBarry Smith ap[j-fshift] = ap[j]; 95417ab2063SBarry Smith } 95517ab2063SBarry Smith } 95617ab2063SBarry Smith ai[i] = ai[i-1] + ailen[i-1]; 95717ab2063SBarry Smith } 95817ab2063SBarry Smith if (m) { 95917ab2063SBarry Smith fshift += imax[m-1] - ailen[m-1]; 96017ab2063SBarry Smith ai[m] = ai[m-1] + ailen[m-1]; 96117ab2063SBarry Smith } 9627b083b7cSBarry Smith 96317ab2063SBarry Smith /* reset ilen and imax for each row */ 9647b083b7cSBarry Smith a->nonzerorowcnt = 0; 96517ab2063SBarry Smith for (i=0; i<m; i++) { 96617ab2063SBarry Smith ailen[i] = imax[i] = ai[i+1] - ai[i]; 9677b083b7cSBarry Smith a->nonzerorowcnt += ((ai[i+1] - ai[i]) > 0); 96817ab2063SBarry Smith } 969bfeeae90SHong Zhang a->nz = ai[m]; 97065e19b50SBarry Smith if (fshift && a->nounused == -1) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB, "Unused space detected in matrix: %D X %D, %D unneeded", m, A->cmap->n, fshift); 97117ab2063SBarry Smith 97209f38230SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 973d0f46423SBarry Smith ierr = PetscInfo4(A,"Matrix size: %D X %D; storage space: %D unneeded,%D used\n",m,A->cmap->n,fshift,a->nz);CHKERRQ(ierr); 974ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues() is %D\n",a->reallocs);CHKERRQ(ierr); 975ae15b995SBarry Smith ierr = PetscInfo1(A,"Maximum nonzeros in any row is %D\n",rmax);CHKERRQ(ierr); 9762205254eSKarl Rupp 9778e58a170SBarry Smith A->info.mallocs += a->reallocs; 978dd5f02e7SSatish Balay a->reallocs = 0; 9796712e2f1SBarry Smith A->info.nz_unneeded = (PetscReal)fshift; 98036db0b34SBarry Smith a->rmax = rmax; 9814e220ebcSLois Curfman McInnes 98211e456e1SBarry Smith ierr = MatCheckCompressedRow(A,a->nonzerorowcnt,&a->compressedrow,a->i,m,ratio);CHKERRQ(ierr); 9834108e4d5SBarry Smith ierr = MatAssemblyEnd_SeqAIJ_Inode(A,mode);CHKERRQ(ierr); 984acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(A);CHKERRQ(ierr); 9853a40ed3dSBarry Smith PetscFunctionReturn(0); 98617ab2063SBarry Smith } 98717ab2063SBarry Smith 98899cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqAIJ(Mat A) 98999cafbc1SBarry Smith { 99099cafbc1SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 99199cafbc1SBarry Smith PetscInt i,nz = a->nz; 99254f21887SBarry Smith MatScalar *aa = a->a; 993acf2f550SJed Brown PetscErrorCode ierr; 99499cafbc1SBarry Smith 99599cafbc1SBarry Smith PetscFunctionBegin; 99699cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 997acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(A);CHKERRQ(ierr); 99899cafbc1SBarry Smith PetscFunctionReturn(0); 99999cafbc1SBarry Smith } 100099cafbc1SBarry Smith 100199cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqAIJ(Mat A) 100299cafbc1SBarry Smith { 100399cafbc1SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 100499cafbc1SBarry Smith PetscInt i,nz = a->nz; 100554f21887SBarry Smith MatScalar *aa = a->a; 1006acf2f550SJed Brown PetscErrorCode ierr; 100799cafbc1SBarry Smith 100899cafbc1SBarry Smith PetscFunctionBegin; 100999cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 1010acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(A);CHKERRQ(ierr); 101199cafbc1SBarry Smith PetscFunctionReturn(0); 101299cafbc1SBarry Smith } 101399cafbc1SBarry Smith 1014dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqAIJ(Mat A) 101517ab2063SBarry Smith { 1016416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1017dfbe8321SBarry Smith PetscErrorCode ierr; 10183a40ed3dSBarry Smith 10193a40ed3dSBarry Smith PetscFunctionBegin; 1020d0f46423SBarry Smith ierr = PetscMemzero(a->a,(a->i[A->rmap->n])*sizeof(PetscScalar));CHKERRQ(ierr); 1021acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(A);CHKERRQ(ierr); 10223a40ed3dSBarry Smith PetscFunctionReturn(0); 102317ab2063SBarry Smith } 1024416022c9SBarry Smith 1025dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqAIJ(Mat A) 102617ab2063SBarry Smith { 1027416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1028dfbe8321SBarry Smith PetscErrorCode ierr; 1029d5d45c9bSBarry Smith 10303a40ed3dSBarry Smith PetscFunctionBegin; 1031aa482453SBarry Smith #if defined(PETSC_USE_LOG) 1032d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, Cols=%D, NZ=%D",A->rmap->n,A->cmap->n,a->nz); 103317ab2063SBarry Smith #endif 1034e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 10356bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 10366bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 103705b42c5fSBarry Smith ierr = PetscFree(a->diag);CHKERRQ(ierr); 1038d48dcb14SBarry Smith ierr = PetscFree(a->ibdiag);CHKERRQ(ierr); 103905b42c5fSBarry Smith ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr); 104071f1c65dSBarry Smith ierr = PetscFree3(a->idiag,a->mdiag,a->ssor_work);CHKERRQ(ierr); 104105b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 10426bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 104305b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 10446bf464f9SBarry Smith ierr = ISColoringDestroy(&a->coloring);CHKERRQ(ierr); 1045cd6b891eSBarry Smith ierr = PetscFree2(a->compressedrow.i,a->compressedrow.rindex);CHKERRQ(ierr); 10460b7e3e3dSHong Zhang ierr = PetscFree(a->matmult_abdense);CHKERRQ(ierr); 1047a30b2313SHong Zhang 10484108e4d5SBarry Smith ierr = MatDestroy_SeqAIJ_Inode(A);CHKERRQ(ierr); 1049bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 1050901853e0SKris Buschelman 1051dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 1052bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqAIJSetColumnIndices_C",NULL);CHKERRQ(ierr); 1053bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr); 1054bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 1055bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_seqsbaij_C",NULL);CHKERRQ(ierr); 1056bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_seqbaij_C",NULL);CHKERRQ(ierr); 1057bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_seqaijperm_C",NULL);CHKERRQ(ierr); 1058af8000cdSHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 1059af8000cdSHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_elemental_C",NULL);CHKERRQ(ierr); 1060af8000cdSHong Zhang #endif 106163c07aadSStefano Zampini #if defined(PETSC_HAVE_HYPRE) 106263c07aadSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_hypre_C",NULL);CHKERRQ(ierr); 10633dad0653Sstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatMatMatMult_transpose_seqaij_seqaij_C",NULL);CHKERRQ(ierr); 106463c07aadSStefano Zampini #endif 1065b49cda9fSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqaij_seqdense_C",NULL);CHKERRQ(ierr); 1066bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatIsTranspose_C",NULL);CHKERRQ(ierr); 1067bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqAIJSetPreallocation_C",NULL);CHKERRQ(ierr); 1068bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr); 1069bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatReorderForNonzeroDiagonal_C",NULL);CHKERRQ(ierr); 10703a40ed3dSBarry Smith PetscFunctionReturn(0); 107117ab2063SBarry Smith } 107217ab2063SBarry Smith 1073ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqAIJ(Mat A,MatOption op,PetscBool flg) 107417ab2063SBarry Smith { 1075416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10764846f1f5SKris Buschelman PetscErrorCode ierr; 10773a40ed3dSBarry Smith 10783a40ed3dSBarry Smith PetscFunctionBegin; 1079a65d3064SKris Buschelman switch (op) { 1080a65d3064SKris Buschelman case MAT_ROW_ORIENTED: 10814e0d8c25SBarry Smith a->roworiented = flg; 1082a65d3064SKris Buschelman break; 1083a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 1084a9817697SBarry Smith a->keepnonzeropattern = flg; 1085a65d3064SKris Buschelman break; 1086512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 1087512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 1088a65d3064SKris Buschelman break; 1089a65d3064SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 10904e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 1091a65d3064SKris Buschelman break; 1092a65d3064SKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 10934e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 1094a65d3064SKris Buschelman break; 109528b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 109628b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 109728b2fa4aSMatthew Knepley break; 1098a65d3064SKris Buschelman case MAT_IGNORE_ZERO_ENTRIES: 10994e0d8c25SBarry Smith a->ignorezeroentries = flg; 11000df259c2SBarry Smith break; 11013d472b54SHong Zhang case MAT_SPD: 1102b1646e73SJed Brown case MAT_SYMMETRIC: 1103b1646e73SJed Brown case MAT_STRUCTURALLY_SYMMETRIC: 1104b1646e73SJed Brown case MAT_HERMITIAN: 1105b1646e73SJed Brown case MAT_SYMMETRY_ETERNAL: 11065021d80fSJed Brown /* These options are handled directly by MatSetOption() */ 11075021d80fSJed Brown break; 11084e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1109a65d3064SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1110a65d3064SKris Buschelman case MAT_USE_HASH_TABLE: 1111290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 1112a65d3064SKris Buschelman break; 1113b87ac2d8SJed Brown case MAT_USE_INODES: 1114b87ac2d8SJed Brown /* Not an error because MatSetOption_SeqAIJ_Inode handles this one */ 1115b87ac2d8SJed Brown break; 1116c10200c1SHong Zhang case MAT_SUBMAT_SINGLEIS: 1117c10200c1SHong Zhang A->submat_singleis = flg; 1118c10200c1SHong Zhang break; 1119a65d3064SKris Buschelman default: 1120e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 1121a65d3064SKris Buschelman } 11224108e4d5SBarry Smith ierr = MatSetOption_SeqAIJ_Inode(A,op,flg);CHKERRQ(ierr); 11233a40ed3dSBarry Smith PetscFunctionReturn(0); 112417ab2063SBarry Smith } 112517ab2063SBarry Smith 1126dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqAIJ(Mat A,Vec v) 112717ab2063SBarry Smith { 1128416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11296849ba73SBarry Smith PetscErrorCode ierr; 1130d3e70bfaSHong Zhang PetscInt i,j,n,*ai=a->i,*aj=a->j,nz; 113135e7444dSHong Zhang PetscScalar *aa=a->a,*x,zero=0.0; 113217ab2063SBarry Smith 11333a40ed3dSBarry Smith PetscFunctionBegin; 1134d3e70bfaSHong Zhang ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 1135e32f2f54SBarry Smith if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 113635e7444dSHong Zhang 1137d5f3da31SBarry Smith if (A->factortype == MAT_FACTOR_ILU || A->factortype == MAT_FACTOR_LU) { 1138d3e70bfaSHong Zhang PetscInt *diag=a->diag; 113935e7444dSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 11402c990fa1SHong Zhang for (i=0; i<n; i++) x[i] = 1.0/aa[diag[i]]; 114135e7444dSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 114235e7444dSHong Zhang PetscFunctionReturn(0); 114335e7444dSHong Zhang } 114435e7444dSHong Zhang 11452dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 11461ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 114735e7444dSHong Zhang for (i=0; i<n; i++) { 114835e7444dSHong Zhang nz = ai[i+1] - ai[i]; 11492f5a7c2eSBarry Smith if (!nz) x[i] = 0.0; 115035e7444dSHong Zhang for (j=ai[i]; j<ai[i+1]; j++) { 115135e7444dSHong Zhang if (aj[j] == i) { 115235e7444dSHong Zhang x[i] = aa[j]; 115317ab2063SBarry Smith break; 115417ab2063SBarry Smith } 115517ab2063SBarry Smith } 115617ab2063SBarry Smith } 11571ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 11583a40ed3dSBarry Smith PetscFunctionReturn(0); 115917ab2063SBarry Smith } 116017ab2063SBarry Smith 1161c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fmult.h> 1162dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy) 116317ab2063SBarry Smith { 1164416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1165d9ca1df4SBarry Smith PetscScalar *y; 1166d9ca1df4SBarry Smith const PetscScalar *x; 1167dfbe8321SBarry Smith PetscErrorCode ierr; 1168d0f46423SBarry Smith PetscInt m = A->rmap->n; 11695c897100SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_MULTTRANSPOSEAIJ) 1170d9ca1df4SBarry Smith const MatScalar *v; 1171a77337e4SBarry Smith PetscScalar alpha; 1172d9ca1df4SBarry Smith PetscInt n,i,j; 1173d9ca1df4SBarry Smith const PetscInt *idx,*ii,*ridx=NULL; 11743447b6efSHong Zhang Mat_CompressedRow cprow = a->compressedrow; 1175ace3abfcSBarry Smith PetscBool usecprow = cprow.use; 11765c897100SBarry Smith #endif 117717ab2063SBarry Smith 11783a40ed3dSBarry Smith PetscFunctionBegin; 11792e8a6d31SBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 1180d9ca1df4SBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 11811ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 11825c897100SBarry Smith 11835c897100SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_MULTTRANSPOSEAIJ) 1184bfeeae90SHong Zhang fortranmulttransposeaddaij_(&m,x,a->i,a->j,a->a,y); 11855c897100SBarry Smith #else 11863447b6efSHong Zhang if (usecprow) { 11873447b6efSHong Zhang m = cprow.nrows; 11883447b6efSHong Zhang ii = cprow.i; 11897b2bb3b9SHong Zhang ridx = cprow.rindex; 11903447b6efSHong Zhang } else { 11913447b6efSHong Zhang ii = a->i; 11923447b6efSHong Zhang } 119317ab2063SBarry Smith for (i=0; i<m; i++) { 11943447b6efSHong Zhang idx = a->j + ii[i]; 11953447b6efSHong Zhang v = a->a + ii[i]; 11963447b6efSHong Zhang n = ii[i+1] - ii[i]; 11973447b6efSHong Zhang if (usecprow) { 11987b2bb3b9SHong Zhang alpha = x[ridx[i]]; 11993447b6efSHong Zhang } else { 120017ab2063SBarry Smith alpha = x[i]; 12013447b6efSHong Zhang } 120204fbf559SBarry Smith for (j=0; j<n; j++) y[idx[j]] += alpha*v[j]; 120317ab2063SBarry Smith } 12045c897100SBarry Smith #endif 1205dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1206d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 12071ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 12083a40ed3dSBarry Smith PetscFunctionReturn(0); 120917ab2063SBarry Smith } 121017ab2063SBarry Smith 1211dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqAIJ(Mat A,Vec xx,Vec yy) 12125c897100SBarry Smith { 1213dfbe8321SBarry Smith PetscErrorCode ierr; 12145c897100SBarry Smith 12155c897100SBarry Smith PetscFunctionBegin; 1216170fe5c8SBarry Smith ierr = VecSet(yy,0.0);CHKERRQ(ierr); 12175c897100SBarry Smith ierr = MatMultTransposeAdd_SeqAIJ(A,xx,yy,yy);CHKERRQ(ierr); 12185c897100SBarry Smith PetscFunctionReturn(0); 12195c897100SBarry Smith } 12205c897100SBarry Smith 1221c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fmult.h> 122278b84d54SShri Abhyankar 1223dfbe8321SBarry Smith PetscErrorCode MatMult_SeqAIJ(Mat A,Vec xx,Vec yy) 122417ab2063SBarry Smith { 1225416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1226d9fead3dSBarry Smith PetscScalar *y; 122754f21887SBarry Smith const PetscScalar *x; 122854f21887SBarry Smith const MatScalar *aa; 1229dfbe8321SBarry Smith PetscErrorCode ierr; 1230003131ecSBarry Smith PetscInt m=A->rmap->n; 12310298fd71SBarry Smith const PetscInt *aj,*ii,*ridx=NULL; 12327b083b7cSBarry Smith PetscInt n,i; 1233362ced78SSatish Balay PetscScalar sum; 1234ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 123517ab2063SBarry Smith 1236b6410449SSatish Balay #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 123797952fefSHong Zhang #pragma disjoint(*x,*y,*aa) 1238fee21e36SBarry Smith #endif 1239fee21e36SBarry Smith 12403a40ed3dSBarry Smith PetscFunctionBegin; 12413649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 12421ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 1243416022c9SBarry Smith ii = a->i; 12444eb6d288SHong Zhang if (usecprow) { /* use compressed row format */ 12454f390cb1SBarry Smith ierr = PetscMemzero(y,m*sizeof(PetscScalar));CHKERRQ(ierr); 124697952fefSHong Zhang m = a->compressedrow.nrows; 124797952fefSHong Zhang ii = a->compressedrow.i; 124897952fefSHong Zhang ridx = a->compressedrow.rindex; 124997952fefSHong Zhang for (i=0; i<m; i++) { 125097952fefSHong Zhang n = ii[i+1] - ii[i]; 125197952fefSHong Zhang aj = a->j + ii[i]; 125297952fefSHong Zhang aa = a->a + ii[i]; 125397952fefSHong Zhang sum = 0.0; 1254003131ecSBarry Smith PetscSparseDensePlusDot(sum,x,aa,aj,n); 1255003131ecSBarry Smith /* for (j=0; j<n; j++) sum += (*aa++)*x[*aj++]; */ 125697952fefSHong Zhang y[*ridx++] = sum; 125797952fefSHong Zhang } 125897952fefSHong Zhang } else { /* do not use compressed row format */ 1259b05257ddSBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_MULTAIJ) 12603d3eaba7SBarry Smith aj = a->j; 12613d3eaba7SBarry Smith aa = a->a; 1262b05257ddSBarry Smith fortranmultaij_(&m,x,ii,aj,aa,y); 1263b05257ddSBarry Smith #else 126417ab2063SBarry Smith for (i=0; i<m; i++) { 1265003131ecSBarry Smith n = ii[i+1] - ii[i]; 1266003131ecSBarry Smith aj = a->j + ii[i]; 1267003131ecSBarry Smith aa = a->a + ii[i]; 126817ab2063SBarry Smith sum = 0.0; 1269003131ecSBarry Smith PetscSparseDensePlusDot(sum,x,aa,aj,n); 127017ab2063SBarry Smith y[i] = sum; 127117ab2063SBarry Smith } 12728d195f9aSBarry Smith #endif 1273b05257ddSBarry Smith } 12747b083b7cSBarry Smith ierr = PetscLogFlops(2.0*a->nz - a->nonzerorowcnt);CHKERRQ(ierr); 12753649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 12761ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 12773a40ed3dSBarry Smith PetscFunctionReturn(0); 127817ab2063SBarry Smith } 127917ab2063SBarry Smith 1280b434eb95SMatthew G. Knepley PetscErrorCode MatMultMax_SeqAIJ(Mat A,Vec xx,Vec yy) 1281b434eb95SMatthew G. Knepley { 1282b434eb95SMatthew G. Knepley Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1283b434eb95SMatthew G. Knepley PetscScalar *y; 1284b434eb95SMatthew G. Knepley const PetscScalar *x; 1285b434eb95SMatthew G. Knepley const MatScalar *aa; 1286b434eb95SMatthew G. Knepley PetscErrorCode ierr; 1287b434eb95SMatthew G. Knepley PetscInt m=A->rmap->n; 1288b434eb95SMatthew G. Knepley const PetscInt *aj,*ii,*ridx=NULL; 1289b434eb95SMatthew G. Knepley PetscInt n,i,nonzerorow=0; 1290b434eb95SMatthew G. Knepley PetscScalar sum; 1291b434eb95SMatthew G. Knepley PetscBool usecprow=a->compressedrow.use; 1292b434eb95SMatthew G. Knepley 1293b434eb95SMatthew G. Knepley #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 1294b434eb95SMatthew G. Knepley #pragma disjoint(*x,*y,*aa) 1295b434eb95SMatthew G. Knepley #endif 1296b434eb95SMatthew G. Knepley 1297b434eb95SMatthew G. Knepley PetscFunctionBegin; 1298b434eb95SMatthew G. Knepley ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1299b434eb95SMatthew G. Knepley ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 1300b434eb95SMatthew G. Knepley if (usecprow) { /* use compressed row format */ 1301b434eb95SMatthew G. Knepley m = a->compressedrow.nrows; 1302b434eb95SMatthew G. Knepley ii = a->compressedrow.i; 1303b434eb95SMatthew G. Knepley ridx = a->compressedrow.rindex; 1304b434eb95SMatthew G. Knepley for (i=0; i<m; i++) { 1305b434eb95SMatthew G. Knepley n = ii[i+1] - ii[i]; 1306b434eb95SMatthew G. Knepley aj = a->j + ii[i]; 1307b434eb95SMatthew G. Knepley aa = a->a + ii[i]; 1308b434eb95SMatthew G. Knepley sum = 0.0; 1309b434eb95SMatthew G. Knepley nonzerorow += (n>0); 1310b434eb95SMatthew G. Knepley PetscSparseDenseMaxDot(sum,x,aa,aj,n); 1311b434eb95SMatthew G. Knepley /* for (j=0; j<n; j++) sum += (*aa++)*x[*aj++]; */ 1312b434eb95SMatthew G. Knepley y[*ridx++] = sum; 1313b434eb95SMatthew G. Knepley } 1314b434eb95SMatthew G. Knepley } else { /* do not use compressed row format */ 13153d3eaba7SBarry Smith ii = a->i; 1316b434eb95SMatthew G. Knepley for (i=0; i<m; i++) { 1317b434eb95SMatthew G. Knepley n = ii[i+1] - ii[i]; 1318b434eb95SMatthew G. Knepley aj = a->j + ii[i]; 1319b434eb95SMatthew G. Knepley aa = a->a + ii[i]; 1320b434eb95SMatthew G. Knepley sum = 0.0; 1321b434eb95SMatthew G. Knepley nonzerorow += (n>0); 1322b434eb95SMatthew G. Knepley PetscSparseDenseMaxDot(sum,x,aa,aj,n); 1323b434eb95SMatthew G. Knepley y[i] = sum; 1324b434eb95SMatthew G. Knepley } 1325b434eb95SMatthew G. Knepley } 1326b434eb95SMatthew G. Knepley ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 1327b434eb95SMatthew G. Knepley ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1328b434eb95SMatthew G. Knepley ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 1329b434eb95SMatthew G. Knepley PetscFunctionReturn(0); 1330b434eb95SMatthew G. Knepley } 1331b434eb95SMatthew G. Knepley 1332b434eb95SMatthew G. Knepley PetscErrorCode MatMultAddMax_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz) 1333b434eb95SMatthew G. Knepley { 1334b434eb95SMatthew G. Knepley Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1335b434eb95SMatthew G. Knepley PetscScalar *y,*z; 1336b434eb95SMatthew G. Knepley const PetscScalar *x; 1337b434eb95SMatthew G. Knepley const MatScalar *aa; 1338b434eb95SMatthew G. Knepley PetscErrorCode ierr; 1339b434eb95SMatthew G. Knepley PetscInt m = A->rmap->n,*aj,*ii; 1340b434eb95SMatthew G. Knepley PetscInt n,i,*ridx=NULL; 1341b434eb95SMatthew G. Knepley PetscScalar sum; 1342b434eb95SMatthew G. Knepley PetscBool usecprow=a->compressedrow.use; 1343b434eb95SMatthew G. Knepley 1344b434eb95SMatthew G. Knepley PetscFunctionBegin; 1345b434eb95SMatthew G. Knepley ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1346d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 1347b434eb95SMatthew G. Knepley if (usecprow) { /* use compressed row format */ 1348b434eb95SMatthew G. Knepley if (zz != yy) { 1349b434eb95SMatthew G. Knepley ierr = PetscMemcpy(z,y,m*sizeof(PetscScalar));CHKERRQ(ierr); 1350b434eb95SMatthew G. Knepley } 1351b434eb95SMatthew G. Knepley m = a->compressedrow.nrows; 1352b434eb95SMatthew G. Knepley ii = a->compressedrow.i; 1353b434eb95SMatthew G. Knepley ridx = a->compressedrow.rindex; 1354b434eb95SMatthew G. Knepley for (i=0; i<m; i++) { 1355b434eb95SMatthew G. Knepley n = ii[i+1] - ii[i]; 1356b434eb95SMatthew G. Knepley aj = a->j + ii[i]; 1357b434eb95SMatthew G. Knepley aa = a->a + ii[i]; 1358b434eb95SMatthew G. Knepley sum = y[*ridx]; 1359b434eb95SMatthew G. Knepley PetscSparseDenseMaxDot(sum,x,aa,aj,n); 1360b434eb95SMatthew G. Knepley z[*ridx++] = sum; 1361b434eb95SMatthew G. Knepley } 1362b434eb95SMatthew G. Knepley } else { /* do not use compressed row format */ 13633d3eaba7SBarry Smith ii = a->i; 1364b434eb95SMatthew G. Knepley for (i=0; i<m; i++) { 1365b434eb95SMatthew G. Knepley n = ii[i+1] - ii[i]; 1366b434eb95SMatthew G. Knepley aj = a->j + ii[i]; 1367b434eb95SMatthew G. Knepley aa = a->a + ii[i]; 1368b434eb95SMatthew G. Knepley sum = y[i]; 1369b434eb95SMatthew G. Knepley PetscSparseDenseMaxDot(sum,x,aa,aj,n); 1370b434eb95SMatthew G. Knepley z[i] = sum; 1371b434eb95SMatthew G. Knepley } 1372b434eb95SMatthew G. Knepley } 1373b434eb95SMatthew G. Knepley ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1374b434eb95SMatthew G. Knepley ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1375d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 1376b434eb95SMatthew G. Knepley PetscFunctionReturn(0); 1377b434eb95SMatthew G. Knepley } 1378b434eb95SMatthew G. Knepley 1379c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fmultadd.h> 1380dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz) 138117ab2063SBarry Smith { 1382416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1383f15663dcSBarry Smith PetscScalar *y,*z; 1384f15663dcSBarry Smith const PetscScalar *x; 138554f21887SBarry Smith const MatScalar *aa; 1386dfbe8321SBarry Smith PetscErrorCode ierr; 1387d9ca1df4SBarry Smith const PetscInt *aj,*ii,*ridx=NULL; 1388d9ca1df4SBarry Smith PetscInt m = A->rmap->n,n,i; 1389362ced78SSatish Balay PetscScalar sum; 1390ace3abfcSBarry Smith PetscBool usecprow=a->compressedrow.use; 13919ea0dfa2SSatish Balay 13923a40ed3dSBarry Smith PetscFunctionBegin; 1393f15663dcSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 1394d9ca1df4SBarry Smith ierr = VecGetArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 13954eb6d288SHong Zhang if (usecprow) { /* use compressed row format */ 13964eb6d288SHong Zhang if (zz != yy) { 13974eb6d288SHong Zhang ierr = PetscMemcpy(z,y,m*sizeof(PetscScalar));CHKERRQ(ierr); 13984eb6d288SHong Zhang } 139997952fefSHong Zhang m = a->compressedrow.nrows; 140097952fefSHong Zhang ii = a->compressedrow.i; 140197952fefSHong Zhang ridx = a->compressedrow.rindex; 140297952fefSHong Zhang for (i=0; i<m; i++) { 140397952fefSHong Zhang n = ii[i+1] - ii[i]; 140497952fefSHong Zhang aj = a->j + ii[i]; 140597952fefSHong Zhang aa = a->a + ii[i]; 140697952fefSHong Zhang sum = y[*ridx]; 1407f15663dcSBarry Smith PetscSparseDensePlusDot(sum,x,aa,aj,n); 140897952fefSHong Zhang z[*ridx++] = sum; 140997952fefSHong Zhang } 141097952fefSHong Zhang } else { /* do not use compressed row format */ 14113d3eaba7SBarry Smith ii = a->i; 1412f15663dcSBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_MULTADDAIJ) 14133d3eaba7SBarry Smith aj = a->j; 14143d3eaba7SBarry Smith aa = a->a; 1415f15663dcSBarry Smith fortranmultaddaij_(&m,x,ii,aj,aa,y,z); 1416f15663dcSBarry Smith #else 141717ab2063SBarry Smith for (i=0; i<m; i++) { 1418f15663dcSBarry Smith n = ii[i+1] - ii[i]; 1419f15663dcSBarry Smith aj = a->j + ii[i]; 1420f15663dcSBarry Smith aa = a->a + ii[i]; 142117ab2063SBarry Smith sum = y[i]; 1422f15663dcSBarry Smith PetscSparseDensePlusDot(sum,x,aa,aj,n); 142317ab2063SBarry Smith z[i] = sum; 142417ab2063SBarry Smith } 142502ab625aSSatish Balay #endif 1426f15663dcSBarry Smith } 1427dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1428f15663dcSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 1429d9ca1df4SBarry Smith ierr = VecRestoreArrayPair(yy,zz,&y,&z);CHKERRQ(ierr); 14308154be41SBarry Smith #if defined(PETSC_HAVE_CUSP) 14316b375ea7SVictor Minden /* 1432918e98c3SVictor Minden ierr = VecView(xx,0);CHKERRQ(ierr); 1433918e98c3SVictor Minden ierr = VecView(zz,0);CHKERRQ(ierr); 1434918e98c3SVictor Minden ierr = MatView(A,0);CHKERRQ(ierr); 14356b375ea7SVictor Minden */ 1436918e98c3SVictor Minden #endif 14373a40ed3dSBarry Smith PetscFunctionReturn(0); 143817ab2063SBarry Smith } 143917ab2063SBarry Smith 144017ab2063SBarry Smith /* 144117ab2063SBarry Smith Adds diagonal pointers to sparse matrix structure. 144217ab2063SBarry Smith */ 1443dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqAIJ(Mat A) 144417ab2063SBarry Smith { 1445416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14466849ba73SBarry Smith PetscErrorCode ierr; 1447d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n; 144817ab2063SBarry Smith 14493a40ed3dSBarry Smith PetscFunctionBegin; 145009f38230SBarry Smith if (!a->diag) { 1451785e854fSJed Brown ierr = PetscMalloc1(m,&a->diag);CHKERRQ(ierr); 14523bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A, m*sizeof(PetscInt));CHKERRQ(ierr); 145309f38230SBarry Smith } 1454d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 145509f38230SBarry Smith a->diag[i] = a->i[i+1]; 1456bfeeae90SHong Zhang for (j=a->i[i]; j<a->i[i+1]; j++) { 1457bfeeae90SHong Zhang if (a->j[j] == i) { 145809f38230SBarry Smith a->diag[i] = j; 145917ab2063SBarry Smith break; 146017ab2063SBarry Smith } 146117ab2063SBarry Smith } 146217ab2063SBarry Smith } 14633a40ed3dSBarry Smith PetscFunctionReturn(0); 146417ab2063SBarry Smith } 146517ab2063SBarry Smith 1466be5855fcSBarry Smith /* 1467be5855fcSBarry Smith Checks for missing diagonals 1468be5855fcSBarry Smith */ 1469ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqAIJ(Mat A,PetscBool *missing,PetscInt *d) 1470be5855fcSBarry Smith { 1471be5855fcSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14727734d3b5SMatthew G. Knepley PetscInt *diag,*ii = a->i,i; 1473be5855fcSBarry Smith 1474be5855fcSBarry Smith PetscFunctionBegin; 147509f38230SBarry Smith *missing = PETSC_FALSE; 14767734d3b5SMatthew G. Knepley if (A->rmap->n > 0 && !ii) { 147709f38230SBarry Smith *missing = PETSC_TRUE; 147809f38230SBarry Smith if (d) *d = 0; 1479955c1f14SBarry Smith PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n"); 148009f38230SBarry Smith } else { 1481f1e2ffcdSBarry Smith diag = a->diag; 1482d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 14837734d3b5SMatthew G. Knepley if (diag[i] >= ii[i+1]) { 148409f38230SBarry Smith *missing = PETSC_TRUE; 148509f38230SBarry Smith if (d) *d = i; 1486955c1f14SBarry Smith PetscInfo1(A,"Matrix is missing diagonal number %D\n",i); 1487358d2f5dSShri Abhyankar break; 148809f38230SBarry Smith } 1489be5855fcSBarry Smith } 1490be5855fcSBarry Smith } 1491be5855fcSBarry Smith PetscFunctionReturn(0); 1492be5855fcSBarry Smith } 1493be5855fcSBarry Smith 1494422a814eSBarry Smith /* 1495422a814eSBarry Smith Negative shift indicates do not generate an error if there is a zero diagonal, just invert it anyways 1496422a814eSBarry Smith */ 14977087cfbeSBarry Smith PetscErrorCode MatInvertDiagonal_SeqAIJ(Mat A,PetscScalar omega,PetscScalar fshift) 149871f1c65dSBarry Smith { 149971f1c65dSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*) A->data; 150071f1c65dSBarry Smith PetscErrorCode ierr; 1501d0f46423SBarry Smith PetscInt i,*diag,m = A->rmap->n; 150254f21887SBarry Smith MatScalar *v = a->a; 150354f21887SBarry Smith PetscScalar *idiag,*mdiag; 150471f1c65dSBarry Smith 150571f1c65dSBarry Smith PetscFunctionBegin; 150671f1c65dSBarry Smith if (a->idiagvalid) PetscFunctionReturn(0); 150771f1c65dSBarry Smith ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 150871f1c65dSBarry Smith diag = a->diag; 150971f1c65dSBarry Smith if (!a->idiag) { 1510dcca6d9dSJed Brown ierr = PetscMalloc3(m,&a->idiag,m,&a->mdiag,m,&a->ssor_work);CHKERRQ(ierr); 15113bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A, 3*m*sizeof(PetscScalar));CHKERRQ(ierr); 151271f1c65dSBarry Smith v = a->a; 151371f1c65dSBarry Smith } 151471f1c65dSBarry Smith mdiag = a->mdiag; 151571f1c65dSBarry Smith idiag = a->idiag; 151671f1c65dSBarry Smith 1517422a814eSBarry Smith if (omega == 1.0 && PetscRealPart(fshift) <= 0.0) { 151871f1c65dSBarry Smith for (i=0; i<m; i++) { 151971f1c65dSBarry Smith mdiag[i] = v[diag[i]]; 1520899639b0SHong Zhang if (!PetscAbsScalar(mdiag[i])) { /* zero diagonal */ 1521899639b0SHong Zhang if (PetscRealPart(fshift)) { 1522899639b0SHong Zhang ierr = PetscInfo1(A,"Zero diagonal on row %D\n",i);CHKERRQ(ierr); 15237b6c816cSBarry Smith A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 15247b6c816cSBarry Smith A->factorerror_zeropivot_value = 0.0; 15257b6c816cSBarry Smith A->factorerror_zeropivot_row = i; 15267b6c816cSBarry Smith } SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Zero diagonal on row %D",i); 1527899639b0SHong Zhang } 152871f1c65dSBarry Smith idiag[i] = 1.0/v[diag[i]]; 152971f1c65dSBarry Smith } 153071f1c65dSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 153171f1c65dSBarry Smith } else { 153271f1c65dSBarry Smith for (i=0; i<m; i++) { 153371f1c65dSBarry Smith mdiag[i] = v[diag[i]]; 153471f1c65dSBarry Smith idiag[i] = omega/(fshift + v[diag[i]]); 153571f1c65dSBarry Smith } 1536dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*m);CHKERRQ(ierr); 153771f1c65dSBarry Smith } 153871f1c65dSBarry Smith a->idiagvalid = PETSC_TRUE; 153971f1c65dSBarry Smith PetscFunctionReturn(0); 154071f1c65dSBarry Smith } 154171f1c65dSBarry Smith 1542c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/frelax.h> 154341f059aeSBarry Smith PetscErrorCode MatSOR_SeqAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 154417ab2063SBarry Smith { 1545416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1546e6d1f457SBarry Smith PetscScalar *x,d,sum,*t,scale; 15473d3eaba7SBarry Smith const MatScalar *v,*idiag=0,*mdiag; 154854f21887SBarry Smith const PetscScalar *b, *bs,*xb, *ts; 1549dfbe8321SBarry Smith PetscErrorCode ierr; 15503d3eaba7SBarry Smith PetscInt n,m = A->rmap->n,i; 155197f1f81fSBarry Smith const PetscInt *idx,*diag; 155217ab2063SBarry Smith 15533a40ed3dSBarry Smith PetscFunctionBegin; 1554b965ef7fSBarry Smith its = its*lits; 155591723122SBarry Smith 155671f1c65dSBarry Smith if (fshift != a->fshift || omega != a->omega) a->idiagvalid = PETSC_FALSE; /* must recompute idiag[] */ 155771f1c65dSBarry Smith if (!a->idiagvalid) {ierr = MatInvertDiagonal_SeqAIJ(A,omega,fshift);CHKERRQ(ierr);} 155871f1c65dSBarry Smith a->fshift = fshift; 155971f1c65dSBarry Smith a->omega = omega; 1560ed480e8bSBarry Smith 156171f1c65dSBarry Smith diag = a->diag; 156271f1c65dSBarry Smith t = a->ssor_work; 1563ed480e8bSBarry Smith idiag = a->idiag; 156471f1c65dSBarry Smith mdiag = a->mdiag; 1565ed480e8bSBarry Smith 15661ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 15673649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1568ed480e8bSBarry Smith /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 156917ab2063SBarry Smith if (flag == SOR_APPLY_UPPER) { 157017ab2063SBarry Smith /* apply (U + D/omega) to the vector */ 1571ed480e8bSBarry Smith bs = b; 157217ab2063SBarry Smith for (i=0; i<m; i++) { 157371f1c65dSBarry Smith d = fshift + mdiag[i]; 1574416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 1575ed480e8bSBarry Smith idx = a->j + diag[i] + 1; 1576ed480e8bSBarry Smith v = a->a + diag[i] + 1; 157717ab2063SBarry Smith sum = b[i]*d/omega; 1578003131ecSBarry Smith PetscSparseDensePlusDot(sum,bs,v,idx,n); 157917ab2063SBarry Smith x[i] = sum; 158017ab2063SBarry Smith } 15811ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15823649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1583efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 15843a40ed3dSBarry Smith PetscFunctionReturn(0); 158517ab2063SBarry Smith } 1586c783ea89SBarry Smith 15872205254eSKarl Rupp if (flag == SOR_APPLY_LOWER) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SOR_APPLY_LOWER is not implemented"); 15882205254eSKarl Rupp else if (flag & SOR_EISENSTAT) { 158917ab2063SBarry Smith /* Let A = L + U + D; where L is lower trianglar, 1590887ee2caSBarry Smith U is upper triangular, E = D/omega; This routine applies 159117ab2063SBarry Smith 159217ab2063SBarry Smith (L + E)^{-1} A (U + E)^{-1} 159317ab2063SBarry Smith 1594887ee2caSBarry Smith to a vector efficiently using Eisenstat's trick. 159517ab2063SBarry Smith */ 159617ab2063SBarry Smith scale = (2.0/omega) - 1.0; 159717ab2063SBarry Smith 159817ab2063SBarry Smith /* x = (E + U)^{-1} b */ 159917ab2063SBarry Smith for (i=m-1; i>=0; i--) { 1600416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 1601ed480e8bSBarry Smith idx = a->j + diag[i] + 1; 1602ed480e8bSBarry Smith v = a->a + diag[i] + 1; 160317ab2063SBarry Smith sum = b[i]; 1604e6d1f457SBarry Smith PetscSparseDenseMinusDot(sum,x,v,idx,n); 1605ed480e8bSBarry Smith x[i] = sum*idiag[i]; 160617ab2063SBarry Smith } 160717ab2063SBarry Smith 160817ab2063SBarry Smith /* t = b - (2*E - D)x */ 1609416022c9SBarry Smith v = a->a; 16102205254eSKarl Rupp for (i=0; i<m; i++) t[i] = b[i] - scale*(v[*diag++])*x[i]; 161117ab2063SBarry Smith 161217ab2063SBarry Smith /* t = (E + L)^{-1}t */ 1613ed480e8bSBarry Smith ts = t; 1614416022c9SBarry Smith diag = a->diag; 161517ab2063SBarry Smith for (i=0; i<m; i++) { 1616416022c9SBarry Smith n = diag[i] - a->i[i]; 1617ed480e8bSBarry Smith idx = a->j + a->i[i]; 1618ed480e8bSBarry Smith v = a->a + a->i[i]; 161917ab2063SBarry Smith sum = t[i]; 1620003131ecSBarry Smith PetscSparseDenseMinusDot(sum,ts,v,idx,n); 1621ed480e8bSBarry Smith t[i] = sum*idiag[i]; 1622733d66baSBarry Smith /* x = x + t */ 1623733d66baSBarry Smith x[i] += t[i]; 162417ab2063SBarry Smith } 162517ab2063SBarry Smith 1626dc0b31edSSatish Balay ierr = PetscLogFlops(6.0*m-1 + 2.0*a->nz);CHKERRQ(ierr); 16271ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 16283649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 16293a40ed3dSBarry Smith PetscFunctionReturn(0); 163017ab2063SBarry Smith } 163117ab2063SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 163217ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 163317ab2063SBarry Smith for (i=0; i<m; i++) { 1634416022c9SBarry Smith n = diag[i] - a->i[i]; 1635ed480e8bSBarry Smith idx = a->j + a->i[i]; 1636ed480e8bSBarry Smith v = a->a + a->i[i]; 163717ab2063SBarry Smith sum = b[i]; 1638e6d1f457SBarry Smith PetscSparseDenseMinusDot(sum,x,v,idx,n); 16395c99c7daSBarry Smith t[i] = sum; 1640ed480e8bSBarry Smith x[i] = sum*idiag[i]; 164117ab2063SBarry Smith } 16425c99c7daSBarry Smith xb = t; 1643efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 16443a40ed3dSBarry Smith } else xb = b; 164517ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 164617ab2063SBarry Smith for (i=m-1; i>=0; i--) { 1647416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 1648ed480e8bSBarry Smith idx = a->j + diag[i] + 1; 1649ed480e8bSBarry Smith v = a->a + diag[i] + 1; 165017ab2063SBarry Smith sum = xb[i]; 1651e6d1f457SBarry Smith PetscSparseDenseMinusDot(sum,x,v,idx,n); 16525c99c7daSBarry Smith if (xb == b) { 1653ed480e8bSBarry Smith x[i] = sum*idiag[i]; 16545c99c7daSBarry Smith } else { 1655b19a5dc2SMark Adams x[i] = (1-omega)*x[i] + sum*idiag[i]; /* omega in idiag */ 165617ab2063SBarry Smith } 16575c99c7daSBarry Smith } 1658b19a5dc2SMark Adams ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 165917ab2063SBarry Smith } 166017ab2063SBarry Smith its--; 166117ab2063SBarry Smith } 166217ab2063SBarry Smith while (its--) { 166317ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 166417ab2063SBarry Smith for (i=0; i<m; i++) { 1665b19a5dc2SMark Adams /* lower */ 1666b19a5dc2SMark Adams n = diag[i] - a->i[i]; 1667ed480e8bSBarry Smith idx = a->j + a->i[i]; 1668ed480e8bSBarry Smith v = a->a + a->i[i]; 166917ab2063SBarry Smith sum = b[i]; 1670e6d1f457SBarry Smith PetscSparseDenseMinusDot(sum,x,v,idx,n); 1671b19a5dc2SMark Adams t[i] = sum; /* save application of the lower-triangular part */ 1672b19a5dc2SMark Adams /* upper */ 1673b19a5dc2SMark Adams n = a->i[i+1] - diag[i] - 1; 1674b19a5dc2SMark Adams idx = a->j + diag[i] + 1; 1675b19a5dc2SMark Adams v = a->a + diag[i] + 1; 1676b19a5dc2SMark Adams PetscSparseDenseMinusDot(sum,x,v,idx,n); 1677b19a5dc2SMark Adams x[i] = (1. - omega)*x[i] + sum*idiag[i]; /* omega in idiag */ 167817ab2063SBarry Smith } 1679b19a5dc2SMark Adams xb = t; 16809f863219SBarry Smith ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1681b19a5dc2SMark Adams } else xb = b; 168217ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 168317ab2063SBarry Smith for (i=m-1; i>=0; i--) { 1684b19a5dc2SMark Adams sum = xb[i]; 1685b19a5dc2SMark Adams if (xb == b) { 1686b19a5dc2SMark Adams /* whole matrix (no checkpointing available) */ 1687416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 1688ed480e8bSBarry Smith idx = a->j + a->i[i]; 1689ed480e8bSBarry Smith v = a->a + a->i[i]; 1690e6d1f457SBarry Smith PetscSparseDenseMinusDot(sum,x,v,idx,n); 1691ed480e8bSBarry Smith x[i] = (1. - omega)*x[i] + (sum + mdiag[i]*x[i])*idiag[i]; 1692b19a5dc2SMark Adams } else { /* lower-triangular part has been saved, so only apply upper-triangular */ 1693b19a5dc2SMark Adams n = a->i[i+1] - diag[i] - 1; 1694b19a5dc2SMark Adams idx = a->j + diag[i] + 1; 1695b19a5dc2SMark Adams v = a->a + diag[i] + 1; 1696b19a5dc2SMark Adams PetscSparseDenseMinusDot(sum,x,v,idx,n); 1697b19a5dc2SMark Adams x[i] = (1. - omega)*x[i] + sum*idiag[i]; /* omega in idiag */ 169817ab2063SBarry Smith } 1699b19a5dc2SMark Adams } 1700b19a5dc2SMark Adams if (xb == b) { 17019f863219SBarry Smith ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1702b19a5dc2SMark Adams } else { 1703b19a5dc2SMark Adams ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper */ 1704b19a5dc2SMark Adams } 170517ab2063SBarry Smith } 170617ab2063SBarry Smith } 17071ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 17083649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1709365a8a9eSBarry Smith PetscFunctionReturn(0); 171017ab2063SBarry Smith } 171117ab2063SBarry Smith 17122af78befSBarry Smith 1713dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info) 171417ab2063SBarry Smith { 1715416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 17164e220ebcSLois Curfman McInnes 17173a40ed3dSBarry Smith PetscFunctionBegin; 17184e220ebcSLois Curfman McInnes info->block_size = 1.0; 17194e220ebcSLois Curfman McInnes info->nz_allocated = (double)a->maxnz; 17204e220ebcSLois Curfman McInnes info->nz_used = (double)a->nz; 17214e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(a->maxnz - a->nz); 17224e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 17238e58a170SBarry Smith info->mallocs = (double)A->info.mallocs; 17247adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 1725d5f3da31SBarry Smith if (A->factortype) { 17264e220ebcSLois Curfman McInnes info->fill_ratio_given = A->info.fill_ratio_given; 17274e220ebcSLois Curfman McInnes info->fill_ratio_needed = A->info.fill_ratio_needed; 17284e220ebcSLois Curfman McInnes info->factor_mallocs = A->info.factor_mallocs; 17294e220ebcSLois Curfman McInnes } else { 17304e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 17314e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 17324e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 17334e220ebcSLois Curfman McInnes } 17343a40ed3dSBarry Smith PetscFunctionReturn(0); 173517ab2063SBarry Smith } 173617ab2063SBarry Smith 17372b40b63fSBarry Smith PetscErrorCode MatZeroRows_SeqAIJ(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 173817ab2063SBarry Smith { 1739416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1740c7da8527SEric Chamberland PetscInt i,m = A->rmap->n - 1; 17416849ba73SBarry Smith PetscErrorCode ierr; 174297b48c8fSBarry Smith const PetscScalar *xx; 174397b48c8fSBarry Smith PetscScalar *bb; 1744c7da8527SEric Chamberland PetscInt d = 0; 174517ab2063SBarry Smith 17463a40ed3dSBarry Smith PetscFunctionBegin; 174797b48c8fSBarry Smith if (x && b) { 174897b48c8fSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 174997b48c8fSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 175097b48c8fSBarry Smith for (i=0; i<N; i++) { 175197b48c8fSBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range", rows[i]); 175297b48c8fSBarry Smith bb[rows[i]] = diag*xx[rows[i]]; 175397b48c8fSBarry Smith } 175497b48c8fSBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 175597b48c8fSBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 175697b48c8fSBarry Smith } 175797b48c8fSBarry Smith 1758a9817697SBarry Smith if (a->keepnonzeropattern) { 1759f1e2ffcdSBarry Smith for (i=0; i<N; i++) { 1760e32f2f54SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range", rows[i]); 1761bfeeae90SHong Zhang ierr = PetscMemzero(&a->a[a->i[rows[i]]],a->ilen[rows[i]]*sizeof(PetscScalar));CHKERRQ(ierr); 1762f1e2ffcdSBarry Smith } 1763f4df32b1SMatthew Knepley if (diag != 0.0) { 1764c7da8527SEric Chamberland for (i=0; i<N; i++) { 1765c7da8527SEric Chamberland d = rows[i]; 1766c7da8527SEric Chamberland if (a->diag[d] >= a->i[d+1]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry in the zeroed row %D",d); 1767c7da8527SEric Chamberland } 1768f1e2ffcdSBarry Smith for (i=0; i<N; i++) { 1769f4df32b1SMatthew Knepley a->a[a->diag[rows[i]]] = diag; 1770f1e2ffcdSBarry Smith } 1771f1e2ffcdSBarry Smith } 1772f1e2ffcdSBarry Smith } else { 1773f4df32b1SMatthew Knepley if (diag != 0.0) { 177417ab2063SBarry Smith for (i=0; i<N; i++) { 1775e32f2f54SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range", rows[i]); 17767ae801bdSBarry Smith if (a->ilen[rows[i]] > 0) { 1777416022c9SBarry Smith a->ilen[rows[i]] = 1; 1778f4df32b1SMatthew Knepley a->a[a->i[rows[i]]] = diag; 1779bfeeae90SHong Zhang a->j[a->i[rows[i]]] = rows[i]; 17807ae801bdSBarry Smith } else { /* in case row was completely empty */ 1781f4df32b1SMatthew Knepley ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],&diag,INSERT_VALUES);CHKERRQ(ierr); 178217ab2063SBarry Smith } 178317ab2063SBarry Smith } 17843a40ed3dSBarry Smith } else { 178517ab2063SBarry Smith for (i=0; i<N; i++) { 1786e32f2f54SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range", rows[i]); 1787416022c9SBarry Smith a->ilen[rows[i]] = 0; 178817ab2063SBarry Smith } 178917ab2063SBarry Smith } 1790e56f5c9eSBarry Smith A->nonzerostate++; 1791f1e2ffcdSBarry Smith } 179243a90d84SBarry Smith ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 17933a40ed3dSBarry Smith PetscFunctionReturn(0); 179417ab2063SBarry Smith } 179517ab2063SBarry Smith 17966e169961SBarry Smith PetscErrorCode MatZeroRowsColumns_SeqAIJ(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 17976e169961SBarry Smith { 17986e169961SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 17996e169961SBarry Smith PetscInt i,j,m = A->rmap->n - 1,d = 0; 18006e169961SBarry Smith PetscErrorCode ierr; 18012b40b63fSBarry Smith PetscBool missing,*zeroed,vecs = PETSC_FALSE; 18026e169961SBarry Smith const PetscScalar *xx; 18036e169961SBarry Smith PetscScalar *bb; 18046e169961SBarry Smith 18056e169961SBarry Smith PetscFunctionBegin; 18066e169961SBarry Smith if (x && b) { 18076e169961SBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 18086e169961SBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 18092b40b63fSBarry Smith vecs = PETSC_TRUE; 18106e169961SBarry Smith } 18111795a4d1SJed Brown ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr); 18126e169961SBarry Smith for (i=0; i<N; i++) { 18136e169961SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range", rows[i]); 18146e169961SBarry Smith ierr = PetscMemzero(&a->a[a->i[rows[i]]],a->ilen[rows[i]]*sizeof(PetscScalar));CHKERRQ(ierr); 18152205254eSKarl Rupp 18166e169961SBarry Smith zeroed[rows[i]] = PETSC_TRUE; 18176e169961SBarry Smith } 18186e169961SBarry Smith for (i=0; i<A->rmap->n; i++) { 18196e169961SBarry Smith if (!zeroed[i]) { 18206e169961SBarry Smith for (j=a->i[i]; j<a->i[i+1]; j++) { 18216e169961SBarry Smith if (zeroed[a->j[j]]) { 18222b40b63fSBarry Smith if (vecs) bb[i] -= a->a[j]*xx[a->j[j]]; 18236e169961SBarry Smith a->a[j] = 0.0; 18246e169961SBarry Smith } 18256e169961SBarry Smith } 18262b40b63fSBarry Smith } else if (vecs) bb[i] = diag*xx[i]; 18276e169961SBarry Smith } 18286e169961SBarry Smith if (x && b) { 18296e169961SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 18306e169961SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 18316e169961SBarry Smith } 18326e169961SBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 18336e169961SBarry Smith if (diag != 0.0) { 18346e169961SBarry Smith ierr = MatMissingDiagonal_SeqAIJ(A,&missing,&d);CHKERRQ(ierr); 18351d5a398dSstefano_zampini if (missing) { 18361d5a398dSstefano_zampini if (a->nonew) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry in row %D",d); 18371d5a398dSstefano_zampini else { 18381d5a398dSstefano_zampini for (i=0; i<N; i++) { 18391d5a398dSstefano_zampini ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],&diag,INSERT_VALUES);CHKERRQ(ierr); 18401d5a398dSstefano_zampini } 18411d5a398dSstefano_zampini } 18421d5a398dSstefano_zampini } else { 18436e169961SBarry Smith for (i=0; i<N; i++) { 18446e169961SBarry Smith a->a[a->diag[rows[i]]] = diag; 18456e169961SBarry Smith } 18466e169961SBarry Smith } 18471d5a398dSstefano_zampini } 18486e169961SBarry Smith ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18496e169961SBarry Smith PetscFunctionReturn(0); 18506e169961SBarry Smith } 18516e169961SBarry Smith 1852a77337e4SBarry Smith PetscErrorCode MatGetRow_SeqAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 185317ab2063SBarry Smith { 1854416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 185597f1f81fSBarry Smith PetscInt *itmp; 185617ab2063SBarry Smith 18573a40ed3dSBarry Smith PetscFunctionBegin; 1858e32f2f54SBarry Smith if (row < 0 || row >= A->rmap->n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D out of range",row); 185917ab2063SBarry Smith 1860416022c9SBarry Smith *nz = a->i[row+1] - a->i[row]; 1861bfeeae90SHong Zhang if (v) *v = a->a + a->i[row]; 186217ab2063SBarry Smith if (idx) { 1863bfeeae90SHong Zhang itmp = a->j + a->i[row]; 186426fbe8dcSKarl Rupp if (*nz) *idx = itmp; 186517ab2063SBarry Smith else *idx = 0; 186617ab2063SBarry Smith } 18673a40ed3dSBarry Smith PetscFunctionReturn(0); 186817ab2063SBarry Smith } 186917ab2063SBarry Smith 1870bfeeae90SHong Zhang /* remove this function? */ 1871a77337e4SBarry Smith PetscErrorCode MatRestoreRow_SeqAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 187217ab2063SBarry Smith { 18733a40ed3dSBarry Smith PetscFunctionBegin; 18743a40ed3dSBarry Smith PetscFunctionReturn(0); 187517ab2063SBarry Smith } 187617ab2063SBarry Smith 1877dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqAIJ(Mat A,NormType type,PetscReal *nrm) 187817ab2063SBarry Smith { 1879416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 188054f21887SBarry Smith MatScalar *v = a->a; 188136db0b34SBarry Smith PetscReal sum = 0.0; 18826849ba73SBarry Smith PetscErrorCode ierr; 188397f1f81fSBarry Smith PetscInt i,j; 188417ab2063SBarry Smith 18853a40ed3dSBarry Smith PetscFunctionBegin; 188617ab2063SBarry Smith if (type == NORM_FROBENIUS) { 1887570b7f6dSBarry Smith #if defined(PETSC_USE_REAL___FP16) 1888570b7f6dSBarry Smith PetscBLASInt one = 1,nz = a->nz; 1889570b7f6dSBarry Smith *nrm = BLASnrm2_(&nz,v,&one); 1890570b7f6dSBarry Smith #else 1891416022c9SBarry Smith for (i=0; i<a->nz; i++) { 189236db0b34SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 189317ab2063SBarry Smith } 18948f1a2a5eSBarry Smith *nrm = PetscSqrtReal(sum); 1895570b7f6dSBarry Smith #endif 189651f70360SJed Brown ierr = PetscLogFlops(2*a->nz);CHKERRQ(ierr); 18973a40ed3dSBarry Smith } else if (type == NORM_1) { 189836db0b34SBarry Smith PetscReal *tmp; 189997f1f81fSBarry Smith PetscInt *jj = a->j; 19001795a4d1SJed Brown ierr = PetscCalloc1(A->cmap->n+1,&tmp);CHKERRQ(ierr); 1901064f8208SBarry Smith *nrm = 0.0; 1902416022c9SBarry Smith for (j=0; j<a->nz; j++) { 1903bfeeae90SHong Zhang tmp[*jj++] += PetscAbsScalar(*v); v++; 190417ab2063SBarry Smith } 1905d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 1906064f8208SBarry Smith if (tmp[j] > *nrm) *nrm = tmp[j]; 190717ab2063SBarry Smith } 1908606d414cSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 190951f70360SJed Brown ierr = PetscLogFlops(PetscMax(a->nz-1,0));CHKERRQ(ierr); 19103a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 1911064f8208SBarry Smith *nrm = 0.0; 1912d0f46423SBarry Smith for (j=0; j<A->rmap->n; j++) { 1913bfeeae90SHong Zhang v = a->a + a->i[j]; 191417ab2063SBarry Smith sum = 0.0; 1915416022c9SBarry Smith for (i=0; i<a->i[j+1]-a->i[j]; i++) { 1916cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v++; 191717ab2063SBarry Smith } 1918064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 191917ab2063SBarry Smith } 192051f70360SJed Brown ierr = PetscLogFlops(PetscMax(a->nz-1,0));CHKERRQ(ierr); 1921f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for two norm"); 19223a40ed3dSBarry Smith PetscFunctionReturn(0); 192317ab2063SBarry Smith } 192417ab2063SBarry Smith 19254e938277SHong Zhang /* Merged from MatGetSymbolicTranspose_SeqAIJ() - replace MatGetSymbolicTranspose_SeqAIJ()? */ 19264e938277SHong Zhang PetscErrorCode MatTransposeSymbolic_SeqAIJ(Mat A,Mat *B) 19274e938277SHong Zhang { 19284e938277SHong Zhang PetscErrorCode ierr; 19294e938277SHong Zhang PetscInt i,j,anzj; 19304e938277SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 19314e938277SHong Zhang PetscInt an=A->cmap->N,am=A->rmap->N; 19324e938277SHong Zhang PetscInt *ati,*atj,*atfill,*ai=a->i,*aj=a->j; 19334e938277SHong Zhang 19344e938277SHong Zhang PetscFunctionBegin; 19354e938277SHong Zhang /* Allocate space for symbolic transpose info and work array */ 1936854ce69bSBarry Smith ierr = PetscCalloc1(an+1,&ati);CHKERRQ(ierr); 1937785e854fSJed Brown ierr = PetscMalloc1(ai[am],&atj);CHKERRQ(ierr); 1938785e854fSJed Brown ierr = PetscMalloc1(an,&atfill);CHKERRQ(ierr); 19394e938277SHong Zhang 19404e938277SHong Zhang /* Walk through aj and count ## of non-zeros in each row of A^T. */ 19414e938277SHong Zhang /* Note: offset by 1 for fast conversion into csr format. */ 194226fbe8dcSKarl Rupp for (i=0;i<ai[am];i++) ati[aj[i]+1] += 1; 19434e938277SHong Zhang /* Form ati for csr format of A^T. */ 194426fbe8dcSKarl Rupp for (i=0;i<an;i++) ati[i+1] += ati[i]; 19454e938277SHong Zhang 19464e938277SHong Zhang /* Copy ati into atfill so we have locations of the next free space in atj */ 19474e938277SHong Zhang ierr = PetscMemcpy(atfill,ati,an*sizeof(PetscInt));CHKERRQ(ierr); 19484e938277SHong Zhang 19494e938277SHong Zhang /* Walk through A row-wise and mark nonzero entries of A^T. */ 19504e938277SHong Zhang for (i=0;i<am;i++) { 19514e938277SHong Zhang anzj = ai[i+1] - ai[i]; 19524e938277SHong Zhang for (j=0;j<anzj;j++) { 19534e938277SHong Zhang atj[atfill[*aj]] = i; 19544e938277SHong Zhang atfill[*aj++] += 1; 19554e938277SHong Zhang } 19564e938277SHong Zhang } 19574e938277SHong Zhang 19584e938277SHong Zhang /* Clean up temporary space and complete requests. */ 19594e938277SHong Zhang ierr = PetscFree(atfill);CHKERRQ(ierr); 1960ce94432eSBarry Smith ierr = MatCreateSeqAIJWithArrays(PetscObjectComm((PetscObject)A),an,am,ati,atj,NULL,B);CHKERRQ(ierr); 196133d57670SJed Brown ierr = MatSetBlockSizes(*B,PetscAbs(A->cmap->bs),PetscAbs(A->rmap->bs));CHKERRQ(ierr); 1962a2f3521dSMark F. Adams 19634e938277SHong Zhang b = (Mat_SeqAIJ*)((*B)->data); 19644e938277SHong Zhang b->free_a = PETSC_FALSE; 19654e938277SHong Zhang b->free_ij = PETSC_TRUE; 19664e938277SHong Zhang b->nonew = 0; 19674e938277SHong Zhang PetscFunctionReturn(0); 19684e938277SHong Zhang } 19694e938277SHong Zhang 1970fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqAIJ(Mat A,MatReuse reuse,Mat *B) 197117ab2063SBarry Smith { 1972416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1973416022c9SBarry Smith Mat C; 19746849ba73SBarry Smith PetscErrorCode ierr; 1975d0f46423SBarry Smith PetscInt i,*aj = a->j,*ai = a->i,m = A->rmap->n,len,*col; 197654f21887SBarry Smith MatScalar *array = a->a; 197717ab2063SBarry Smith 19783a40ed3dSBarry Smith PetscFunctionBegin; 1979cf37664fSBarry Smith if (reuse == MAT_INPLACE_MATRIX && m != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Square matrix only for in-place"); 1980fc4dec0aSBarry Smith 1981cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_INPLACE_MATRIX) { 1982854ce69bSBarry Smith ierr = PetscCalloc1(1+A->cmap->n,&col);CHKERRQ(ierr); 1983bfeeae90SHong Zhang 1984bfeeae90SHong Zhang for (i=0; i<ai[m]; i++) col[aj[i]] += 1; 1985ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 1986d0f46423SBarry Smith ierr = MatSetSizes(C,A->cmap->n,m,A->cmap->n,m);CHKERRQ(ierr); 198733d57670SJed Brown ierr = MatSetBlockSizes(C,PetscAbs(A->cmap->bs),PetscAbs(A->rmap->bs));CHKERRQ(ierr); 19887adad957SLisandro Dalcin ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr); 1989ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(C,0,col);CHKERRQ(ierr); 1990606d414cSSatish Balay ierr = PetscFree(col);CHKERRQ(ierr); 1991a541d17aSBarry Smith } else { 1992a541d17aSBarry Smith C = *B; 1993a541d17aSBarry Smith } 1994a541d17aSBarry Smith 199517ab2063SBarry Smith for (i=0; i<m; i++) { 199617ab2063SBarry Smith len = ai[i+1]-ai[i]; 199787d4246cSBarry Smith ierr = MatSetValues_SeqAIJ(C,len,aj,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 1998b9b97703SBarry Smith array += len; 1999b9b97703SBarry Smith aj += len; 200017ab2063SBarry Smith } 20016d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 20026d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 200317ab2063SBarry Smith 2004cf37664fSBarry Smith if (reuse == MAT_INITIAL_MATRIX || reuse == MAT_REUSE_MATRIX) { 2005416022c9SBarry Smith *B = C; 200617ab2063SBarry Smith } else { 200728be2f97SBarry Smith ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr); 200817ab2063SBarry Smith } 20093a40ed3dSBarry Smith PetscFunctionReturn(0); 201017ab2063SBarry Smith } 201117ab2063SBarry Smith 20127087cfbeSBarry Smith PetscErrorCode MatIsTranspose_SeqAIJ(Mat A,Mat B,PetscReal tol,PetscBool *f) 2013cd0d46ebSvictorle { 20143d3eaba7SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*) A->data,*bij = (Mat_SeqAIJ*) B->data; 201554f21887SBarry Smith PetscInt *adx,*bdx,*aii,*bii,*aptr,*bptr; 201654f21887SBarry Smith MatScalar *va,*vb; 20176849ba73SBarry Smith PetscErrorCode ierr; 201897f1f81fSBarry Smith PetscInt ma,na,mb,nb, i; 2019cd0d46ebSvictorle 2020cd0d46ebSvictorle PetscFunctionBegin; 2021cd0d46ebSvictorle ierr = MatGetSize(A,&ma,&na);CHKERRQ(ierr); 2022cd0d46ebSvictorle ierr = MatGetSize(B,&mb,&nb);CHKERRQ(ierr); 20235485867bSBarry Smith if (ma!=nb || na!=mb) { 20245485867bSBarry Smith *f = PETSC_FALSE; 20255485867bSBarry Smith PetscFunctionReturn(0); 20265485867bSBarry Smith } 2027cd0d46ebSvictorle aii = aij->i; bii = bij->i; 2028cd0d46ebSvictorle adx = aij->j; bdx = bij->j; 2029cd0d46ebSvictorle va = aij->a; vb = bij->a; 2030785e854fSJed Brown ierr = PetscMalloc1(ma,&aptr);CHKERRQ(ierr); 2031785e854fSJed Brown ierr = PetscMalloc1(mb,&bptr);CHKERRQ(ierr); 2032cd0d46ebSvictorle for (i=0; i<ma; i++) aptr[i] = aii[i]; 2033cd0d46ebSvictorle for (i=0; i<mb; i++) bptr[i] = bii[i]; 2034cd0d46ebSvictorle 2035cd0d46ebSvictorle *f = PETSC_TRUE; 2036cd0d46ebSvictorle for (i=0; i<ma; i++) { 2037cd0d46ebSvictorle while (aptr[i]<aii[i+1]) { 203897f1f81fSBarry Smith PetscInt idc,idr; 20395485867bSBarry Smith PetscScalar vc,vr; 2040cd0d46ebSvictorle /* column/row index/value */ 20415485867bSBarry Smith idc = adx[aptr[i]]; 20425485867bSBarry Smith idr = bdx[bptr[idc]]; 20435485867bSBarry Smith vc = va[aptr[i]]; 20445485867bSBarry Smith vr = vb[bptr[idc]]; 20455485867bSBarry Smith if (i!=idr || PetscAbsScalar(vc-vr) > tol) { 20465485867bSBarry Smith *f = PETSC_FALSE; 20475485867bSBarry Smith goto done; 2048cd0d46ebSvictorle } else { 20495485867bSBarry Smith aptr[i]++; 20505485867bSBarry Smith if (B || i!=idc) bptr[idc]++; 2051cd0d46ebSvictorle } 2052cd0d46ebSvictorle } 2053cd0d46ebSvictorle } 2054cd0d46ebSvictorle done: 2055cd0d46ebSvictorle ierr = PetscFree(aptr);CHKERRQ(ierr); 20563aeef889SHong Zhang ierr = PetscFree(bptr);CHKERRQ(ierr); 2057cd0d46ebSvictorle PetscFunctionReturn(0); 2058cd0d46ebSvictorle } 2059cd0d46ebSvictorle 20607087cfbeSBarry Smith PetscErrorCode MatIsHermitianTranspose_SeqAIJ(Mat A,Mat B,PetscReal tol,PetscBool *f) 20611cbb95d3SBarry Smith { 20623d3eaba7SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*) A->data,*bij = (Mat_SeqAIJ*) B->data; 206354f21887SBarry Smith PetscInt *adx,*bdx,*aii,*bii,*aptr,*bptr; 206454f21887SBarry Smith MatScalar *va,*vb; 20651cbb95d3SBarry Smith PetscErrorCode ierr; 20661cbb95d3SBarry Smith PetscInt ma,na,mb,nb, i; 20671cbb95d3SBarry Smith 20681cbb95d3SBarry Smith PetscFunctionBegin; 20691cbb95d3SBarry Smith ierr = MatGetSize(A,&ma,&na);CHKERRQ(ierr); 20701cbb95d3SBarry Smith ierr = MatGetSize(B,&mb,&nb);CHKERRQ(ierr); 20711cbb95d3SBarry Smith if (ma!=nb || na!=mb) { 20721cbb95d3SBarry Smith *f = PETSC_FALSE; 20731cbb95d3SBarry Smith PetscFunctionReturn(0); 20741cbb95d3SBarry Smith } 20751cbb95d3SBarry Smith aii = aij->i; bii = bij->i; 20761cbb95d3SBarry Smith adx = aij->j; bdx = bij->j; 20771cbb95d3SBarry Smith va = aij->a; vb = bij->a; 2078785e854fSJed Brown ierr = PetscMalloc1(ma,&aptr);CHKERRQ(ierr); 2079785e854fSJed Brown ierr = PetscMalloc1(mb,&bptr);CHKERRQ(ierr); 20801cbb95d3SBarry Smith for (i=0; i<ma; i++) aptr[i] = aii[i]; 20811cbb95d3SBarry Smith for (i=0; i<mb; i++) bptr[i] = bii[i]; 20821cbb95d3SBarry Smith 20831cbb95d3SBarry Smith *f = PETSC_TRUE; 20841cbb95d3SBarry Smith for (i=0; i<ma; i++) { 20851cbb95d3SBarry Smith while (aptr[i]<aii[i+1]) { 20861cbb95d3SBarry Smith PetscInt idc,idr; 20871cbb95d3SBarry Smith PetscScalar vc,vr; 20881cbb95d3SBarry Smith /* column/row index/value */ 20891cbb95d3SBarry Smith idc = adx[aptr[i]]; 20901cbb95d3SBarry Smith idr = bdx[bptr[idc]]; 20911cbb95d3SBarry Smith vc = va[aptr[i]]; 20921cbb95d3SBarry Smith vr = vb[bptr[idc]]; 20931cbb95d3SBarry Smith if (i!=idr || PetscAbsScalar(vc-PetscConj(vr)) > tol) { 20941cbb95d3SBarry Smith *f = PETSC_FALSE; 20951cbb95d3SBarry Smith goto done; 20961cbb95d3SBarry Smith } else { 20971cbb95d3SBarry Smith aptr[i]++; 20981cbb95d3SBarry Smith if (B || i!=idc) bptr[idc]++; 20991cbb95d3SBarry Smith } 21001cbb95d3SBarry Smith } 21011cbb95d3SBarry Smith } 21021cbb95d3SBarry Smith done: 21031cbb95d3SBarry Smith ierr = PetscFree(aptr);CHKERRQ(ierr); 21041cbb95d3SBarry Smith ierr = PetscFree(bptr);CHKERRQ(ierr); 21051cbb95d3SBarry Smith PetscFunctionReturn(0); 21061cbb95d3SBarry Smith } 21071cbb95d3SBarry Smith 2108ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqAIJ(Mat A,PetscReal tol,PetscBool *f) 21099e29f15eSvictorle { 2110dfbe8321SBarry Smith PetscErrorCode ierr; 21116e111a19SKarl Rupp 21129e29f15eSvictorle PetscFunctionBegin; 21135485867bSBarry Smith ierr = MatIsTranspose_SeqAIJ(A,A,tol,f);CHKERRQ(ierr); 21149e29f15eSvictorle PetscFunctionReturn(0); 21159e29f15eSvictorle } 21169e29f15eSvictorle 2117ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqAIJ(Mat A,PetscReal tol,PetscBool *f) 21181cbb95d3SBarry Smith { 21191cbb95d3SBarry Smith PetscErrorCode ierr; 21206e111a19SKarl Rupp 21211cbb95d3SBarry Smith PetscFunctionBegin; 21221cbb95d3SBarry Smith ierr = MatIsHermitianTranspose_SeqAIJ(A,A,tol,f);CHKERRQ(ierr); 21231cbb95d3SBarry Smith PetscFunctionReturn(0); 21241cbb95d3SBarry Smith } 21251cbb95d3SBarry Smith 2126dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr) 212717ab2063SBarry Smith { 2128416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 212954f21887SBarry Smith PetscScalar *l,*r,x; 213054f21887SBarry Smith MatScalar *v; 2131dfbe8321SBarry Smith PetscErrorCode ierr; 2132d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,n = A->cmap->n,M,nz = a->nz,*jj; 213317ab2063SBarry Smith 21343a40ed3dSBarry Smith PetscFunctionBegin; 213517ab2063SBarry Smith if (ll) { 21363ea7c6a1SSatish Balay /* The local size is used so that VecMPI can be passed to this routine 21373ea7c6a1SSatish Balay by MatDiagonalScale_MPIAIJ */ 2138e1311b90SBarry Smith ierr = VecGetLocalSize(ll,&m);CHKERRQ(ierr); 2139e32f2f54SBarry Smith if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vector wrong length"); 21401ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 2141416022c9SBarry Smith v = a->a; 214217ab2063SBarry Smith for (i=0; i<m; i++) { 214317ab2063SBarry Smith x = l[i]; 2144416022c9SBarry Smith M = a->i[i+1] - a->i[i]; 21452205254eSKarl Rupp for (j=0; j<M; j++) (*v++) *= x; 214617ab2063SBarry Smith } 21471ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 2148efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 214917ab2063SBarry Smith } 215017ab2063SBarry Smith if (rr) { 2151e1311b90SBarry Smith ierr = VecGetLocalSize(rr,&n);CHKERRQ(ierr); 2152e32f2f54SBarry Smith if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vector wrong length"); 21531ebc52fbSHong Zhang ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 2154416022c9SBarry Smith v = a->a; jj = a->j; 21552205254eSKarl Rupp for (i=0; i<nz; i++) (*v++) *= r[*jj++]; 21561ebc52fbSHong Zhang ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 2157efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 215817ab2063SBarry Smith } 2159acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(A);CHKERRQ(ierr); 21603a40ed3dSBarry Smith PetscFunctionReturn(0); 216117ab2063SBarry Smith } 216217ab2063SBarry Smith 216397f1f81fSBarry Smith PetscErrorCode MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,PetscInt csize,MatReuse scall,Mat *B) 216417ab2063SBarry Smith { 2165db02288aSLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*c; 21666849ba73SBarry Smith PetscErrorCode ierr; 2167d0f46423SBarry Smith PetscInt *smap,i,k,kstart,kend,oldcols = A->cmap->n,*lens; 216897f1f81fSBarry Smith PetscInt row,mat_i,*mat_j,tcol,first,step,*mat_ilen,sum,lensi; 21695d0c19d7SBarry Smith const PetscInt *irow,*icol; 21705d0c19d7SBarry Smith PetscInt nrows,ncols; 217197f1f81fSBarry Smith PetscInt *starts,*j_new,*i_new,*aj = a->j,*ai = a->i,ii,*ailen = a->ilen; 217254f21887SBarry Smith MatScalar *a_new,*mat_a; 2173416022c9SBarry Smith Mat C; 2174cdc6f3adSToby Isaac PetscBool stride; 217517ab2063SBarry Smith 21763a40ed3dSBarry Smith PetscFunctionBegin; 217799141d43SSatish Balay 217817ab2063SBarry Smith ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 2179b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 2180b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 218117ab2063SBarry Smith 2182251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)iscol,ISSTRIDE,&stride);CHKERRQ(ierr); 2183ff718158SBarry Smith if (stride) { 2184ff718158SBarry Smith ierr = ISStrideGetInfo(iscol,&first,&step);CHKERRQ(ierr); 2185ff718158SBarry Smith } else { 2186ff718158SBarry Smith first = 0; 2187ff718158SBarry Smith step = 0; 2188ff718158SBarry Smith } 2189fee21e36SBarry Smith if (stride && step == 1) { 219002834360SBarry Smith /* special case of contiguous rows */ 2191dcca6d9dSJed Brown ierr = PetscMalloc2(nrows,&lens,nrows,&starts);CHKERRQ(ierr); 219202834360SBarry Smith /* loop over new rows determining lens and starting points */ 219302834360SBarry Smith for (i=0; i<nrows; i++) { 2194bfeeae90SHong Zhang kstart = ai[irow[i]]; 2195a2744918SBarry Smith kend = kstart + ailen[irow[i]]; 2196a91a9bebSLisandro Dalcin starts[i] = kstart; 219702834360SBarry Smith for (k=kstart; k<kend; k++) { 2198bfeeae90SHong Zhang if (aj[k] >= first) { 219902834360SBarry Smith starts[i] = k; 220002834360SBarry Smith break; 220102834360SBarry Smith } 220202834360SBarry Smith } 2203a2744918SBarry Smith sum = 0; 220402834360SBarry Smith while (k < kend) { 2205bfeeae90SHong Zhang if (aj[k++] >= first+ncols) break; 2206a2744918SBarry Smith sum++; 220702834360SBarry Smith } 2208a2744918SBarry Smith lens[i] = sum; 220902834360SBarry Smith } 221002834360SBarry Smith /* create submatrix */ 2211cddf8d76SBarry Smith if (scall == MAT_REUSE_MATRIX) { 221297f1f81fSBarry Smith PetscInt n_cols,n_rows; 221308480c60SBarry Smith ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr); 2214e32f2f54SBarry Smith if (n_rows != nrows || n_cols != ncols) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); 2215d8ced48eSBarry Smith ierr = MatZeroEntries(*B);CHKERRQ(ierr); 221608480c60SBarry Smith C = *B; 22173a40ed3dSBarry Smith } else { 22183bef6203SJed Brown PetscInt rbs,cbs; 2219ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2220f69a0ea3SMatthew Knepley ierr = MatSetSizes(C,nrows,ncols,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 22213bef6203SJed Brown ierr = ISGetBlockSize(isrow,&rbs);CHKERRQ(ierr); 22223bef6203SJed Brown ierr = ISGetBlockSize(iscol,&cbs);CHKERRQ(ierr); 22233bef6203SJed Brown ierr = MatSetBlockSizes(C,rbs,cbs);CHKERRQ(ierr); 22247adad957SLisandro Dalcin ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr); 2225ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(C,0,lens);CHKERRQ(ierr); 222608480c60SBarry Smith } 2227db02288aSLois Curfman McInnes c = (Mat_SeqAIJ*)C->data; 2228db02288aSLois Curfman McInnes 222902834360SBarry Smith /* loop over rows inserting into submatrix */ 2230db02288aSLois Curfman McInnes a_new = c->a; 2231db02288aSLois Curfman McInnes j_new = c->j; 2232db02288aSLois Curfman McInnes i_new = c->i; 2233bfeeae90SHong Zhang 223402834360SBarry Smith for (i=0; i<nrows; i++) { 2235a2744918SBarry Smith ii = starts[i]; 2236a2744918SBarry Smith lensi = lens[i]; 2237a2744918SBarry Smith for (k=0; k<lensi; k++) { 2238a2744918SBarry Smith *j_new++ = aj[ii+k] - first; 223902834360SBarry Smith } 224087828ca2SBarry Smith ierr = PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(PetscScalar));CHKERRQ(ierr); 2241a2744918SBarry Smith a_new += lensi; 2242a2744918SBarry Smith i_new[i+1] = i_new[i] + lensi; 2243a2744918SBarry Smith c->ilen[i] = lensi; 224402834360SBarry Smith } 22450e83c824SBarry Smith ierr = PetscFree2(lens,starts);CHKERRQ(ierr); 22463a40ed3dSBarry Smith } else { 224702834360SBarry Smith ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 22481795a4d1SJed Brown ierr = PetscCalloc1(oldcols,&smap);CHKERRQ(ierr); 2249854ce69bSBarry Smith ierr = PetscMalloc1(1+nrows,&lens);CHKERRQ(ierr); 22504dcab191SBarry Smith for (i=0; i<ncols; i++) { 22514dcab191SBarry Smith #if defined(PETSC_USE_DEBUG) 22524dcab191SBarry Smith if (icol[i] >= oldcols) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Requesting column beyond largest column icol[%D] %D <= A->cmap->n %D",i,icol[i],oldcols); 22534dcab191SBarry Smith #endif 22544dcab191SBarry Smith smap[icol[i]] = i+1; 22554dcab191SBarry Smith } 22564dcab191SBarry Smith 225702834360SBarry Smith /* determine lens of each row */ 225802834360SBarry Smith for (i=0; i<nrows; i++) { 2259bfeeae90SHong Zhang kstart = ai[irow[i]]; 226002834360SBarry Smith kend = kstart + a->ilen[irow[i]]; 226102834360SBarry Smith lens[i] = 0; 226202834360SBarry Smith for (k=kstart; k<kend; k++) { 2263bfeeae90SHong Zhang if (smap[aj[k]]) { 226402834360SBarry Smith lens[i]++; 226502834360SBarry Smith } 226602834360SBarry Smith } 226702834360SBarry Smith } 226817ab2063SBarry Smith /* Create and fill new matrix */ 2269a2744918SBarry Smith if (scall == MAT_REUSE_MATRIX) { 2270ace3abfcSBarry Smith PetscBool equal; 22710f5bd95cSBarry Smith 227299141d43SSatish Balay c = (Mat_SeqAIJ*)((*B)->data); 2273e32f2f54SBarry Smith if ((*B)->rmap->n != nrows || (*B)->cmap->n != ncols) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong size"); 2274d0f46423SBarry Smith ierr = PetscMemcmp(c->ilen,lens,(*B)->rmap->n*sizeof(PetscInt),&equal);CHKERRQ(ierr); 2275f23aa3ddSBarry Smith if (!equal) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Cannot reuse matrix. wrong no of nonzeros"); 2276d0f46423SBarry Smith ierr = PetscMemzero(c->ilen,(*B)->rmap->n*sizeof(PetscInt));CHKERRQ(ierr); 227708480c60SBarry Smith C = *B; 22783a40ed3dSBarry Smith } else { 22793bef6203SJed Brown PetscInt rbs,cbs; 2280ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr); 2281f69a0ea3SMatthew Knepley ierr = MatSetSizes(C,nrows,ncols,PETSC_DETERMINE,PETSC_DETERMINE);CHKERRQ(ierr); 22823bef6203SJed Brown ierr = ISGetBlockSize(isrow,&rbs);CHKERRQ(ierr); 22833bef6203SJed Brown ierr = ISGetBlockSize(iscol,&cbs);CHKERRQ(ierr); 22843bef6203SJed Brown ierr = MatSetBlockSizes(C,rbs,cbs);CHKERRQ(ierr); 22857adad957SLisandro Dalcin ierr = MatSetType(C,((PetscObject)A)->type_name);CHKERRQ(ierr); 2286ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(C,0,lens);CHKERRQ(ierr); 228708480c60SBarry Smith } 228899141d43SSatish Balay c = (Mat_SeqAIJ*)(C->data); 228917ab2063SBarry Smith for (i=0; i<nrows; i++) { 229099141d43SSatish Balay row = irow[i]; 2291bfeeae90SHong Zhang kstart = ai[row]; 229299141d43SSatish Balay kend = kstart + a->ilen[row]; 2293bfeeae90SHong Zhang mat_i = c->i[i]; 229499141d43SSatish Balay mat_j = c->j + mat_i; 229599141d43SSatish Balay mat_a = c->a + mat_i; 229699141d43SSatish Balay mat_ilen = c->ilen + i; 229717ab2063SBarry Smith for (k=kstart; k<kend; k++) { 2298bfeeae90SHong Zhang if ((tcol=smap[a->j[k]])) { 2299ed480e8bSBarry Smith *mat_j++ = tcol - 1; 230099141d43SSatish Balay *mat_a++ = a->a[k]; 230199141d43SSatish Balay (*mat_ilen)++; 230299141d43SSatish Balay 230317ab2063SBarry Smith } 230417ab2063SBarry Smith } 230517ab2063SBarry Smith } 230602834360SBarry Smith /* Free work space */ 230702834360SBarry Smith ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 2308606d414cSSatish Balay ierr = PetscFree(smap);CHKERRQ(ierr); 2309606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 2310cdc6f3adSToby Isaac /* sort */ 2311cdc6f3adSToby Isaac for (i = 0; i < nrows; i++) { 2312cdc6f3adSToby Isaac PetscInt ilen; 2313cdc6f3adSToby Isaac 2314cdc6f3adSToby Isaac mat_i = c->i[i]; 2315cdc6f3adSToby Isaac mat_j = c->j + mat_i; 2316cdc6f3adSToby Isaac mat_a = c->a + mat_i; 2317cdc6f3adSToby Isaac ilen = c->ilen[i]; 2318390e1bf2SBarry Smith ierr = PetscSortIntWithScalarArray(ilen,mat_j,mat_a);CHKERRQ(ierr); 2319cdc6f3adSToby Isaac } 232002834360SBarry Smith } 23216d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23226d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 232317ab2063SBarry Smith 232417ab2063SBarry Smith ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 2325416022c9SBarry Smith *B = C; 23263a40ed3dSBarry Smith PetscFunctionReturn(0); 232717ab2063SBarry Smith } 232817ab2063SBarry Smith 2329fc08c53fSHong Zhang PetscErrorCode MatGetMultiProcBlock_SeqAIJ(Mat mat,MPI_Comm subComm,MatReuse scall,Mat *subMat) 233082d44351SHong Zhang { 233182d44351SHong Zhang PetscErrorCode ierr; 233282d44351SHong Zhang Mat B; 233382d44351SHong Zhang 233482d44351SHong Zhang PetscFunctionBegin; 2335c2d650bdSHong Zhang if (scall == MAT_INITIAL_MATRIX) { 233682d44351SHong Zhang ierr = MatCreate(subComm,&B);CHKERRQ(ierr); 233782d44351SHong Zhang ierr = MatSetSizes(B,mat->rmap->n,mat->cmap->n,mat->rmap->n,mat->cmap->n);CHKERRQ(ierr); 233833d57670SJed Brown ierr = MatSetBlockSizesFromMats(B,mat,mat);CHKERRQ(ierr); 233982d44351SHong Zhang ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr); 234082d44351SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(B,mat,MAT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 234182d44351SHong Zhang *subMat = B; 2342c2d650bdSHong Zhang } else { 2343c2d650bdSHong Zhang ierr = MatCopy_SeqAIJ(mat,*subMat,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 2344c2d650bdSHong Zhang } 234582d44351SHong Zhang PetscFunctionReturn(0); 234682d44351SHong Zhang } 234782d44351SHong Zhang 23489a625307SHong Zhang PetscErrorCode MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,const MatFactorInfo *info) 2349a871dcd8SBarry Smith { 235063b91edcSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)inA->data; 2351dfbe8321SBarry Smith PetscErrorCode ierr; 235263b91edcSBarry Smith Mat outA; 2353ace3abfcSBarry Smith PetscBool row_identity,col_identity; 235463b91edcSBarry Smith 23553a40ed3dSBarry Smith PetscFunctionBegin; 2356e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 supported for in-place ilu"); 23571df811f5SHong Zhang 2358b8a78c4aSBarry Smith ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 2359b8a78c4aSBarry Smith ierr = ISIdentity(col,&col_identity);CHKERRQ(ierr); 2360a871dcd8SBarry Smith 236163b91edcSBarry Smith outA = inA; 2362d5f3da31SBarry Smith outA->factortype = MAT_FACTOR_LU; 2363f6224b95SHong Zhang ierr = PetscFree(inA->solvertype);CHKERRQ(ierr); 2364f6224b95SHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr); 23652205254eSKarl Rupp 2366c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 23676bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 23682205254eSKarl Rupp 2369c3122656SLisandro Dalcin a->row = row; 23702205254eSKarl Rupp 2371c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)col);CHKERRQ(ierr); 23726bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 23732205254eSKarl Rupp 2374c3122656SLisandro Dalcin a->col = col; 237563b91edcSBarry Smith 237636db0b34SBarry Smith /* Create the inverse permutation so that it can be used in MatLUFactorNumeric() */ 23776bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 23784c49b128SBarry Smith ierr = ISInvertPermutation(col,PETSC_DECIDE,&a->icol);CHKERRQ(ierr); 23793bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr); 2380f0ec6fceSSatish Balay 238194a9d846SBarry Smith if (!a->solve_work) { /* this matrix may have been factored before */ 2382854ce69bSBarry Smith ierr = PetscMalloc1(inA->rmap->n+1,&a->solve_work);CHKERRQ(ierr); 23833bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)inA, (inA->rmap->n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 238494a9d846SBarry Smith } 238563b91edcSBarry Smith 2386f1e2ffcdSBarry Smith ierr = MatMarkDiagonal_SeqAIJ(inA);CHKERRQ(ierr); 2387137fb511SHong Zhang if (row_identity && col_identity) { 2388ad04f41aSHong Zhang ierr = MatLUFactorNumeric_SeqAIJ_inplace(outA,inA,info);CHKERRQ(ierr); 2389137fb511SHong Zhang } else { 2390719d5645SBarry Smith ierr = MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(outA,inA,info);CHKERRQ(ierr); 2391137fb511SHong Zhang } 23923a40ed3dSBarry Smith PetscFunctionReturn(0); 2393a871dcd8SBarry Smith } 2394a871dcd8SBarry Smith 2395f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqAIJ(Mat inA,PetscScalar alpha) 2396f0b747eeSBarry Smith { 2397f0b747eeSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)inA->data; 2398f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 2399efee365bSSatish Balay PetscErrorCode ierr; 2400c5df96a5SBarry Smith PetscBLASInt one = 1,bnz; 24013a40ed3dSBarry Smith 24023a40ed3dSBarry Smith PetscFunctionBegin; 2403c5df96a5SBarry Smith ierr = PetscBLASIntCast(a->nz,&bnz);CHKERRQ(ierr); 24048b83055fSJed Brown PetscStackCallBLAS("BLASscal",BLASscal_(&bnz,&oalpha,a->a,&one)); 2405efee365bSSatish Balay ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 2406acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(inA);CHKERRQ(ierr); 24073a40ed3dSBarry Smith PetscFunctionReturn(0); 2408f0b747eeSBarry Smith } 2409f0b747eeSBarry Smith 241097f1f81fSBarry Smith PetscErrorCode MatGetSubMatrices_SeqAIJ(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 2411cddf8d76SBarry Smith { 2412dfbe8321SBarry Smith PetscErrorCode ierr; 241397f1f81fSBarry Smith PetscInt i; 2414cddf8d76SBarry Smith 24153a40ed3dSBarry Smith PetscFunctionBegin; 2416cddf8d76SBarry Smith if (scall == MAT_INITIAL_MATRIX) { 2417*df750dc8SHong Zhang ierr = PetscCalloc1(n+1,B);CHKERRQ(ierr); 2418cddf8d76SBarry Smith } 2419cddf8d76SBarry Smith 2420cddf8d76SBarry Smith for (i=0; i<n; i++) { 24216a6a5d1dSBarry Smith ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr); 2422cddf8d76SBarry Smith } 24233a40ed3dSBarry Smith PetscFunctionReturn(0); 2424cddf8d76SBarry Smith } 2425cddf8d76SBarry Smith 242697f1f81fSBarry Smith PetscErrorCode MatIncreaseOverlap_SeqAIJ(Mat A,PetscInt is_max,IS is[],PetscInt ov) 24274dcbc457SBarry Smith { 2428e4d965acSSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 24296849ba73SBarry Smith PetscErrorCode ierr; 24305d0c19d7SBarry Smith PetscInt row,i,j,k,l,m,n,*nidx,isz,val; 24315d0c19d7SBarry Smith const PetscInt *idx; 243297f1f81fSBarry Smith PetscInt start,end,*ai,*aj; 2433f1af5d2fSBarry Smith PetscBT table; 2434bbd702dbSSatish Balay 24353a40ed3dSBarry Smith PetscFunctionBegin; 2436d0f46423SBarry Smith m = A->rmap->n; 2437e4d965acSSatish Balay ai = a->i; 2438bfeeae90SHong Zhang aj = a->j; 24398a047759SSatish Balay 2440e32f2f54SBarry Smith if (ov < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"illegal negative overlap value used"); 244106763907SSatish Balay 2442854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&nidx);CHKERRQ(ierr); 244353b8de81SBarry Smith ierr = PetscBTCreate(m,&table);CHKERRQ(ierr); 244406763907SSatish Balay 2445e4d965acSSatish Balay for (i=0; i<is_max; i++) { 2446b97fc60eSLois Curfman McInnes /* Initialize the two local arrays */ 2447e4d965acSSatish Balay isz = 0; 24486831982aSBarry Smith ierr = PetscBTMemzero(m,table);CHKERRQ(ierr); 2449e4d965acSSatish Balay 2450e4d965acSSatish Balay /* Extract the indices, assume there can be duplicate entries */ 24514dcbc457SBarry Smith ierr = ISGetIndices(is[i],&idx);CHKERRQ(ierr); 2452b9b97703SBarry Smith ierr = ISGetLocalSize(is[i],&n);CHKERRQ(ierr); 2453e4d965acSSatish Balay 2454dd097bc3SLois Curfman McInnes /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */ 2455e4d965acSSatish Balay for (j=0; j<n; ++j) { 24562205254eSKarl Rupp if (!PetscBTLookupSet(table,idx[j])) nidx[isz++] = idx[j]; 24574dcbc457SBarry Smith } 245806763907SSatish Balay ierr = ISRestoreIndices(is[i],&idx);CHKERRQ(ierr); 24596bf464f9SBarry Smith ierr = ISDestroy(&is[i]);CHKERRQ(ierr); 2460e4d965acSSatish Balay 246104a348a9SBarry Smith k = 0; 246204a348a9SBarry Smith for (j=0; j<ov; j++) { /* for each overlap */ 246304a348a9SBarry Smith n = isz; 246406763907SSatish Balay for (; k<n; k++) { /* do only those rows in nidx[k], which are not done yet */ 2465e4d965acSSatish Balay row = nidx[k]; 2466e4d965acSSatish Balay start = ai[row]; 2467e4d965acSSatish Balay end = ai[row+1]; 246804a348a9SBarry Smith for (l = start; l<end; l++) { 2469efb16452SHong Zhang val = aj[l]; 24702205254eSKarl Rupp if (!PetscBTLookupSet(table,val)) nidx[isz++] = val; 2471e4d965acSSatish Balay } 2472e4d965acSSatish Balay } 2473e4d965acSSatish Balay } 247470b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,isz,nidx,PETSC_COPY_VALUES,(is+i));CHKERRQ(ierr); 2475e4d965acSSatish Balay } 247694bacf5dSBarry Smith ierr = PetscBTDestroy(&table);CHKERRQ(ierr); 2477606d414cSSatish Balay ierr = PetscFree(nidx);CHKERRQ(ierr); 24783a40ed3dSBarry Smith PetscFunctionReturn(0); 24794dcbc457SBarry Smith } 248017ab2063SBarry Smith 24810513a670SBarry Smith /* -------------------------------------------------------------- */ 2482dfbe8321SBarry Smith PetscErrorCode MatPermute_SeqAIJ(Mat A,IS rowp,IS colp,Mat *B) 24830513a670SBarry Smith { 24840513a670SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 24856849ba73SBarry Smith PetscErrorCode ierr; 24863b98c0a2SBarry Smith PetscInt i,nz = 0,m = A->rmap->n,n = A->cmap->n; 24875d0c19d7SBarry Smith const PetscInt *row,*col; 24885d0c19d7SBarry Smith PetscInt *cnew,j,*lens; 248956cd22aeSBarry Smith IS icolp,irowp; 24900298fd71SBarry Smith PetscInt *cwork = NULL; 24910298fd71SBarry Smith PetscScalar *vwork = NULL; 24920513a670SBarry Smith 24933a40ed3dSBarry Smith PetscFunctionBegin; 24944c49b128SBarry Smith ierr = ISInvertPermutation(rowp,PETSC_DECIDE,&irowp);CHKERRQ(ierr); 249556cd22aeSBarry Smith ierr = ISGetIndices(irowp,&row);CHKERRQ(ierr); 24964c49b128SBarry Smith ierr = ISInvertPermutation(colp,PETSC_DECIDE,&icolp);CHKERRQ(ierr); 249756cd22aeSBarry Smith ierr = ISGetIndices(icolp,&col);CHKERRQ(ierr); 24980513a670SBarry Smith 24990513a670SBarry Smith /* determine lengths of permuted rows */ 2500854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&lens);CHKERRQ(ierr); 25012205254eSKarl Rupp for (i=0; i<m; i++) lens[row[i]] = a->i[i+1] - a->i[i]; 2502ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 2503f69a0ea3SMatthew Knepley ierr = MatSetSizes(*B,m,n,m,n);CHKERRQ(ierr); 250433d57670SJed Brown ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr); 25057adad957SLisandro Dalcin ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr); 2506ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(*B,0,lens);CHKERRQ(ierr); 2507606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 25080513a670SBarry Smith 2509785e854fSJed Brown ierr = PetscMalloc1(n,&cnew);CHKERRQ(ierr); 25100513a670SBarry Smith for (i=0; i<m; i++) { 251132ec9ce4SBarry Smith ierr = MatGetRow_SeqAIJ(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr); 25122205254eSKarl Rupp for (j=0; j<nz; j++) cnew[j] = col[cwork[j]]; 2513cdc0ba36SBarry Smith ierr = MatSetValues_SeqAIJ(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES);CHKERRQ(ierr); 251432ec9ce4SBarry Smith ierr = MatRestoreRow_SeqAIJ(A,i,&nz,&cwork,&vwork);CHKERRQ(ierr); 25150513a670SBarry Smith } 2516606d414cSSatish Balay ierr = PetscFree(cnew);CHKERRQ(ierr); 25172205254eSKarl Rupp 25183c7d62e4SBarry Smith (*B)->assembled = PETSC_FALSE; 25192205254eSKarl Rupp 25200513a670SBarry Smith ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 25210513a670SBarry Smith ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 252256cd22aeSBarry Smith ierr = ISRestoreIndices(irowp,&row);CHKERRQ(ierr); 252356cd22aeSBarry Smith ierr = ISRestoreIndices(icolp,&col);CHKERRQ(ierr); 25246bf464f9SBarry Smith ierr = ISDestroy(&irowp);CHKERRQ(ierr); 25256bf464f9SBarry Smith ierr = ISDestroy(&icolp);CHKERRQ(ierr); 25263a40ed3dSBarry Smith PetscFunctionReturn(0); 25270513a670SBarry Smith } 25280513a670SBarry Smith 2529dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqAIJ(Mat A,Mat B,MatStructure str) 2530cb5b572fSBarry Smith { 2531dfbe8321SBarry Smith PetscErrorCode ierr; 2532cb5b572fSBarry Smith 2533cb5b572fSBarry Smith PetscFunctionBegin; 253433f4a19fSKris Buschelman /* If the two matrices have the same copy implementation, use fast copy. */ 253533f4a19fSKris Buschelman if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 2536be6bf707SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2537be6bf707SBarry Smith Mat_SeqAIJ *b = (Mat_SeqAIJ*)B->data; 2538be6bf707SBarry Smith 2539700c5bfcSBarry Smith if (a->i[A->rmap->n] != b->i[B->rmap->n]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 2540d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->n])*sizeof(PetscScalar));CHKERRQ(ierr); 2541cb5b572fSBarry Smith } else { 2542cb5b572fSBarry Smith ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 2543cb5b572fSBarry Smith } 2544cb5b572fSBarry Smith PetscFunctionReturn(0); 2545cb5b572fSBarry Smith } 2546cb5b572fSBarry Smith 25474994cf47SJed Brown PetscErrorCode MatSetUp_SeqAIJ(Mat A) 2548273d9f13SBarry Smith { 2549dfbe8321SBarry Smith PetscErrorCode ierr; 2550273d9f13SBarry Smith 2551273d9f13SBarry Smith PetscFunctionBegin; 2552ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(A,PETSC_DEFAULT,0);CHKERRQ(ierr); 2553273d9f13SBarry Smith PetscFunctionReturn(0); 2554273d9f13SBarry Smith } 2555273d9f13SBarry Smith 25568c778c55SBarry Smith PetscErrorCode MatSeqAIJGetArray_SeqAIJ(Mat A,PetscScalar *array[]) 25576c0721eeSBarry Smith { 25586c0721eeSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 25596e111a19SKarl Rupp 25606c0721eeSBarry Smith PetscFunctionBegin; 25616c0721eeSBarry Smith *array = a->a; 25626c0721eeSBarry Smith PetscFunctionReturn(0); 25636c0721eeSBarry Smith } 25646c0721eeSBarry Smith 25658c778c55SBarry Smith PetscErrorCode MatSeqAIJRestoreArray_SeqAIJ(Mat A,PetscScalar *array[]) 25666c0721eeSBarry Smith { 25676c0721eeSBarry Smith PetscFunctionBegin; 25686c0721eeSBarry Smith PetscFunctionReturn(0); 25696c0721eeSBarry Smith } 2570273d9f13SBarry Smith 25718229c054SShri Abhyankar /* 25728229c054SShri Abhyankar Computes the number of nonzeros per row needed for preallocation when X and Y 25738229c054SShri Abhyankar have different nonzero structure. 25748229c054SShri Abhyankar */ 2575b264fe52SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqX_private(PetscInt m,const PetscInt *xi,const PetscInt *xj,const PetscInt *yi,const PetscInt *yj,PetscInt *nnz) 2576ec7775f6SShri Abhyankar { 2577b264fe52SHong Zhang PetscInt i,j,k,nzx,nzy; 2578ec7775f6SShri Abhyankar 2579ec7775f6SShri Abhyankar PetscFunctionBegin; 2580ec7775f6SShri Abhyankar /* Set the number of nonzeros in the new matrix */ 2581ec7775f6SShri Abhyankar for (i=0; i<m; i++) { 2582b264fe52SHong Zhang const PetscInt *xjj = xj+xi[i],*yjj = yj+yi[i]; 2583b264fe52SHong Zhang nzx = xi[i+1] - xi[i]; 2584b264fe52SHong Zhang nzy = yi[i+1] - yi[i]; 25858af7cee1SJed Brown nnz[i] = 0; 25868af7cee1SJed Brown for (j=0,k=0; j<nzx; j++) { /* Point in X */ 2587b264fe52SHong Zhang for (; k<nzy && yjj[k]<xjj[j]; k++) nnz[i]++; /* Catch up to X */ 2588b264fe52SHong Zhang if (k<nzy && yjj[k]==xjj[j]) k++; /* Skip duplicate */ 25898af7cee1SJed Brown nnz[i]++; 25908af7cee1SJed Brown } 25918af7cee1SJed Brown for (; k<nzy; k++) nnz[i]++; 2592ec7775f6SShri Abhyankar } 2593ec7775f6SShri Abhyankar PetscFunctionReturn(0); 2594ec7775f6SShri Abhyankar } 2595ec7775f6SShri Abhyankar 2596b264fe52SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqAIJ(Mat Y,Mat X,PetscInt *nnz) 2597b264fe52SHong Zhang { 2598b264fe52SHong Zhang PetscInt m = Y->rmap->N; 2599b264fe52SHong Zhang Mat_SeqAIJ *x = (Mat_SeqAIJ*)X->data; 2600b264fe52SHong Zhang Mat_SeqAIJ *y = (Mat_SeqAIJ*)Y->data; 2601b264fe52SHong Zhang PetscErrorCode ierr; 2602b264fe52SHong Zhang 2603b264fe52SHong Zhang PetscFunctionBegin; 2604b264fe52SHong Zhang /* Set the number of nonzeros in the new matrix */ 2605b264fe52SHong Zhang ierr = MatAXPYGetPreallocation_SeqX_private(m,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr); 2606b264fe52SHong Zhang PetscFunctionReturn(0); 2607b264fe52SHong Zhang } 2608b264fe52SHong Zhang 2609f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 2610ac90fabeSBarry Smith { 2611dfbe8321SBarry Smith PetscErrorCode ierr; 2612ac90fabeSBarry Smith Mat_SeqAIJ *x = (Mat_SeqAIJ*)X->data,*y = (Mat_SeqAIJ*)Y->data; 2613c5df96a5SBarry Smith PetscBLASInt one=1,bnz; 2614ac90fabeSBarry Smith 2615ac90fabeSBarry Smith PetscFunctionBegin; 2616c5df96a5SBarry Smith ierr = PetscBLASIntCast(x->nz,&bnz);CHKERRQ(ierr); 2617ac90fabeSBarry Smith if (str == SAME_NONZERO_PATTERN) { 2618f4df32b1SMatthew Knepley PetscScalar alpha = a; 26198b83055fSJed Brown PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one)); 2620acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal(Y);CHKERRQ(ierr); 2621a3fa217bSJose E. Roman ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 2622ab784542SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 2623ab784542SHong Zhang ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 2624ac90fabeSBarry Smith } else { 26258229c054SShri Abhyankar Mat B; 26268229c054SShri Abhyankar PetscInt *nnz; 2627785e854fSJed Brown ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr); 2628ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr); 2629bc5a2726SShri Abhyankar ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr); 26304aa94f47SShri Abhyankar ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr); 263133d57670SJed Brown ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr); 2632176df525SBarry Smith ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr); 26338229c054SShri Abhyankar ierr = MatAXPYGetPreallocation_SeqAIJ(Y,X,nnz);CHKERRQ(ierr); 2634ecd8bba6SJed Brown ierr = MatSeqAIJSetPreallocation(B,0,nnz);CHKERRQ(ierr); 2635ec7775f6SShri Abhyankar ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 263628be2f97SBarry Smith ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr); 26378229c054SShri Abhyankar ierr = PetscFree(nnz);CHKERRQ(ierr); 2638ac90fabeSBarry Smith } 2639ac90fabeSBarry Smith PetscFunctionReturn(0); 2640ac90fabeSBarry Smith } 2641ac90fabeSBarry Smith 26427087cfbeSBarry Smith PetscErrorCode MatConjugate_SeqAIJ(Mat mat) 2643354c94deSBarry Smith { 2644354c94deSBarry Smith #if defined(PETSC_USE_COMPLEX) 2645354c94deSBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)mat->data; 2646354c94deSBarry Smith PetscInt i,nz; 2647354c94deSBarry Smith PetscScalar *a; 2648354c94deSBarry Smith 2649354c94deSBarry Smith PetscFunctionBegin; 2650354c94deSBarry Smith nz = aij->nz; 2651354c94deSBarry Smith a = aij->a; 26522205254eSKarl Rupp for (i=0; i<nz; i++) a[i] = PetscConj(a[i]); 2653354c94deSBarry Smith #else 2654354c94deSBarry Smith PetscFunctionBegin; 2655354c94deSBarry Smith #endif 2656354c94deSBarry Smith PetscFunctionReturn(0); 2657354c94deSBarry Smith } 2658354c94deSBarry Smith 2659985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqAIJ(Mat A,Vec v,PetscInt idx[]) 2660e34fafa9SBarry Smith { 2661e34fafa9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2662e34fafa9SBarry Smith PetscErrorCode ierr; 2663d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,*ai,*aj,ncols,n; 2664e34fafa9SBarry Smith PetscReal atmp; 2665985db425SBarry Smith PetscScalar *x; 2666e34fafa9SBarry Smith MatScalar *aa; 2667e34fafa9SBarry Smith 2668e34fafa9SBarry Smith PetscFunctionBegin; 2669e32f2f54SBarry Smith if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 2670e34fafa9SBarry Smith aa = a->a; 2671e34fafa9SBarry Smith ai = a->i; 2672e34fafa9SBarry Smith aj = a->j; 2673e34fafa9SBarry Smith 2674985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 2675e34fafa9SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 2676e34fafa9SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 2677e32f2f54SBarry Smith if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 2678e34fafa9SBarry Smith for (i=0; i<m; i++) { 2679e34fafa9SBarry Smith ncols = ai[1] - ai[0]; ai++; 26809189402eSHong Zhang x[i] = 0.0; 2681e34fafa9SBarry Smith for (j=0; j<ncols; j++) { 2682985db425SBarry Smith atmp = PetscAbsScalar(*aa); 2683985db425SBarry Smith if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = *aj;} 2684985db425SBarry Smith aa++; aj++; 2685985db425SBarry Smith } 2686985db425SBarry Smith } 2687985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 2688985db425SBarry Smith PetscFunctionReturn(0); 2689985db425SBarry Smith } 2690985db425SBarry Smith 2691985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqAIJ(Mat A,Vec v,PetscInt idx[]) 2692985db425SBarry Smith { 2693985db425SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2694985db425SBarry Smith PetscErrorCode ierr; 2695d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,*ai,*aj,ncols,n; 2696985db425SBarry Smith PetscScalar *x; 2697985db425SBarry Smith MatScalar *aa; 2698985db425SBarry Smith 2699985db425SBarry Smith PetscFunctionBegin; 2700e32f2f54SBarry Smith if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 2701985db425SBarry Smith aa = a->a; 2702985db425SBarry Smith ai = a->i; 2703985db425SBarry Smith aj = a->j; 2704985db425SBarry Smith 2705985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 2706985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 2707985db425SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 2708e32f2f54SBarry Smith if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 2709985db425SBarry Smith for (i=0; i<m; i++) { 2710985db425SBarry Smith ncols = ai[1] - ai[0]; ai++; 2711d0f46423SBarry Smith if (ncols == A->cmap->n) { /* row is dense */ 2712985db425SBarry Smith x[i] = *aa; if (idx) idx[i] = 0; 2713985db425SBarry Smith } else { /* row is sparse so already KNOW maximum is 0.0 or higher */ 2714985db425SBarry Smith x[i] = 0.0; 2715985db425SBarry Smith if (idx) { 2716985db425SBarry Smith idx[i] = 0; /* in case ncols is zero */ 2717985db425SBarry Smith for (j=0;j<ncols;j++) { /* find first implicit 0.0 in the row */ 2718985db425SBarry Smith if (aj[j] > j) { 2719985db425SBarry Smith idx[i] = j; 2720985db425SBarry Smith break; 2721985db425SBarry Smith } 2722985db425SBarry Smith } 2723985db425SBarry Smith } 2724985db425SBarry Smith } 2725985db425SBarry Smith for (j=0; j<ncols; j++) { 2726985db425SBarry Smith if (PetscRealPart(x[i]) < PetscRealPart(*aa)) {x[i] = *aa; if (idx) idx[i] = *aj;} 2727985db425SBarry Smith aa++; aj++; 2728985db425SBarry Smith } 2729985db425SBarry Smith } 2730985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 2731985db425SBarry Smith PetscFunctionReturn(0); 2732985db425SBarry Smith } 2733985db425SBarry Smith 2734c87e5d42SMatthew Knepley PetscErrorCode MatGetRowMinAbs_SeqAIJ(Mat A,Vec v,PetscInt idx[]) 2735c87e5d42SMatthew Knepley { 2736c87e5d42SMatthew Knepley Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2737c87e5d42SMatthew Knepley PetscErrorCode ierr; 2738c87e5d42SMatthew Knepley PetscInt i,j,m = A->rmap->n,*ai,*aj,ncols,n; 2739c87e5d42SMatthew Knepley PetscReal atmp; 2740c87e5d42SMatthew Knepley PetscScalar *x; 2741c87e5d42SMatthew Knepley MatScalar *aa; 2742c87e5d42SMatthew Knepley 2743c87e5d42SMatthew Knepley PetscFunctionBegin; 2744e32f2f54SBarry Smith if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 2745c87e5d42SMatthew Knepley aa = a->a; 2746c87e5d42SMatthew Knepley ai = a->i; 2747c87e5d42SMatthew Knepley aj = a->j; 2748c87e5d42SMatthew Knepley 2749c87e5d42SMatthew Knepley ierr = VecSet(v,0.0);CHKERRQ(ierr); 2750c87e5d42SMatthew Knepley ierr = VecGetArray(v,&x);CHKERRQ(ierr); 2751c87e5d42SMatthew Knepley ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 275260e0710aSBarry Smith if (n != A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector, %D vs. %D rows", A->rmap->n, n); 2753c87e5d42SMatthew Knepley for (i=0; i<m; i++) { 2754c87e5d42SMatthew Knepley ncols = ai[1] - ai[0]; ai++; 2755289a08f5SMatthew Knepley if (ncols) { 2756289a08f5SMatthew Knepley /* Get first nonzero */ 2757289a08f5SMatthew Knepley for (j = 0; j < ncols; j++) { 2758289a08f5SMatthew Knepley atmp = PetscAbsScalar(aa[j]); 27592205254eSKarl Rupp if (atmp > 1.0e-12) { 27602205254eSKarl Rupp x[i] = atmp; 27612205254eSKarl Rupp if (idx) idx[i] = aj[j]; 27622205254eSKarl Rupp break; 27632205254eSKarl Rupp } 2764289a08f5SMatthew Knepley } 276512431cb0SMatthew G Knepley if (j == ncols) {x[i] = PetscAbsScalar(*aa); if (idx) idx[i] = *aj;} 2766289a08f5SMatthew Knepley } else { 2767289a08f5SMatthew Knepley x[i] = 0.0; if (idx) idx[i] = 0; 2768289a08f5SMatthew Knepley } 2769c87e5d42SMatthew Knepley for (j = 0; j < ncols; j++) { 2770c87e5d42SMatthew Knepley atmp = PetscAbsScalar(*aa); 2771289a08f5SMatthew Knepley if (atmp > 1.0e-12 && PetscAbsScalar(x[i]) > atmp) {x[i] = atmp; if (idx) idx[i] = *aj;} 2772c87e5d42SMatthew Knepley aa++; aj++; 2773c87e5d42SMatthew Knepley } 2774c87e5d42SMatthew Knepley } 2775c87e5d42SMatthew Knepley ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 2776c87e5d42SMatthew Knepley PetscFunctionReturn(0); 2777c87e5d42SMatthew Knepley } 2778c87e5d42SMatthew Knepley 2779985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqAIJ(Mat A,Vec v,PetscInt idx[]) 2780985db425SBarry Smith { 2781985db425SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2782985db425SBarry Smith PetscErrorCode ierr; 2783d9ca1df4SBarry Smith PetscInt i,j,m = A->rmap->n,ncols,n; 2784d9ca1df4SBarry Smith const PetscInt *ai,*aj; 2785985db425SBarry Smith PetscScalar *x; 2786d9ca1df4SBarry Smith const MatScalar *aa; 2787985db425SBarry Smith 2788985db425SBarry Smith PetscFunctionBegin; 2789e32f2f54SBarry Smith if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 2790985db425SBarry Smith aa = a->a; 2791985db425SBarry Smith ai = a->i; 2792985db425SBarry Smith aj = a->j; 2793985db425SBarry Smith 2794985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 2795985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 2796985db425SBarry Smith ierr = VecGetLocalSize(v,&n);CHKERRQ(ierr); 2797e32f2f54SBarry Smith if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 2798985db425SBarry Smith for (i=0; i<m; i++) { 2799985db425SBarry Smith ncols = ai[1] - ai[0]; ai++; 2800d0f46423SBarry Smith if (ncols == A->cmap->n) { /* row is dense */ 2801985db425SBarry Smith x[i] = *aa; if (idx) idx[i] = 0; 2802985db425SBarry Smith } else { /* row is sparse so already KNOW minimum is 0.0 or lower */ 2803985db425SBarry Smith x[i] = 0.0; 2804985db425SBarry Smith if (idx) { /* find first implicit 0.0 in the row */ 2805985db425SBarry Smith idx[i] = 0; /* in case ncols is zero */ 2806985db425SBarry Smith for (j=0; j<ncols; j++) { 2807985db425SBarry Smith if (aj[j] > j) { 2808985db425SBarry Smith idx[i] = j; 2809985db425SBarry Smith break; 2810985db425SBarry Smith } 2811985db425SBarry Smith } 2812985db425SBarry Smith } 2813985db425SBarry Smith } 2814985db425SBarry Smith for (j=0; j<ncols; j++) { 2815985db425SBarry Smith if (PetscRealPart(x[i]) > PetscRealPart(*aa)) {x[i] = *aa; if (idx) idx[i] = *aj;} 2816985db425SBarry Smith aa++; aj++; 2817e34fafa9SBarry Smith } 2818e34fafa9SBarry Smith } 2819e34fafa9SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 2820e34fafa9SBarry Smith PetscFunctionReturn(0); 2821e34fafa9SBarry Smith } 2822bbead8a2SBarry Smith 2823bbead8a2SBarry Smith #include <petscblaslapack.h> 2824af0996ceSBarry Smith #include <petsc/private/kernels/blockinvert.h> 2825bbead8a2SBarry Smith 2826713ccfa9SJed Brown PetscErrorCode MatInvertBlockDiagonal_SeqAIJ(Mat A,const PetscScalar **values) 2827bbead8a2SBarry Smith { 2828bbead8a2SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*) A->data; 2829bbead8a2SBarry Smith PetscErrorCode ierr; 283033d57670SJed Brown PetscInt i,bs = PetscAbs(A->rmap->bs),mbs = A->rmap->n/bs,ipvt[5],bs2 = bs*bs,*v_pivots,ij[7],*IJ,j; 2831bbead8a2SBarry Smith MatScalar *diag,work[25],*v_work; 2832bbead8a2SBarry Smith PetscReal shift = 0.0; 28331a9391e3SHong Zhang PetscBool allowzeropivot,zeropivotdetected=PETSC_FALSE; 2834bbead8a2SBarry Smith 2835bbead8a2SBarry Smith PetscFunctionBegin; 2836a455e926SHong Zhang allowzeropivot = PetscNot(A->erroriffailure); 28374a0d0026SBarry Smith if (a->ibdiagvalid) { 28384a0d0026SBarry Smith if (values) *values = a->ibdiag; 28394a0d0026SBarry Smith PetscFunctionReturn(0); 28404a0d0026SBarry Smith } 2841bbead8a2SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 2842bbead8a2SBarry Smith if (!a->ibdiag) { 2843785e854fSJed Brown ierr = PetscMalloc1(bs2*mbs,&a->ibdiag);CHKERRQ(ierr); 28443bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)A,bs2*mbs*sizeof(PetscScalar));CHKERRQ(ierr); 2845bbead8a2SBarry Smith } 2846bbead8a2SBarry Smith diag = a->ibdiag; 2847bbead8a2SBarry Smith if (values) *values = a->ibdiag; 2848bbead8a2SBarry Smith /* factor and invert each block */ 2849bbead8a2SBarry Smith switch (bs) { 2850bbead8a2SBarry Smith case 1: 2851bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2852bbead8a2SBarry Smith ierr = MatGetValues(A,1,&i,1,&i,diag+i);CHKERRQ(ierr); 2853ec1892c8SHong Zhang if (PetscAbsScalar(diag[i] + shift) < PETSC_MACHINE_EPSILON) { 2854ec1892c8SHong Zhang if (allowzeropivot) { 28557b6c816cSBarry Smith A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 28567b6c816cSBarry Smith A->factorerror_zeropivot_value = PetscAbsScalar(diag[i]); 28577b6c816cSBarry Smith A->factorerror_zeropivot_row = i; 28587b6c816cSBarry Smith ierr = PetscInfo3(A,"Zero pivot, row %D pivot %g tolerance %g\n",i,(double)PetscAbsScalar(diag[i]),(double)PETSC_MACHINE_EPSILON);CHKERRQ(ierr); 28597b6c816cSBarry Smith } else SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot, row %D pivot %g tolerance %g",i,(double)PetscAbsScalar(diag[i]),(double)PETSC_MACHINE_EPSILON); 2860ec1892c8SHong Zhang } 2861bbead8a2SBarry Smith diag[i] = (PetscScalar)1.0 / (diag[i] + shift); 2862bbead8a2SBarry Smith } 2863bbead8a2SBarry Smith break; 2864bbead8a2SBarry Smith case 2: 2865bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2866bbead8a2SBarry Smith ij[0] = 2*i; ij[1] = 2*i + 1; 2867bbead8a2SBarry Smith ierr = MatGetValues(A,2,ij,2,ij,diag);CHKERRQ(ierr); 2868a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_2(diag,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 28697b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 287096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_2(diag);CHKERRQ(ierr); 2871bbead8a2SBarry Smith diag += 4; 2872bbead8a2SBarry Smith } 2873bbead8a2SBarry Smith break; 2874bbead8a2SBarry Smith case 3: 2875bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2876bbead8a2SBarry Smith ij[0] = 3*i; ij[1] = 3*i + 1; ij[2] = 3*i + 2; 2877bbead8a2SBarry Smith ierr = MatGetValues(A,3,ij,3,ij,diag);CHKERRQ(ierr); 2878a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_3(diag,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 28797b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 288096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_3(diag);CHKERRQ(ierr); 2881bbead8a2SBarry Smith diag += 9; 2882bbead8a2SBarry Smith } 2883bbead8a2SBarry Smith break; 2884bbead8a2SBarry Smith case 4: 2885bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2886bbead8a2SBarry Smith ij[0] = 4*i; ij[1] = 4*i + 1; ij[2] = 4*i + 2; ij[3] = 4*i + 3; 2887bbead8a2SBarry Smith ierr = MatGetValues(A,4,ij,4,ij,diag);CHKERRQ(ierr); 2888a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_4(diag,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 28897b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 289096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_4(diag);CHKERRQ(ierr); 2891bbead8a2SBarry Smith diag += 16; 2892bbead8a2SBarry Smith } 2893bbead8a2SBarry Smith break; 2894bbead8a2SBarry Smith case 5: 2895bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2896bbead8a2SBarry Smith ij[0] = 5*i; ij[1] = 5*i + 1; ij[2] = 5*i + 2; ij[3] = 5*i + 3; ij[4] = 5*i + 4; 2897bbead8a2SBarry Smith ierr = MatGetValues(A,5,ij,5,ij,diag);CHKERRQ(ierr); 2898a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_5(diag,ipvt,work,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 28997b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 290096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_5(diag);CHKERRQ(ierr); 2901bbead8a2SBarry Smith diag += 25; 2902bbead8a2SBarry Smith } 2903bbead8a2SBarry Smith break; 2904bbead8a2SBarry Smith case 6: 2905bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2906bbead8a2SBarry Smith ij[0] = 6*i; ij[1] = 6*i + 1; ij[2] = 6*i + 2; ij[3] = 6*i + 3; ij[4] = 6*i + 4; ij[5] = 6*i + 5; 2907bbead8a2SBarry Smith ierr = MatGetValues(A,6,ij,6,ij,diag);CHKERRQ(ierr); 2908a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_6(diag,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 29097b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 291096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_6(diag);CHKERRQ(ierr); 2911bbead8a2SBarry Smith diag += 36; 2912bbead8a2SBarry Smith } 2913bbead8a2SBarry Smith break; 2914bbead8a2SBarry Smith case 7: 2915bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2916bbead8a2SBarry Smith ij[0] = 7*i; ij[1] = 7*i + 1; ij[2] = 7*i + 2; ij[3] = 7*i + 3; ij[4] = 7*i + 4; ij[5] = 7*i + 5; ij[5] = 7*i + 6; 2917bbead8a2SBarry Smith ierr = MatGetValues(A,7,ij,7,ij,diag);CHKERRQ(ierr); 2918a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_7(diag,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 29197b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 292096b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_7(diag);CHKERRQ(ierr); 2921bbead8a2SBarry Smith diag += 49; 2922bbead8a2SBarry Smith } 2923bbead8a2SBarry Smith break; 2924bbead8a2SBarry Smith default: 2925dcca6d9dSJed Brown ierr = PetscMalloc3(bs,&v_work,bs,&v_pivots,bs,&IJ);CHKERRQ(ierr); 2926bbead8a2SBarry Smith for (i=0; i<mbs; i++) { 2927bbead8a2SBarry Smith for (j=0; j<bs; j++) { 2928bbead8a2SBarry Smith IJ[j] = bs*i + j; 2929bbead8a2SBarry Smith } 2930bbead8a2SBarry Smith ierr = MatGetValues(A,bs,IJ,bs,IJ,diag);CHKERRQ(ierr); 29315f8bbccaSHong Zhang ierr = PetscKernel_A_gets_inverse_A(bs,diag,v_pivots,v_work,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 29327b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 293396b95a6bSBarry Smith ierr = PetscKernel_A_gets_transpose_A_N(diag,bs);CHKERRQ(ierr); 2934bbead8a2SBarry Smith diag += bs2; 2935bbead8a2SBarry Smith } 2936bbead8a2SBarry Smith ierr = PetscFree3(v_work,v_pivots,IJ);CHKERRQ(ierr); 2937bbead8a2SBarry Smith } 2938bbead8a2SBarry Smith a->ibdiagvalid = PETSC_TRUE; 2939bbead8a2SBarry Smith PetscFunctionReturn(0); 2940bbead8a2SBarry Smith } 2941bbead8a2SBarry Smith 294273a71a0fSBarry Smith static PetscErrorCode MatSetRandom_SeqAIJ(Mat x,PetscRandom rctx) 294373a71a0fSBarry Smith { 294473a71a0fSBarry Smith PetscErrorCode ierr; 294573a71a0fSBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)x->data; 294673a71a0fSBarry Smith PetscScalar a; 294773a71a0fSBarry Smith PetscInt m,n,i,j,col; 294873a71a0fSBarry Smith 294973a71a0fSBarry Smith PetscFunctionBegin; 295073a71a0fSBarry Smith if (!x->assembled) { 295173a71a0fSBarry Smith ierr = MatGetSize(x,&m,&n);CHKERRQ(ierr); 295273a71a0fSBarry Smith for (i=0; i<m; i++) { 295373a71a0fSBarry Smith for (j=0; j<aij->imax[i]; j++) { 295473a71a0fSBarry Smith ierr = PetscRandomGetValue(rctx,&a);CHKERRQ(ierr); 295573a71a0fSBarry Smith col = (PetscInt)(n*PetscRealPart(a)); 295673a71a0fSBarry Smith ierr = MatSetValues(x,1,&i,1,&col,&a,ADD_VALUES);CHKERRQ(ierr); 295773a71a0fSBarry Smith } 295873a71a0fSBarry Smith } 295973a71a0fSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not yet coded"); 296073a71a0fSBarry Smith ierr = MatAssemblyBegin(x,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 296173a71a0fSBarry Smith ierr = MatAssemblyEnd(x,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 296273a71a0fSBarry Smith PetscFunctionReturn(0); 296373a71a0fSBarry Smith } 296473a71a0fSBarry Smith 29657d68702bSBarry Smith PetscErrorCode MatShift_SeqAIJ(Mat Y,PetscScalar a) 29667d68702bSBarry Smith { 29677d68702bSBarry Smith PetscErrorCode ierr; 29687d68702bSBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)Y->data; 29697d68702bSBarry Smith 29707d68702bSBarry Smith PetscFunctionBegin; 29716f33a894SBarry Smith if (!Y->preallocated || !aij->nz) { 29727d68702bSBarry Smith ierr = MatSeqAIJSetPreallocation(Y,1,NULL);CHKERRQ(ierr); 29737d68702bSBarry Smith } 29747d68702bSBarry Smith ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 29757d68702bSBarry Smith PetscFunctionReturn(0); 29767d68702bSBarry Smith } 29777d68702bSBarry Smith 2978682d7d0cSBarry Smith /* -------------------------------------------------------------------*/ 29790a6ffc59SBarry Smith static struct _MatOps MatOps_Values = { MatSetValues_SeqAIJ, 2980cb5b572fSBarry Smith MatGetRow_SeqAIJ, 2981cb5b572fSBarry Smith MatRestoreRow_SeqAIJ, 2982cb5b572fSBarry Smith MatMult_SeqAIJ, 298397304618SKris Buschelman /* 4*/ MatMultAdd_SeqAIJ, 29847c922b88SBarry Smith MatMultTranspose_SeqAIJ, 29857c922b88SBarry Smith MatMultTransposeAdd_SeqAIJ, 2986db4efbfdSBarry Smith 0, 2987db4efbfdSBarry Smith 0, 2988db4efbfdSBarry Smith 0, 2989db4efbfdSBarry Smith /* 10*/ 0, 2990cb5b572fSBarry Smith MatLUFactor_SeqAIJ, 2991cb5b572fSBarry Smith 0, 299241f059aeSBarry Smith MatSOR_SeqAIJ, 299317ab2063SBarry Smith MatTranspose_SeqAIJ, 299497304618SKris Buschelman /*1 5*/ MatGetInfo_SeqAIJ, 2995cb5b572fSBarry Smith MatEqual_SeqAIJ, 2996cb5b572fSBarry Smith MatGetDiagonal_SeqAIJ, 2997cb5b572fSBarry Smith MatDiagonalScale_SeqAIJ, 2998cb5b572fSBarry Smith MatNorm_SeqAIJ, 299997304618SKris Buschelman /* 20*/ 0, 3000cb5b572fSBarry Smith MatAssemblyEnd_SeqAIJ, 3001cb5b572fSBarry Smith MatSetOption_SeqAIJ, 3002cb5b572fSBarry Smith MatZeroEntries_SeqAIJ, 3003d519adbfSMatthew Knepley /* 24*/ MatZeroRows_SeqAIJ, 3004db4efbfdSBarry Smith 0, 3005db4efbfdSBarry Smith 0, 3006db4efbfdSBarry Smith 0, 3007db4efbfdSBarry Smith 0, 30084994cf47SJed Brown /* 29*/ MatSetUp_SeqAIJ, 3009db4efbfdSBarry Smith 0, 3010db4efbfdSBarry Smith 0, 30118c778c55SBarry Smith 0, 30128c778c55SBarry Smith 0, 3013d519adbfSMatthew Knepley /* 34*/ MatDuplicate_SeqAIJ, 3014cb5b572fSBarry Smith 0, 3015cb5b572fSBarry Smith 0, 3016cb5b572fSBarry Smith MatILUFactor_SeqAIJ, 3017cb5b572fSBarry Smith 0, 3018d519adbfSMatthew Knepley /* 39*/ MatAXPY_SeqAIJ, 3019cb5b572fSBarry Smith MatGetSubMatrices_SeqAIJ, 3020cb5b572fSBarry Smith MatIncreaseOverlap_SeqAIJ, 3021cb5b572fSBarry Smith MatGetValues_SeqAIJ, 3022cb5b572fSBarry Smith MatCopy_SeqAIJ, 3023d519adbfSMatthew Knepley /* 44*/ MatGetRowMax_SeqAIJ, 3024cb5b572fSBarry Smith MatScale_SeqAIJ, 30257d68702bSBarry Smith MatShift_SeqAIJ, 302679299369SBarry Smith MatDiagonalSet_SeqAIJ, 30276e169961SBarry Smith MatZeroRowsColumns_SeqAIJ, 302873a71a0fSBarry Smith /* 49*/ MatSetRandom_SeqAIJ, 30293b2fbd54SBarry Smith MatGetRowIJ_SeqAIJ, 30303b2fbd54SBarry Smith MatRestoreRowIJ_SeqAIJ, 30313b2fbd54SBarry Smith MatGetColumnIJ_SeqAIJ, 3032a93ec695SBarry Smith MatRestoreColumnIJ_SeqAIJ, 303393dfae19SHong Zhang /* 54*/ MatFDColoringCreate_SeqXAIJ, 3034b9617806SBarry Smith 0, 30350513a670SBarry Smith 0, 3036cda55fadSBarry Smith MatPermute_SeqAIJ, 3037cda55fadSBarry Smith 0, 3038d519adbfSMatthew Knepley /* 59*/ 0, 3039b9b97703SBarry Smith MatDestroy_SeqAIJ, 3040b9b97703SBarry Smith MatView_SeqAIJ, 3041357abbc8SBarry Smith 0, 3042321b30b9SSatish Balay MatMatMatMult_SeqAIJ_SeqAIJ_SeqAIJ, 3043321b30b9SSatish Balay /* 64*/ MatMatMatMultSymbolic_SeqAIJ_SeqAIJ_SeqAIJ, 3044321b30b9SSatish Balay MatMatMatMultNumeric_SeqAIJ_SeqAIJ_SeqAIJ, 3045ee4f033dSBarry Smith 0, 3046ee4f033dSBarry Smith 0, 3047ee4f033dSBarry Smith 0, 3048d519adbfSMatthew Knepley /* 69*/ MatGetRowMaxAbs_SeqAIJ, 3049c87e5d42SMatthew Knepley MatGetRowMinAbs_SeqAIJ, 3050ee4f033dSBarry Smith 0, 3051dcf5cc72SBarry Smith 0, 30522c93a97aSBarry Smith 0, 30532c93a97aSBarry Smith /* 74*/ 0, 30543acb8795SBarry Smith MatFDColoringApply_AIJ, 305597304618SKris Buschelman 0, 305697304618SKris Buschelman 0, 305797304618SKris Buschelman 0, 30586ce1633cSBarry Smith /* 79*/ MatFindZeroDiagonals_SeqAIJ, 305997304618SKris Buschelman 0, 306097304618SKris Buschelman 0, 306197304618SKris Buschelman 0, 3062bc011b1eSHong Zhang MatLoad_SeqAIJ, 3063d519adbfSMatthew Knepley /* 84*/ MatIsSymmetric_SeqAIJ, 30641cbb95d3SBarry Smith MatIsHermitian_SeqAIJ, 30656284ec50SHong Zhang 0, 30666284ec50SHong Zhang 0, 3067bc011b1eSHong Zhang 0, 3068d519adbfSMatthew Knepley /* 89*/ MatMatMult_SeqAIJ_SeqAIJ, 306926be0446SHong Zhang MatMatMultSymbolic_SeqAIJ_SeqAIJ, 307026be0446SHong Zhang MatMatMultNumeric_SeqAIJ_SeqAIJ, 307165e8a0caSHong Zhang MatPtAP_SeqAIJ_SeqAIJ, 30724a1b09b7SHong Zhang MatPtAPSymbolic_SeqAIJ_SeqAIJ_DenseAxpy, 307365e8a0caSHong Zhang /* 94*/ MatPtAPNumeric_SeqAIJ_SeqAIJ, 30746fc122caSHong Zhang MatMatTransposeMult_SeqAIJ_SeqAIJ, 30756fc122caSHong Zhang MatMatTransposeMultSymbolic_SeqAIJ_SeqAIJ, 30766fc122caSHong Zhang MatMatTransposeMultNumeric_SeqAIJ_SeqAIJ, 30772121bac1SHong Zhang 0, 30782121bac1SHong Zhang /* 99*/ 0, 3079609c6c4dSKris Buschelman 0, 3080609c6c4dSKris Buschelman 0, 308187d4246cSBarry Smith MatConjugate_SeqAIJ, 308287d4246cSBarry Smith 0, 3083d519adbfSMatthew Knepley /*104*/ MatSetValuesRow_SeqAIJ, 308499cafbc1SBarry Smith MatRealPart_SeqAIJ, 3085f5edf698SHong Zhang MatImaginaryPart_SeqAIJ, 3086f5edf698SHong Zhang 0, 30872bebee5dSHong Zhang 0, 3088cbd44569SHong Zhang /*109*/ MatMatSolve_SeqAIJ, 3089985db425SBarry Smith 0, 30902af78befSBarry Smith MatGetRowMin_SeqAIJ, 30912af78befSBarry Smith 0, 3092599ef60dSHong Zhang MatMissingDiagonal_SeqAIJ, 3093d519adbfSMatthew Knepley /*114*/ 0, 3094599ef60dSHong Zhang 0, 30953c2a7987SHong Zhang 0, 3096fe97e370SBarry Smith 0, 3097fbdbba38SShri Abhyankar 0, 3098fbdbba38SShri Abhyankar /*119*/ 0, 3099fbdbba38SShri Abhyankar 0, 3100fbdbba38SShri Abhyankar 0, 310182d44351SHong Zhang 0, 3102b3a44c85SBarry Smith MatGetMultiProcBlock_SeqAIJ, 31030716a85fSBarry Smith /*124*/ MatFindNonzeroRows_SeqAIJ, 3104bbead8a2SBarry Smith MatGetColumnNorms_SeqAIJ, 310537868618SMatthew G Knepley MatInvertBlockDiagonal_SeqAIJ, 310637868618SMatthew G Knepley 0, 310737868618SMatthew G Knepley 0, 31085df89d91SHong Zhang /*129*/ 0, 310975648e8dSHong Zhang MatTransposeMatMult_SeqAIJ_SeqAIJ, 311075648e8dSHong Zhang MatTransposeMatMultSymbolic_SeqAIJ_SeqAIJ, 311175648e8dSHong Zhang MatTransposeMatMultNumeric_SeqAIJ_SeqAIJ, 3112b9af6bddSHong Zhang MatTransposeColoringCreate_SeqAIJ, 3113b9af6bddSHong Zhang /*134*/ MatTransColoringApplySpToDen_SeqAIJ, 31142b8ad9a3SHong Zhang MatTransColoringApplyDenToSp_SeqAIJ, 31152b8ad9a3SHong Zhang MatRARt_SeqAIJ_SeqAIJ, 31162b8ad9a3SHong Zhang MatRARtSymbolic_SeqAIJ_SeqAIJ, 31173964eb88SJed Brown MatRARtNumeric_SeqAIJ_SeqAIJ, 31183964eb88SJed Brown /*139*/0, 3119f9426fe0SMark Adams 0, 31201919a2e2SJed Brown 0, 31213a062f41SBarry Smith MatFDColoringSetUp_SeqXAIJ, 31229c8f2541SHong Zhang MatFindOffBlockDiagonalEntries_SeqAIJ, 31239c8f2541SHong Zhang /*144*/MatCreateMPIMatConcatenateSeqMat_SeqAIJ 31249e29f15eSvictorle }; 312517ab2063SBarry Smith 31267087cfbeSBarry Smith PetscErrorCode MatSeqAIJSetColumnIndices_SeqAIJ(Mat mat,PetscInt *indices) 3127bef8e0ddSBarry Smith { 3128bef8e0ddSBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)mat->data; 312997f1f81fSBarry Smith PetscInt i,nz,n; 3130bef8e0ddSBarry Smith 3131bef8e0ddSBarry Smith PetscFunctionBegin; 3132bef8e0ddSBarry Smith nz = aij->maxnz; 3133d0f46423SBarry Smith n = mat->rmap->n; 3134bef8e0ddSBarry Smith for (i=0; i<nz; i++) { 3135bef8e0ddSBarry Smith aij->j[i] = indices[i]; 3136bef8e0ddSBarry Smith } 3137bef8e0ddSBarry Smith aij->nz = nz; 3138bef8e0ddSBarry Smith for (i=0; i<n; i++) { 3139bef8e0ddSBarry Smith aij->ilen[i] = aij->imax[i]; 3140bef8e0ddSBarry Smith } 3141bef8e0ddSBarry Smith PetscFunctionReturn(0); 3142bef8e0ddSBarry Smith } 3143bef8e0ddSBarry Smith 3144bef8e0ddSBarry Smith /*@ 3145bef8e0ddSBarry Smith MatSeqAIJSetColumnIndices - Set the column indices for all the rows 3146bef8e0ddSBarry Smith in the matrix. 3147bef8e0ddSBarry Smith 3148bef8e0ddSBarry Smith Input Parameters: 3149bef8e0ddSBarry Smith + mat - the SeqAIJ matrix 3150bef8e0ddSBarry Smith - indices - the column indices 3151bef8e0ddSBarry Smith 315215091d37SBarry Smith Level: advanced 315315091d37SBarry Smith 3154bef8e0ddSBarry Smith Notes: 3155bef8e0ddSBarry Smith This can be called if you have precomputed the nonzero structure of the 3156bef8e0ddSBarry Smith matrix and want to provide it to the matrix object to improve the performance 3157bef8e0ddSBarry Smith of the MatSetValues() operation. 3158bef8e0ddSBarry Smith 3159bef8e0ddSBarry Smith You MUST have set the correct numbers of nonzeros per row in the call to 3160d1be2dadSMatthew Knepley MatCreateSeqAIJ(), and the columns indices MUST be sorted. 3161bef8e0ddSBarry Smith 3162bef8e0ddSBarry Smith MUST be called before any calls to MatSetValues(); 3163bef8e0ddSBarry Smith 3164b9617806SBarry Smith The indices should start with zero, not one. 3165b9617806SBarry Smith 3166bef8e0ddSBarry Smith @*/ 31677087cfbeSBarry Smith PetscErrorCode MatSeqAIJSetColumnIndices(Mat mat,PetscInt *indices) 3168bef8e0ddSBarry Smith { 31694ac538c5SBarry Smith PetscErrorCode ierr; 3170bef8e0ddSBarry Smith 3171bef8e0ddSBarry Smith PetscFunctionBegin; 31720700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 31734482741eSBarry Smith PetscValidPointer(indices,2); 31744ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr); 3175bef8e0ddSBarry Smith PetscFunctionReturn(0); 3176bef8e0ddSBarry Smith } 3177bef8e0ddSBarry Smith 3178be6bf707SBarry Smith /* ----------------------------------------------------------------------------------------*/ 3179be6bf707SBarry Smith 31807087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqAIJ(Mat mat) 3181be6bf707SBarry Smith { 3182be6bf707SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)mat->data; 31836849ba73SBarry Smith PetscErrorCode ierr; 3184d0f46423SBarry Smith size_t nz = aij->i[mat->rmap->n]; 3185be6bf707SBarry Smith 3186be6bf707SBarry Smith PetscFunctionBegin; 3187169f6850SBarry Smith if (!aij->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 3188be6bf707SBarry Smith 3189be6bf707SBarry Smith /* allocate space for values if not already there */ 3190be6bf707SBarry Smith if (!aij->saved_values) { 3191854ce69bSBarry Smith ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr); 31923bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)mat,(nz+1)*sizeof(PetscScalar));CHKERRQ(ierr); 3193be6bf707SBarry Smith } 3194be6bf707SBarry Smith 3195be6bf707SBarry Smith /* copy values over */ 319687828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 3197be6bf707SBarry Smith PetscFunctionReturn(0); 3198be6bf707SBarry Smith } 3199be6bf707SBarry Smith 3200be6bf707SBarry Smith /*@ 3201be6bf707SBarry Smith MatStoreValues - Stashes a copy of the matrix values; this allows, for 3202be6bf707SBarry Smith example, reuse of the linear part of a Jacobian, while recomputing the 3203be6bf707SBarry Smith nonlinear portion. 3204be6bf707SBarry Smith 3205be6bf707SBarry Smith Collect on Mat 3206be6bf707SBarry Smith 3207be6bf707SBarry Smith Input Parameters: 32080e609b76SBarry Smith . mat - the matrix (currently only AIJ matrices support this option) 3209be6bf707SBarry Smith 321015091d37SBarry Smith Level: advanced 321115091d37SBarry Smith 3212be6bf707SBarry Smith Common Usage, with SNESSolve(): 3213be6bf707SBarry Smith $ Create Jacobian matrix 3214be6bf707SBarry Smith $ Set linear terms into matrix 3215be6bf707SBarry Smith $ Apply boundary conditions to matrix, at this time matrix must have 3216be6bf707SBarry Smith $ final nonzero structure (i.e. setting the nonlinear terms and applying 3217be6bf707SBarry Smith $ boundary conditions again will not change the nonzero structure 3218512a5fc5SBarry Smith $ ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE); 3219be6bf707SBarry Smith $ ierr = MatStoreValues(mat); 3220be6bf707SBarry Smith $ Call SNESSetJacobian() with matrix 3221be6bf707SBarry Smith $ In your Jacobian routine 3222be6bf707SBarry Smith $ ierr = MatRetrieveValues(mat); 3223be6bf707SBarry Smith $ Set nonlinear terms in matrix 3224be6bf707SBarry Smith 3225be6bf707SBarry Smith Common Usage without SNESSolve(), i.e. when you handle nonlinear solve yourself: 3226be6bf707SBarry Smith $ // build linear portion of Jacobian 3227512a5fc5SBarry Smith $ ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE); 3228be6bf707SBarry Smith $ ierr = MatStoreValues(mat); 3229be6bf707SBarry Smith $ loop over nonlinear iterations 3230be6bf707SBarry Smith $ ierr = MatRetrieveValues(mat); 3231be6bf707SBarry Smith $ // call MatSetValues(mat,...) to set nonliner portion of Jacobian 3232be6bf707SBarry Smith $ // call MatAssemblyBegin/End() on matrix 3233be6bf707SBarry Smith $ Solve linear system with Jacobian 3234be6bf707SBarry Smith $ endloop 3235be6bf707SBarry Smith 3236be6bf707SBarry Smith Notes: 3237be6bf707SBarry Smith Matrix must already be assemblied before calling this routine 3238512a5fc5SBarry Smith Must set the matrix option MatSetOption(mat,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE); before 3239be6bf707SBarry Smith calling this routine. 3240be6bf707SBarry Smith 32410c468ba9SBarry Smith When this is called multiple times it overwrites the previous set of stored values 32420c468ba9SBarry Smith and does not allocated additional space. 32430c468ba9SBarry Smith 3244be6bf707SBarry Smith .seealso: MatRetrieveValues() 3245be6bf707SBarry Smith 3246be6bf707SBarry Smith @*/ 32477087cfbeSBarry Smith PetscErrorCode MatStoreValues(Mat mat) 3248be6bf707SBarry Smith { 32494ac538c5SBarry Smith PetscErrorCode ierr; 3250be6bf707SBarry Smith 3251be6bf707SBarry Smith PetscFunctionBegin; 32520700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 3253e32f2f54SBarry Smith if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 3254e32f2f54SBarry Smith if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 32554ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatStoreValues_C",(Mat),(mat));CHKERRQ(ierr); 3256be6bf707SBarry Smith PetscFunctionReturn(0); 3257be6bf707SBarry Smith } 3258be6bf707SBarry Smith 32597087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqAIJ(Mat mat) 3260be6bf707SBarry Smith { 3261be6bf707SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)mat->data; 32626849ba73SBarry Smith PetscErrorCode ierr; 3263d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->n]; 3264be6bf707SBarry Smith 3265be6bf707SBarry Smith PetscFunctionBegin; 3266169f6850SBarry Smith if (!aij->nonew) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 3267f23aa3ddSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 3268be6bf707SBarry Smith /* copy values over */ 326987828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 3270be6bf707SBarry Smith PetscFunctionReturn(0); 3271be6bf707SBarry Smith } 3272be6bf707SBarry Smith 3273be6bf707SBarry Smith /*@ 3274be6bf707SBarry Smith MatRetrieveValues - Retrieves the copy of the matrix values; this allows, for 3275be6bf707SBarry Smith example, reuse of the linear part of a Jacobian, while recomputing the 3276be6bf707SBarry Smith nonlinear portion. 3277be6bf707SBarry Smith 3278be6bf707SBarry Smith Collect on Mat 3279be6bf707SBarry Smith 3280be6bf707SBarry Smith Input Parameters: 3281386f7cf9SBarry Smith . mat - the matrix (currently only AIJ matrices support this option) 3282be6bf707SBarry Smith 328315091d37SBarry Smith Level: advanced 328415091d37SBarry Smith 3285be6bf707SBarry Smith .seealso: MatStoreValues() 3286be6bf707SBarry Smith 3287be6bf707SBarry Smith @*/ 32887087cfbeSBarry Smith PetscErrorCode MatRetrieveValues(Mat mat) 3289be6bf707SBarry Smith { 32904ac538c5SBarry Smith PetscErrorCode ierr; 3291be6bf707SBarry Smith 3292be6bf707SBarry Smith PetscFunctionBegin; 32930700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 3294e32f2f54SBarry Smith if (!mat->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 3295e32f2f54SBarry Smith if (mat->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 32964ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatRetrieveValues_C",(Mat),(mat));CHKERRQ(ierr); 3297be6bf707SBarry Smith PetscFunctionReturn(0); 3298be6bf707SBarry Smith } 3299be6bf707SBarry Smith 3300f83d6046SBarry Smith 3301be6bf707SBarry Smith /* --------------------------------------------------------------------------------*/ 330217ab2063SBarry Smith /*@C 3303682d7d0cSBarry Smith MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format 33040d15e28bSLois Curfman McInnes (the default parallel PETSc format). For good matrix assembly performance 33056e62573dSLois Curfman McInnes the user should preallocate the matrix storage by setting the parameter nz 330651c19458SBarry Smith (or the array nnz). By setting these parameters accurately, performance 33072bd5e0b2SLois Curfman McInnes during matrix assembly can be increased by more than a factor of 50. 330817ab2063SBarry Smith 3309db81eaa0SLois Curfman McInnes Collective on MPI_Comm 3310db81eaa0SLois Curfman McInnes 331117ab2063SBarry Smith Input Parameters: 3312db81eaa0SLois Curfman McInnes + comm - MPI communicator, set to PETSC_COMM_SELF 331317ab2063SBarry Smith . m - number of rows 331417ab2063SBarry Smith . n - number of columns 331517ab2063SBarry Smith . nz - number of nonzeros per row (same for all rows) 331651c19458SBarry Smith - nnz - array containing the number of nonzeros in the various rows 33170298fd71SBarry Smith (possibly different for each row) or NULL 331817ab2063SBarry Smith 331917ab2063SBarry Smith Output Parameter: 3320416022c9SBarry Smith . A - the matrix 332117ab2063SBarry Smith 3322175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 3323ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 3324175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 3325175b88e8SBarry Smith 3326b259b22eSLois Curfman McInnes Notes: 332749a6f317SBarry Smith If nnz is given then nz is ignored 332849a6f317SBarry Smith 332917ab2063SBarry Smith The AIJ format (also called the Yale sparse matrix format or 333017ab2063SBarry Smith compressed row storage), is fully compatible with standard Fortran 77 33310002213bSLois Curfman McInnes storage. That is, the stored row and column indices can begin at 333244cd7ae7SLois Curfman McInnes either one (as in Fortran) or zero. See the users' manual for details. 333317ab2063SBarry Smith 333417ab2063SBarry Smith Specify the preallocated storage with either nz or nnz (not both). 33350298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 33363d323bbdSBarry Smith allocation. For large problems you MUST preallocate memory or you 33376da5968aSLois Curfman McInnes will get TERRIBLE performance, see the users' manual chapter on matrices. 333817ab2063SBarry Smith 3339682d7d0cSBarry Smith By default, this format uses inodes (identical nodes) when possible, to 33404fca80b9SLois Curfman McInnes improve numerical efficiency of matrix-vector products and solves. We 3341682d7d0cSBarry Smith search for consecutive rows with the same nonzero structure, thereby 33426c7ebb05SLois Curfman McInnes reusing matrix information to achieve increased efficiency. 33436c7ebb05SLois Curfman McInnes 33446c7ebb05SLois Curfman McInnes Options Database Keys: 3345698d4c6aSKris Buschelman + -mat_no_inode - Do not use inodes 33469db58ca8SBarry Smith - -mat_inode_limit <limit> - Sets inode limit (max limit=5) 334717ab2063SBarry Smith 3348027ccd11SLois Curfman McInnes Level: intermediate 3349027ccd11SLois Curfman McInnes 335069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateAIJ(), MatSetValues(), MatSeqAIJSetColumnIndices(), MatCreateSeqAIJWithArrays() 335136db0b34SBarry Smith 335217ab2063SBarry Smith @*/ 33537087cfbeSBarry Smith PetscErrorCode MatCreateSeqAIJ(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 335417ab2063SBarry Smith { 3355dfbe8321SBarry Smith PetscErrorCode ierr; 33566945ee14SBarry Smith 33573a40ed3dSBarry Smith PetscFunctionBegin; 3358f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 3359117016b1SBarry Smith ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 3360c4752a88SBarry Smith ierr = MatSetType(*A,MATSEQAIJ);CHKERRQ(ierr); 3361d28bb7d2SJed Brown ierr = MatSeqAIJSetPreallocation_SeqAIJ(*A,nz,nnz);CHKERRQ(ierr); 3362273d9f13SBarry Smith PetscFunctionReturn(0); 3363273d9f13SBarry Smith } 3364273d9f13SBarry Smith 3365273d9f13SBarry Smith /*@C 3366273d9f13SBarry Smith MatSeqAIJSetPreallocation - For good matrix assembly performance 3367273d9f13SBarry Smith the user should preallocate the matrix storage by setting the parameter nz 3368273d9f13SBarry Smith (or the array nnz). By setting these parameters accurately, performance 3369273d9f13SBarry Smith during matrix assembly can be increased by more than a factor of 50. 3370273d9f13SBarry Smith 3371273d9f13SBarry Smith Collective on MPI_Comm 3372273d9f13SBarry Smith 3373273d9f13SBarry Smith Input Parameters: 33741c4f3114SJed Brown + B - The matrix 3375273d9f13SBarry Smith . nz - number of nonzeros per row (same for all rows) 3376273d9f13SBarry Smith - nnz - array containing the number of nonzeros in the various rows 33770298fd71SBarry Smith (possibly different for each row) or NULL 3378273d9f13SBarry Smith 3379273d9f13SBarry Smith Notes: 338049a6f317SBarry Smith If nnz is given then nz is ignored 338149a6f317SBarry Smith 3382273d9f13SBarry Smith The AIJ format (also called the Yale sparse matrix format or 3383273d9f13SBarry Smith compressed row storage), is fully compatible with standard Fortran 77 3384273d9f13SBarry Smith storage. That is, the stored row and column indices can begin at 3385273d9f13SBarry Smith either one (as in Fortran) or zero. See the users' manual for details. 3386273d9f13SBarry Smith 3387273d9f13SBarry Smith Specify the preallocated storage with either nz or nnz (not both). 33880298fd71SBarry Smith Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory 3389273d9f13SBarry Smith allocation. For large problems you MUST preallocate memory or you 3390273d9f13SBarry Smith will get TERRIBLE performance, see the users' manual chapter on matrices. 3391273d9f13SBarry Smith 3392aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 3393aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 3394aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 3395aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 3396aa95bbe8SBarry Smith 3397a96a251dSBarry Smith Developers: Use nz of MAT_SKIP_ALLOCATION to not allocate any space for the matrix 3398a96a251dSBarry Smith entries or columns indices 3399a96a251dSBarry Smith 3400273d9f13SBarry Smith By default, this format uses inodes (identical nodes) when possible, to 3401273d9f13SBarry Smith improve numerical efficiency of matrix-vector products and solves. We 3402273d9f13SBarry Smith search for consecutive rows with the same nonzero structure, thereby 3403273d9f13SBarry Smith reusing matrix information to achieve increased efficiency. 3404273d9f13SBarry Smith 3405273d9f13SBarry Smith Options Database Keys: 3406698d4c6aSKris Buschelman + -mat_no_inode - Do not use inodes 3407698d4c6aSKris Buschelman . -mat_inode_limit <limit> - Sets inode limit (max limit=5) 3408273d9f13SBarry Smith - -mat_aij_oneindex - Internally use indexing starting at 1 3409273d9f13SBarry Smith rather than 0. Note that when calling MatSetValues(), 3410273d9f13SBarry Smith the user still MUST index entries starting at 0! 3411273d9f13SBarry Smith 3412273d9f13SBarry Smith Level: intermediate 3413273d9f13SBarry Smith 341469b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateAIJ(), MatSetValues(), MatSeqAIJSetColumnIndices(), MatCreateSeqAIJWithArrays(), MatGetInfo() 3415273d9f13SBarry Smith 3416273d9f13SBarry Smith @*/ 34177087cfbeSBarry Smith PetscErrorCode MatSeqAIJSetPreallocation(Mat B,PetscInt nz,const PetscInt nnz[]) 3418273d9f13SBarry Smith { 34194ac538c5SBarry Smith PetscErrorCode ierr; 3420a23d5eceSKris Buschelman 3421a23d5eceSKris Buschelman PetscFunctionBegin; 34226ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 34236ba663aaSJed Brown PetscValidType(B,1); 34244ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqAIJSetPreallocation_C",(Mat,PetscInt,const PetscInt[]),(B,nz,nnz));CHKERRQ(ierr); 3425a23d5eceSKris Buschelman PetscFunctionReturn(0); 3426a23d5eceSKris Buschelman } 3427a23d5eceSKris Buschelman 34287087cfbeSBarry Smith PetscErrorCode MatSeqAIJSetPreallocation_SeqAIJ(Mat B,PetscInt nz,const PetscInt *nnz) 3429a23d5eceSKris Buschelman { 3430273d9f13SBarry Smith Mat_SeqAIJ *b; 34312576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,realalloc = PETSC_FALSE; 34326849ba73SBarry Smith PetscErrorCode ierr; 343397f1f81fSBarry Smith PetscInt i; 3434273d9f13SBarry Smith 3435273d9f13SBarry Smith PetscFunctionBegin; 34362576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 3437a96a251dSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 3438c461c341SBarry Smith skipallocation = PETSC_TRUE; 3439c461c341SBarry Smith nz = 0; 3440c461c341SBarry Smith } 3441c461c341SBarry Smith 344226283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 344326283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 3444899cda47SBarry Smith 3445435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 344660e0710aSBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 3447b73539f3SBarry Smith if (nnz) { 3448d0f46423SBarry Smith for (i=0; i<B->rmap->n; i++) { 344960e0710aSBarry Smith if (nnz[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]); 345060e0710aSBarry Smith if (nnz[i] > B->cmap->n) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than row length: local row %D value %d rowlength %D",i,nnz[i],B->cmap->n); 3451b73539f3SBarry Smith } 3452b73539f3SBarry Smith } 3453b73539f3SBarry Smith 3454273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 34552205254eSKarl Rupp 3456273d9f13SBarry Smith b = (Mat_SeqAIJ*)B->data; 3457273d9f13SBarry Smith 3458ab93d7beSBarry Smith if (!skipallocation) { 34592ee49352SLisandro Dalcin if (!b->imax) { 3460dcca6d9dSJed Brown ierr = PetscMalloc2(B->rmap->n,&b->imax,B->rmap->n,&b->ilen);CHKERRQ(ierr); 34613bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,2*B->rmap->n*sizeof(PetscInt));CHKERRQ(ierr); 34622ee49352SLisandro Dalcin } 3463273d9f13SBarry Smith if (!nnz) { 3464435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 10; 3465c62bd62aSJed Brown else if (nz < 0) nz = 1; 3466d0f46423SBarry Smith for (i=0; i<B->rmap->n; i++) b->imax[i] = nz; 3467d0f46423SBarry Smith nz = nz*B->rmap->n; 3468273d9f13SBarry Smith } else { 3469273d9f13SBarry Smith nz = 0; 3470d0f46423SBarry Smith for (i=0; i<B->rmap->n; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 3471273d9f13SBarry Smith } 3472ab93d7beSBarry Smith /* b->ilen will count nonzeros in each row so far. */ 34732205254eSKarl Rupp for (i=0; i<B->rmap->n; i++) b->ilen[i] = 0; 3474ab93d7beSBarry Smith 3475273d9f13SBarry Smith /* allocate the matrix space */ 347653dd7562SDmitry Karpeev /* FIXME: should B's old memory be unlogged? */ 34772ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 3478dcca6d9dSJed Brown ierr = PetscMalloc3(nz,&b->a,nz,&b->j,B->rmap->n+1,&b->i);CHKERRQ(ierr); 34793bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->n+1)*sizeof(PetscInt)+nz*(sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 3480bfeeae90SHong Zhang b->i[0] = 0; 3481d0f46423SBarry Smith for (i=1; i<B->rmap->n+1; i++) { 34825da197adSKris Buschelman b->i[i] = b->i[i-1] + b->imax[i-1]; 34835da197adSKris Buschelman } 3484273d9f13SBarry Smith b->singlemalloc = PETSC_TRUE; 3485e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3486e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 3487c461c341SBarry Smith } else { 3488e6b907acSBarry Smith b->free_a = PETSC_FALSE; 3489e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 3490c461c341SBarry Smith } 3491273d9f13SBarry Smith 3492273d9f13SBarry Smith b->nz = 0; 3493273d9f13SBarry Smith b->maxnz = nz; 3494273d9f13SBarry Smith B->info.nz_unneeded = (double)b->maxnz; 34952205254eSKarl Rupp if (realalloc) { 34962205254eSKarl Rupp ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 34972205254eSKarl Rupp } 3498cb7b82ddSBarry Smith B->was_assembled = PETSC_FALSE; 3499cb7b82ddSBarry Smith B->assembled = PETSC_FALSE; 3500273d9f13SBarry Smith PetscFunctionReturn(0); 3501273d9f13SBarry Smith } 3502273d9f13SBarry Smith 350358d36128SBarry Smith /*@ 3504a1661176SMatthew Knepley MatSeqAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in AIJ format. 3505a1661176SMatthew Knepley 3506a1661176SMatthew Knepley Input Parameters: 3507a1661176SMatthew Knepley + B - the matrix 3508a1661176SMatthew Knepley . i - the indices into j for the start of each row (starts with zero) 3509a1661176SMatthew Knepley . j - the column indices for each row (starts with zero) these must be sorted for each row 3510a1661176SMatthew Knepley - v - optional values in the matrix 3511a1661176SMatthew Knepley 3512a1661176SMatthew Knepley Level: developer 3513a1661176SMatthew Knepley 351458d36128SBarry Smith The i,j,v values are COPIED with this routine; to avoid the copy use MatCreateSeqAIJWithArrays() 351558d36128SBarry Smith 3516a1661176SMatthew Knepley .keywords: matrix, aij, compressed row, sparse, sequential 3517a1661176SMatthew Knepley 3518a1661176SMatthew Knepley .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatSeqAIJSetPreallocation(), MatCreateSeqAIJ(), SeqAIJ 3519a1661176SMatthew Knepley @*/ 3520a1661176SMatthew Knepley PetscErrorCode MatSeqAIJSetPreallocationCSR(Mat B,const PetscInt i[],const PetscInt j[],const PetscScalar v[]) 3521a1661176SMatthew Knepley { 3522a1661176SMatthew Knepley PetscErrorCode ierr; 3523a1661176SMatthew Knepley 3524a1661176SMatthew Knepley PetscFunctionBegin; 35250700a824SBarry Smith PetscValidHeaderSpecific(B,MAT_CLASSID,1); 35266ba663aaSJed Brown PetscValidType(B,1); 35274ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqAIJSetPreallocationCSR_C",(Mat,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,i,j,v));CHKERRQ(ierr); 3528a1661176SMatthew Knepley PetscFunctionReturn(0); 3529a1661176SMatthew Knepley } 3530a1661176SMatthew Knepley 35317087cfbeSBarry Smith PetscErrorCode MatSeqAIJSetPreallocationCSR_SeqAIJ(Mat B,const PetscInt Ii[],const PetscInt J[],const PetscScalar v[]) 3532a1661176SMatthew Knepley { 3533a1661176SMatthew Knepley PetscInt i; 3534a1661176SMatthew Knepley PetscInt m,n; 3535a1661176SMatthew Knepley PetscInt nz; 3536a1661176SMatthew Knepley PetscInt *nnz, nz_max = 0; 3537a1661176SMatthew Knepley PetscScalar *values; 3538a1661176SMatthew Knepley PetscErrorCode ierr; 3539a1661176SMatthew Knepley 3540a1661176SMatthew Knepley PetscFunctionBegin; 354165e19b50SBarry Smith if (Ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Ii[0] must be 0 it is %D", Ii[0]); 3542779a8d59SSatish Balay 3543779a8d59SSatish Balay ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 3544779a8d59SSatish Balay ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 3545779a8d59SSatish Balay 3546779a8d59SSatish Balay ierr = MatGetSize(B, &m, &n);CHKERRQ(ierr); 3547854ce69bSBarry Smith ierr = PetscMalloc1(m+1, &nnz);CHKERRQ(ierr); 3548a1661176SMatthew Knepley for (i = 0; i < m; i++) { 3549b7940d39SSatish Balay nz = Ii[i+1]- Ii[i]; 3550a1661176SMatthew Knepley nz_max = PetscMax(nz_max, nz); 355165e19b50SBarry Smith if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Local row %D has a negative number of columns %D", i, nnz); 3552a1661176SMatthew Knepley nnz[i] = nz; 3553a1661176SMatthew Knepley } 3554a1661176SMatthew Knepley ierr = MatSeqAIJSetPreallocation(B, 0, nnz);CHKERRQ(ierr); 3555a1661176SMatthew Knepley ierr = PetscFree(nnz);CHKERRQ(ierr); 3556a1661176SMatthew Knepley 3557a1661176SMatthew Knepley if (v) { 3558a1661176SMatthew Knepley values = (PetscScalar*) v; 3559a1661176SMatthew Knepley } else { 35601795a4d1SJed Brown ierr = PetscCalloc1(nz_max, &values);CHKERRQ(ierr); 3561a1661176SMatthew Knepley } 3562a1661176SMatthew Knepley 3563a1661176SMatthew Knepley for (i = 0; i < m; i++) { 3564b7940d39SSatish Balay nz = Ii[i+1] - Ii[i]; 3565b7940d39SSatish Balay ierr = MatSetValues_SeqAIJ(B, 1, &i, nz, J+Ii[i], values + (v ? Ii[i] : 0), INSERT_VALUES);CHKERRQ(ierr); 3566a1661176SMatthew Knepley } 3567a1661176SMatthew Knepley 3568a1661176SMatthew Knepley ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3569a1661176SMatthew Knepley ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3570a1661176SMatthew Knepley 3571a1661176SMatthew Knepley if (!v) { 3572a1661176SMatthew Knepley ierr = PetscFree(values);CHKERRQ(ierr); 3573a1661176SMatthew Knepley } 35747827cd58SJed Brown ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 3575a1661176SMatthew Knepley PetscFunctionReturn(0); 3576a1661176SMatthew Knepley } 3577a1661176SMatthew Knepley 3578c6db04a5SJed Brown #include <../src/mat/impls/dense/seq/dense.h> 3579af0996ceSBarry Smith #include <petsc/private/kernels/petscaxpy.h> 3580170fe5c8SBarry Smith 3581170fe5c8SBarry Smith /* 3582170fe5c8SBarry Smith Computes (B'*A')' since computing B*A directly is untenable 3583170fe5c8SBarry Smith 3584170fe5c8SBarry Smith n p p 3585170fe5c8SBarry Smith ( ) ( ) ( ) 3586170fe5c8SBarry Smith m ( A ) * n ( B ) = m ( C ) 3587170fe5c8SBarry Smith ( ) ( ) ( ) 3588170fe5c8SBarry Smith 3589170fe5c8SBarry Smith */ 3590170fe5c8SBarry Smith PetscErrorCode MatMatMultNumeric_SeqDense_SeqAIJ(Mat A,Mat B,Mat C) 3591170fe5c8SBarry Smith { 3592170fe5c8SBarry Smith PetscErrorCode ierr; 3593170fe5c8SBarry Smith Mat_SeqDense *sub_a = (Mat_SeqDense*)A->data; 3594170fe5c8SBarry Smith Mat_SeqAIJ *sub_b = (Mat_SeqAIJ*)B->data; 3595170fe5c8SBarry Smith Mat_SeqDense *sub_c = (Mat_SeqDense*)C->data; 35961de00fd4SBarry Smith PetscInt i,n,m,q,p; 3597170fe5c8SBarry Smith const PetscInt *ii,*idx; 3598170fe5c8SBarry Smith const PetscScalar *b,*a,*a_q; 3599170fe5c8SBarry Smith PetscScalar *c,*c_q; 3600170fe5c8SBarry Smith 3601170fe5c8SBarry Smith PetscFunctionBegin; 3602d0f46423SBarry Smith m = A->rmap->n; 3603d0f46423SBarry Smith n = A->cmap->n; 3604d0f46423SBarry Smith p = B->cmap->n; 3605170fe5c8SBarry Smith a = sub_a->v; 3606170fe5c8SBarry Smith b = sub_b->a; 3607170fe5c8SBarry Smith c = sub_c->v; 3608170fe5c8SBarry Smith ierr = PetscMemzero(c,m*p*sizeof(PetscScalar));CHKERRQ(ierr); 3609170fe5c8SBarry Smith 3610170fe5c8SBarry Smith ii = sub_b->i; 3611170fe5c8SBarry Smith idx = sub_b->j; 3612170fe5c8SBarry Smith for (i=0; i<n; i++) { 3613170fe5c8SBarry Smith q = ii[i+1] - ii[i]; 3614170fe5c8SBarry Smith while (q-->0) { 3615170fe5c8SBarry Smith c_q = c + m*(*idx); 3616170fe5c8SBarry Smith a_q = a + m*i; 3617854c7f52SBarry Smith PetscKernelAXPY(c_q,*b,a_q,m); 3618170fe5c8SBarry Smith idx++; 3619170fe5c8SBarry Smith b++; 3620170fe5c8SBarry Smith } 3621170fe5c8SBarry Smith } 3622170fe5c8SBarry Smith PetscFunctionReturn(0); 3623170fe5c8SBarry Smith } 3624170fe5c8SBarry Smith 3625170fe5c8SBarry Smith PetscErrorCode MatMatMultSymbolic_SeqDense_SeqAIJ(Mat A,Mat B,PetscReal fill,Mat *C) 3626170fe5c8SBarry Smith { 3627170fe5c8SBarry Smith PetscErrorCode ierr; 3628d0f46423SBarry Smith PetscInt m=A->rmap->n,n=B->cmap->n; 3629170fe5c8SBarry Smith Mat Cmat; 3630170fe5c8SBarry Smith 3631170fe5c8SBarry Smith PetscFunctionBegin; 363260e0710aSBarry Smith if (A->cmap->n != B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"A->cmap->n %D != B->rmap->n %D\n",A->cmap->n,B->rmap->n); 3633ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&Cmat);CHKERRQ(ierr); 3634170fe5c8SBarry Smith ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr); 363533d57670SJed Brown ierr = MatSetBlockSizesFromMats(Cmat,A,B);CHKERRQ(ierr); 3636170fe5c8SBarry Smith ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr); 36370298fd71SBarry Smith ierr = MatSeqDenseSetPreallocation(Cmat,NULL);CHKERRQ(ierr); 3638d73949e8SHong Zhang 3639d73949e8SHong Zhang Cmat->ops->matmultnumeric = MatMatMultNumeric_SeqDense_SeqAIJ; 36402205254eSKarl Rupp 3641170fe5c8SBarry Smith *C = Cmat; 3642170fe5c8SBarry Smith PetscFunctionReturn(0); 3643170fe5c8SBarry Smith } 3644170fe5c8SBarry Smith 3645170fe5c8SBarry Smith /* ----------------------------------------------------------------*/ 3646150d2497SBarry Smith PETSC_INTERN PetscErrorCode MatMatMult_SeqDense_SeqAIJ(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3647170fe5c8SBarry Smith { 3648170fe5c8SBarry Smith PetscErrorCode ierr; 3649170fe5c8SBarry Smith 3650170fe5c8SBarry Smith PetscFunctionBegin; 3651170fe5c8SBarry Smith if (scall == MAT_INITIAL_MATRIX) { 36523ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3653170fe5c8SBarry Smith ierr = MatMatMultSymbolic_SeqDense_SeqAIJ(A,B,fill,C);CHKERRQ(ierr); 36543ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3655170fe5c8SBarry Smith } 36563ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3657170fe5c8SBarry Smith ierr = MatMatMultNumeric_SeqDense_SeqAIJ(A,B,*C);CHKERRQ(ierr); 36583ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3659170fe5c8SBarry Smith PetscFunctionReturn(0); 3660170fe5c8SBarry Smith } 3661170fe5c8SBarry Smith 3662170fe5c8SBarry Smith 36630bad9183SKris Buschelman /*MC 3664fafad747SKris Buschelman MATSEQAIJ - MATSEQAIJ = "seqaij" - A matrix type to be used for sequential sparse matrices, 36650bad9183SKris Buschelman based on compressed sparse row format. 36660bad9183SKris Buschelman 36670bad9183SKris Buschelman Options Database Keys: 36680bad9183SKris Buschelman . -mat_type seqaij - sets the matrix type to "seqaij" during a call to MatSetFromOptions() 36690bad9183SKris Buschelman 36700bad9183SKris Buschelman Level: beginner 36710bad9183SKris Buschelman 3672f587520bSBarry Smith .seealso: MatCreateSeqAIJ(), MatSetFromOptions(), MatSetType(), MatCreate(), MatType 36730bad9183SKris Buschelman M*/ 36740bad9183SKris Buschelman 3675ccd284c7SBarry Smith /*MC 3676ccd284c7SBarry Smith MATAIJ - MATAIJ = "aij" - A matrix type to be used for sparse matrices. 3677ccd284c7SBarry Smith 3678ccd284c7SBarry Smith This matrix type is identical to MATSEQAIJ when constructed with a single process communicator, 3679ccd284c7SBarry Smith and MATMPIAIJ otherwise. As a result, for single process communicators, 3680ccd284c7SBarry Smith MatSeqAIJSetPreallocation is supported, and similarly MatMPIAIJSetPreallocation is supported 3681ccd284c7SBarry Smith for communicators controlling multiple processes. It is recommended that you call both of 3682ccd284c7SBarry Smith the above preallocation routines for simplicity. 3683ccd284c7SBarry Smith 3684ccd284c7SBarry Smith Options Database Keys: 3685ccd284c7SBarry Smith . -mat_type aij - sets the matrix type to "aij" during a call to MatSetFromOptions() 3686ccd284c7SBarry Smith 3687ccd284c7SBarry Smith Developer Notes: Subclasses include MATAIJCUSP, MATAIJPERM, MATAIJCRL, and also automatically switches over to use inodes when 3688ccd284c7SBarry Smith enough exist. 3689ccd284c7SBarry Smith 3690ccd284c7SBarry Smith Level: beginner 3691ccd284c7SBarry Smith 3692ccd284c7SBarry Smith .seealso: MatCreateAIJ(), MatCreateSeqAIJ(), MATSEQAIJ,MATMPIAIJ 3693ccd284c7SBarry Smith M*/ 3694ccd284c7SBarry Smith 3695ccd284c7SBarry Smith /*MC 3696ccd284c7SBarry Smith MATAIJCRL - MATAIJCRL = "aijcrl" - A matrix type to be used for sparse matrices. 3697ccd284c7SBarry Smith 3698ccd284c7SBarry Smith This matrix type is identical to MATSEQAIJCRL when constructed with a single process communicator, 3699ccd284c7SBarry Smith and MATMPIAIJCRL otherwise. As a result, for single process communicators, 3700ccd284c7SBarry Smith MatSeqAIJSetPreallocation() is supported, and similarly MatMPIAIJSetPreallocation() is supported 3701ccd284c7SBarry Smith for communicators controlling multiple processes. It is recommended that you call both of 3702ccd284c7SBarry Smith the above preallocation routines for simplicity. 3703ccd284c7SBarry Smith 3704ccd284c7SBarry Smith Options Database Keys: 3705ccd284c7SBarry Smith . -mat_type aijcrl - sets the matrix type to "aijcrl" during a call to MatSetFromOptions() 3706ccd284c7SBarry Smith 3707ccd284c7SBarry Smith Level: beginner 3708ccd284c7SBarry Smith 3709ccd284c7SBarry Smith .seealso: MatCreateMPIAIJCRL,MATSEQAIJCRL,MATMPIAIJCRL, MATSEQAIJCRL, MATMPIAIJCRL 3710ccd284c7SBarry Smith M*/ 3711ccd284c7SBarry Smith 3712cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_SeqAIJCRL(Mat,MatType,MatReuse,Mat*); 3713af8000cdSHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 3714cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_Elemental(Mat,MatType,MatReuse,Mat*); 3715af8000cdSHong Zhang #endif 371663c07aadSStefano Zampini #if defined(PETSC_HAVE_HYPRE) 371763c07aadSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_AIJ_HYPRE(Mat A,MatType,MatReuse,Mat*); 37183dad0653Sstefano_zampini PETSC_INTERN PetscErrorCode MatMatMatMult_Transpose_AIJ_AIJ(Mat,Mat,Mat,MatReuse,PetscReal,Mat*); 371963c07aadSStefano Zampini #endif 3720cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_SeqDense(Mat,MatType,MatReuse,Mat*); 372142c9c57cSBarry Smith 3722b3866ffcSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 372329b38603SBarry Smith PETSC_EXTERN PetscErrorCode MatlabEnginePut_SeqAIJ(PetscObject,void*); 372429b38603SBarry Smith PETSC_EXTERN PetscErrorCode MatlabEngineGet_SeqAIJ(PetscObject,void*); 3725b3866ffcSBarry Smith #endif 372617667f90SBarry Smith 3727c0c8ee5eSDmitry Karpeev 37288c778c55SBarry Smith /*@C 37298397e458SBarry Smith MatSeqAIJGetArray - gives access to the array where the data for a MATSEQAIJ matrix is stored 37308c778c55SBarry Smith 37318c778c55SBarry Smith Not Collective 37328c778c55SBarry Smith 37338c778c55SBarry Smith Input Parameter: 3734579dbff0SBarry Smith . mat - a MATSEQAIJ matrix 37358c778c55SBarry Smith 37368c778c55SBarry Smith Output Parameter: 37378c778c55SBarry Smith . array - pointer to the data 37388c778c55SBarry Smith 37398c778c55SBarry Smith Level: intermediate 37408c778c55SBarry Smith 3741774cf152SJed Brown .seealso: MatSeqAIJRestoreArray(), MatSeqAIJGetArrayF90() 37428c778c55SBarry Smith @*/ 37438c778c55SBarry Smith PetscErrorCode MatSeqAIJGetArray(Mat A,PetscScalar **array) 37448c778c55SBarry Smith { 37458c778c55SBarry Smith PetscErrorCode ierr; 37468c778c55SBarry Smith 37478c778c55SBarry Smith PetscFunctionBegin; 37488c778c55SBarry Smith ierr = PetscUseMethod(A,"MatSeqAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 37498c778c55SBarry Smith PetscFunctionReturn(0); 37508c778c55SBarry Smith } 37518c778c55SBarry Smith 375221e72a00SBarry Smith /*@C 375321e72a00SBarry Smith MatSeqAIJGetMaxRowNonzeros - returns the maximum number of nonzeros in any row 375421e72a00SBarry Smith 375521e72a00SBarry Smith Not Collective 375621e72a00SBarry Smith 375721e72a00SBarry Smith Input Parameter: 3758579dbff0SBarry Smith . mat - a MATSEQAIJ matrix 375921e72a00SBarry Smith 376021e72a00SBarry Smith Output Parameter: 376121e72a00SBarry Smith . nz - the maximum number of nonzeros in any row 376221e72a00SBarry Smith 376321e72a00SBarry Smith Level: intermediate 376421e72a00SBarry Smith 376521e72a00SBarry Smith .seealso: MatSeqAIJRestoreArray(), MatSeqAIJGetArrayF90() 376621e72a00SBarry Smith @*/ 376721e72a00SBarry Smith PetscErrorCode MatSeqAIJGetMaxRowNonzeros(Mat A,PetscInt *nz) 376821e72a00SBarry Smith { 376921e72a00SBarry Smith Mat_SeqAIJ *aij = (Mat_SeqAIJ*)A->data; 377021e72a00SBarry Smith 377121e72a00SBarry Smith PetscFunctionBegin; 377221e72a00SBarry Smith *nz = aij->rmax; 377321e72a00SBarry Smith PetscFunctionReturn(0); 377421e72a00SBarry Smith } 377521e72a00SBarry Smith 37768c778c55SBarry Smith /*@C 3777579dbff0SBarry Smith MatSeqAIJRestoreArray - returns access to the array where the data for a MATSEQAIJ matrix is stored obtained by MatSeqAIJGetArray() 37788c778c55SBarry Smith 37798c778c55SBarry Smith Not Collective 37808c778c55SBarry Smith 37818c778c55SBarry Smith Input Parameters: 3782579dbff0SBarry Smith . mat - a MATSEQAIJ matrix 37838c778c55SBarry Smith . array - pointer to the data 37848c778c55SBarry Smith 37858c778c55SBarry Smith Level: intermediate 37868c778c55SBarry Smith 3787774cf152SJed Brown .seealso: MatSeqAIJGetArray(), MatSeqAIJRestoreArrayF90() 37888c778c55SBarry Smith @*/ 37898c778c55SBarry Smith PetscErrorCode MatSeqAIJRestoreArray(Mat A,PetscScalar **array) 37908c778c55SBarry Smith { 37918c778c55SBarry Smith PetscErrorCode ierr; 37928c778c55SBarry Smith 37938c778c55SBarry Smith PetscFunctionBegin; 37948c778c55SBarry Smith ierr = PetscUseMethod(A,"MatSeqAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr); 37958c778c55SBarry Smith PetscFunctionReturn(0); 37968c778c55SBarry Smith } 37978c778c55SBarry Smith 37988cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqAIJ(Mat B) 3799273d9f13SBarry Smith { 3800273d9f13SBarry Smith Mat_SeqAIJ *b; 3801dfbe8321SBarry Smith PetscErrorCode ierr; 380238baddfdSBarry Smith PetscMPIInt size; 3803273d9f13SBarry Smith 3804273d9f13SBarry Smith PetscFunctionBegin; 3805ce94432eSBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 3806e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Comm must be of size 1"); 3807273d9f13SBarry Smith 3808b00a9115SJed Brown ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 38092205254eSKarl Rupp 3810b0a32e0cSBarry Smith B->data = (void*)b; 38112205254eSKarl Rupp 3812549d3d68SSatish Balay ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 38132205254eSKarl Rupp 3814416022c9SBarry Smith b->row = 0; 3815416022c9SBarry Smith b->col = 0; 381682bf6240SBarry Smith b->icol = 0; 3817b810aeb4SBarry Smith b->reallocs = 0; 381836db0b34SBarry Smith b->ignorezeroentries = PETSC_FALSE; 3819f1e2ffcdSBarry Smith b->roworiented = PETSC_TRUE; 3820416022c9SBarry Smith b->nonew = 0; 3821416022c9SBarry Smith b->diag = 0; 3822416022c9SBarry Smith b->solve_work = 0; 38232a1b7f2aSHong Zhang B->spptr = 0; 3824be6bf707SBarry Smith b->saved_values = 0; 3825d7f994e1SBarry Smith b->idiag = 0; 382671f1c65dSBarry Smith b->mdiag = 0; 382771f1c65dSBarry Smith b->ssor_work = 0; 382871f1c65dSBarry Smith b->omega = 1.0; 382971f1c65dSBarry Smith b->fshift = 0.0; 383071f1c65dSBarry Smith b->idiagvalid = PETSC_FALSE; 3831bbead8a2SBarry Smith b->ibdiagvalid = PETSC_FALSE; 3832a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 383317ab2063SBarry Smith 383435d8aa7fSBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQAIJ);CHKERRQ(ierr); 3835bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqAIJGetArray_C",MatSeqAIJGetArray_SeqAIJ);CHKERRQ(ierr); 3836bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqAIJRestoreArray_C",MatSeqAIJRestoreArray_SeqAIJ);CHKERRQ(ierr); 38378c778c55SBarry Smith 3838b3866ffcSBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 3839bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"PetscMatlabEnginePut_C",MatlabEnginePut_SeqAIJ);CHKERRQ(ierr); 3840bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"PetscMatlabEngineGet_C",MatlabEngineGet_SeqAIJ);CHKERRQ(ierr); 3841b3866ffcSBarry Smith #endif 384217f1a0eaSHong Zhang 3843bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqAIJSetColumnIndices_C",MatSeqAIJSetColumnIndices_SeqAIJ);CHKERRQ(ierr); 3844bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqAIJ);CHKERRQ(ierr); 3845bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqAIJ);CHKERRQ(ierr); 3846bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_seqsbaij_C",MatConvert_SeqAIJ_SeqSBAIJ);CHKERRQ(ierr); 3847bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_seqbaij_C",MatConvert_SeqAIJ_SeqBAIJ);CHKERRQ(ierr); 3848bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_seqaijperm_C",MatConvert_SeqAIJ_SeqAIJPERM);CHKERRQ(ierr); 3849bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_seqaijcrl_C",MatConvert_SeqAIJ_SeqAIJCRL);CHKERRQ(ierr); 3850af8000cdSHong Zhang #if defined(PETSC_HAVE_ELEMENTAL) 3851af8000cdSHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_elemental_C",MatConvert_SeqAIJ_Elemental);CHKERRQ(ierr); 3852af8000cdSHong Zhang #endif 385363c07aadSStefano Zampini #if defined(PETSC_HAVE_HYPRE) 385463c07aadSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_hypre_C",MatConvert_AIJ_HYPRE);CHKERRQ(ierr); 38553dad0653Sstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMatMult_transpose_seqaij_seqaij_C",MatMatMatMult_Transpose_AIJ_AIJ);CHKERRQ(ierr); 385663c07aadSStefano Zampini #endif 3857b49cda9fSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqaij_seqdense_C",MatConvert_SeqAIJ_SeqDense);CHKERRQ(ierr); 3858bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatIsTranspose_C",MatIsTranspose_SeqAIJ);CHKERRQ(ierr); 3859bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatIsHermitianTranspose_C",MatIsTranspose_SeqAIJ);CHKERRQ(ierr); 3860bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqAIJSetPreallocation_C",MatSeqAIJSetPreallocation_SeqAIJ);CHKERRQ(ierr); 3861bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqAIJSetPreallocationCSR_C",MatSeqAIJSetPreallocationCSR_SeqAIJ);CHKERRQ(ierr); 3862bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatReorderForNonzeroDiagonal_C",MatReorderForNonzeroDiagonal_SeqAIJ);CHKERRQ(ierr); 3863bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMult_seqdense_seqaij_C",MatMatMult_SeqDense_SeqAIJ);CHKERRQ(ierr); 3864bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMultSymbolic_seqdense_seqaij_C",MatMatMultSymbolic_SeqDense_SeqAIJ);CHKERRQ(ierr); 3865bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMultNumeric_seqdense_seqaij_C",MatMatMultNumeric_SeqDense_SeqAIJ);CHKERRQ(ierr); 38664108e4d5SBarry Smith ierr = MatCreate_SeqAIJ_Inode(B);CHKERRQ(ierr); 386717667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQAIJ);CHKERRQ(ierr); 38683a40ed3dSBarry Smith PetscFunctionReturn(0); 386917ab2063SBarry Smith } 387017ab2063SBarry Smith 3871b24902e0SBarry Smith /* 3872b24902e0SBarry Smith Given a matrix generated with MatGetFactor() duplicates all the information in A into B 3873b24902e0SBarry Smith */ 3874ace3abfcSBarry Smith PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat C,Mat A,MatDuplicateOption cpvalues,PetscBool mallocmatspace) 387517ab2063SBarry Smith { 3876416022c9SBarry Smith Mat_SeqAIJ *c,*a = (Mat_SeqAIJ*)A->data; 38776849ba73SBarry Smith PetscErrorCode ierr; 3878d0f46423SBarry Smith PetscInt i,m = A->rmap->n; 387917ab2063SBarry Smith 38803a40ed3dSBarry Smith PetscFunctionBegin; 3881273d9f13SBarry Smith c = (Mat_SeqAIJ*)C->data; 3882273d9f13SBarry Smith 3883d5f3da31SBarry Smith C->factortype = A->factortype; 3884416022c9SBarry Smith c->row = 0; 3885416022c9SBarry Smith c->col = 0; 388682bf6240SBarry Smith c->icol = 0; 38876ad4291fSHong Zhang c->reallocs = 0; 388817ab2063SBarry Smith 38896ad4291fSHong Zhang C->assembled = PETSC_TRUE; 389017ab2063SBarry Smith 3891aa5ea44dSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 3892aa5ea44dSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 3893eec197d1SBarry Smith 3894dcca6d9dSJed Brown ierr = PetscMalloc2(m,&c->imax,m,&c->ilen);CHKERRQ(ierr); 38953bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C, 2*m*sizeof(PetscInt));CHKERRQ(ierr); 389617ab2063SBarry Smith for (i=0; i<m; i++) { 3897416022c9SBarry Smith c->imax[i] = a->imax[i]; 3898416022c9SBarry Smith c->ilen[i] = a->ilen[i]; 389917ab2063SBarry Smith } 390017ab2063SBarry Smith 390117ab2063SBarry Smith /* allocate the matrix space */ 3902f77e22a1SHong Zhang if (mallocmatspace) { 3903dcca6d9dSJed Brown ierr = PetscMalloc3(a->i[m],&c->a,a->i[m],&c->j,m+1,&c->i);CHKERRQ(ierr); 39043bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C, a->i[m]*(sizeof(PetscScalar)+sizeof(PetscInt))+(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 39052205254eSKarl Rupp 3906f1e2ffcdSBarry Smith c->singlemalloc = PETSC_TRUE; 39072205254eSKarl Rupp 390897f1f81fSBarry Smith ierr = PetscMemcpy(c->i,a->i,(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 390917ab2063SBarry Smith if (m > 0) { 391097f1f81fSBarry Smith ierr = PetscMemcpy(c->j,a->j,(a->i[m])*sizeof(PetscInt));CHKERRQ(ierr); 3911be6bf707SBarry Smith if (cpvalues == MAT_COPY_VALUES) { 3912bfeeae90SHong Zhang ierr = PetscMemcpy(c->a,a->a,(a->i[m])*sizeof(PetscScalar));CHKERRQ(ierr); 3913be6bf707SBarry Smith } else { 3914bfeeae90SHong Zhang ierr = PetscMemzero(c->a,(a->i[m])*sizeof(PetscScalar));CHKERRQ(ierr); 391517ab2063SBarry Smith } 391608480c60SBarry Smith } 3917f77e22a1SHong Zhang } 391817ab2063SBarry Smith 39196ad4291fSHong Zhang c->ignorezeroentries = a->ignorezeroentries; 3920416022c9SBarry Smith c->roworiented = a->roworiented; 3921416022c9SBarry Smith c->nonew = a->nonew; 3922416022c9SBarry Smith if (a->diag) { 3923854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&c->diag);CHKERRQ(ierr); 39243bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)C,(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 392517ab2063SBarry Smith for (i=0; i<m; i++) { 3926416022c9SBarry Smith c->diag[i] = a->diag[i]; 392717ab2063SBarry Smith } 39283a40ed3dSBarry Smith } else c->diag = 0; 39292205254eSKarl Rupp 39306ad4291fSHong Zhang c->solve_work = 0; 39316ad4291fSHong Zhang c->saved_values = 0; 39326ad4291fSHong Zhang c->idiag = 0; 393371f1c65dSBarry Smith c->ssor_work = 0; 3934a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 3935e6b907acSBarry Smith c->free_a = PETSC_TRUE; 3936e6b907acSBarry Smith c->free_ij = PETSC_TRUE; 39376ad4291fSHong Zhang 3938893ad86cSHong Zhang c->rmax = a->rmax; 3939416022c9SBarry Smith c->nz = a->nz; 39408ed568f8SMatthew G Knepley c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 3941273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 3942754ec7b1SSatish Balay 39436ad4291fSHong Zhang c->compressedrow.use = a->compressedrow.use; 39446ad4291fSHong Zhang c->compressedrow.nrows = a->compressedrow.nrows; 3945cd6b891eSBarry Smith if (a->compressedrow.use) { 39466ad4291fSHong Zhang i = a->compressedrow.nrows; 3947dcca6d9dSJed Brown ierr = PetscMalloc2(i+1,&c->compressedrow.i,i,&c->compressedrow.rindex);CHKERRQ(ierr); 39486ad4291fSHong Zhang ierr = PetscMemcpy(c->compressedrow.i,a->compressedrow.i,(i+1)*sizeof(PetscInt));CHKERRQ(ierr); 39496ad4291fSHong Zhang ierr = PetscMemcpy(c->compressedrow.rindex,a->compressedrow.rindex,i*sizeof(PetscInt));CHKERRQ(ierr); 395027ea64f8SHong Zhang } else { 395127ea64f8SHong Zhang c->compressedrow.use = PETSC_FALSE; 39520298fd71SBarry Smith c->compressedrow.i = NULL; 39530298fd71SBarry Smith c->compressedrow.rindex = NULL; 39546ad4291fSHong Zhang } 3955ea632784SBarry Smith c->nonzerorowcnt = a->nonzerorowcnt; 3956e56f5c9eSBarry Smith C->nonzerostate = A->nonzerostate; 39574846f1f5SKris Buschelman 39582205254eSKarl Rupp ierr = MatDuplicate_SeqAIJ_Inode(A,cpvalues,&C);CHKERRQ(ierr); 3959140e18c1SBarry Smith ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 39603a40ed3dSBarry Smith PetscFunctionReturn(0); 396117ab2063SBarry Smith } 396217ab2063SBarry Smith 3963b24902e0SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 3964b24902e0SBarry Smith { 3965b24902e0SBarry Smith PetscErrorCode ierr; 3966b24902e0SBarry Smith 3967b24902e0SBarry Smith PetscFunctionBegin; 3968ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 39694b6263acSBarry Smith ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 3970cfd3f464SBarry Smith if (!(A->rmap->n % A->rmap->bs) && !(A->cmap->n % A->cmap->bs)) { 397133d57670SJed Brown ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr); 3972cfd3f464SBarry Smith } 3973a54f2f98SBarry Smith ierr = MatSetType(*B,((PetscObject)A)->type_name);CHKERRQ(ierr); 3974f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(*B,A,cpvalues,PETSC_TRUE);CHKERRQ(ierr); 3975b24902e0SBarry Smith PetscFunctionReturn(0); 3976b24902e0SBarry Smith } 3977b24902e0SBarry Smith 3978112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqAIJ(Mat newMat, PetscViewer viewer) 3979fbdbba38SShri Abhyankar { 3980fbdbba38SShri Abhyankar Mat_SeqAIJ *a; 3981fbdbba38SShri Abhyankar PetscErrorCode ierr; 3982fbdbba38SShri Abhyankar PetscInt i,sum,nz,header[4],*rowlengths = 0,M,N,rows,cols; 3983fbdbba38SShri Abhyankar int fd; 3984fbdbba38SShri Abhyankar PetscMPIInt size; 3985fbdbba38SShri Abhyankar MPI_Comm comm; 39863059b6faSBarry Smith PetscInt bs = newMat->rmap->bs; 3987fbdbba38SShri Abhyankar 3988fbdbba38SShri Abhyankar PetscFunctionBegin; 3989c98fd787SBarry Smith /* force binary viewer to load .info file if it has not yet done so */ 3990c98fd787SBarry Smith ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr); 3991fbdbba38SShri Abhyankar ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr); 3992fbdbba38SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 3993fbdbba38SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"view must have one processor"); 3994bbead8a2SBarry Smith 39950298fd71SBarry Smith ierr = PetscOptionsBegin(comm,NULL,"Options for loading SEQAIJ matrix","Mat");CHKERRQ(ierr); 39960298fd71SBarry Smith ierr = PetscOptionsInt("-matload_block_size","Set the blocksize used to store the matrix","MatLoad",bs,&bs,NULL);CHKERRQ(ierr); 3997bbead8a2SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 39983059b6faSBarry Smith if (bs < 0) bs = 1; 39993059b6faSBarry Smith ierr = MatSetBlockSize(newMat,bs);CHKERRQ(ierr); 4000bbead8a2SBarry Smith 4001fbdbba38SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 4002fbdbba38SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 4003fbdbba38SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not matrix object in file"); 4004fbdbba38SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 4005fbdbba38SShri Abhyankar 4006bbead8a2SBarry Smith if (nz < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format on disk,cannot load as SeqAIJ"); 4007fbdbba38SShri Abhyankar 4008fbdbba38SShri Abhyankar /* read in row lengths */ 4009785e854fSJed Brown ierr = PetscMalloc1(M,&rowlengths);CHKERRQ(ierr); 4010fbdbba38SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 4011fbdbba38SShri Abhyankar 4012fbdbba38SShri Abhyankar /* check if sum of rowlengths is same as nz */ 4013fbdbba38SShri Abhyankar for (i=0,sum=0; i< M; i++) sum +=rowlengths[i]; 401460e0710aSBarry Smith if (sum != nz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_FILE_READ,"Inconsistant matrix data in file. no-nonzeros = %dD, sum-row-lengths = %D\n",nz,sum); 4015fbdbba38SShri Abhyankar 4016fbdbba38SShri Abhyankar /* set global size if not set already*/ 4017f501eaabSShri Abhyankar if (newMat->rmap->n < 0 && newMat->rmap->N < 0 && newMat->cmap->n < 0 && newMat->cmap->N < 0) { 4018fbdbba38SShri Abhyankar ierr = MatSetSizes(newMat,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 4019aabbc4fbSShri Abhyankar } else { 40209d36ed5fSBarry Smith /* if sizes and type are already set, check if the matrix global sizes are correct */ 4021fbdbba38SShri Abhyankar ierr = MatGetSize(newMat,&rows,&cols);CHKERRQ(ierr); 40224c5b953cSHong Zhang if (rows < 0 && cols < 0) { /* user might provide local size instead of global size */ 40234c5b953cSHong Zhang ierr = MatGetLocalSize(newMat,&rows,&cols);CHKERRQ(ierr); 40244c5b953cSHong Zhang } 402560e0710aSBarry Smith if (M != rows || N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED, "Matrix in file of different length (%D, %D) than the input matrix (%D, %D)",M,N,rows,cols); 4026aabbc4fbSShri Abhyankar } 4027fbdbba38SShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(newMat,0,rowlengths);CHKERRQ(ierr); 4028fbdbba38SShri Abhyankar a = (Mat_SeqAIJ*)newMat->data; 4029fbdbba38SShri Abhyankar 4030fbdbba38SShri Abhyankar ierr = PetscBinaryRead(fd,a->j,nz,PETSC_INT);CHKERRQ(ierr); 4031fbdbba38SShri Abhyankar 4032fbdbba38SShri Abhyankar /* read in nonzero values */ 4033fbdbba38SShri Abhyankar ierr = PetscBinaryRead(fd,a->a,nz,PETSC_SCALAR);CHKERRQ(ierr); 4034fbdbba38SShri Abhyankar 4035fbdbba38SShri Abhyankar /* set matrix "i" values */ 4036fbdbba38SShri Abhyankar a->i[0] = 0; 4037fbdbba38SShri Abhyankar for (i=1; i<= M; i++) { 4038fbdbba38SShri Abhyankar a->i[i] = a->i[i-1] + rowlengths[i-1]; 4039fbdbba38SShri Abhyankar a->ilen[i-1] = rowlengths[i-1]; 4040fbdbba38SShri Abhyankar } 4041fbdbba38SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 4042fbdbba38SShri Abhyankar 4043fbdbba38SShri Abhyankar ierr = MatAssemblyBegin(newMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 4044fbdbba38SShri Abhyankar ierr = MatAssemblyEnd(newMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 4045fbdbba38SShri Abhyankar PetscFunctionReturn(0); 4046fbdbba38SShri Abhyankar } 4047fbdbba38SShri Abhyankar 4048ace3abfcSBarry Smith PetscErrorCode MatEqual_SeqAIJ(Mat A,Mat B,PetscBool * flg) 40497264ac53SSatish Balay { 40507264ac53SSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)B->data; 4051dfbe8321SBarry Smith PetscErrorCode ierr; 4052eeffb40dSHong Zhang #if defined(PETSC_USE_COMPLEX) 4053eeffb40dSHong Zhang PetscInt k; 4054eeffb40dSHong Zhang #endif 40557264ac53SSatish Balay 40563a40ed3dSBarry Smith PetscFunctionBegin; 4057bfeeae90SHong Zhang /* If the matrix dimensions are not equal,or no of nonzeros */ 4058d0f46423SBarry Smith if ((A->rmap->n != B->rmap->n) || (A->cmap->n != B->cmap->n) ||(a->nz != b->nz)) { 4059ca44d042SBarry Smith *flg = PETSC_FALSE; 4060ca44d042SBarry Smith PetscFunctionReturn(0); 4061bcd2baecSBarry Smith } 40627264ac53SSatish Balay 40637264ac53SSatish Balay /* if the a->i are the same */ 4064d0f46423SBarry Smith ierr = PetscMemcmp(a->i,b->i,(A->rmap->n+1)*sizeof(PetscInt),flg);CHKERRQ(ierr); 4065abc0a331SBarry Smith if (!*flg) PetscFunctionReturn(0); 40667264ac53SSatish Balay 40677264ac53SSatish Balay /* if a->j are the same */ 406897f1f81fSBarry Smith ierr = PetscMemcmp(a->j,b->j,(a->nz)*sizeof(PetscInt),flg);CHKERRQ(ierr); 4069abc0a331SBarry Smith if (!*flg) PetscFunctionReturn(0); 4070bcd2baecSBarry Smith 4071bcd2baecSBarry Smith /* if a->a are the same */ 4072eeffb40dSHong Zhang #if defined(PETSC_USE_COMPLEX) 4073eeffb40dSHong Zhang for (k=0; k<a->nz; k++) { 4074eeffb40dSHong Zhang if (PetscRealPart(a->a[k]) != PetscRealPart(b->a[k]) || PetscImaginaryPart(a->a[k]) != PetscImaginaryPart(b->a[k])) { 4075eeffb40dSHong Zhang *flg = PETSC_FALSE; 40763a40ed3dSBarry Smith PetscFunctionReturn(0); 4077eeffb40dSHong Zhang } 4078eeffb40dSHong Zhang } 4079eeffb40dSHong Zhang #else 4080eeffb40dSHong Zhang ierr = PetscMemcmp(a->a,b->a,(a->nz)*sizeof(PetscScalar),flg);CHKERRQ(ierr); 4081eeffb40dSHong Zhang #endif 4082eeffb40dSHong Zhang PetscFunctionReturn(0); 40837264ac53SSatish Balay } 408436db0b34SBarry Smith 408505869f15SSatish Balay /*@ 408636db0b34SBarry Smith MatCreateSeqAIJWithArrays - Creates an sequential AIJ matrix using matrix elements (in CSR format) 408736db0b34SBarry Smith provided by the user. 408836db0b34SBarry Smith 4089c75a6043SHong Zhang Collective on MPI_Comm 409036db0b34SBarry Smith 409136db0b34SBarry Smith Input Parameters: 409236db0b34SBarry Smith + comm - must be an MPI communicator of size 1 409336db0b34SBarry Smith . m - number of rows 409436db0b34SBarry Smith . n - number of columns 409536db0b34SBarry Smith . i - row indices 409636db0b34SBarry Smith . j - column indices 409736db0b34SBarry Smith - a - matrix values 409836db0b34SBarry Smith 409936db0b34SBarry Smith Output Parameter: 410036db0b34SBarry Smith . mat - the matrix 410136db0b34SBarry Smith 410236db0b34SBarry Smith Level: intermediate 410336db0b34SBarry Smith 410436db0b34SBarry Smith Notes: 41050551d7c0SBarry Smith The i, j, and a arrays are not copied by this routine, the user must free these arrays 4106292fb18eSBarry Smith once the matrix is destroyed and not before 410736db0b34SBarry Smith 410836db0b34SBarry Smith You cannot set new nonzero locations into this matrix, that will generate an error. 410936db0b34SBarry Smith 4110bfeeae90SHong Zhang The i and j indices are 0 based 411136db0b34SBarry Smith 4112a4552177SSatish Balay The format which is used for the sparse matrix input, is equivalent to a 4113a4552177SSatish Balay row-major ordering.. i.e for the following matrix, the input data expected is 41148eef79e4SBarry Smith as shown 4115a4552177SSatish Balay 41168eef79e4SBarry Smith $ 1 0 0 41178eef79e4SBarry Smith $ 2 0 3 41188eef79e4SBarry Smith $ 4 5 6 41198eef79e4SBarry Smith $ 41208eef79e4SBarry Smith $ i = {0,1,3,6} [size = nrow+1 = 3+1] 41218eef79e4SBarry Smith $ j = {0,0,2,0,1,2} [size = 6]; values must be sorted for each row 41228eef79e4SBarry Smith $ v = {1,2,3,4,5,6} [size = 6] 4123a4552177SSatish Balay 41249985e31cSBarry Smith 412569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateAIJ(), MatCreateSeqAIJ(), MatCreateMPIAIJWithArrays(), MatMPIAIJSetPreallocationCSR() 412636db0b34SBarry Smith 412736db0b34SBarry Smith @*/ 4128c3c607ccSBarry Smith PetscErrorCode MatCreateSeqAIJWithArrays(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat) 412936db0b34SBarry Smith { 4130dfbe8321SBarry Smith PetscErrorCode ierr; 4131cbcfb4deSHong Zhang PetscInt ii; 413236db0b34SBarry Smith Mat_SeqAIJ *aij; 4133cbcfb4deSHong Zhang #if defined(PETSC_USE_DEBUG) 4134cbcfb4deSHong Zhang PetscInt jj; 4135cbcfb4deSHong Zhang #endif 413636db0b34SBarry Smith 413736db0b34SBarry Smith PetscFunctionBegin; 413841096f02SStefano Zampini if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 4139f69a0ea3SMatthew Knepley ierr = MatCreate(comm,mat);CHKERRQ(ierr); 4140f69a0ea3SMatthew Knepley ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 4141a2f3521dSMark F. Adams /* ierr = MatSetBlockSizes(*mat,,);CHKERRQ(ierr); */ 4142ab93d7beSBarry Smith ierr = MatSetType(*mat,MATSEQAIJ);CHKERRQ(ierr); 4143ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(*mat,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 4144ab93d7beSBarry Smith aij = (Mat_SeqAIJ*)(*mat)->data; 4145dcca6d9dSJed Brown ierr = PetscMalloc2(m,&aij->imax,m,&aij->ilen);CHKERRQ(ierr); 4146ab93d7beSBarry Smith 414736db0b34SBarry Smith aij->i = i; 414836db0b34SBarry Smith aij->j = j; 414936db0b34SBarry Smith aij->a = a; 415036db0b34SBarry Smith aij->singlemalloc = PETSC_FALSE; 415136db0b34SBarry Smith aij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 4152e6b907acSBarry Smith aij->free_a = PETSC_FALSE; 4153e6b907acSBarry Smith aij->free_ij = PETSC_FALSE; 415436db0b34SBarry Smith 415536db0b34SBarry Smith for (ii=0; ii<m; ii++) { 415636db0b34SBarry Smith aij->ilen[ii] = aij->imax[ii] = i[ii+1] - i[ii]; 41572515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 415860e0710aSBarry Smith if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %D length = %D",ii,i[ii+1] - i[ii]); 41599985e31cSBarry Smith for (jj=i[ii]+1; jj<i[ii+1]; jj++) { 4160e32f2f54SBarry Smith if (j[jj] < j[jj-1]) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column entry number %D (actual colum %D) in row %D is not sorted",jj-i[ii],j[jj],ii); 4161e32f2f54SBarry Smith if (j[jj] == j[jj]-1) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column entry number %D (actual colum %D) in row %D is identical to previous entry",jj-i[ii],j[jj],ii); 41629985e31cSBarry Smith } 416336db0b34SBarry Smith #endif 416436db0b34SBarry Smith } 41652515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 416636db0b34SBarry Smith for (ii=0; ii<aij->i[m]; ii++) { 416760e0710aSBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %D index = %D",ii,j[ii]); 416860e0710aSBarry Smith if (j[ii] > n - 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %D index = %D",ii,j[ii]); 416936db0b34SBarry Smith } 417036db0b34SBarry Smith #endif 417136db0b34SBarry Smith 4172b65db4caSBarry Smith ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 4173b65db4caSBarry Smith ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 417436db0b34SBarry Smith PetscFunctionReturn(0); 417536db0b34SBarry Smith } 417680ef6e79SMatthew G Knepley /*@C 4177d021a1c5SVictor Minden MatCreateSeqAIJFromTriple - Creates an sequential AIJ matrix using matrix elements (in COO format) 41788a0b0e6bSVictor Minden provided by the user. 41798a0b0e6bSVictor Minden 41808a0b0e6bSVictor Minden Collective on MPI_Comm 41818a0b0e6bSVictor Minden 41828a0b0e6bSVictor Minden Input Parameters: 41838a0b0e6bSVictor Minden + comm - must be an MPI communicator of size 1 41848a0b0e6bSVictor Minden . m - number of rows 41858a0b0e6bSVictor Minden . n - number of columns 41868a0b0e6bSVictor Minden . i - row indices 41878a0b0e6bSVictor Minden . j - column indices 41881230e6d1SVictor Minden . a - matrix values 41891230e6d1SVictor Minden . nz - number of nonzeros 41901230e6d1SVictor Minden - idx - 0 or 1 based 41918a0b0e6bSVictor Minden 41928a0b0e6bSVictor Minden Output Parameter: 41938a0b0e6bSVictor Minden . mat - the matrix 41948a0b0e6bSVictor Minden 41958a0b0e6bSVictor Minden Level: intermediate 41968a0b0e6bSVictor Minden 41978a0b0e6bSVictor Minden Notes: 41988a0b0e6bSVictor Minden The i and j indices are 0 based 41998a0b0e6bSVictor Minden 42008a0b0e6bSVictor Minden The format which is used for the sparse matrix input, is equivalent to a 42018a0b0e6bSVictor Minden row-major ordering.. i.e for the following matrix, the input data expected is 42028a0b0e6bSVictor Minden as shown: 42038a0b0e6bSVictor Minden 42048a0b0e6bSVictor Minden 1 0 0 42058a0b0e6bSVictor Minden 2 0 3 42068a0b0e6bSVictor Minden 4 5 6 42078a0b0e6bSVictor Minden 42088a0b0e6bSVictor Minden i = {0,1,1,2,2,2} 42098a0b0e6bSVictor Minden j = {0,0,2,0,1,2} 42108a0b0e6bSVictor Minden v = {1,2,3,4,5,6} 42118a0b0e6bSVictor Minden 42128a0b0e6bSVictor Minden 421369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateAIJ(), MatCreateSeqAIJ(), MatCreateSeqAIJWithArrays(), MatMPIAIJSetPreallocationCSR() 42148a0b0e6bSVictor Minden 42158a0b0e6bSVictor Minden @*/ 4216c3c607ccSBarry Smith PetscErrorCode MatCreateSeqAIJFromTriple(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat,PetscInt nz,PetscBool idx) 42178a0b0e6bSVictor Minden { 42188a0b0e6bSVictor Minden PetscErrorCode ierr; 4219d021a1c5SVictor Minden PetscInt ii, *nnz, one = 1,row,col; 42208a0b0e6bSVictor Minden 42218a0b0e6bSVictor Minden 42228a0b0e6bSVictor Minden PetscFunctionBegin; 42231795a4d1SJed Brown ierr = PetscCalloc1(m,&nnz);CHKERRQ(ierr); 42241230e6d1SVictor Minden for (ii = 0; ii < nz; ii++) { 4225c8d679ebSHong Zhang nnz[i[ii] - !!idx] += 1; 42261230e6d1SVictor Minden } 42278a0b0e6bSVictor Minden ierr = MatCreate(comm,mat);CHKERRQ(ierr); 42288a0b0e6bSVictor Minden ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 42298a0b0e6bSVictor Minden ierr = MatSetType(*mat,MATSEQAIJ);CHKERRQ(ierr); 42301230e6d1SVictor Minden ierr = MatSeqAIJSetPreallocation_SeqAIJ(*mat,0,nnz);CHKERRQ(ierr); 42311230e6d1SVictor Minden for (ii = 0; ii < nz; ii++) { 42321230e6d1SVictor Minden if (idx) { 42331230e6d1SVictor Minden row = i[ii] - 1; 42341230e6d1SVictor Minden col = j[ii] - 1; 42351230e6d1SVictor Minden } else { 42361230e6d1SVictor Minden row = i[ii]; 42371230e6d1SVictor Minden col = j[ii]; 42388a0b0e6bSVictor Minden } 42391230e6d1SVictor Minden ierr = MatSetValues(*mat,one,&row,one,&col,&a[ii],ADD_VALUES);CHKERRQ(ierr); 42408a0b0e6bSVictor Minden } 42418a0b0e6bSVictor Minden ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 42428a0b0e6bSVictor Minden ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 4243d021a1c5SVictor Minden ierr = PetscFree(nnz);CHKERRQ(ierr); 42448a0b0e6bSVictor Minden PetscFunctionReturn(0); 42458a0b0e6bSVictor Minden } 424636db0b34SBarry Smith 4247acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal(Mat A) 4248acf2f550SJed Brown { 4249acf2f550SJed Brown Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 4250acf2f550SJed Brown PetscErrorCode ierr; 4251acf2f550SJed Brown 4252acf2f550SJed Brown PetscFunctionBegin; 4253acf2f550SJed Brown a->idiagvalid = PETSC_FALSE; 4254acf2f550SJed Brown a->ibdiagvalid = PETSC_FALSE; 42552205254eSKarl Rupp 4256acf2f550SJed Brown ierr = MatSeqAIJInvalidateDiagonal_Inode(A);CHKERRQ(ierr); 4257acf2f550SJed Brown PetscFunctionReturn(0); 4258acf2f550SJed Brown } 4259acf2f550SJed Brown 42609c8f2541SHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat) 42619c8f2541SHong Zhang { 42629c8f2541SHong Zhang PetscErrorCode ierr; 42638761c3d6SHong Zhang PetscMPIInt size; 42649c8f2541SHong Zhang 42659c8f2541SHong Zhang PetscFunctionBegin; 42668761c3d6SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 42678761c3d6SHong Zhang if (size == 1 && scall == MAT_REUSE_MATRIX) { 42688761c3d6SHong Zhang ierr = MatCopy(inmat,*outmat,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 42698761c3d6SHong Zhang } else { 42709c8f2541SHong Zhang ierr = MatCreateMPIMatConcatenateSeqMat_MPIAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr); 42718761c3d6SHong Zhang } 42729c8f2541SHong Zhang PetscFunctionReturn(0); 42739c8f2541SHong Zhang } 42749c8f2541SHong Zhang 427581824310SBarry Smith /* 427653dd7562SDmitry Karpeev Permute A into C's *local* index space using rowemb,colemb. 427753dd7562SDmitry Karpeev The embedding are supposed to be injections and the above implies that the range of rowemb is a subset 427853dd7562SDmitry Karpeev of [0,m), colemb is in [0,n). 427953dd7562SDmitry Karpeev If pattern == DIFFERENT_NONZERO_PATTERN, C is preallocated according to A. 428053dd7562SDmitry Karpeev */ 428153dd7562SDmitry Karpeev PetscErrorCode MatSetSeqMat_SeqAIJ(Mat C,IS rowemb,IS colemb,MatStructure pattern,Mat B) 428253dd7562SDmitry Karpeev { 428353dd7562SDmitry Karpeev /* If making this function public, change the error returned in this function away from _PLIB. */ 428453dd7562SDmitry Karpeev PetscErrorCode ierr; 428553dd7562SDmitry Karpeev Mat_SeqAIJ *Baij; 428653dd7562SDmitry Karpeev PetscBool seqaij; 428753dd7562SDmitry Karpeev PetscInt m,n,*nz,i,j,count; 428853dd7562SDmitry Karpeev PetscScalar v; 428953dd7562SDmitry Karpeev const PetscInt *rowindices,*colindices; 429053dd7562SDmitry Karpeev 429153dd7562SDmitry Karpeev PetscFunctionBegin; 429253dd7562SDmitry Karpeev if (!B) PetscFunctionReturn(0); 429353dd7562SDmitry Karpeev /* Check to make sure the target matrix (and embeddings) are compatible with C and each other. */ 429453dd7562SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQAIJ,&seqaij);CHKERRQ(ierr); 429553dd7562SDmitry Karpeev if (!seqaij) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Input matrix is of wrong type"); 429653dd7562SDmitry Karpeev if (rowemb) { 429753dd7562SDmitry Karpeev ierr = ISGetLocalSize(rowemb,&m);CHKERRQ(ierr); 429853dd7562SDmitry Karpeev if (m != B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Row IS of size %D is incompatible with matrix row size %D",m,B->rmap->n); 429953dd7562SDmitry Karpeev } else { 43006c4ed002SBarry Smith if (C->rmap->n != B->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Input matrix is row-incompatible with the target matrix"); 430153dd7562SDmitry Karpeev } 430253dd7562SDmitry Karpeev if (colemb) { 430353dd7562SDmitry Karpeev ierr = ISGetLocalSize(colemb,&n);CHKERRQ(ierr); 430453dd7562SDmitry Karpeev if (n != B->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Diag col IS of size %D is incompatible with input matrix col size %D",n,B->cmap->n); 430553dd7562SDmitry Karpeev } else { 430653dd7562SDmitry Karpeev if (C->cmap->n != B->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Input matrix is col-incompatible with the target matrix"); 430753dd7562SDmitry Karpeev } 430853dd7562SDmitry Karpeev 430953dd7562SDmitry Karpeev Baij = (Mat_SeqAIJ*)(B->data); 431053dd7562SDmitry Karpeev if (pattern == DIFFERENT_NONZERO_PATTERN) { 431153dd7562SDmitry Karpeev ierr = PetscMalloc1(B->rmap->n,&nz);CHKERRQ(ierr); 431253dd7562SDmitry Karpeev for (i=0; i<B->rmap->n; i++) { 431353dd7562SDmitry Karpeev nz[i] = Baij->i[i+1] - Baij->i[i]; 431453dd7562SDmitry Karpeev } 431553dd7562SDmitry Karpeev ierr = MatSeqAIJSetPreallocation(C,0,nz);CHKERRQ(ierr); 431653dd7562SDmitry Karpeev ierr = PetscFree(nz);CHKERRQ(ierr); 431753dd7562SDmitry Karpeev } 431853dd7562SDmitry Karpeev if (pattern == SUBSET_NONZERO_PATTERN) { 431953dd7562SDmitry Karpeev ierr = MatZeroEntries(C);CHKERRQ(ierr); 432053dd7562SDmitry Karpeev } 432153dd7562SDmitry Karpeev count = 0; 432253dd7562SDmitry Karpeev rowindices = NULL; 432353dd7562SDmitry Karpeev colindices = NULL; 432453dd7562SDmitry Karpeev if (rowemb) { 432553dd7562SDmitry Karpeev ierr = ISGetIndices(rowemb,&rowindices);CHKERRQ(ierr); 432653dd7562SDmitry Karpeev } 432753dd7562SDmitry Karpeev if (colemb) { 432853dd7562SDmitry Karpeev ierr = ISGetIndices(colemb,&colindices);CHKERRQ(ierr); 432953dd7562SDmitry Karpeev } 433053dd7562SDmitry Karpeev for (i=0; i<B->rmap->n; i++) { 433153dd7562SDmitry Karpeev PetscInt row; 433253dd7562SDmitry Karpeev row = i; 433353dd7562SDmitry Karpeev if (rowindices) row = rowindices[i]; 433453dd7562SDmitry Karpeev for (j=Baij->i[i]; j<Baij->i[i+1]; j++) { 433553dd7562SDmitry Karpeev PetscInt col; 433653dd7562SDmitry Karpeev col = Baij->j[count]; 433753dd7562SDmitry Karpeev if (colindices) col = colindices[col]; 433853dd7562SDmitry Karpeev v = Baij->a[count]; 433953dd7562SDmitry Karpeev ierr = MatSetValues(C,1,&row,1,&col,&v,INSERT_VALUES);CHKERRQ(ierr); 434053dd7562SDmitry Karpeev ++count; 434153dd7562SDmitry Karpeev } 434253dd7562SDmitry Karpeev } 434353dd7562SDmitry Karpeev /* FIXME: set C's nonzerostate correctly. */ 434453dd7562SDmitry Karpeev /* Assembly for C is necessary. */ 434553dd7562SDmitry Karpeev C->preallocated = PETSC_TRUE; 434653dd7562SDmitry Karpeev C->assembled = PETSC_TRUE; 434753dd7562SDmitry Karpeev C->was_assembled = PETSC_FALSE; 434853dd7562SDmitry Karpeev PetscFunctionReturn(0); 434953dd7562SDmitry Karpeev } 435053dd7562SDmitry Karpeev 435153dd7562SDmitry Karpeev 435253dd7562SDmitry Karpeev /* 435381824310SBarry Smith Special version for direct calls from Fortran 435481824310SBarry Smith */ 4355af0996ceSBarry Smith #include <petsc/private/fortranimpl.h> 435681824310SBarry Smith #if defined(PETSC_HAVE_FORTRAN_CAPS) 435781824310SBarry Smith #define matsetvaluesseqaij_ MATSETVALUESSEQAIJ 435881824310SBarry Smith #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE) 435981824310SBarry Smith #define matsetvaluesseqaij_ matsetvaluesseqaij 436081824310SBarry Smith #endif 436181824310SBarry Smith 436281824310SBarry Smith /* Change these macros so can be used in void function */ 436381824310SBarry Smith #undef CHKERRQ 4364ce94432eSBarry Smith #define CHKERRQ(ierr) CHKERRABORT(PetscObjectComm((PetscObject)A),ierr) 436581824310SBarry Smith #undef SETERRQ2 4366e32f2f54SBarry Smith #define SETERRQ2(comm,ierr,b,c,d) CHKERRABORT(comm,ierr) 43674994cf47SJed Brown #undef SETERRQ3 43684994cf47SJed Brown #define SETERRQ3(comm,ierr,b,c,d,e) CHKERRABORT(comm,ierr) 436981824310SBarry Smith 43708cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL matsetvaluesseqaij_(Mat *AA,PetscInt *mm,const PetscInt im[],PetscInt *nn,const PetscInt in[],const PetscScalar v[],InsertMode *isis, PetscErrorCode *_ierr) 437181824310SBarry Smith { 437281824310SBarry Smith Mat A = *AA; 437381824310SBarry Smith PetscInt m = *mm, n = *nn; 437481824310SBarry Smith InsertMode is = *isis; 437581824310SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 437681824310SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N; 437781824310SBarry Smith PetscInt *imax,*ai,*ailen; 437881824310SBarry Smith PetscErrorCode ierr; 437981824310SBarry Smith PetscInt *aj,nonew = a->nonew,lastcol = -1; 438054f21887SBarry Smith MatScalar *ap,value,*aa; 4381ace3abfcSBarry Smith PetscBool ignorezeroentries = a->ignorezeroentries; 4382ace3abfcSBarry Smith PetscBool roworiented = a->roworiented; 438381824310SBarry Smith 438481824310SBarry Smith PetscFunctionBegin; 43854994cf47SJed Brown MatCheckPreallocated(A,1); 438681824310SBarry Smith imax = a->imax; 438781824310SBarry Smith ai = a->i; 438881824310SBarry Smith ailen = a->ilen; 438981824310SBarry Smith aj = a->j; 439081824310SBarry Smith aa = a->a; 439181824310SBarry Smith 439281824310SBarry Smith for (k=0; k<m; k++) { /* loop over added rows */ 439381824310SBarry Smith row = im[k]; 439481824310SBarry Smith if (row < 0) continue; 439581824310SBarry Smith #if defined(PETSC_USE_DEBUG) 4396ce94432eSBarry Smith if (row >= A->rmap->n) SETERRABORT(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_OUTOFRANGE,"Row too large"); 439781824310SBarry Smith #endif 439881824310SBarry Smith rp = aj + ai[row]; ap = aa + ai[row]; 439981824310SBarry Smith rmax = imax[row]; nrow = ailen[row]; 440081824310SBarry Smith low = 0; 440181824310SBarry Smith high = nrow; 440281824310SBarry Smith for (l=0; l<n; l++) { /* loop over added columns */ 440381824310SBarry Smith if (in[l] < 0) continue; 440481824310SBarry Smith #if defined(PETSC_USE_DEBUG) 4405ce94432eSBarry Smith if (in[l] >= A->cmap->n) SETERRABORT(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_OUTOFRANGE,"Column too large"); 440681824310SBarry Smith #endif 440781824310SBarry Smith col = in[l]; 44082205254eSKarl Rupp if (roworiented) value = v[l + k*n]; 44092205254eSKarl Rupp else value = v[k + l*m]; 44102205254eSKarl Rupp 441181824310SBarry Smith if (value == 0.0 && ignorezeroentries && (is == ADD_VALUES)) continue; 441281824310SBarry Smith 44132205254eSKarl Rupp if (col <= lastcol) low = 0; 44142205254eSKarl Rupp else high = nrow; 441581824310SBarry Smith lastcol = col; 441681824310SBarry Smith while (high-low > 5) { 441781824310SBarry Smith t = (low+high)/2; 441881824310SBarry Smith if (rp[t] > col) high = t; 441981824310SBarry Smith else low = t; 442081824310SBarry Smith } 442181824310SBarry Smith for (i=low; i<high; i++) { 442281824310SBarry Smith if (rp[i] > col) break; 442381824310SBarry Smith if (rp[i] == col) { 442481824310SBarry Smith if (is == ADD_VALUES) ap[i] += value; 442581824310SBarry Smith else ap[i] = value; 442681824310SBarry Smith goto noinsert; 442781824310SBarry Smith } 442881824310SBarry Smith } 442981824310SBarry Smith if (value == 0.0 && ignorezeroentries) goto noinsert; 443081824310SBarry Smith if (nonew == 1) goto noinsert; 4431ce94432eSBarry Smith if (nonew == -1) SETERRABORT(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero in the matrix"); 4432fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,A->rmap->n,1,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 443381824310SBarry Smith N = nrow++ - 1; a->nz++; high++; 443481824310SBarry Smith /* shift up all the later entries in this row */ 443581824310SBarry Smith for (ii=N; ii>=i; ii--) { 443681824310SBarry Smith rp[ii+1] = rp[ii]; 443781824310SBarry Smith ap[ii+1] = ap[ii]; 443881824310SBarry Smith } 443981824310SBarry Smith rp[i] = col; 444081824310SBarry Smith ap[i] = value; 4441e56f5c9eSBarry Smith A->nonzerostate++; 444281824310SBarry Smith noinsert:; 444381824310SBarry Smith low = i + 1; 444481824310SBarry Smith } 444581824310SBarry Smith ailen[row] = nrow; 444681824310SBarry Smith } 444781824310SBarry Smith PetscFunctionReturnVoid(); 444881824310SBarry Smith } 44499f7953f8SBarry Smith 4450