xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 941593c847bb92c54fe3ea55b33506f4fd829627)
149b5e25fSSatish Balay 
249b5e25fSSatish Balay /*
3a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
449b5e25fSSatish Balay   matrix storage format.
549b5e25fSSatish Balay */
69e070d67SMatthew Knepley #include "src/mat/impls/baij/seq/baij.h"         /*I "petscmat.h" I*/
749b5e25fSSatish Balay #include "src/inline/spops.h"
8aec82049SSatish Balay #include "src/mat/impls/sbaij/seq/sbaij.h"
949b5e25fSSatish Balay 
1049b5e25fSSatish Balay #define CHUNKSIZE  10
1149b5e25fSSatish Balay 
1249b5e25fSSatish Balay /*
1349b5e25fSSatish Balay      Checks for missing diagonals
1449b5e25fSSatish Balay */
154a2ae208SSatish Balay #undef __FUNCT__
164a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
17dfbe8321SBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A)
1849b5e25fSSatish Balay {
19045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
206849ba73SBarry Smith   PetscErrorCode ierr;
2113f74950SBarry Smith   PetscInt       *diag,*jj = a->j,i;
2249b5e25fSSatish Balay 
2349b5e25fSSatish Balay   PetscFunctionBegin;
24045c9aa0SHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
2549b5e25fSSatish Balay   diag = a->diag;
2649b5e25fSSatish Balay   for (i=0; i<a->mbs; i++) {
2777431f27SBarry Smith     if (jj[diag[i]] != i) SETERRQ1(PETSC_ERR_ARG_CORRUPT,"Matrix is missing diagonal number %D",i);
2849b5e25fSSatish Balay   }
2949b5e25fSSatish Balay   PetscFunctionReturn(0);
3049b5e25fSSatish Balay }
3149b5e25fSSatish Balay 
324a2ae208SSatish Balay #undef __FUNCT__
334a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
34dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
3549b5e25fSSatish Balay {
36045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
376849ba73SBarry Smith   PetscErrorCode ierr;
3813f74950SBarry Smith   PetscInt       i,mbs = a->mbs;
3949b5e25fSSatish Balay 
4049b5e25fSSatish Balay   PetscFunctionBegin;
4149b5e25fSSatish Balay   if (a->diag) PetscFunctionReturn(0);
4249b5e25fSSatish Balay 
4313f74950SBarry Smith   ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&a->diag);CHKERRQ(ierr);
4452e6d16bSBarry Smith   ierr = PetscLogObjectMemory(A,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
45b424e231SHong Zhang   for (i=0; i<mbs; i++) a->diag[i] = a->i[i];
4649b5e25fSSatish Balay   PetscFunctionReturn(0);
4749b5e25fSSatish Balay }
4849b5e25fSSatish Balay 
494a2ae208SSatish Balay #undef __FUNCT__
504a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
5113f74950SBarry Smith static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
5249b5e25fSSatish Balay {
53a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
5413f74950SBarry Smith   PetscInt     n = a->mbs,i;
5549b5e25fSSatish Balay 
5649b5e25fSSatish Balay   PetscFunctionBegin;
57d3e5a4abSHong Zhang   *nn = n;
58a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
59a1373b80SHong Zhang 
60a6ece127SHong Zhang   if (oshift == 1) {
61a6ece127SHong Zhang     /* temporarily add 1 to i and j indices */
6213f74950SBarry Smith     PetscInt nz = a->i[n];
636c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]++;
64a1373b80SHong Zhang     for (i=0; i<n+1; i++) a->i[i]++;
65a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
66a1373b80SHong Zhang   } else {
67a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
68a6ece127SHong Zhang   }
6949b5e25fSSatish Balay   PetscFunctionReturn(0);
7049b5e25fSSatish Balay }
7149b5e25fSSatish Balay 
724a2ae208SSatish Balay #undef __FUNCT__
734a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
7413f74950SBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
7549b5e25fSSatish Balay {
76b7aaefc3SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
7713f74950SBarry Smith   PetscInt     i,n = a->mbs;
78a6ece127SHong Zhang 
7949b5e25fSSatish Balay   PetscFunctionBegin;
8049b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
81a6ece127SHong Zhang 
82a6ece127SHong Zhang   if (oshift == 1) {
8313f74950SBarry Smith     PetscInt nz = a->i[n]-1;
846c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]--;
85a6ece127SHong Zhang     for (i=0; i<n+1; i++) a->i[i]--;
86a6ece127SHong Zhang   }
87a6ece127SHong Zhang   PetscFunctionReturn(0);
8849b5e25fSSatish Balay }
8949b5e25fSSatish Balay 
904a2ae208SSatish Balay #undef __FUNCT__
914a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
92dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
9349b5e25fSSatish Balay {
9449b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
95dfbe8321SBarry Smith   PetscErrorCode ierr;
9649b5e25fSSatish Balay 
9749b5e25fSSatish Balay   PetscFunctionBegin;
9877431f27SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->m,a->nz);
9949b5e25fSSatish Balay   ierr = PetscFree(a->a);CHKERRQ(ierr);
10049b5e25fSSatish Balay   if (!a->singlemalloc) {
10149b5e25fSSatish Balay     ierr = PetscFree(a->i);CHKERRQ(ierr);
10263c8bf9fSHong Zhang     ierr = PetscFree(a->j);CHKERRQ(ierr);
10349b5e25fSSatish Balay   }
10449b5e25fSSatish Balay   if (a->row) {
10549b5e25fSSatish Balay     ierr = ISDestroy(a->row);CHKERRQ(ierr);
10649b5e25fSSatish Balay   }
10749b5e25fSSatish Balay   if (a->diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
10849b5e25fSSatish Balay   if (a->ilen) {ierr = PetscFree(a->ilen);CHKERRQ(ierr);}
10949b5e25fSSatish Balay   if (a->imax) {ierr = PetscFree(a->imax);CHKERRQ(ierr);}
11049b5e25fSSatish Balay   if (a->icol) {ierr = ISDestroy(a->icol);CHKERRQ(ierr);}
111d59c15a7SBarry Smith   if (a->solve_work)  {ierr = PetscFree(a->solve_work);CHKERRQ(ierr);}
112d59c15a7SBarry Smith   if (a->solves_work) {ierr = PetscFree(a->solves_work);CHKERRQ(ierr);}
113d59c15a7SBarry Smith   if (a->mult_work)   {ierr = PetscFree(a->mult_work);CHKERRQ(ierr);}
11449b5e25fSSatish Balay   if (a->saved_values) {ierr = PetscFree(a->saved_values);CHKERRQ(ierr);}
115c4319e64SHong Zhang   if (a->xtoy) {ierr = PetscFree(a->xtoy);CHKERRQ(ierr);}
1161a3463dfSHong Zhang 
1171a3463dfSHong Zhang   if (a->inew){
1181a3463dfSHong Zhang     ierr = PetscFree(a->inew);CHKERRQ(ierr);
1191a3463dfSHong Zhang     a->inew = 0;
1201a3463dfSHong Zhang   }
12149b5e25fSSatish Balay   ierr = PetscFree(a);CHKERRQ(ierr);
122901853e0SKris Buschelman 
123901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr);
124901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr);
125901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr);
126901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr);
127901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr);
128901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
12949b5e25fSSatish Balay   PetscFunctionReturn(0);
13049b5e25fSSatish Balay }
13149b5e25fSSatish Balay 
1324a2ae208SSatish Balay #undef __FUNCT__
1334a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
134dfbe8321SBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op)
13549b5e25fSSatish Balay {
136045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
13749b5e25fSSatish Balay 
13849b5e25fSSatish Balay   PetscFunctionBegin;
1394d9d31abSKris Buschelman   switch (op) {
1404d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1414d9d31abSKris Buschelman     a->roworiented = PETSC_TRUE;
1424d9d31abSKris Buschelman     break;
1434d9d31abSKris Buschelman   case MAT_COLUMN_ORIENTED:
1444d9d31abSKris Buschelman     a->roworiented = PETSC_FALSE;
1454d9d31abSKris Buschelman     break;
1464d9d31abSKris Buschelman   case MAT_COLUMNS_SORTED:
1474d9d31abSKris Buschelman     a->sorted = PETSC_TRUE;
1484d9d31abSKris Buschelman     break;
1494d9d31abSKris Buschelman   case MAT_COLUMNS_UNSORTED:
1504d9d31abSKris Buschelman     a->sorted = PETSC_FALSE;
1514d9d31abSKris Buschelman     break;
1524d9d31abSKris Buschelman   case MAT_KEEP_ZEROED_ROWS:
1534d9d31abSKris Buschelman     a->keepzeroedrows = PETSC_TRUE;
1544d9d31abSKris Buschelman     break;
1554d9d31abSKris Buschelman   case MAT_NO_NEW_NONZERO_LOCATIONS:
1564d9d31abSKris Buschelman     a->nonew = 1;
1574d9d31abSKris Buschelman     break;
1584d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1594d9d31abSKris Buschelman     a->nonew = -1;
1604d9d31abSKris Buschelman     break;
1614d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1624d9d31abSKris Buschelman     a->nonew = -2;
1634d9d31abSKris Buschelman     break;
1644d9d31abSKris Buschelman   case MAT_YES_NEW_NONZERO_LOCATIONS:
1654d9d31abSKris Buschelman     a->nonew = 0;
1664d9d31abSKris Buschelman     break;
1674d9d31abSKris Buschelman   case MAT_ROWS_SORTED:
1684d9d31abSKris Buschelman   case MAT_ROWS_UNSORTED:
1694d9d31abSKris Buschelman   case MAT_YES_NEW_DIAGONALS:
1704d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1714d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
172b0a32e0cSBarry Smith     PetscLogInfo(A,"MatSetOption_SeqSBAIJ:Option ignored\n");
1734d9d31abSKris Buschelman     break;
1744d9d31abSKris Buschelman   case MAT_NO_NEW_DIAGONALS:
17529bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"MAT_NO_NEW_DIAGONALS");
1769a4540c5SBarry Smith   case MAT_NOT_SYMMETRIC:
1779a4540c5SBarry Smith   case MAT_NOT_STRUCTURALLY_SYMMETRIC:
1789a4540c5SBarry Smith   case MAT_HERMITIAN:
1799a4540c5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Matrix must be symmetric");
18077e54ba9SKris Buschelman   case MAT_SYMMETRIC:
18177e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
1829a4540c5SBarry Smith   case MAT_NOT_HERMITIAN:
1839a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1849a4540c5SBarry Smith   case MAT_NOT_SYMMETRY_ETERNAL:
185*941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
186*941593c8SHong Zhang     a->ignore_ltriangular = PETSC_TRUE;
187*941593c8SHong Zhang     break;
188*941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
189*941593c8SHong Zhang     a->ignore_ltriangular = PETSC_FALSE;
19077e54ba9SKris Buschelman     break;
1914d9d31abSKris Buschelman   default:
19229bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"unknown option");
19349b5e25fSSatish Balay   }
19449b5e25fSSatish Balay   PetscFunctionReturn(0);
19549b5e25fSSatish Balay  }
19649b5e25fSSatish Balay 
1974a2ae208SSatish Balay  #undef __FUNCT__
1984a2ae208SSatish Balay  #define __FUNCT__ "MatGetRow_SeqSBAIJ"
19913f74950SBarry Smith  PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v)
20049b5e25fSSatish Balay  {
20149b5e25fSSatish Balay    Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2026849ba73SBarry Smith    PetscErrorCode ierr;
20313f74950SBarry Smith    PetscInt       itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2;
20449b5e25fSSatish Balay    MatScalar      *aa,*aa_i;
20587828ca2SBarry Smith    PetscScalar    *v_i;
20649b5e25fSSatish Balay 
20749b5e25fSSatish Balay    PetscFunctionBegin;
208521d7252SBarry Smith    bs  = A->bs;
20949b5e25fSSatish Balay    ai  = a->i;
21049b5e25fSSatish Balay    aj  = a->j;
21149b5e25fSSatish Balay    aa  = a->a;
21249b5e25fSSatish Balay    bs2 = a->bs2;
21349b5e25fSSatish Balay 
21477431f27SBarry Smith    if (row < 0 || row >= A->m) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row);
21549b5e25fSSatish Balay 
21649b5e25fSSatish Balay    bn  = row/bs;   /* Block number */
21749b5e25fSSatish Balay    bp  = row % bs; /* Block position */
21849b5e25fSSatish Balay    M   = ai[bn+1] - ai[bn];
21949b5e25fSSatish Balay    *ncols = bs*M;
22049b5e25fSSatish Balay 
22149b5e25fSSatish Balay    if (v) {
22249b5e25fSSatish Balay      *v = 0;
22349b5e25fSSatish Balay      if (*ncols) {
22487828ca2SBarry Smith        ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr);
22549b5e25fSSatish Balay        for (i=0; i<M; i++) { /* for each block in the block row */
22649b5e25fSSatish Balay          v_i  = *v + i*bs;
22749b5e25fSSatish Balay          aa_i = aa + bs2*(ai[bn] + i);
22849b5e25fSSatish Balay          for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];}
22949b5e25fSSatish Balay        }
23049b5e25fSSatish Balay      }
23149b5e25fSSatish Balay    }
23249b5e25fSSatish Balay 
23349b5e25fSSatish Balay    if (cols) {
23449b5e25fSSatish Balay      *cols = 0;
23549b5e25fSSatish Balay      if (*ncols) {
23613f74950SBarry Smith        ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr);
23749b5e25fSSatish Balay        for (i=0; i<M; i++) { /* for each block in the block row */
23849b5e25fSSatish Balay          cols_i = *cols + i*bs;
23949b5e25fSSatish Balay          itmp  = bs*aj[ai[bn] + i];
24049b5e25fSSatish Balay          for (j=0; j<bs; j++) {cols_i[j] = itmp++;}
24149b5e25fSSatish Balay        }
24249b5e25fSSatish Balay      }
24349b5e25fSSatish Balay    }
24449b5e25fSSatish Balay 
24549b5e25fSSatish Balay    /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */
2465ddb2528SHong Zhang    /* this segment is currently removed, so only entries in the upper triangle are obtained */
2475ddb2528SHong Zhang  #ifdef column_search
24849b5e25fSSatish Balay    v_i    = *v    + M*bs;
24949b5e25fSSatish Balay    cols_i = *cols + M*bs;
25049b5e25fSSatish Balay    for (i=0; i<bn; i++){ /* for each block row */
25149b5e25fSSatish Balay      M = ai[i+1] - ai[i];
25249b5e25fSSatish Balay      for (j=0; j<M; j++){
25349b5e25fSSatish Balay        itmp = aj[ai[i] + j];    /* block column value */
25449b5e25fSSatish Balay        if (itmp == bn){
25549b5e25fSSatish Balay          aa_i   = aa    + bs2*(ai[i] + j) + bs*bp;
25649b5e25fSSatish Balay          for (k=0; k<bs; k++) {
25749b5e25fSSatish Balay            *cols_i++ = i*bs+k;
25849b5e25fSSatish Balay            *v_i++    = aa_i[k];
25949b5e25fSSatish Balay          }
26049b5e25fSSatish Balay          *ncols += bs;
26149b5e25fSSatish Balay          break;
26249b5e25fSSatish Balay        }
26349b5e25fSSatish Balay      }
26449b5e25fSSatish Balay    }
2655ddb2528SHong Zhang  #endif
26649b5e25fSSatish Balay    PetscFunctionReturn(0);
26749b5e25fSSatish Balay  }
26849b5e25fSSatish Balay 
2694a2ae208SSatish Balay  #undef __FUNCT__
2704a2ae208SSatish Balay  #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
27113f74950SBarry Smith  PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
27249b5e25fSSatish Balay  {
273dfbe8321SBarry Smith    PetscErrorCode ierr;
27449b5e25fSSatish Balay 
27549b5e25fSSatish Balay    PetscFunctionBegin;
27649b5e25fSSatish Balay    if (idx) {if (*idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}}
27749b5e25fSSatish Balay    if (v)   {if (*v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}}
27849b5e25fSSatish Balay    PetscFunctionReturn(0);
27949b5e25fSSatish Balay  }
28049b5e25fSSatish Balay 
2814a2ae208SSatish Balay  #undef __FUNCT__
2824a2ae208SSatish Balay  #define __FUNCT__ "MatTranspose_SeqSBAIJ"
283dfbe8321SBarry Smith  PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,Mat *B)
28449b5e25fSSatish Balay  {
285dfbe8321SBarry Smith    PetscErrorCode ierr;
28649b5e25fSSatish Balay    PetscFunctionBegin;
287999d9058SBarry Smith    ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
2888115998fSBarry Smith    PetscFunctionReturn(0);
28949b5e25fSSatish Balay  }
29049b5e25fSSatish Balay 
2914a2ae208SSatish Balay  #undef __FUNCT__
2924a2ae208SSatish Balay  #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
2936849ba73SBarry Smith  static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
29449b5e25fSSatish Balay  {
29549b5e25fSSatish Balay    Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
296dfbe8321SBarry Smith    PetscErrorCode    ierr;
297521d7252SBarry Smith    PetscInt          i,j,bs = A->bs,k,l,bs2=a->bs2;
298fb9695e5SSatish Balay    char              *name;
299f3ef73ceSBarry Smith    PetscViewerFormat format;
30049b5e25fSSatish Balay 
30149b5e25fSSatish Balay    PetscFunctionBegin;
30280fe4e49SBarry Smith    ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
303b0a32e0cSBarry Smith    ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
304456192e2SBarry Smith    if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
30577431f27SBarry Smith      ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
306fb9695e5SSatish Balay    } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
30729bbc08cSBarry Smith      SETERRQ(PETSC_ERR_SUP,"Matlab format not supported");
308fb9695e5SSatish Balay    } else if (format == PETSC_VIEWER_ASCII_COMMON) {
309b0a32e0cSBarry Smith      ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
31049b5e25fSSatish Balay      for (i=0; i<a->mbs; i++) {
31149b5e25fSSatish Balay        for (j=0; j<bs; j++) {
31277431f27SBarry Smith          ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
31349b5e25fSSatish Balay          for (k=a->i[i]; k<a->i[i+1]; k++) {
31449b5e25fSSatish Balay            for (l=0; l<bs; l++) {
31549b5e25fSSatish Balay  #if defined(PETSC_USE_COMPLEX)
31649b5e25fSSatish Balay              if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
31777431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
31849b5e25fSSatish Balay                        PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
31949b5e25fSSatish Balay              } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
32077431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
32149b5e25fSSatish Balay                        PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
32249b5e25fSSatish Balay              } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
32377431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
32449b5e25fSSatish Balay              }
32549b5e25fSSatish Balay  #else
32649b5e25fSSatish Balay              if (a->a[bs2*k + l*bs + j] != 0.0) {
32777431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
32849b5e25fSSatish Balay              }
32949b5e25fSSatish Balay  #endif
33049b5e25fSSatish Balay            }
33149b5e25fSSatish Balay          }
332b0a32e0cSBarry Smith          ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
33349b5e25fSSatish Balay        }
33449b5e25fSSatish Balay      }
335b0a32e0cSBarry Smith      ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
33649b5e25fSSatish Balay    } else {
337b0a32e0cSBarry Smith      ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
33849b5e25fSSatish Balay      for (i=0; i<a->mbs; i++) {
33949b5e25fSSatish Balay        for (j=0; j<bs; j++) {
34077431f27SBarry Smith          ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
34149b5e25fSSatish Balay          for (k=a->i[i]; k<a->i[i+1]; k++) {
34249b5e25fSSatish Balay            for (l=0; l<bs; l++) {
34349b5e25fSSatish Balay  #if defined(PETSC_USE_COMPLEX)
34449b5e25fSSatish Balay              if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
34577431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
34649b5e25fSSatish Balay                  PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34749b5e25fSSatish Balay              } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
34877431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
34949b5e25fSSatish Balay                  PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
35049b5e25fSSatish Balay              } else {
35177431f27SBarry Smith                ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
35249b5e25fSSatish Balay              }
35349b5e25fSSatish Balay  #else
35477431f27SBarry Smith              ierr = PetscViewerASCIIPrintf(viewer," %D %g ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
35549b5e25fSSatish Balay  #endif
35649b5e25fSSatish Balay            }
35749b5e25fSSatish Balay          }
358b0a32e0cSBarry Smith          ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
35949b5e25fSSatish Balay        }
36049b5e25fSSatish Balay      }
361b0a32e0cSBarry Smith      ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
36249b5e25fSSatish Balay    }
363b0a32e0cSBarry Smith    ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
36449b5e25fSSatish Balay    PetscFunctionReturn(0);
36549b5e25fSSatish Balay  }
36649b5e25fSSatish Balay 
3674a2ae208SSatish Balay  #undef __FUNCT__
3684a2ae208SSatish Balay  #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
3696849ba73SBarry Smith  static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
37049b5e25fSSatish Balay  {
37149b5e25fSSatish Balay    Mat            A = (Mat) Aa;
37249b5e25fSSatish Balay    Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
3736849ba73SBarry Smith    PetscErrorCode ierr;
37413f74950SBarry Smith    PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->bs,bs2=a->bs2;
37513f74950SBarry Smith    PetscMPIInt    rank;
37649b5e25fSSatish Balay    PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
37749b5e25fSSatish Balay    MatScalar      *aa;
37849b5e25fSSatish Balay    MPI_Comm       comm;
379b0a32e0cSBarry Smith    PetscViewer    viewer;
38049b5e25fSSatish Balay 
38149b5e25fSSatish Balay    PetscFunctionBegin;
38249b5e25fSSatish Balay    /*
38349b5e25fSSatish Balay        This is nasty. If this is called from an originally parallel matrix
38449b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
38549b5e25fSSatish Balay     rest should return immediately.
38649b5e25fSSatish Balay    */
38749b5e25fSSatish Balay    ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
38849b5e25fSSatish Balay    ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
38949b5e25fSSatish Balay    if (rank) PetscFunctionReturn(0);
39049b5e25fSSatish Balay 
39149b5e25fSSatish Balay    ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
39249b5e25fSSatish Balay 
393b0a32e0cSBarry Smith    ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
394b0a32e0cSBarry Smith    PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
39549b5e25fSSatish Balay 
39649b5e25fSSatish Balay    /* loop over matrix elements drawing boxes */
397b0a32e0cSBarry Smith    color = PETSC_DRAW_BLUE;
39849b5e25fSSatish Balay    for (i=0,row=0; i<mbs; i++,row+=bs) {
39949b5e25fSSatish Balay      for (j=a->i[i]; j<a->i[i+1]; j++) {
400c464158bSHong Zhang        y_l = A->m - row - 1.0; y_r = y_l + 1.0;
40149b5e25fSSatish Balay        x_l = a->j[j]*bs; x_r = x_l + 1.0;
40249b5e25fSSatish Balay        aa = a->a + j*bs2;
40349b5e25fSSatish Balay        for (k=0; k<bs; k++) {
40449b5e25fSSatish Balay          for (l=0; l<bs; l++) {
40549b5e25fSSatish Balay            if (PetscRealPart(*aa++) >=  0.) continue;
406b0a32e0cSBarry Smith            ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
40749b5e25fSSatish Balay          }
40849b5e25fSSatish Balay        }
40949b5e25fSSatish Balay      }
41049b5e25fSSatish Balay    }
411b0a32e0cSBarry Smith    color = PETSC_DRAW_CYAN;
41249b5e25fSSatish Balay    for (i=0,row=0; i<mbs; i++,row+=bs) {
41349b5e25fSSatish Balay      for (j=a->i[i]; j<a->i[i+1]; j++) {
414c464158bSHong Zhang        y_l = A->m - row - 1.0; y_r = y_l + 1.0;
41549b5e25fSSatish Balay        x_l = a->j[j]*bs; x_r = x_l + 1.0;
41649b5e25fSSatish Balay        aa = a->a + j*bs2;
41749b5e25fSSatish Balay        for (k=0; k<bs; k++) {
41849b5e25fSSatish Balay          for (l=0; l<bs; l++) {
41949b5e25fSSatish Balay            if (PetscRealPart(*aa++) != 0.) continue;
420b0a32e0cSBarry Smith            ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
42149b5e25fSSatish Balay          }
42249b5e25fSSatish Balay        }
42349b5e25fSSatish Balay      }
42449b5e25fSSatish Balay    }
42549b5e25fSSatish Balay 
426b0a32e0cSBarry Smith    color = PETSC_DRAW_RED;
42749b5e25fSSatish Balay    for (i=0,row=0; i<mbs; i++,row+=bs) {
42849b5e25fSSatish Balay      for (j=a->i[i]; j<a->i[i+1]; j++) {
429c464158bSHong Zhang        y_l = A->m - row - 1.0; y_r = y_l + 1.0;
43049b5e25fSSatish Balay        x_l = a->j[j]*bs; x_r = x_l + 1.0;
43149b5e25fSSatish Balay        aa = a->a + j*bs2;
43249b5e25fSSatish Balay        for (k=0; k<bs; k++) {
43349b5e25fSSatish Balay          for (l=0; l<bs; l++) {
43449b5e25fSSatish Balay            if (PetscRealPart(*aa++) <= 0.) continue;
435b0a32e0cSBarry Smith            ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
43649b5e25fSSatish Balay          }
43749b5e25fSSatish Balay        }
43849b5e25fSSatish Balay      }
43949b5e25fSSatish Balay    }
44049b5e25fSSatish Balay    PetscFunctionReturn(0);
44149b5e25fSSatish Balay  }
44249b5e25fSSatish Balay 
4434a2ae208SSatish Balay  #undef __FUNCT__
4444a2ae208SSatish Balay  #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
4456849ba73SBarry Smith  static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
44649b5e25fSSatish Balay  {
447dfbe8321SBarry Smith    PetscErrorCode ierr;
44849b5e25fSSatish Balay    PetscReal      xl,yl,xr,yr,w,h;
449b0a32e0cSBarry Smith    PetscDraw      draw;
45049b5e25fSSatish Balay    PetscTruth     isnull;
45149b5e25fSSatish Balay 
45249b5e25fSSatish Balay    PetscFunctionBegin;
45349b5e25fSSatish Balay 
454b0a32e0cSBarry Smith    ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
455b0a32e0cSBarry Smith    ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
45649b5e25fSSatish Balay 
45749b5e25fSSatish Balay    ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
458c464158bSHong Zhang    xr  = A->m; yr = A->m; h = yr/10.0; w = xr/10.0;
45949b5e25fSSatish Balay    xr += w;    yr += h;  xl = -w;     yl = -h;
460b0a32e0cSBarry Smith    ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
461b0a32e0cSBarry Smith    ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
46249b5e25fSSatish Balay    ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
46349b5e25fSSatish Balay    PetscFunctionReturn(0);
46449b5e25fSSatish Balay  }
46549b5e25fSSatish Balay 
4664a2ae208SSatish Balay  #undef __FUNCT__
4674a2ae208SSatish Balay  #define __FUNCT__ "MatView_SeqSBAIJ"
468dfbe8321SBarry Smith  PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
46949b5e25fSSatish Balay  {
470dfbe8321SBarry Smith    PetscErrorCode ierr;
47132077d6dSBarry Smith    PetscTruth     iascii,isdraw;
47249b5e25fSSatish Balay 
47349b5e25fSSatish Balay    PetscFunctionBegin;
47432077d6dSBarry Smith    ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
475fb9695e5SSatish Balay    ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
47632077d6dSBarry Smith    if (iascii){
47749b5e25fSSatish Balay      ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
47849b5e25fSSatish Balay    } else if (isdraw) {
47949b5e25fSSatish Balay      ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
48049b5e25fSSatish Balay    } else {
481a5e6ed63SBarry Smith      Mat B;
482ceb03754SKris Buschelman      ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
483a5e6ed63SBarry Smith      ierr = MatView(B,viewer);CHKERRQ(ierr);
484a5e6ed63SBarry Smith      ierr = MatDestroy(B);CHKERRQ(ierr);
48549b5e25fSSatish Balay    }
48649b5e25fSSatish Balay    PetscFunctionReturn(0);
48749b5e25fSSatish Balay  }
48849b5e25fSSatish Balay 
48949b5e25fSSatish Balay 
4904a2ae208SSatish Balay  #undef __FUNCT__
4914a2ae208SSatish Balay  #define __FUNCT__ "MatGetValues_SeqSBAIJ"
49213f74950SBarry Smith  PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
49349b5e25fSSatish Balay  {
494045c9aa0SHong Zhang    Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
49513f74950SBarry Smith    PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
49613f74950SBarry Smith    PetscInt     *ai = a->i,*ailen = a->ilen;
49713f74950SBarry Smith    PetscInt     brow,bcol,ridx,cidx,bs=A->bs,bs2=a->bs2;
49849b5e25fSSatish Balay    MatScalar    *ap,*aa = a->a,zero = 0.0;
49949b5e25fSSatish Balay 
50049b5e25fSSatish Balay    PetscFunctionBegin;
50149b5e25fSSatish Balay    for (k=0; k<m; k++) { /* loop over rows */
50249b5e25fSSatish Balay      row  = im[k]; brow = row/bs;
50377431f27SBarry Smith      if (row < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row);
50477431f27SBarry Smith      if (row >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->m-1);
50549b5e25fSSatish Balay      rp   = aj + ai[brow] ; ap = aa + bs2*ai[brow] ;
50649b5e25fSSatish Balay      nrow = ailen[brow];
50749b5e25fSSatish Balay      for (l=0; l<n; l++) { /* loop over columns */
50877431f27SBarry Smith        if (in[l] < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]);
50977431f27SBarry Smith        if (in[l] >= A->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->n-1);
51049b5e25fSSatish Balay        col  = in[l] ;
51149b5e25fSSatish Balay        bcol = col/bs;
51249b5e25fSSatish Balay        cidx = col%bs;
51349b5e25fSSatish Balay        ridx = row%bs;
51449b5e25fSSatish Balay        high = nrow;
51549b5e25fSSatish Balay        low  = 0; /* assume unsorted */
51649b5e25fSSatish Balay        while (high-low > 5) {
51749b5e25fSSatish Balay          t = (low+high)/2;
51849b5e25fSSatish Balay          if (rp[t] > bcol) high = t;
51949b5e25fSSatish Balay          else             low  = t;
52049b5e25fSSatish Balay        }
52149b5e25fSSatish Balay        for (i=low; i<high; i++) {
52249b5e25fSSatish Balay          if (rp[i] > bcol) break;
52349b5e25fSSatish Balay          if (rp[i] == bcol) {
52449b5e25fSSatish Balay            *v++ = ap[bs2*i+bs*cidx+ridx];
52549b5e25fSSatish Balay            goto finished;
52649b5e25fSSatish Balay          }
52749b5e25fSSatish Balay        }
52849b5e25fSSatish Balay        *v++ = zero;
52949b5e25fSSatish Balay        finished:;
53049b5e25fSSatish Balay      }
53149b5e25fSSatish Balay    }
53249b5e25fSSatish Balay    PetscFunctionReturn(0);
53349b5e25fSSatish Balay  }
53449b5e25fSSatish Balay 
53549b5e25fSSatish Balay 
5364a2ae208SSatish Balay  #undef __FUNCT__
5374a2ae208SSatish Balay  #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
53813f74950SBarry Smith  PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
53949b5e25fSSatish Balay  {
5400880e062SHong Zhang    Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
5416849ba73SBarry Smith    PetscErrorCode  ierr;
542e2ee6c50SBarry Smith    PetscInt        *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
54313f74950SBarry Smith    PetscInt        *imax=a->imax,*ai=a->i,*ailen=a->ilen;
54413f74950SBarry Smith    PetscInt        *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->bs,stepval;
5450880e062SHong Zhang    PetscTruth      roworiented=a->roworiented;
546f15d580aSBarry Smith    const MatScalar *value = v;
547f15d580aSBarry Smith    MatScalar       *ap,*aa = a->a,*bap;
5480880e062SHong Zhang 
54949b5e25fSSatish Balay    PetscFunctionBegin;
5500880e062SHong Zhang    if (roworiented) {
5510880e062SHong Zhang      stepval = (n-1)*bs;
5520880e062SHong Zhang    } else {
5530880e062SHong Zhang      stepval = (m-1)*bs;
5540880e062SHong Zhang    }
5550880e062SHong Zhang    for (k=0; k<m; k++) { /* loop over added rows */
5560880e062SHong Zhang      row  = im[k];
5570880e062SHong Zhang      if (row < 0) continue;
5582515c552SBarry Smith  #if defined(PETSC_USE_DEBUG)
55977431f27SBarry Smith      if (row >= a->mbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
5600880e062SHong Zhang  #endif
5610880e062SHong Zhang      rp   = aj + ai[row];
5620880e062SHong Zhang      ap   = aa + bs2*ai[row];
5630880e062SHong Zhang      rmax = imax[row];
5640880e062SHong Zhang      nrow = ailen[row];
5650880e062SHong Zhang      low  = 0;
566818f2c47SBarry Smith      high = nrow;
5670880e062SHong Zhang      for (l=0; l<n; l++) { /* loop over added columns */
5680880e062SHong Zhang        if (in[l] < 0) continue;
5690880e062SHong Zhang        col = in[l];
5702515c552SBarry Smith  #if defined(PETSC_USE_DEBUG)
57177431f27SBarry Smith        if (col >= a->nbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
572b1823623SSatish Balay  #endif
5730880e062SHong Zhang        if (col < row) continue; /* ignore lower triangular block */
5740880e062SHong Zhang        if (roworiented) {
5750880e062SHong Zhang          value = v + k*(stepval+bs)*bs + l*bs;
5760880e062SHong Zhang        } else {
5770880e062SHong Zhang          value = v + l*(stepval+bs)*bs + k*bs;
5780880e062SHong Zhang        }
579818f2c47SBarry Smith        if (col < lastcol) low = 0; else high = nrow;
580e2ee6c50SBarry Smith        lastcol = col;
5810880e062SHong Zhang        while (high-low > 7) {
5820880e062SHong Zhang          t = (low+high)/2;
5830880e062SHong Zhang          if (rp[t] > col) high = t;
5840880e062SHong Zhang          else             low  = t;
5850880e062SHong Zhang        }
5860880e062SHong Zhang        for (i=low; i<high; i++) {
5870880e062SHong Zhang          if (rp[i] > col) break;
5880880e062SHong Zhang          if (rp[i] == col) {
5890880e062SHong Zhang            bap  = ap +  bs2*i;
5900880e062SHong Zhang            if (roworiented) {
5910880e062SHong Zhang              if (is == ADD_VALUES) {
5920880e062SHong Zhang                for (ii=0; ii<bs; ii++,value+=stepval) {
5930880e062SHong Zhang                  for (jj=ii; jj<bs2; jj+=bs) {
5940880e062SHong Zhang                    bap[jj] += *value++;
5950880e062SHong Zhang                  }
5960880e062SHong Zhang                }
5970880e062SHong Zhang              } else {
5980880e062SHong Zhang                for (ii=0; ii<bs; ii++,value+=stepval) {
5990880e062SHong Zhang                  for (jj=ii; jj<bs2; jj+=bs) {
6000880e062SHong Zhang                    bap[jj] = *value++;
6010880e062SHong Zhang                  }
6020880e062SHong Zhang                }
6030880e062SHong Zhang              }
6040880e062SHong Zhang            } else {
6050880e062SHong Zhang              if (is == ADD_VALUES) {
6060880e062SHong Zhang                for (ii=0; ii<bs; ii++,value+=stepval) {
6070880e062SHong Zhang                  for (jj=0; jj<bs; jj++) {
6080880e062SHong Zhang                    *bap++ += *value++;
6090880e062SHong Zhang                  }
6100880e062SHong Zhang                }
6110880e062SHong Zhang              } else {
6120880e062SHong Zhang                for (ii=0; ii<bs; ii++,value+=stepval) {
6130880e062SHong Zhang                  for (jj=0; jj<bs; jj++) {
6140880e062SHong Zhang                    *bap++  = *value++;
6150880e062SHong Zhang                  }
6160880e062SHong Zhang                }
6170880e062SHong Zhang              }
6180880e062SHong Zhang            }
6190880e062SHong Zhang            goto noinsert2;
6200880e062SHong Zhang          }
6210880e062SHong Zhang        }
6220880e062SHong Zhang        if (nonew == 1) goto noinsert2;
62377431f27SBarry Smith        else if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
6240880e062SHong Zhang        if (nrow >= rmax) {
6250880e062SHong Zhang          /* there is no extra room in row, therefore enlarge */
62613f74950SBarry Smith          PetscInt       new_nz = ai[a->mbs] + CHUNKSIZE,len,*new_i,*new_j;
6270880e062SHong Zhang          MatScalar *new_a;
6280880e062SHong Zhang 
62977431f27SBarry Smith          if (nonew == -2) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
6300880e062SHong Zhang 
6310880e062SHong Zhang          /* malloc new storage space */
63213f74950SBarry Smith          len     = new_nz*(sizeof(PetscInt)+bs2*sizeof(MatScalar))+(a->mbs+1)*sizeof(PetscInt);
6330880e062SHong Zhang          ierr    = PetscMalloc(len,&new_a);CHKERRQ(ierr);
63413f74950SBarry Smith          new_j   = (PetscInt*)(new_a + bs2*new_nz);
6350880e062SHong Zhang          new_i   = new_j + new_nz;
6360880e062SHong Zhang 
6370880e062SHong Zhang          /* copy over old data into new slots */
6380880e062SHong Zhang          for (ii=0; ii<row+1; ii++) {new_i[ii] = ai[ii];}
6390880e062SHong Zhang          for (ii=row+1; ii<a->mbs+1; ii++) {new_i[ii] = ai[ii]+CHUNKSIZE;}
64013f74950SBarry Smith          ierr = PetscMemcpy(new_j,aj,(ai[row]+nrow)*sizeof(PetscInt));CHKERRQ(ierr);
6410880e062SHong Zhang          len  = (new_nz - CHUNKSIZE - ai[row] - nrow);
64213f74950SBarry Smith          ierr = PetscMemcpy(new_j+ai[row]+nrow+CHUNKSIZE,aj+ai[row]+nrow,len*sizeof(PetscInt));CHKERRQ(ierr);
6430880e062SHong Zhang          ierr = PetscMemcpy(new_a,aa,(ai[row]+nrow)*bs2*sizeof(MatScalar));CHKERRQ(ierr);
6440880e062SHong Zhang          ierr = PetscMemzero(new_a+bs2*(ai[row]+nrow),bs2*CHUNKSIZE*sizeof(MatScalar));CHKERRQ(ierr);
6450880e062SHong Zhang          ierr = PetscMemcpy(new_a+bs2*(ai[row]+nrow+CHUNKSIZE),aa+bs2*(ai[row]+nrow),bs2*len*sizeof(MatScalar));CHKERRQ(ierr);
6460880e062SHong Zhang          /* free up old matrix storage */
6470880e062SHong Zhang          ierr = PetscFree(a->a);CHKERRQ(ierr);
6480880e062SHong Zhang          if (!a->singlemalloc) {
6490880e062SHong Zhang            ierr = PetscFree(a->i);CHKERRQ(ierr);
6500880e062SHong Zhang            ierr = PetscFree(a->j);CHKERRQ(ierr);
6510880e062SHong Zhang          }
6520880e062SHong Zhang          aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
6530880e062SHong Zhang          a->singlemalloc = PETSC_TRUE;
6540880e062SHong Zhang 
6550880e062SHong Zhang          rp   = aj + ai[row]; ap = aa + bs2*ai[row];
6560880e062SHong Zhang          rmax = imax[row] = imax[row] + CHUNKSIZE;
65752e6d16bSBarry Smith          ierr = PetscLogObjectMemory(A,CHUNKSIZE*(sizeof(PetscInt) + bs2*sizeof(MatScalar)));CHKERRQ(ierr);
6586c6c5352SBarry Smith          a->maxnz += bs2*CHUNKSIZE;
6590880e062SHong Zhang          a->reallocs++;
6606c6c5352SBarry Smith          a->nz++;
6610880e062SHong Zhang        }
6620880e062SHong Zhang        N = nrow++ - 1;
6630880e062SHong Zhang        /* shift up all the later entries in this row */
6640880e062SHong Zhang        for (ii=N; ii>=i; ii--) {
6650880e062SHong Zhang          rp[ii+1] = rp[ii];
6660880e062SHong Zhang          ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6670880e062SHong Zhang        }
6680880e062SHong Zhang        if (N >= i) {
6690880e062SHong Zhang          ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6700880e062SHong Zhang        }
6710880e062SHong Zhang        rp[i] = col;
6720880e062SHong Zhang        bap   = ap +  bs2*i;
6730880e062SHong Zhang        if (roworiented) {
6740880e062SHong Zhang          for (ii=0; ii<bs; ii++,value+=stepval) {
6750880e062SHong Zhang            for (jj=ii; jj<bs2; jj+=bs) {
6760880e062SHong Zhang              bap[jj] = *value++;
6770880e062SHong Zhang            }
6780880e062SHong Zhang          }
6790880e062SHong Zhang        } else {
6800880e062SHong Zhang          for (ii=0; ii<bs; ii++,value+=stepval) {
6810880e062SHong Zhang            for (jj=0; jj<bs; jj++) {
6820880e062SHong Zhang              *bap++  = *value++;
6830880e062SHong Zhang            }
6840880e062SHong Zhang          }
6850880e062SHong Zhang        }
6860880e062SHong Zhang        noinsert2:;
6870880e062SHong Zhang        low = i;
6880880e062SHong Zhang      }
6890880e062SHong Zhang      ailen[row] = nrow;
6900880e062SHong Zhang    }
6910880e062SHong Zhang    PetscFunctionReturn(0);
69249b5e25fSSatish Balay  }
69349b5e25fSSatish Balay 
6944a2ae208SSatish Balay  #undef __FUNCT__
6954a2ae208SSatish Balay  #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
696dfbe8321SBarry Smith  PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
69749b5e25fSSatish Balay  {
69849b5e25fSSatish Balay    Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
6996849ba73SBarry Smith    PetscErrorCode ierr;
70013f74950SBarry Smith    PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
70113f74950SBarry Smith    PetscInt       m = A->m,*ip,N,*ailen = a->ilen;
70213f74950SBarry Smith    PetscInt       mbs = a->mbs,bs2 = a->bs2,rmax = 0;
70349b5e25fSSatish Balay    MatScalar      *aa = a->a,*ap;
70449b5e25fSSatish Balay 
70549b5e25fSSatish Balay    PetscFunctionBegin;
70649b5e25fSSatish Balay    if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
70749b5e25fSSatish Balay 
70849b5e25fSSatish Balay    if (m) rmax = ailen[0];
70949b5e25fSSatish Balay    for (i=1; i<mbs; i++) {
71049b5e25fSSatish Balay      /* move each row back by the amount of empty slots (fshift) before it*/
71149b5e25fSSatish Balay      fshift += imax[i-1] - ailen[i-1];
71249b5e25fSSatish Balay      rmax   = PetscMax(rmax,ailen[i]);
71349b5e25fSSatish Balay      if (fshift) {
71449b5e25fSSatish Balay        ip = aj + ai[i]; ap = aa + bs2*ai[i];
71549b5e25fSSatish Balay        N = ailen[i];
71649b5e25fSSatish Balay        for (j=0; j<N; j++) {
71749b5e25fSSatish Balay          ip[j-fshift] = ip[j];
71849b5e25fSSatish Balay          ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
71949b5e25fSSatish Balay        }
72049b5e25fSSatish Balay      }
72149b5e25fSSatish Balay      ai[i] = ai[i-1] + ailen[i-1];
72249b5e25fSSatish Balay    }
72349b5e25fSSatish Balay    if (mbs) {
72449b5e25fSSatish Balay      fshift += imax[mbs-1] - ailen[mbs-1];
72549b5e25fSSatish Balay      ai[mbs] = ai[mbs-1] + ailen[mbs-1];
72649b5e25fSSatish Balay    }
72749b5e25fSSatish Balay    /* reset ilen and imax for each row */
72849b5e25fSSatish Balay    for (i=0; i<mbs; i++) {
72949b5e25fSSatish Balay      ailen[i] = imax[i] = ai[i+1] - ai[i];
73049b5e25fSSatish Balay    }
7316c6c5352SBarry Smith    a->nz = ai[mbs];
73249b5e25fSSatish Balay 
733b424e231SHong Zhang    /* diagonals may have moved, reset it */
734b424e231SHong Zhang    if (a->diag) {
73513f74950SBarry Smith      ierr = PetscMemcpy(a->diag,ai,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
73649b5e25fSSatish Balay    }
73777431f27SBarry Smith    PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Matrix size: %D X %D, block size %D; storage space: %D unneeded, %D used\n",
738521d7252SBarry Smith             m,A->m,A->bs,fshift*bs2,a->nz*bs2);
73977431f27SBarry Smith    PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Number of mallocs during MatSetValues is %D\n",
74049b5e25fSSatish Balay             a->reallocs);
74177431f27SBarry Smith    PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Most nonzeros blocks in any row is %D\n",rmax);
74249b5e25fSSatish Balay    a->reallocs          = 0;
74349b5e25fSSatish Balay    A->info.nz_unneeded  = (PetscReal)fshift*bs2;
74449b5e25fSSatish Balay    PetscFunctionReturn(0);
74549b5e25fSSatish Balay  }
74649b5e25fSSatish Balay 
74749b5e25fSSatish Balay  /*
74849b5e25fSSatish Balay     This function returns an array of flags which indicate the locations of contiguous
74949b5e25fSSatish Balay     blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
75049b5e25fSSatish Balay     then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
75149b5e25fSSatish Balay     Assume: sizes should be long enough to hold all the values.
75249b5e25fSSatish Balay  */
7534a2ae208SSatish Balay  #undef __FUNCT__
7544a2ae208SSatish Balay  #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
75513f74950SBarry Smith  PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
75649b5e25fSSatish Balay  {
75713f74950SBarry Smith    PetscInt   i,j,k,row;
75849b5e25fSSatish Balay    PetscTruth flg;
75949b5e25fSSatish Balay 
76049b5e25fSSatish Balay    PetscFunctionBegin;
76149b5e25fSSatish Balay    for (i=0,j=0; i<n; j++) {
76249b5e25fSSatish Balay      row = idx[i];
76349b5e25fSSatish Balay      if (row%bs!=0) { /* Not the begining of a block */
76449b5e25fSSatish Balay        sizes[j] = 1;
76549b5e25fSSatish Balay        i++;
76649b5e25fSSatish Balay      } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
76749b5e25fSSatish Balay        sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
76849b5e25fSSatish Balay        i++;
76949b5e25fSSatish Balay      } else { /* Begining of the block, so check if the complete block exists */
77049b5e25fSSatish Balay        flg = PETSC_TRUE;
77149b5e25fSSatish Balay        for (k=1; k<bs; k++) {
77249b5e25fSSatish Balay          if (row+k != idx[i+k]) { /* break in the block */
77349b5e25fSSatish Balay            flg = PETSC_FALSE;
77449b5e25fSSatish Balay            break;
77549b5e25fSSatish Balay          }
77649b5e25fSSatish Balay        }
777abc0a331SBarry Smith        if (flg) { /* No break in the bs */
77849b5e25fSSatish Balay          sizes[j] = bs;
77949b5e25fSSatish Balay          i+= bs;
78049b5e25fSSatish Balay        } else {
78149b5e25fSSatish Balay          sizes[j] = 1;
78249b5e25fSSatish Balay          i++;
78349b5e25fSSatish Balay        }
78449b5e25fSSatish Balay      }
78549b5e25fSSatish Balay    }
78649b5e25fSSatish Balay    *bs_max = j;
78749b5e25fSSatish Balay    PetscFunctionReturn(0);
78849b5e25fSSatish Balay  }
78949b5e25fSSatish Balay 
79049b5e25fSSatish Balay 
79149b5e25fSSatish Balay  /* Only add/insert a(i,j) with i<=j (blocks).
79249b5e25fSSatish Balay     Any a(i,j) with i>j input by user is ingored.
79349b5e25fSSatish Balay  */
79449b5e25fSSatish Balay 
7954a2ae208SSatish Balay  #undef __FUNCT__
7964a2ae208SSatish Balay  #define __FUNCT__ "MatSetValues_SeqSBAIJ"
79713f74950SBarry Smith  PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
79849b5e25fSSatish Balay  {
79949b5e25fSSatish Balay    Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8006849ba73SBarry Smith    PetscErrorCode ierr;
801e2ee6c50SBarry Smith    PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
80213f74950SBarry Smith    PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
80313f74950SBarry Smith    PetscInt       *aj=a->j,nonew=a->nonew,bs=A->bs,brow,bcol;
80413f74950SBarry Smith    PetscInt       ridx,cidx,bs2=a->bs2;
80549b5e25fSSatish Balay    MatScalar      *ap,value,*aa=a->a,*bap;
80649b5e25fSSatish Balay 
80749b5e25fSSatish Balay    PetscFunctionBegin;
80849b5e25fSSatish Balay    for (k=0; k<m; k++) { /* loop over added rows */
80949b5e25fSSatish Balay      row  = im[k];       /* row number */
81049b5e25fSSatish Balay      brow = row/bs;      /* block row number */
81149b5e25fSSatish Balay      if (row < 0) continue;
8122515c552SBarry Smith  #if defined(PETSC_USE_DEBUG)
81377431f27SBarry Smith      if (row >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->m-1);
81449b5e25fSSatish Balay  #endif
81549b5e25fSSatish Balay      rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
81649b5e25fSSatish Balay      ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
81749b5e25fSSatish Balay      rmax = imax[brow];  /* maximum space allocated for this row */
81849b5e25fSSatish Balay      nrow = ailen[brow]; /* actual length of this row */
81949b5e25fSSatish Balay      low  = 0;
82049b5e25fSSatish Balay 
82149b5e25fSSatish Balay      for (l=0; l<n; l++) { /* loop over added columns */
82249b5e25fSSatish Balay        if (in[l] < 0) continue;
8232515c552SBarry Smith  #if defined(PETSC_USE_DEBUG)
82477431f27SBarry Smith        if (in[l] >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->m-1);
82549b5e25fSSatish Balay  #endif
82649b5e25fSSatish Balay        col = in[l];
82749b5e25fSSatish Balay        bcol = col/bs;              /* block col number */
82849b5e25fSSatish Balay 
829*941593c8SHong Zhang        if (brow > bcol) {
830*941593c8SHong Zhang          if (a->ignore_ltriangular){
831*941593c8SHong Zhang            continue; /* ignore lower triangular values */
832*941593c8SHong Zhang          } else {
833*941593c8SHong Zhang            SETERRQ(PETSC_ERR_USER,"Lower triangular value cannot be set for sbaij format. Ignoring these values, run with -mat_ignore_lower_triangular or call MatSetOption(mat,MAT_IGNORE_LOWER_TRIANGULAR)");
834*941593c8SHong Zhang          }
835*941593c8SHong Zhang        }
836f4989cb3SHong Zhang 
83749b5e25fSSatish Balay        ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
8388549e402SHong Zhang        if ((brow==bcol && ridx<=cidx) || (brow<bcol)){
83949b5e25fSSatish Balay          /* element value a(k,l) */
84049b5e25fSSatish Balay          if (roworiented) {
84149b5e25fSSatish Balay            value = v[l + k*n];
84249b5e25fSSatish Balay          } else {
84349b5e25fSSatish Balay            value = v[k + l*m];
84449b5e25fSSatish Balay          }
84549b5e25fSSatish Balay 
84649b5e25fSSatish Balay          /* move pointer bap to a(k,l) quickly and add/insert value */
847e2ee6c50SBarry Smith          if (col < lastcol) low = 0; high = nrow;
848e2ee6c50SBarry Smith          lastcol = col;
84949b5e25fSSatish Balay          while (high-low > 7) {
85049b5e25fSSatish Balay            t = (low+high)/2;
85149b5e25fSSatish Balay            if (rp[t] > bcol) high = t;
85249b5e25fSSatish Balay            else              low  = t;
85349b5e25fSSatish Balay          }
85449b5e25fSSatish Balay          for (i=low; i<high; i++) {
85549b5e25fSSatish Balay            if (rp[i] > bcol) break;
85649b5e25fSSatish Balay            if (rp[i] == bcol) {
85749b5e25fSSatish Balay              bap  = ap +  bs2*i + bs*cidx + ridx;
85849b5e25fSSatish Balay              if (is == ADD_VALUES) *bap += value;
85949b5e25fSSatish Balay              else                  *bap  = value;
8608549e402SHong Zhang              /* for diag block, add/insert its symmetric element a(cidx,ridx) */
8618549e402SHong Zhang              if (brow == bcol && ridx < cidx){
8628549e402SHong Zhang                bap  = ap +  bs2*i + bs*ridx + cidx;
8638549e402SHong Zhang                if (is == ADD_VALUES) *bap += value;
8648549e402SHong Zhang                else                  *bap  = value;
8658549e402SHong Zhang              }
86649b5e25fSSatish Balay              goto noinsert1;
86749b5e25fSSatish Balay            }
86849b5e25fSSatish Balay          }
86949b5e25fSSatish Balay 
87049b5e25fSSatish Balay          if (nonew == 1) goto noinsert1;
87177431f27SBarry Smith          else if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
87249b5e25fSSatish Balay          if (nrow >= rmax) {
87349b5e25fSSatish Balay            /* there is no extra room in row, therefore enlarge */
87413f74950SBarry Smith            PetscInt       new_nz = ai[a->mbs] + CHUNKSIZE,len,*new_i,*new_j;
87549b5e25fSSatish Balay            MatScalar *new_a;
87649b5e25fSSatish Balay 
87777431f27SBarry Smith            if (nonew == -2) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
87849b5e25fSSatish Balay 
87949b5e25fSSatish Balay            /* Malloc new storage space */
88013f74950SBarry Smith            len   = new_nz*(sizeof(PetscInt)+bs2*sizeof(MatScalar))+(a->mbs+1)*sizeof(PetscInt);
88182502324SSatish Balay            ierr  = PetscMalloc(len,&new_a);CHKERRQ(ierr);
88213f74950SBarry Smith            new_j = (PetscInt*)(new_a + bs2*new_nz);
88349b5e25fSSatish Balay            new_i = new_j + new_nz;
88449b5e25fSSatish Balay 
88549b5e25fSSatish Balay            /* copy over old data into new slots */
88649b5e25fSSatish Balay            for (ii=0; ii<brow+1; ii++) {new_i[ii] = ai[ii];}
88749b5e25fSSatish Balay            for (ii=brow+1; ii<a->mbs+1; ii++) {new_i[ii] = ai[ii]+CHUNKSIZE;}
88813f74950SBarry Smith            ierr = PetscMemcpy(new_j,aj,(ai[brow]+nrow)*sizeof(PetscInt));CHKERRQ(ierr);
88949b5e25fSSatish Balay            len  = (new_nz - CHUNKSIZE - ai[brow] - nrow);
89013f74950SBarry Smith            ierr = PetscMemcpy(new_j+ai[brow]+nrow+CHUNKSIZE,aj+ai[brow]+nrow,len*sizeof(PetscInt));CHKERRQ(ierr);
89149b5e25fSSatish Balay            ierr = PetscMemcpy(new_a,aa,(ai[brow]+nrow)*bs2*sizeof(MatScalar));CHKERRQ(ierr);
89249b5e25fSSatish Balay            ierr = PetscMemzero(new_a+bs2*(ai[brow]+nrow),bs2*CHUNKSIZE*sizeof(MatScalar));CHKERRQ(ierr);
89349b5e25fSSatish Balay            ierr = PetscMemcpy(new_a+bs2*(ai[brow]+nrow+CHUNKSIZE),aa+bs2*(ai[brow]+nrow),bs2*len*sizeof(MatScalar));CHKERRQ(ierr);
89449b5e25fSSatish Balay            /* free up old matrix storage */
89549b5e25fSSatish Balay            ierr = PetscFree(a->a);CHKERRQ(ierr);
89649b5e25fSSatish Balay            if (!a->singlemalloc) {
89749b5e25fSSatish Balay              ierr = PetscFree(a->i);CHKERRQ(ierr);
89849b5e25fSSatish Balay              ierr = PetscFree(a->j);CHKERRQ(ierr);
89949b5e25fSSatish Balay            }
90049b5e25fSSatish Balay            aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
90149b5e25fSSatish Balay            a->singlemalloc = PETSC_TRUE;
90249b5e25fSSatish Balay 
90349b5e25fSSatish Balay            rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
90449b5e25fSSatish Balay            rmax = imax[brow] = imax[brow] + CHUNKSIZE;
90552e6d16bSBarry Smith            ierr = PetscLogObjectMemory(A,CHUNKSIZE*(sizeof(PetscInt) + bs2*sizeof(MatScalar)));CHKERRQ(ierr);
9066c6c5352SBarry Smith            a->maxnz += bs2*CHUNKSIZE;
90749b5e25fSSatish Balay            a->reallocs++;
9086c6c5352SBarry Smith            a->nz++;
90949b5e25fSSatish Balay          }
91049b5e25fSSatish Balay 
91149b5e25fSSatish Balay          N = nrow++ - 1;
91249b5e25fSSatish Balay          /* shift up all the later entries in this row */
91349b5e25fSSatish Balay          for (ii=N; ii>=i; ii--) {
91449b5e25fSSatish Balay            rp[ii+1] = rp[ii];
91549b5e25fSSatish Balay            ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
91649b5e25fSSatish Balay          }
91749b5e25fSSatish Balay          if (N>=i) {
91849b5e25fSSatish Balay            ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
91949b5e25fSSatish Balay          }
92049b5e25fSSatish Balay          rp[i]                      = bcol;
92149b5e25fSSatish Balay          ap[bs2*i + bs*cidx + ridx] = value;
92249b5e25fSSatish Balay        noinsert1:;
92349b5e25fSSatish Balay          low = i;
9248549e402SHong Zhang        }
92549b5e25fSSatish Balay      }   /* end of loop over added columns */
92649b5e25fSSatish Balay      ailen[brow] = nrow;
92749b5e25fSSatish Balay    }   /* end of loop over added rows */
92849b5e25fSSatish Balay    PetscFunctionReturn(0);
92949b5e25fSSatish Balay  }
93049b5e25fSSatish Balay 
9316849ba73SBarry Smith  EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*);
9326849ba73SBarry Smith  EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*);
93313f74950SBarry Smith  EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt);
93413f74950SBarry Smith  EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*);
93513f74950SBarry Smith  EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]);
9366849ba73SBarry Smith  EXTERN PetscErrorCode MatScale_SeqSBAIJ(const PetscScalar*,Mat);
9376849ba73SBarry Smith  EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *);
9386849ba73SBarry Smith  EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*);
9396849ba73SBarry Smith  EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec);
9406849ba73SBarry Smith  EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec);
9416849ba73SBarry Smith  EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *);
9426849ba73SBarry Smith  EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat);
9436849ba73SBarry Smith  EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec);
94449b5e25fSSatish Balay 
9456849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec);
9466849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec);
9476849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec);
9486849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec);
9496849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec);
9506849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec);
9516849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec);
9526849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec);
95349b5e25fSSatish Balay 
9546849ba73SBarry Smith  EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs);
955d59c15a7SBarry Smith 
9566849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
9576849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
9586849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
9596849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
9606849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
9616849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
9626849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
9636849ba73SBarry Smith  EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
96407f98182SSatish Balay 
965af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*);
966af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*);
967af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*);
968af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*);
969af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*);
970af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*);
971af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*);
972af281ebdSHong Zhang  EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*);
97313f74950SBarry Smith  EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*);
97449b5e25fSSatish Balay 
9756849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec);
9766849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec);
9776849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec);
9786849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec);
9796849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec);
9806849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec);
9816849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec);
9826849ba73SBarry Smith  EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec);
98349b5e25fSSatish Balay 
9846849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec);
9856849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec);
9866849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec);
9876849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec);
9886849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec);
9896849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec);
9906849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec);
9916849ba73SBarry Smith  EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec);
99249b5e25fSSatish Balay 
9934d101231SSatish Balay  #ifdef HAVE_MatICCFactor
9946849ba73SBarry Smith  /* modified from MatILUFactor_SeqSBAIJ, needs further work!  */
9954a2ae208SSatish Balay  #undef __FUNCT__
9964d101231SSatish Balay  #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
997dfbe8321SBarry Smith  PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,MatFactorInfo *info)
99849b5e25fSSatish Balay  {
9994ccecd49SHong Zhang    Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
100049b5e25fSSatish Balay    Mat            outA;
1001dfbe8321SBarry Smith    PetscErrorCode ierr;
100249b5e25fSSatish Balay    PetscTruth     row_identity,col_identity;
100349b5e25fSSatish Balay 
100449b5e25fSSatish Balay    PetscFunctionBegin;
100549b5e25fSSatish Balay    outA          = inA;
10061260e1f8SHong Zhang    inA->factor   = FACTOR_CHOLESKY;
100749b5e25fSSatish Balay 
100849b5e25fSSatish Balay    if (!a->diag) {
10091a3463dfSHong Zhang      ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
101049b5e25fSSatish Balay    }
101149b5e25fSSatish Balay    /*
101249b5e25fSSatish Balay        Blocksize 2, 3, 4, 5, 6 and 7 have a special faster factorization/solver
101349b5e25fSSatish Balay        for ILU(0) factorization with natural ordering
101449b5e25fSSatish Balay    */
101549b5e25fSSatish Balay    switch (a->bs) {
101649b5e25fSSatish Balay    case 1:
10170fe9e5beSBarry Smith      inA->ops->solve            = MatSolve_SeqSBAIJ_1_NaturalOrdering;
101807f98182SSatish Balay      inA->ops->solvetranspose   = MatSolve_SeqSBAIJ_1_NaturalOrdering;
1019d59c15a7SBarry Smith      inA->ops->solves           = MatSolves_SeqSBAIJ_1;
10204d101231SSatish Balay      PetscLoginfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering solvetrans BS=1\n");
102149b5e25fSSatish Balay    case 2:
10221a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
10231a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_2_NaturalOrdering;
10241a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_2_NaturalOrdering;
10254d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=2\n");
102649b5e25fSSatish Balay      break;
102749b5e25fSSatish Balay    case 3:
10281a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
10291a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_3_NaturalOrdering;
10301a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_3_NaturalOrdering;
10314d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=3\n");
103249b5e25fSSatish Balay      break;
103349b5e25fSSatish Balay    case 4:
10341a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
10351a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_4_NaturalOrdering;
10361a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_4_NaturalOrdering;
10374d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=4\n");
103849b5e25fSSatish Balay      break;
103949b5e25fSSatish Balay    case 5:
10401a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
10411a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_5_NaturalOrdering;
10421a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_5_NaturalOrdering;
10434d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=5\n");
104449b5e25fSSatish Balay      break;
104549b5e25fSSatish Balay    case 6:
10461a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
10471a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_6_NaturalOrdering;
10481a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_6_NaturalOrdering;
10494d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=6\n");
105049b5e25fSSatish Balay      break;
105149b5e25fSSatish Balay    case 7:
10521a3463dfSHong Zhang      inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
10531a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_7_NaturalOrdering;
10541a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_7_NaturalOrdering;
10554d101231SSatish Balay      PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=7\n");
105649b5e25fSSatish Balay      break;
105749b5e25fSSatish Balay    default:
105849b5e25fSSatish Balay      a->row        = row;
10591a3463dfSHong Zhang      a->icol       = col;
106049b5e25fSSatish Balay      ierr          = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
106149b5e25fSSatish Balay      ierr          = PetscObjectReference((PetscObject)col);CHKERRQ(ierr);
106249b5e25fSSatish Balay 
106349b5e25fSSatish Balay      /* Create the invert permutation so that it can be used in MatLUFactorNumeric() */
106449b5e25fSSatish Balay      ierr = ISInvertPermutation(col,PETSC_DECIDE, &(a->icol));CHKERRQ(ierr);
106552e6d16bSBarry Smith      ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr);
106649b5e25fSSatish Balay 
106749b5e25fSSatish Balay      if (!a->solve_work) {
106887828ca2SBarry Smith        ierr = PetscMalloc((A->m+a->bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr);
106952e6d16bSBarry Smith        ierr = PetscLogObjectMemory(inA,(A->m+a->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
107049b5e25fSSatish Balay      }
107149b5e25fSSatish Balay    }
107249b5e25fSSatish Balay 
1073af281ebdSHong Zhang    ierr = MatCholeskyFactorNumeric(inA,info,&outA);CHKERRQ(ierr);
107449b5e25fSSatish Balay    PetscFunctionReturn(0);
107549b5e25fSSatish Balay  }
1076950f1e5bSHong Zhang  #endif
1077950f1e5bSHong Zhang 
10784a2ae208SSatish Balay  #undef __FUNCT__
10794a2ae208SSatish Balay  #define __FUNCT__ "MatPrintHelp_SeqSBAIJ"
1080dfbe8321SBarry Smith  PetscErrorCode MatPrintHelp_SeqSBAIJ(Mat A)
108149b5e25fSSatish Balay  {
108249b5e25fSSatish Balay    static PetscTruth called = PETSC_FALSE;
108349b5e25fSSatish Balay    MPI_Comm          comm = A->comm;
1084dfbe8321SBarry Smith    PetscErrorCode    ierr;
108549b5e25fSSatish Balay 
108649b5e25fSSatish Balay    PetscFunctionBegin;
108749b5e25fSSatish Balay    if (called) {PetscFunctionReturn(0);} else called = PETSC_TRUE;
108849b5e25fSSatish Balay    ierr = (*PetscHelpPrintf)(comm," Options for MATSEQSBAIJ and MATMPISBAIJ matrix formats (the defaults):\n");CHKERRQ(ierr);
108949b5e25fSSatish Balay    ierr = (*PetscHelpPrintf)(comm,"  -mat_block_size <block_size>\n");CHKERRQ(ierr);
1090*941593c8SHong Zhang    ierr = (*PetscHelpPrintf)(comm,"  -mat_ignore_ltriangular: Ignore lower triangular values set by user\n");CHKERRQ(ierr);
109149b5e25fSSatish Balay    PetscFunctionReturn(0);
109249b5e25fSSatish Balay  }
109349b5e25fSSatish Balay 
109449b5e25fSSatish Balay  EXTERN_C_BEGIN
10954a2ae208SSatish Balay  #undef __FUNCT__
10964a2ae208SSatish Balay  #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
109713f74950SBarry Smith  PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
109849b5e25fSSatish Balay  {
1099045c9aa0SHong Zhang    Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data;
110013f74950SBarry Smith    PetscInt     i,nz,n;
110149b5e25fSSatish Balay 
110249b5e25fSSatish Balay    PetscFunctionBegin;
11036c6c5352SBarry Smith    nz = baij->maxnz;
1104c464158bSHong Zhang    n  = mat->n;
110549b5e25fSSatish Balay    for (i=0; i<nz; i++) {
110649b5e25fSSatish Balay      baij->j[i] = indices[i];
110749b5e25fSSatish Balay    }
11086c6c5352SBarry Smith    baij->nz = nz;
110949b5e25fSSatish Balay    for (i=0; i<n; i++) {
111049b5e25fSSatish Balay      baij->ilen[i] = baij->imax[i];
111149b5e25fSSatish Balay    }
111249b5e25fSSatish Balay    PetscFunctionReturn(0);
111349b5e25fSSatish Balay  }
111449b5e25fSSatish Balay  EXTERN_C_END
111549b5e25fSSatish Balay 
11164a2ae208SSatish Balay  #undef __FUNCT__
11174a2ae208SSatish Balay  #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
111849b5e25fSSatish Balay  /*@
111919585528SSatish Balay      MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
112049b5e25fSSatish Balay         in the matrix.
112149b5e25fSSatish Balay 
112249b5e25fSSatish Balay    Input Parameters:
112319585528SSatish Balay  +  mat     - the SeqSBAIJ matrix
112449b5e25fSSatish Balay  -  indices - the column indices
112549b5e25fSSatish Balay 
112649b5e25fSSatish Balay    Level: advanced
112749b5e25fSSatish Balay 
112849b5e25fSSatish Balay    Notes:
112949b5e25fSSatish Balay      This can be called if you have precomputed the nonzero structure of the
113049b5e25fSSatish Balay    matrix and want to provide it to the matrix object to improve the performance
113149b5e25fSSatish Balay    of the MatSetValues() operation.
113249b5e25fSSatish Balay 
113349b5e25fSSatish Balay      You MUST have set the correct numbers of nonzeros per row in the call to
1134d1be2dadSMatthew Knepley    MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
113549b5e25fSSatish Balay 
1136ab9f2c04SSatish Balay      MUST be called before any calls to MatSetValues()
113749b5e25fSSatish Balay 
1138ab9f2c04SSatish Balay  .seealso: MatCreateSeqSBAIJ
113949b5e25fSSatish Balay  @*/
114013f74950SBarry Smith  PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
114149b5e25fSSatish Balay  {
114213f74950SBarry Smith    PetscErrorCode ierr,(*f)(Mat,PetscInt *);
114349b5e25fSSatish Balay 
114449b5e25fSSatish Balay    PetscFunctionBegin;
11454482741eSBarry Smith    PetscValidHeaderSpecific(mat,MAT_COOKIE,1);
11464482741eSBarry Smith    PetscValidPointer(indices,2);
1147c134de8dSSatish Balay    ierr = PetscObjectQueryFunction((PetscObject)mat,"MatSeqSBAIJSetColumnIndices_C",(void (**)(void))&f);CHKERRQ(ierr);
114849b5e25fSSatish Balay    if (f) {
114949b5e25fSSatish Balay      ierr = (*f)(mat,indices);CHKERRQ(ierr);
115049b5e25fSSatish Balay    } else {
1151e005ede5SBarry Smith      SETERRQ(PETSC_ERR_SUP,"Wrong type of matrix to set column indices");
115249b5e25fSSatish Balay    }
115349b5e25fSSatish Balay    PetscFunctionReturn(0);
115449b5e25fSSatish Balay  }
115549b5e25fSSatish Balay 
11564a2ae208SSatish Balay  #undef __FUNCT__
11574a2ae208SSatish Balay  #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ"
1158dfbe8321SBarry Smith  PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A)
1159273d9f13SBarry Smith  {
1160dfbe8321SBarry Smith    PetscErrorCode ierr;
1161273d9f13SBarry Smith 
1162273d9f13SBarry Smith    PetscFunctionBegin;
1163273d9f13SBarry Smith    ierr =  MatSeqSBAIJSetPreallocation(A,1,PETSC_DEFAULT,0);CHKERRQ(ierr);
1164273d9f13SBarry Smith    PetscFunctionReturn(0);
1165273d9f13SBarry Smith  }
1166273d9f13SBarry Smith 
1167a6ece127SHong Zhang  #undef __FUNCT__
1168a6ece127SHong Zhang  #define __FUNCT__ "MatGetArray_SeqSBAIJ"
1169dfbe8321SBarry Smith  PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1170a6ece127SHong Zhang  {
1171a6ece127SHong Zhang    Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1172a6ece127SHong Zhang    PetscFunctionBegin;
1173a6ece127SHong Zhang    *array = a->a;
1174a6ece127SHong Zhang    PetscFunctionReturn(0);
1175a6ece127SHong Zhang  }
1176a6ece127SHong Zhang 
1177a6ece127SHong Zhang  #undef __FUNCT__
1178a6ece127SHong Zhang  #define __FUNCT__ "MatRestoreArray_SeqSBAIJ"
1179dfbe8321SBarry Smith  PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1180a6ece127SHong Zhang  {
1181a6ece127SHong Zhang    PetscFunctionBegin;
1182a6ece127SHong Zhang    PetscFunctionReturn(0);
1183a6ece127SHong Zhang  }
1184a6ece127SHong Zhang 
118542ee4b1aSHong Zhang  #include "petscblaslapack.h"
118642ee4b1aSHong Zhang  #undef __FUNCT__
118742ee4b1aSHong Zhang  #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1188dfbe8321SBarry Smith  PetscErrorCode MatAXPY_SeqSBAIJ(const PetscScalar *a,Mat X,Mat Y,MatStructure str)
118942ee4b1aSHong Zhang  {
119042ee4b1aSHong Zhang    Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data;
1191dfbe8321SBarry Smith    PetscErrorCode ierr;
1192521d7252SBarry Smith    PetscInt       i,bs=Y->bs,bs2,j;
11934ce68768SBarry Smith    PetscBLASInt   bnz = (PetscBLASInt)x->nz,one = 1;
119442ee4b1aSHong Zhang 
119542ee4b1aSHong Zhang    PetscFunctionBegin;
119642ee4b1aSHong Zhang    if (str == SAME_NONZERO_PATTERN) {
119771044d3cSBarry Smith      BLASaxpy_(&bnz,(PetscScalar*)a,x->a,&one,y->a,&one);
1198c537a176SHong Zhang    } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1199c4319e64SHong Zhang      if (y->xtoy && y->XtoY != X) {
1200c4319e64SHong Zhang        ierr = PetscFree(y->xtoy);CHKERRQ(ierr);
1201c4319e64SHong Zhang        ierr = MatDestroy(y->XtoY);CHKERRQ(ierr);
1202c537a176SHong Zhang      }
1203c4319e64SHong Zhang      if (!y->xtoy) { /* get xtoy */
1204c4319e64SHong Zhang        ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr);
1205c4319e64SHong Zhang        y->XtoY = X;
1206c537a176SHong Zhang      }
1207c4319e64SHong Zhang      bs2 = bs*bs;
12086c6c5352SBarry Smith      for (i=0; i<x->nz; i++) {
1209c4319e64SHong Zhang        j = 0;
1210c4319e64SHong Zhang        while (j < bs2){
12116550863bSHong Zhang          y->a[bs2*y->xtoy[i]+j] += (*a)*(x->a[bs2*i+j]);
1212c4319e64SHong Zhang          j++;
1213c537a176SHong Zhang        }
1214c4319e64SHong Zhang      }
121577431f27SBarry Smith      PetscLogInfo(0,"MatAXPY_SeqSBAIJ: ratio of nnz_s(X)/nnz_s(Y): %D/%D = %g\n",bs2*x->nz,bs2*y->nz,(PetscReal)(bs2*x->nz)/(bs2*y->nz));
121642ee4b1aSHong Zhang    } else {
121742ee4b1aSHong Zhang      ierr = MatAXPY_Basic(a,X,Y,str);CHKERRQ(ierr);
121842ee4b1aSHong Zhang    }
121942ee4b1aSHong Zhang    PetscFunctionReturn(0);
122042ee4b1aSHong Zhang  }
122142ee4b1aSHong Zhang 
1222efcf0fc3SBarry Smith  #undef __FUNCT__
1223efcf0fc3SBarry Smith  #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1224dfbe8321SBarry Smith  PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscTruth *flg)
1225efcf0fc3SBarry Smith  {
1226efcf0fc3SBarry Smith    PetscFunctionBegin;
1227efcf0fc3SBarry Smith    *flg = PETSC_TRUE;
1228efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1229efcf0fc3SBarry Smith  }
1230efcf0fc3SBarry Smith 
1231efcf0fc3SBarry Smith  #undef __FUNCT__
1232efcf0fc3SBarry Smith  #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1233dfbe8321SBarry Smith  PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscTruth *flg)
1234efcf0fc3SBarry Smith  {
1235efcf0fc3SBarry Smith    PetscFunctionBegin;
1236efcf0fc3SBarry Smith    *flg = PETSC_TRUE;
1237efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1238efcf0fc3SBarry Smith  }
1239efcf0fc3SBarry Smith 
1240efcf0fc3SBarry Smith  #undef __FUNCT__
1241efcf0fc3SBarry Smith  #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1242dfbe8321SBarry Smith  PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscTruth *flg)
1243efcf0fc3SBarry Smith  {
1244efcf0fc3SBarry Smith    PetscFunctionBegin;
1245efcf0fc3SBarry Smith    *flg = PETSC_FALSE;
1246efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1247efcf0fc3SBarry Smith  }
1248efcf0fc3SBarry Smith 
124949b5e25fSSatish Balay  /* -------------------------------------------------------------------*/
125049b5e25fSSatish Balay  static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
125149b5e25fSSatish Balay         MatGetRow_SeqSBAIJ,
125249b5e25fSSatish Balay         MatRestoreRow_SeqSBAIJ,
125349b5e25fSSatish Balay         MatMult_SeqSBAIJ_N,
125497304618SKris Buschelman  /* 4*/ MatMultAdd_SeqSBAIJ_N,
1255e005ede5SBarry Smith         MatMult_SeqSBAIJ_N,
1256e005ede5SBarry Smith         MatMultAdd_SeqSBAIJ_N,
125749b5e25fSSatish Balay         MatSolve_SeqSBAIJ_N,
125849b5e25fSSatish Balay         0,
125949b5e25fSSatish Balay         0,
126097304618SKris Buschelman  /*10*/ 0,
126149b5e25fSSatish Balay         0,
12625f4ef2deSHong Zhang         MatCholeskyFactor_SeqSBAIJ,
1263d06b337dSHong Zhang         MatRelax_SeqSBAIJ,
126449b5e25fSSatish Balay         MatTranspose_SeqSBAIJ,
126597304618SKris Buschelman  /*15*/ MatGetInfo_SeqSBAIJ,
126649b5e25fSSatish Balay         MatEqual_SeqSBAIJ,
126749b5e25fSSatish Balay         MatGetDiagonal_SeqSBAIJ,
126849b5e25fSSatish Balay         MatDiagonalScale_SeqSBAIJ,
126949b5e25fSSatish Balay         MatNorm_SeqSBAIJ,
127097304618SKris Buschelman  /*20*/ 0,
127149b5e25fSSatish Balay         MatAssemblyEnd_SeqSBAIJ,
127249b5e25fSSatish Balay         0,
127349b5e25fSSatish Balay         MatSetOption_SeqSBAIJ,
127449b5e25fSSatish Balay         MatZeroEntries_SeqSBAIJ,
1275dcf5cc72SBarry Smith  /*25*/ 0,
127649b5e25fSSatish Balay         0,
127749b5e25fSSatish Balay         0,
12785f4ef2deSHong Zhang         MatCholeskyFactorSymbolic_SeqSBAIJ,
12795f4ef2deSHong Zhang         MatCholeskyFactorNumeric_SeqSBAIJ_N,
128097304618SKris Buschelman  /*30*/ MatSetUpPreallocation_SeqSBAIJ,
1281c464158bSHong Zhang         0,
12824d101231SSatish Balay         MatICCFactorSymbolic_SeqSBAIJ,
1283a6ece127SHong Zhang         MatGetArray_SeqSBAIJ,
1284a6ece127SHong Zhang         MatRestoreArray_SeqSBAIJ,
128597304618SKris Buschelman  /*35*/ MatDuplicate_SeqSBAIJ,
128649b5e25fSSatish Balay         0,
128749b5e25fSSatish Balay         0,
128849b5e25fSSatish Balay         0,
1289950f1e5bSHong Zhang         0,
129097304618SKris Buschelman  /*40*/ MatAXPY_SeqSBAIJ,
129149b5e25fSSatish Balay         MatGetSubMatrices_SeqSBAIJ,
129249b5e25fSSatish Balay         MatIncreaseOverlap_SeqSBAIJ,
129349b5e25fSSatish Balay         MatGetValues_SeqSBAIJ,
129449b5e25fSSatish Balay         0,
129597304618SKris Buschelman  /*45*/ MatPrintHelp_SeqSBAIJ,
129649b5e25fSSatish Balay         MatScale_SeqSBAIJ,
129749b5e25fSSatish Balay         0,
129849b5e25fSSatish Balay         0,
129949b5e25fSSatish Balay         0,
1300521d7252SBarry Smith  /*50*/ 0,
130149b5e25fSSatish Balay         MatGetRowIJ_SeqSBAIJ,
130249b5e25fSSatish Balay         MatRestoreRowIJ_SeqSBAIJ,
130349b5e25fSSatish Balay         0,
130449b5e25fSSatish Balay         0,
130597304618SKris Buschelman  /*55*/ 0,
130649b5e25fSSatish Balay         0,
130749b5e25fSSatish Balay         0,
130849b5e25fSSatish Balay         0,
130949b5e25fSSatish Balay         MatSetValuesBlocked_SeqSBAIJ,
131097304618SKris Buschelman  /*60*/ MatGetSubMatrix_SeqSBAIJ,
131149b5e25fSSatish Balay         0,
131249b5e25fSSatish Balay         0,
13138a124369SBarry Smith         MatGetPetscMaps_Petsc,
1314d959ec07SHong Zhang         0,
131597304618SKris Buschelman  /*65*/ 0,
1316d959ec07SHong Zhang         0,
1317d959ec07SHong Zhang         0,
1318d959ec07SHong Zhang         0,
1319d959ec07SHong Zhang         0,
132097304618SKris Buschelman  /*70*/ MatGetRowMax_SeqSBAIJ,
13213e0d88b5SBarry Smith         0,
13223e0d88b5SBarry Smith         0,
13233e0d88b5SBarry Smith         0,
13243e0d88b5SBarry Smith         0,
132597304618SKris Buschelman  /*75*/ 0,
13263e0d88b5SBarry Smith         0,
13273e0d88b5SBarry Smith         0,
13283e0d88b5SBarry Smith         0,
13293e0d88b5SBarry Smith         0,
133097304618SKris Buschelman  /*80*/ 0,
13313e0d88b5SBarry Smith         0,
13323e0d88b5SBarry Smith         0,
13333e0d88b5SBarry Smith  #if !defined(PETSC_USE_COMPLEX)
133497304618SKris Buschelman         MatGetInertia_SeqSBAIJ,
13353e0d88b5SBarry Smith  #else
133697304618SKris Buschelman         0,
13373e0d88b5SBarry Smith  #endif
1338865e5f61SKris Buschelman         MatLoad_SeqSBAIJ,
1339865e5f61SKris Buschelman  /*85*/ MatIsSymmetric_SeqSBAIJ,
1340865e5f61SKris Buschelman         MatIsHermitian_SeqSBAIJ,
1341efcf0fc3SBarry Smith         MatIsStructurallySymmetric_SeqSBAIJ,
1342865e5f61SKris Buschelman         0,
1343865e5f61SKris Buschelman         0,
1344865e5f61SKris Buschelman  /*90*/ 0,
1345865e5f61SKris Buschelman         0,
1346865e5f61SKris Buschelman         0,
1347865e5f61SKris Buschelman         0,
1348865e5f61SKris Buschelman         0,
1349865e5f61SKris Buschelman  /*95*/ 0,
1350865e5f61SKris Buschelman         0,
1351865e5f61SKris Buschelman         0,
1352865e5f61SKris Buschelman         0};
135349b5e25fSSatish Balay 
135449b5e25fSSatish Balay  EXTERN_C_BEGIN
13554a2ae208SSatish Balay  #undef __FUNCT__
13564a2ae208SSatish Balay  #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
1357dfbe8321SBarry Smith  PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat)
135849b5e25fSSatish Balay  {
13594afc71dfSHong Zhang    Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
1360521d7252SBarry Smith    PetscInt       nz = aij->i[mat->m]*mat->bs*aij->bs2;
1361dfbe8321SBarry Smith    PetscErrorCode ierr;
136249b5e25fSSatish Balay 
136349b5e25fSSatish Balay    PetscFunctionBegin;
136449b5e25fSSatish Balay    if (aij->nonew != 1) {
1365e005ede5SBarry Smith      SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
136649b5e25fSSatish Balay    }
136749b5e25fSSatish Balay 
136849b5e25fSSatish Balay    /* allocate space for values if not already there */
136949b5e25fSSatish Balay    if (!aij->saved_values) {
137087828ca2SBarry Smith      ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr);
137149b5e25fSSatish Balay    }
137249b5e25fSSatish Balay 
137349b5e25fSSatish Balay    /* copy values over */
137487828ca2SBarry Smith    ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
137549b5e25fSSatish Balay    PetscFunctionReturn(0);
137649b5e25fSSatish Balay  }
137749b5e25fSSatish Balay  EXTERN_C_END
137849b5e25fSSatish Balay 
137949b5e25fSSatish Balay  EXTERN_C_BEGIN
13804a2ae208SSatish Balay  #undef __FUNCT__
13814a2ae208SSatish Balay  #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
1382dfbe8321SBarry Smith  PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat)
138349b5e25fSSatish Balay  {
13844afc71dfSHong Zhang    Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
13856849ba73SBarry Smith    PetscErrorCode ierr;
1386521d7252SBarry Smith    PetscInt       nz = aij->i[mat->m]*mat->bs*aij->bs2;
138749b5e25fSSatish Balay 
138849b5e25fSSatish Balay    PetscFunctionBegin;
138949b5e25fSSatish Balay    if (aij->nonew != 1) {
1390e005ede5SBarry Smith      SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
139149b5e25fSSatish Balay    }
139249b5e25fSSatish Balay    if (!aij->saved_values) {
1393e005ede5SBarry Smith      SETERRQ(PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
139449b5e25fSSatish Balay    }
139549b5e25fSSatish Balay 
139649b5e25fSSatish Balay    /* copy values over */
139787828ca2SBarry Smith    ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
139849b5e25fSSatish Balay    PetscFunctionReturn(0);
139949b5e25fSSatish Balay  }
140049b5e25fSSatish Balay  EXTERN_C_END
140149b5e25fSSatish Balay 
14028549e402SHong Zhang  EXTERN_C_BEGIN
14034a2ae208SSatish Balay  #undef __FUNCT__
1404a23d5eceSKris Buschelman  #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
140513f74950SBarry Smith  PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
140649b5e25fSSatish Balay  {
1407c464158bSHong Zhang    Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
14086849ba73SBarry Smith    PetscErrorCode ierr;
140913f74950SBarry Smith    PetscInt       i,len,mbs,bs2;
141049b5e25fSSatish Balay    PetscTruth     flg;
141149b5e25fSSatish Balay 
141249b5e25fSSatish Balay    PetscFunctionBegin;
1413273d9f13SBarry Smith    B->preallocated = PETSC_TRUE;
1414e82a3eeeSBarry Smith    ierr = PetscOptionsGetInt(B->prefix,"-mat_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
1415c464158bSHong Zhang    mbs  = B->m/bs;
141649b5e25fSSatish Balay    bs2  = bs*bs;
141749b5e25fSSatish Balay 
1418c464158bSHong Zhang    if (mbs*bs != B->m) {
141929bbc08cSBarry Smith      SETERRQ(PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
142049b5e25fSSatish Balay    }
142149b5e25fSSatish Balay 
1422435da068SBarry Smith    if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
142377431f27SBarry Smith    if (nz < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
142449b5e25fSSatish Balay    if (nnz) {
142549b5e25fSSatish Balay      for (i=0; i<mbs; i++) {
142677431f27SBarry Smith        if (nnz[i] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]);
142777431f27SBarry Smith        if (nnz[i] > mbs) SETERRQ3(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs);
142849b5e25fSSatish Balay      }
142949b5e25fSSatish Balay    }
143049b5e25fSSatish Balay 
1431e82a3eeeSBarry Smith    ierr    = PetscOptionsHasName(B->prefix,"-mat_no_unroll",&flg);CHKERRQ(ierr);
143249b5e25fSSatish Balay    if (!flg) {
143349b5e25fSSatish Balay      switch (bs) {
143449b5e25fSSatish Balay      case 1:
14354ccecd49SHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1;
143649b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_1;
1437d59c15a7SBarry Smith        B->ops->solves          = MatSolves_SeqSBAIJ_1;
1438e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
143949b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_1;
144049b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_1;
144149b5e25fSSatish Balay        break;
144249b5e25fSSatish Balay      case 2:
14434ccecd49SHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
144449b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_2;
1445e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_2;
144649b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_2;
144749b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_2;
144849b5e25fSSatish Balay        break;
144949b5e25fSSatish Balay      case 3:
14505f4ef2deSHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
145149b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_3;
1452e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_3;
145349b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_3;
145449b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_3;
145549b5e25fSSatish Balay        break;
145649b5e25fSSatish Balay      case 4:
14575f4ef2deSHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
145849b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_4;
1459e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_4;
146049b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_4;
146149b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_4;
146249b5e25fSSatish Balay        break;
146349b5e25fSSatish Balay      case 5:
14645f4ef2deSHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
146549b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_5;
1466e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_5;
146749b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_5;
146849b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_5;
146949b5e25fSSatish Balay        break;
147049b5e25fSSatish Balay      case 6:
14715f4ef2deSHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
147249b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_6;
1473e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_6;
147449b5e25fSSatish Balay        B->ops->mult            = MatMult_SeqSBAIJ_6;
147549b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_6;
147649b5e25fSSatish Balay        break;
147749b5e25fSSatish Balay      case 7:
1478de53e5efSHong Zhang        B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
147949b5e25fSSatish Balay        B->ops->solve           = MatSolve_SeqSBAIJ_7;
1480e005ede5SBarry Smith        B->ops->solvetranspose  = MatSolve_SeqSBAIJ_7;
1481de53e5efSHong Zhang        B->ops->mult            = MatMult_SeqSBAIJ_7;
148249b5e25fSSatish Balay        B->ops->multadd         = MatMultAdd_SeqSBAIJ_7;
148349b5e25fSSatish Balay        break;
148449b5e25fSSatish Balay      }
148549b5e25fSSatish Balay    }
148649b5e25fSSatish Balay 
148749b5e25fSSatish Balay    b->mbs = mbs;
14884afc71dfSHong Zhang    b->nbs = mbs;
148913f74950SBarry Smith    ierr   = PetscMalloc((mbs+1)*sizeof(PetscInt),&b->imax);CHKERRQ(ierr);
149049b5e25fSSatish Balay    if (!nnz) {
1491435da068SBarry Smith      if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
149249b5e25fSSatish Balay      else if (nz <= 0)        nz = 1;
149349b5e25fSSatish Balay      for (i=0; i<mbs; i++) {
14948cef66ccSHong Zhang        b->imax[i] = nz;
149549b5e25fSSatish Balay      }
1496153ea458SHong Zhang      nz = nz*mbs; /* total nz */
149749b5e25fSSatish Balay    } else {
149849b5e25fSSatish Balay      nz = 0;
14998cef66ccSHong Zhang      for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
150049b5e25fSSatish Balay    }
15016c6c5352SBarry Smith    /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
150249b5e25fSSatish Balay 
150349b5e25fSSatish Balay    /* allocate the matrix space */
150413f74950SBarry Smith    len  = nz*sizeof(PetscInt) + nz*bs2*sizeof(MatScalar) + (B->m+1)*sizeof(PetscInt);
150582502324SSatish Balay    ierr = PetscMalloc(len,&b->a);CHKERRQ(ierr);
15066c6c5352SBarry Smith    ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
150713f74950SBarry Smith    b->j = (PetscInt*)(b->a + nz*bs2);
150813f74950SBarry Smith    ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
15096c6c5352SBarry Smith    b->i = b->j + nz;
151049b5e25fSSatish Balay    b->singlemalloc = PETSC_TRUE;
151149b5e25fSSatish Balay 
151249b5e25fSSatish Balay    /* pointer to beginning of each row */
151349b5e25fSSatish Balay    b->i[0] = 0;
151449b5e25fSSatish Balay    for (i=1; i<mbs+1; i++) {
151549b5e25fSSatish Balay      b->i[i] = b->i[i-1] + b->imax[i-1];
151649b5e25fSSatish Balay    }
151749b5e25fSSatish Balay 
151849b5e25fSSatish Balay    /* b->ilen will count nonzeros in each block row so far. */
151913f74950SBarry Smith    ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&b->ilen);CHKERRQ(ierr);
152052e6d16bSBarry Smith    ierr = PetscLogObjectMemory(B,len+2*(mbs+1)*sizeof(PetscInt)+sizeof(struct _p_Mat)+sizeof(Mat_SeqSBAIJ));CHKERRQ(ierr);
152149b5e25fSSatish Balay    for (i=0; i<mbs; i++) { b->ilen[i] = 0;}
152249b5e25fSSatish Balay 
1523521d7252SBarry Smith    B->bs               = bs;
152449b5e25fSSatish Balay    b->bs2              = bs2;
15256c6c5352SBarry Smith    b->nz             = 0;
15266c6c5352SBarry Smith    b->maxnz          = nz*bs2;
1527153ea458SHong Zhang 
152816cdd363SHong Zhang    b->inew             = 0;
152916cdd363SHong Zhang    b->jnew             = 0;
153016cdd363SHong Zhang    b->anew             = 0;
153116cdd363SHong Zhang    b->a2anew           = 0;
15321a3463dfSHong Zhang    b->permute          = PETSC_FALSE;
1533c464158bSHong Zhang    PetscFunctionReturn(0);
1534c464158bSHong Zhang  }
1535a23d5eceSKris Buschelman  EXTERN_C_END
1536153ea458SHong Zhang 
1537d769727bSBarry Smith  EXTERN_C_BEGIN
1538ceb03754SKris Buschelman  EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat,const MatType,MatReuse,Mat*);
1539ceb03754SKris Buschelman  EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat,const MatType,MatReuse,Mat*);
1540d769727bSBarry Smith  EXTERN_C_END
1541d769727bSBarry Smith 
15420bad9183SKris Buschelman  /*MC
1543fafad747SKris Buschelman     MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
15440bad9183SKris Buschelman     based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
15450bad9183SKris Buschelman 
15460bad9183SKris Buschelman     Options Database Keys:
15470bad9183SKris Buschelman  . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
15480bad9183SKris Buschelman 
15490bad9183SKris Buschelman    Level: beginner
15500bad9183SKris Buschelman 
15510bad9183SKris Buschelman  .seealso: MatCreateSeqSBAIJ
15520bad9183SKris Buschelman  M*/
15530bad9183SKris Buschelman 
1554a23d5eceSKris Buschelman  EXTERN_C_BEGIN
1555a23d5eceSKris Buschelman  #undef __FUNCT__
1556a23d5eceSKris Buschelman  #define __FUNCT__ "MatCreate_SeqSBAIJ"
1557dfbe8321SBarry Smith  PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1558a23d5eceSKris Buschelman  {
1559a23d5eceSKris Buschelman    Mat_SeqSBAIJ   *b;
1560dfbe8321SBarry Smith    PetscErrorCode ierr;
156113f74950SBarry Smith    PetscMPIInt    size;
1562*941593c8SHong Zhang    PetscTruth     flg;
1563a23d5eceSKris Buschelman 
1564a23d5eceSKris Buschelman    PetscFunctionBegin;
1565a23d5eceSKris Buschelman    ierr = MPI_Comm_size(B->comm,&size);CHKERRQ(ierr);
1566a23d5eceSKris Buschelman    if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1567a23d5eceSKris Buschelman    B->m = B->M = PetscMax(B->m,B->M);
1568a23d5eceSKris Buschelman    B->n = B->N = PetscMax(B->n,B->N);
1569a23d5eceSKris Buschelman 
1570a23d5eceSKris Buschelman    ierr    = PetscNew(Mat_SeqSBAIJ,&b);CHKERRQ(ierr);
1571a23d5eceSKris Buschelman    B->data = (void*)b;
1572a23d5eceSKris Buschelman    ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1573a23d5eceSKris Buschelman    B->ops->destroy     = MatDestroy_SeqSBAIJ;
1574a23d5eceSKris Buschelman    B->ops->view        = MatView_SeqSBAIJ;
1575a23d5eceSKris Buschelman    B->factor           = 0;
1576a23d5eceSKris Buschelman    B->lupivotthreshold = 1.0;
1577a23d5eceSKris Buschelman    B->mapping          = 0;
1578a23d5eceSKris Buschelman    b->row              = 0;
1579a23d5eceSKris Buschelman    b->icol             = 0;
1580a23d5eceSKris Buschelman    b->reallocs         = 0;
1581a23d5eceSKris Buschelman    b->saved_values     = 0;
1582a23d5eceSKris Buschelman 
1583a23d5eceSKris Buschelman    ierr = PetscMapCreateMPI(B->comm,B->m,B->M,&B->rmap);CHKERRQ(ierr);
1584a23d5eceSKris Buschelman    ierr = PetscMapCreateMPI(B->comm,B->n,B->N,&B->cmap);CHKERRQ(ierr);
1585a23d5eceSKris Buschelman 
1586a23d5eceSKris Buschelman    b->sorted           = PETSC_FALSE;
1587a23d5eceSKris Buschelman    b->roworiented      = PETSC_TRUE;
1588a23d5eceSKris Buschelman    b->nonew            = 0;
1589a23d5eceSKris Buschelman    b->diag             = 0;
1590a23d5eceSKris Buschelman    b->solve_work       = 0;
1591a23d5eceSKris Buschelman    b->mult_work        = 0;
1592a23d5eceSKris Buschelman    B->spptr            = 0;
1593a23d5eceSKris Buschelman    b->keepzeroedrows   = PETSC_FALSE;
1594a23d5eceSKris Buschelman   b->xtoy             = 0;
1595a23d5eceSKris Buschelman   b->XtoY             = 0;
1596a23d5eceSKris Buschelman 
1597a23d5eceSKris Buschelman   b->inew             = 0;
1598a23d5eceSKris Buschelman   b->jnew             = 0;
1599a23d5eceSKris Buschelman   b->anew             = 0;
1600a23d5eceSKris Buschelman   b->a2anew           = 0;
1601a23d5eceSKris Buschelman   b->permute          = PETSC_FALSE;
1602a23d5eceSKris Buschelman 
1603*941593c8SHong Zhang   b->ignore_ltriangular = PETSC_FALSE;
1604*941593c8SHong Zhang   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_ignore_lower_triangular",&flg);CHKERRQ(ierr);
1605*941593c8SHong Zhang   if (flg) b->ignore_ltriangular = PETSC_TRUE;
1606*941593c8SHong Zhang 
1607a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C",
1608a23d5eceSKris Buschelman                                      "MatStoreValues_SeqSBAIJ",
1609a23d5eceSKris Buschelman                                      MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1610a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C",
1611a23d5eceSKris Buschelman                                      "MatRetrieveValues_SeqSBAIJ",
1612a23d5eceSKris Buschelman                                      (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1613a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",
1614a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetColumnIndices_SeqSBAIJ",
1615a23d5eceSKris Buschelman                                      MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
16164e5e7fe4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",
16174e5e7fe4SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqAIJ",
16184e5e7fe4SHong Zhang                                       MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1619a0e1a404SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",
1620a0e1a404SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqBAIJ",
1621a0e1a404SHong Zhang                                       MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1622a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",
1623a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetPreallocation_SeqSBAIJ",
1624a23d5eceSKris Buschelman                                      MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
162523ce1328SBarry Smith 
162623ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
162723ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
162823ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
162923ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
1630a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1631a23d5eceSKris Buschelman }
1632a23d5eceSKris Buschelman EXTERN_C_END
1633a23d5eceSKris Buschelman 
1634a23d5eceSKris Buschelman #undef __FUNCT__
1635a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1636a23d5eceSKris Buschelman /*@C
1637a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1638a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1639a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1640a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1641a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1642a23d5eceSKris Buschelman 
1643a23d5eceSKris Buschelman    Collective on Mat
1644a23d5eceSKris Buschelman 
1645a23d5eceSKris Buschelman    Input Parameters:
1646a23d5eceSKris Buschelman +  A - the symmetric matrix
1647a23d5eceSKris Buschelman .  bs - size of block
1648a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1649a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
1650a23d5eceSKris Buschelman          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1651a23d5eceSKris Buschelman 
1652a23d5eceSKris Buschelman    Options Database Keys:
1653a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1654a23d5eceSKris Buschelman                      block calculations (much slower)
1655a23d5eceSKris Buschelman .    -mat_block_size - size of the blocks to use
1656a23d5eceSKris Buschelman 
1657a23d5eceSKris Buschelman    Level: intermediate
1658a23d5eceSKris Buschelman 
1659a23d5eceSKris Buschelman    Notes:
1660a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
1661a23d5eceSKris Buschelman    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1662a23d5eceSKris Buschelman    allocation.  For additional details, see the users manual chapter on
1663a23d5eceSKris Buschelman    matrices.
1664a23d5eceSKris Buschelman 
166549a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
166649a6f317SBarry Smith 
166749a6f317SBarry Smith 
1668a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1669a23d5eceSKris Buschelman @*/
167013f74950SBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
167113f74950SBarry Smith {
167213f74950SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt,PetscInt,const PetscInt[]);
1673a23d5eceSKris Buschelman 
1674a23d5eceSKris Buschelman   PetscFunctionBegin;
1675a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1676a23d5eceSKris Buschelman   if (f) {
1677a23d5eceSKris Buschelman     ierr = (*f)(B,bs,nz,nnz);CHKERRQ(ierr);
1678a23d5eceSKris Buschelman   }
1679a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1680a23d5eceSKris Buschelman }
168149b5e25fSSatish Balay 
16824a2ae208SSatish Balay #undef __FUNCT__
16834a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
1684c464158bSHong Zhang /*@C
1685c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
1686c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
1687c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
1688c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
1689c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
169049b5e25fSSatish Balay 
1691c464158bSHong Zhang    Collective on MPI_Comm
1692c464158bSHong Zhang 
1693c464158bSHong Zhang    Input Parameters:
1694c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
1695c464158bSHong Zhang .  bs - size of block
1696c464158bSHong Zhang .  m - number of rows, or number of columns
1697c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
1698744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
1699744e8345SSatish Balay          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1700c464158bSHong Zhang 
1701c464158bSHong Zhang    Output Parameter:
1702c464158bSHong Zhang .  A - the symmetric matrix
1703c464158bSHong Zhang 
1704c464158bSHong Zhang    Options Database Keys:
1705c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
1706c464158bSHong Zhang                      block calculations (much slower)
1707c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
1708c464158bSHong Zhang 
1709c464158bSHong Zhang    Level: intermediate
1710c464158bSHong Zhang 
1711c464158bSHong Zhang    Notes:
1712c464158bSHong Zhang 
1713c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
1714c464158bSHong Zhang    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1715c464158bSHong Zhang    allocation.  For additional details, see the users manual chapter on
1716c464158bSHong Zhang    matrices.
1717c464158bSHong Zhang 
171849a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
171949a6f317SBarry Smith 
1720c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1721c464158bSHong Zhang @*/
172213f74950SBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
1723c464158bSHong Zhang {
1724dfbe8321SBarry Smith   PetscErrorCode ierr;
1725c464158bSHong Zhang 
1726c464158bSHong Zhang   PetscFunctionBegin;
1727c464158bSHong Zhang   ierr = MatCreate(comm,m,n,m,n,A);CHKERRQ(ierr);
1728c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
1729273d9f13SBarry Smith   ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,nnz);CHKERRQ(ierr);
173049b5e25fSSatish Balay   PetscFunctionReturn(0);
173149b5e25fSSatish Balay }
173249b5e25fSSatish Balay 
17334a2ae208SSatish Balay #undef __FUNCT__
17344a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
1735dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
173649b5e25fSSatish Balay {
173749b5e25fSSatish Balay   Mat            C;
173849b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
17396849ba73SBarry Smith   PetscErrorCode ierr;
174013f74950SBarry Smith   PetscInt       i,len,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
174149b5e25fSSatish Balay 
174249b5e25fSSatish Balay   PetscFunctionBegin;
174329bbc08cSBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_ERR_PLIB,"Corrupt matrix");
174449b5e25fSSatish Balay 
174549b5e25fSSatish Balay   *B = 0;
1746692f9cbeSHong Zhang   ierr = MatCreate(A->comm,A->m,A->n,A->m,A->n,&C);CHKERRQ(ierr);
1747be5d1d56SKris Buschelman   ierr = MatSetType(C,A->type_name);CHKERRQ(ierr);
17481d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
1749692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
1750692f9cbeSHong Zhang 
1751273d9f13SBarry Smith   C->preallocated   = PETSC_TRUE;
175249b5e25fSSatish Balay   C->factor         = A->factor;
175349b5e25fSSatish Balay   c->row            = 0;
175449b5e25fSSatish Balay   c->icol           = 0;
175549b5e25fSSatish Balay   c->saved_values   = 0;
175649b5e25fSSatish Balay   c->keepzeroedrows = a->keepzeroedrows;
175749b5e25fSSatish Balay   C->assembled      = PETSC_TRUE;
175849b5e25fSSatish Balay 
175982327fa8SHong Zhang   C->M    = A->M;
176082327fa8SHong Zhang   C->N    = A->N;
1761521d7252SBarry Smith   C->bs   = A->bs;
176249b5e25fSSatish Balay   c->bs2  = a->bs2;
176349b5e25fSSatish Balay   c->mbs  = a->mbs;
176449b5e25fSSatish Balay   c->nbs  = a->nbs;
176549b5e25fSSatish Balay 
176613f74950SBarry Smith   ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->imax);CHKERRQ(ierr);
176713f74950SBarry Smith   ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->ilen);CHKERRQ(ierr);
176849b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
176949b5e25fSSatish Balay     c->imax[i] = a->imax[i];
177049b5e25fSSatish Balay     c->ilen[i] = a->ilen[i];
177149b5e25fSSatish Balay   }
177249b5e25fSSatish Balay 
177349b5e25fSSatish Balay   /* allocate the matrix space */
177449b5e25fSSatish Balay   c->singlemalloc = PETSC_TRUE;
177513f74950SBarry Smith   len  = (mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt));
177682502324SSatish Balay   ierr = PetscMalloc(len,&c->a);CHKERRQ(ierr);
177713f74950SBarry Smith   c->j = (PetscInt*)(c->a + nz*bs2);
177849b5e25fSSatish Balay   c->i = c->j + nz;
177913f74950SBarry Smith   ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
178049b5e25fSSatish Balay   if (mbs > 0) {
178113f74950SBarry Smith     ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
178249b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
178349b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
178449b5e25fSSatish Balay     } else {
178549b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
178649b5e25fSSatish Balay     }
178749b5e25fSSatish Balay   }
178849b5e25fSSatish Balay 
178952e6d16bSBarry Smith   ierr = PetscLogObjectMemory(C,len+2*(mbs+1)*sizeof(PetscInt)+sizeof(struct _p_Mat)+sizeof(Mat_SeqSBAIJ));CHKERRQ(ierr);
179049b5e25fSSatish Balay   c->sorted      = a->sorted;
179149b5e25fSSatish Balay   c->roworiented = a->roworiented;
179249b5e25fSSatish Balay   c->nonew       = a->nonew;
179349b5e25fSSatish Balay 
179449b5e25fSSatish Balay   if (a->diag) {
179513f74950SBarry Smith     ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->diag);CHKERRQ(ierr);
179652e6d16bSBarry Smith     ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
179749b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
179849b5e25fSSatish Balay       c->diag[i] = a->diag[i];
179949b5e25fSSatish Balay     }
180049b5e25fSSatish Balay   } else c->diag        = 0;
18016c6c5352SBarry Smith   c->nz               = a->nz;
18026c6c5352SBarry Smith   c->maxnz            = a->maxnz;
180349b5e25fSSatish Balay   c->solve_work         = 0;
180449b5e25fSSatish Balay   c->mult_work          = 0;
180549b5e25fSSatish Balay   *B = C;
1806b0a32e0cSBarry Smith   ierr = PetscFListDuplicate(A->qlist,&C->qlist);CHKERRQ(ierr);
180749b5e25fSSatish Balay   PetscFunctionReturn(0);
180849b5e25fSSatish Balay }
180949b5e25fSSatish Balay 
18104a2ae208SSatish Balay #undef __FUNCT__
18114a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqSBAIJ"
1812dfbe8321SBarry Smith PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer viewer,const MatType type,Mat *A)
181349b5e25fSSatish Balay {
181449b5e25fSSatish Balay   Mat_SeqSBAIJ   *a;
181549b5e25fSSatish Balay   Mat            B;
18166849ba73SBarry Smith   PetscErrorCode ierr;
181713f74950SBarry Smith   int            fd;
181813f74950SBarry Smith   PetscMPIInt    size;
181913f74950SBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
182013f74950SBarry Smith   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
182113f74950SBarry Smith   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows;
182213f74950SBarry Smith   PetscInt       *masked,nmask,tmp,bs2,ishift;
182387828ca2SBarry Smith   PetscScalar    *aa;
182449b5e25fSSatish Balay   MPI_Comm       comm = ((PetscObject)viewer)->comm;
182549b5e25fSSatish Balay 
182649b5e25fSSatish Balay   PetscFunctionBegin;
1827b0a32e0cSBarry Smith   ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
182849b5e25fSSatish Balay   bs2  = bs*bs;
182949b5e25fSSatish Balay 
183049b5e25fSSatish Balay   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
183129bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor");
1832b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
183349b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
1834552e946dSBarry Smith   if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
183549b5e25fSSatish Balay   M = header[1]; N = header[2]; nz = header[3];
183649b5e25fSSatish Balay 
183749b5e25fSSatish Balay   if (header[3] < 0) {
183829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
183949b5e25fSSatish Balay   }
184049b5e25fSSatish Balay 
184129bbc08cSBarry Smith   if (M != N) SETERRQ(PETSC_ERR_SUP,"Can only do square matrices");
184249b5e25fSSatish Balay 
184349b5e25fSSatish Balay   /*
184449b5e25fSSatish Balay      This code adds extra rows to make sure the number of rows is
184549b5e25fSSatish Balay     divisible by the blocksize
184649b5e25fSSatish Balay   */
184749b5e25fSSatish Balay   mbs        = M/bs;
184849b5e25fSSatish Balay   extra_rows = bs - M + bs*(mbs);
184949b5e25fSSatish Balay   if (extra_rows == bs) extra_rows = 0;
185049b5e25fSSatish Balay   else                  mbs++;
185149b5e25fSSatish Balay   if (extra_rows) {
1852b0a32e0cSBarry Smith     PetscLogInfo(0,"MatLoad_SeqSBAIJ:Padding loaded matrix to match blocksize\n");
185349b5e25fSSatish Balay   }
185449b5e25fSSatish Balay 
185549b5e25fSSatish Balay   /* read in row lengths */
185613f74950SBarry Smith   ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
185749b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
185849b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
185949b5e25fSSatish Balay 
186049b5e25fSSatish Balay   /* read in column indices */
186113f74950SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr);
186249b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
186349b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
186449b5e25fSSatish Balay 
186549b5e25fSSatish Balay   /* loop over row lengths determining block row lengths */
186613f74950SBarry Smith   ierr     = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr);
186713f74950SBarry Smith   ierr     = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr);
186813f74950SBarry Smith   ierr     = PetscMalloc(2*mbs*sizeof(PetscInt),&mask);CHKERRQ(ierr);
186913f74950SBarry Smith   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
187049b5e25fSSatish Balay   masked   = mask + mbs;
187149b5e25fSSatish Balay   rowcount = 0; nzcount = 0;
187249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
187349b5e25fSSatish Balay     nmask = 0;
187449b5e25fSSatish Balay     for (j=0; j<bs; j++) {
187549b5e25fSSatish Balay       kmax = rowlengths[rowcount];
187649b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
18772d703238SHong Zhang         tmp = jj[nzcount++]/bs;   /* block col. index */
187803630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
187949b5e25fSSatish Balay       }
188049b5e25fSSatish Balay       rowcount++;
188149b5e25fSSatish Balay     }
1882574b2666SHong Zhang     s_browlengths[i] += nmask;
1883574b2666SHong Zhang 
188449b5e25fSSatish Balay     /* zero out the mask elements we set */
188549b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
188649b5e25fSSatish Balay   }
188749b5e25fSSatish Balay 
188849b5e25fSSatish Balay   /* create our matrix */
18899abb65ffSKris Buschelman   ierr = MatCreate(comm,M+extra_rows,N+extra_rows,M+extra_rows,N+extra_rows,&B);CHKERRQ(ierr);
18909abb65ffSKris Buschelman   ierr = MatSetType(B,type);CHKERRQ(ierr);
189178473fd7SKris Buschelman   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,s_browlengths);CHKERRQ(ierr);
189249b5e25fSSatish Balay   a = (Mat_SeqSBAIJ*)B->data;
189349b5e25fSSatish Balay 
189449b5e25fSSatish Balay   /* set matrix "i" values */
189549b5e25fSSatish Balay   a->i[0] = 0;
189649b5e25fSSatish Balay   for (i=1; i<= mbs; i++) {
1897574b2666SHong Zhang     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
1898574b2666SHong Zhang     a->ilen[i-1] = s_browlengths[i-1];
189949b5e25fSSatish Balay   }
19006c6c5352SBarry Smith   a->nz = a->i[mbs];
190149b5e25fSSatish Balay 
190249b5e25fSSatish Balay   /* read in nonzero values */
190387828ca2SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr);
190449b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
190549b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
190649b5e25fSSatish Balay 
190749b5e25fSSatish Balay   /* set "a" and "j" values into matrix */
190849b5e25fSSatish Balay   nzcount = 0; jcount = 0;
190949b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
191049b5e25fSSatish Balay     nzcountb = nzcount;
191149b5e25fSSatish Balay     nmask    = 0;
191249b5e25fSSatish Balay     for (j=0; j<bs; j++) {
191349b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
191449b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
19152d703238SHong Zhang         tmp = jj[nzcount++]/bs; /* block col. index */
191603630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
19172d703238SHong Zhang       }
19182d703238SHong Zhang     }
19192d703238SHong Zhang     /* sort the masked values */
19202d703238SHong Zhang     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
19212d703238SHong Zhang 
19222d703238SHong Zhang     /* set "j" values into matrix */
19232d703238SHong Zhang     maskcount = 1;
19242d703238SHong Zhang     for (j=0; j<nmask; j++) {
192549b5e25fSSatish Balay       a->j[jcount++]  = masked[j];
192649b5e25fSSatish Balay       mask[masked[j]] = maskcount++;
192749b5e25fSSatish Balay     }
1928574b2666SHong Zhang 
192949b5e25fSSatish Balay     /* set "a" values into matrix */
193049b5e25fSSatish Balay     ishift = bs2*a->i[i];
193149b5e25fSSatish Balay     for (j=0; j<bs; j++) {
193249b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
193349b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
1934574b2666SHong Zhang         tmp       = jj[nzcountb]/bs ; /* block col. index */
1935574b2666SHong Zhang         if (tmp >= i){
193649b5e25fSSatish Balay           block     = mask[tmp] - 1;
193749b5e25fSSatish Balay           point     = jj[nzcountb] - bs*tmp;
193849b5e25fSSatish Balay           idx       = ishift + bs2*block + j + bs*point;
1939574b2666SHong Zhang           a->a[idx] = aa[nzcountb];
1940574b2666SHong Zhang         }
1941574b2666SHong Zhang         nzcountb++;
194249b5e25fSSatish Balay       }
194349b5e25fSSatish Balay     }
194449b5e25fSSatish Balay     /* zero out the mask elements we set */
194549b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
194649b5e25fSSatish Balay   }
19476c6c5352SBarry Smith   if (jcount != a->nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
194849b5e25fSSatish Balay 
194949b5e25fSSatish Balay   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
1950574b2666SHong Zhang   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
195149b5e25fSSatish Balay   ierr = PetscFree(aa);CHKERRQ(ierr);
195249b5e25fSSatish Balay   ierr = PetscFree(jj);CHKERRQ(ierr);
195349b5e25fSSatish Balay   ierr = PetscFree(mask);CHKERRQ(ierr);
195449b5e25fSSatish Balay 
19559abb65ffSKris Buschelman   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19569abb65ffSKris Buschelman   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
195749b5e25fSSatish Balay   ierr = MatView_Private(B);CHKERRQ(ierr);
19589abb65ffSKris Buschelman   *A = B;
195949b5e25fSSatish Balay   PetscFunctionReturn(0);
196049b5e25fSSatish Balay }
1961574b2666SHong Zhang 
1962d06b337dSHong Zhang #undef __FUNCT__
1963d06b337dSHong Zhang #define __FUNCT__ "MatRelax_SeqSBAIJ"
196413f74950SBarry Smith PetscErrorCode MatRelax_SeqSBAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1965d06b337dSHong Zhang {
1966d06b337dSHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
1967d06b337dSHong Zhang   MatScalar      *aa=a->a,*v,*v1;
1968d06b337dSHong Zhang   PetscScalar    *x,*b,*t,sum,d;
19696849ba73SBarry Smith   PetscErrorCode ierr;
1970521d7252SBarry Smith   PetscInt       m=a->mbs,bs=A->bs,*ai=a->i,*aj=a->j;
1971521d7252SBarry Smith   PetscInt       nz,nz1,*vj,*vj1,i;
1972d06b337dSHong Zhang 
1973d06b337dSHong Zhang   PetscFunctionBegin;
1974b965ef7fSBarry Smith   its = its*lits;
197577431f27SBarry Smith   if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
1976d06b337dSHong Zhang 
1977d06b337dSHong Zhang   if (bs > 1)
1978d06b337dSHong Zhang     SETERRQ(PETSC_ERR_SUP,"SSOR for block size > 1 is not yet implemented");
1979d06b337dSHong Zhang 
19801ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1981d06b337dSHong Zhang   if (xx != bb) {
19821ebc52fbSHong Zhang     ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1983d06b337dSHong Zhang   } else {
1984d06b337dSHong Zhang     b = x;
1985d06b337dSHong Zhang   }
1986d06b337dSHong Zhang 
1987d06b337dSHong Zhang   ierr = PetscMalloc(m*sizeof(PetscScalar),&t);CHKERRQ(ierr);
1988d06b337dSHong Zhang 
1989d06b337dSHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
19902798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
1991d06b337dSHong Zhang       for (i=0; i<m; i++)
1992d06b337dSHong Zhang         t[i] = b[i];
1993d06b337dSHong Zhang 
1994d06b337dSHong Zhang       for (i=0; i<m; i++){
199544706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
1996d06b337dSHong Zhang         v  = aa + ai[i] + 1;
1997d06b337dSHong Zhang         vj = aj + ai[i] + 1;
1998d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
1999d06b337dSHong Zhang         x[i] = omega*t[i]/d;
2000d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++); /* update rhs */
200144706875SHong Zhang         PetscLogFlops(2*nz-1);
2002d06b337dSHong Zhang       }
2003d06b337dSHong Zhang     }
2004d06b337dSHong Zhang 
20052798e883SHong Zhang     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
2006d06b337dSHong Zhang       for (i=0; i<m; i++)
2007d06b337dSHong Zhang         t[i] = b[i];
2008d06b337dSHong Zhang 
2009d06b337dSHong Zhang       for (i=0; i<m-1; i++){  /* update rhs */
2010d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2011d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2012d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2013d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++);
201444706875SHong Zhang         PetscLogFlops(2*nz-1);
2015d06b337dSHong Zhang       }
2016d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2017d06b337dSHong Zhang         d  = *(aa + ai[i]);
2018d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2019d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2020d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2021d06b337dSHong Zhang         sum = t[i];
2022d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
202344706875SHong Zhang         PetscLogFlops(2*nz-1);
2024d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2025d06b337dSHong Zhang       }
2026d06b337dSHong Zhang     }
2027d06b337dSHong Zhang     its--;
2028d06b337dSHong Zhang   }
2029d06b337dSHong Zhang 
2030d06b337dSHong Zhang   while (its--) {
203144706875SHong Zhang     /*
203244706875SHong Zhang        forward sweep:
203344706875SHong Zhang        for i=0,...,m-1:
203444706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
203544706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
203644706875SHong Zhang          b      = b - x[i]*U^T(i,:);
2037d06b337dSHong Zhang 
203844706875SHong Zhang     */
20392798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
2040d06b337dSHong Zhang       for (i=0; i<m; i++)
2041d06b337dSHong Zhang         t[i] = b[i];
2042d06b337dSHong Zhang 
2043d06b337dSHong Zhang       for (i=0; i<m; i++){
204444706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
2045d06b337dSHong Zhang         v  = aa + ai[i] + 1; v1=v;
2046d06b337dSHong Zhang         vj = aj + ai[i] + 1; vj1=vj;
2047d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1; nz1=nz;
2048d06b337dSHong Zhang         sum = t[i];
2049d06b337dSHong Zhang         while (nz1--) sum -= (*v1++)*x[*vj1++];
2050d06b337dSHong Zhang         x[i] = (1-omega)*x[i] + omega*sum/d;
2051d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++);
205244706875SHong Zhang         PetscLogFlops(4*nz-2);
2053d06b337dSHong Zhang       }
2054d06b337dSHong Zhang     }
2055d06b337dSHong Zhang 
20562798e883SHong Zhang   if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
205744706875SHong Zhang       /*
205844706875SHong Zhang        backward sweep:
205944706875SHong Zhang        b = b - x[i]*U^T(i,:), i=0,...,n-2
206044706875SHong Zhang        for i=m-1,...,0:
206144706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
206244706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
206344706875SHong Zhang       */
2064d06b337dSHong Zhang       for (i=0; i<m; i++)
2065d06b337dSHong Zhang         t[i] = b[i];
2066d06b337dSHong Zhang 
2067d06b337dSHong Zhang       for (i=0; i<m-1; i++){  /* update rhs */
2068d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2069d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2070d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2071d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++);
207244706875SHong Zhang         PetscLogFlops(2*nz-1);
2073d06b337dSHong Zhang       }
2074d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2075d06b337dSHong Zhang         d  = *(aa + ai[i]);
2076d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2077d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2078d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2079d06b337dSHong Zhang         sum = t[i];
2080d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
208144706875SHong Zhang         PetscLogFlops(2*nz-1);
2082d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2083d06b337dSHong Zhang       }
2084d06b337dSHong Zhang     }
2085d06b337dSHong Zhang   }
2086d06b337dSHong Zhang 
2087d06b337dSHong Zhang   ierr = PetscFree(t);CHKERRQ(ierr);
20881ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2089d06b337dSHong Zhang   if (bb != xx) {
20901ebc52fbSHong Zhang     ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
2091d06b337dSHong Zhang   }
2092d06b337dSHong Zhang   PetscFunctionReturn(0);
2093d06b337dSHong Zhang }
2094d06b337dSHong Zhang 
2095d06b337dSHong Zhang 
2096d06b337dSHong Zhang 
2097d06b337dSHong Zhang 
209849b5e25fSSatish Balay 
209949b5e25fSSatish Balay 
2100