Actual source code: ex81.c

petsc-3.5.4 2015-05-23
Report Typos and Errors
  2: static char help[] = "Reads in a PETSc binary matrix and saves in Harwell-Boeing format.\n\
  3:   -fout <output_file> : file to load.\n\
  4:   -fin <input_file> : For example see $PETSC_DIR/share/petsc/datafiles/matrices\n\n";

  6: /*
  7:   Include the private file (not included by most applications) so we have direct
  8:   access to the matrix data structure.

 10:   This code is buggy! What is it doing here?
 11: */
 12: #include <../src/mat/impls/aij/seq/aij.h>

 16: int main(int argc,char **args)
 17: {
 19:   PetscInt       n,m,i,*ai,*aj,nz;
 20:   PetscMPIInt    size;
 21:   Mat            A;
 22:   Vec            x;
 23:   char           bfile[PETSC_MAX_PATH_LEN],hbfile[PETSC_MAX_PATH_LEN];
 24:   PetscViewer    fd;
 25:   Mat_SeqAIJ     *a;
 26:   PetscScalar    *aa,*xx;
 27:   FILE           *file;
 28:   char           head[81];

 30:   PetscInitialize(&argc,&args,(char*)0,help);

 32: #if defined(PETSC_USE_COMPLEX)
 33:   SETERRQ(PETSC_COMM_WORLD,1,"This example does not work with complex numbers");
 34: #endif
 35:   MPI_Comm_size(PETSC_COMM_WORLD,&size);
 36:   if (size > 1) SETERRQ(PETSC_COMM_WORLD,1,"Only runs on one processor");

 38:   PetscOptionsGetString(NULL,"-fin",bfile,PETSC_MAX_PATH_LEN,NULL);
 39:   PetscOptionsGetString(NULL,"-fout",hbfile,PETSC_MAX_PATH_LEN,NULL);

 41:   /* Read matrix and RHS */
 42:   PetscViewerBinaryOpen(PETSC_COMM_WORLD,bfile,FILE_MODE_READ,&fd);
 43:   MatCreate(PETSC_COMM_WORLD,&A);
 44:   MatSetType(A,MATSEQAIJ);
 45:   MatLoad(A,fd);
 46:   VecCreate(PETSC_COMM_WORLD,&x);
 47:   VecLoad(x,fd);
 48:   PetscViewerDestroy(&fd);

 50:   /* Format is in column storage so we print transpose matrix */
 51:   MatTranspose(A,MAT_REUSE_MATRIX,&A);

 53:   m = A->rmap->n;
 54:   n = A->cmap->n;
 55:   if (n != m) SETERRQ(PETSC_COMM_SELF,1,"Only for square matrices");

 57:   /* charrage returns \n may not belong below
 58:     depends on what 80 character fixed format means to Fortran */

 60:   file = fopen(hbfile,"w"); if (!file) SETERRQ(PETSC_COMM_SELF,1,"Cannot open HB file");
 61:   sprintf(head,"%-72s%-8s\n","Title","Key");
 62:   fprintf(file,head);
 63:   a  = (Mat_SeqAIJ*)A->data;
 64:   aa = a->a;
 65:   ai = a->i;
 66:   aj = a->j;
 67:   nz = a->nz;


 70:   sprintf(head,"%14d%14d%14d%14d%14d%10s\n",3*m+1,m+1,nz,nz," ");
 71:   fprintf(file,head);
 72:   sprintf(head,"RUA%14d%14d%14d%14d%13s\n",m,m,nz," ");
 73:   fprintf(file,head);

 75:   fprintf(file,"Formats I don't know\n");

 77:   for (i=0; i<m+1; i++) {
 78:     fprintf(file,"%10d%70s\n",ai[i]," ");
 79:   }
 80:   for (i=0; i<nz; i++) {
 81:     fprintf(file,"%10d%70s\n",aj[i]," ");
 82:   }

 84:   for (i=0; i<nz; i++) {
 85:     fprintf(file,"%16.14e,%64s\n",aa[i]," ");
 86:   }

 88:   /* print the vector to the file */
 89:   VecGetArray(x,&xx);
 90:   for (i=0; i<m; i++) {
 91:     fprintf(file,"%16.14e%64s\n",xx[i]," ");
 92:   }
 93:   VecRestoreArray(x,&xx);

 95:   fclose(file);
 96:   MatDestroy(&A);
 97:   VecDestroy(&x);

 99:   PetscFinalize();
100:   return 0;
101: }