Date: 2022-03-31T04:47:28+00:00

Actual source code: daview.c

      2: /*
      3:   Code for manipulating distributed regular arrays in parallel.
      4: */

      6: #include <petsc/private/dmdaimpl.h>

      8: #if defined(PETSC_HAVE_MATLAB_ENGINE)
      9: #include <mat.h>   /* MATLAB include file */

     11: PetscErrorCode DMView_DA_Matlab(DM da,PetscViewer viewer)
     12: {
     13:   PetscMPIInt      rank;
     14:   PetscInt         dim,m,n,p,dof,swidth;
     15:   DMDAStencilType  stencil;
     16:   DMBoundaryType   bx,by,bz;
     17:   mxArray          *mx;
     18:   const char       *fnames[] = {"dimension","m","n","p","dof","stencil_width","bx","by","bz","stencil_type"};

     20:   MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);
     21:   if (rank == 0) {
     22:     DMDAGetInfo(da,&dim,&m,&n,&p,0,0,0,&dof,&swidth,&bx,&by,&bz,&stencil);
     23:     mx   = mxCreateStructMatrix(1,1,8,(const char**)fnames);
     25:     mxSetFieldByNumber(mx,0,0,mxCreateDoubleScalar((double)dim));
     26:     mxSetFieldByNumber(mx,0,1,mxCreateDoubleScalar((double)m));
     27:     mxSetFieldByNumber(mx,0,2,mxCreateDoubleScalar((double)n));
     28:     mxSetFieldByNumber(mx,0,3,mxCreateDoubleScalar((double)p));
     29:     mxSetFieldByNumber(mx,0,4,mxCreateDoubleScalar((double)dof));
     30:     mxSetFieldByNumber(mx,0,5,mxCreateDoubleScalar((double)swidth));
     31:     mxSetFieldByNumber(mx,0,6,mxCreateDoubleScalar((double)bx));
     32:     mxSetFieldByNumber(mx,0,7,mxCreateDoubleScalar((double)by));
     33:     mxSetFieldByNumber(mx,0,8,mxCreateDoubleScalar((double)bz));
     34:     mxSetFieldByNumber(mx,0,9,mxCreateDoubleScalar((double)stencil));
     35:     PetscObjectName((PetscObject)da);
     36:     PetscViewerMatlabPutVariable(viewer,((PetscObject)da)->name,mx);
     37:   }
     38:   return 0;
     39: }
     40: #endif

     42: PetscErrorCode DMView_DA_Binary(DM da,PetscViewer viewer)
     43: {
     44:   PetscMPIInt      rank;
     45:   PetscInt         dim,m,n,p,dof,swidth,M,N,P;
     46:   DMDAStencilType  stencil;
     47:   DMBoundaryType   bx,by,bz;
     48:   MPI_Comm         comm;
     49:   PetscBool        coors = PETSC_FALSE;

     51:   PetscObjectGetComm((PetscObject)da,&comm);

     53:   DMDAGetInfo(da,&dim,&m,&n,&p,&M,&N,&P,&dof,&swidth,&bx,&by,&bz,&stencil);
     54:   MPI_Comm_rank(comm,&rank);
     55:   if (rank == 0) {
     56:     PetscViewerBinaryWrite(viewer,&dim,1,PETSC_INT);
     57:     PetscViewerBinaryWrite(viewer,&m,1,PETSC_INT);
     58:     PetscViewerBinaryWrite(viewer,&n,1,PETSC_INT);
     59:     PetscViewerBinaryWrite(viewer,&p,1,PETSC_INT);
     60:     PetscViewerBinaryWrite(viewer,&dof,1,PETSC_INT);
     61:     PetscViewerBinaryWrite(viewer,&swidth,1,PETSC_INT);
     62:     PetscViewerBinaryWrite(viewer,&bx,1,PETSC_ENUM);
     63:     PetscViewerBinaryWrite(viewer,&by,1,PETSC_ENUM);
     64:     PetscViewerBinaryWrite(viewer,&bz,1,PETSC_ENUM);
     65:     PetscViewerBinaryWrite(viewer,&stencil,1,PETSC_ENUM);
     66:     if (da->coordinates) coors = PETSC_TRUE;
     67:     PetscViewerBinaryWrite(viewer,&coors,1,PETSC_BOOL);
     68:   }

     70:   /* save the coordinates if they exist to disk (in the natural ordering) */
     71:   if (da->coordinates) {
     72:     VecView(da->coordinates,viewer);
     73:   }
     74:   return 0;
     75: }

     77: PetscErrorCode DMView_DA_VTK(DM da, PetscViewer viewer)
     78: {
     79:   PetscInt       dim, dof, M = 0, N = 0, P = 0;

     81:   DMDAGetInfo(da, &dim, &M, &N, &P, NULL, NULL, NULL, &dof, NULL, NULL, NULL, NULL, NULL);
     83:   /* Write Header */
     84:   PetscViewerASCIIPrintf(viewer,"# vtk DataFile Version 2.0\n");
     85:   PetscViewerASCIIPrintf(viewer,"Structured Mesh Example\n");
     86:   PetscViewerASCIIPrintf(viewer,"ASCII\n");
     87:   PetscViewerASCIIPrintf(viewer,"DATASET STRUCTURED_GRID\n");
     88:   PetscViewerASCIIPrintf(viewer,"DIMENSIONS %d %d %d\n", M, N, P);
     89:   PetscViewerASCIIPrintf(viewer,"POINTS %d double\n", M*N*P);
     90:   if (da->coordinates) {
     91:     DM  dac;
     92:     Vec natural;

     94:     DMGetCoordinateDM(da, &dac);
     95:     DMDACreateNaturalVector(dac, &natural);
     96:     PetscObjectSetOptionsPrefix((PetscObject) natural, "coor_");
     97:     DMDAGlobalToNaturalBegin(dac, da->coordinates, INSERT_VALUES, natural);
     98:     DMDAGlobalToNaturalEnd(dac, da->coordinates, INSERT_VALUES, natural);
     99:     PetscViewerPushFormat(viewer, PETSC_VIEWER_ASCII_VTK_COORDS_DEPRECATED);
    100:     VecView(natural, viewer);
    101:     PetscViewerPopFormat(viewer);
    102:     VecDestroy(&natural);
    103:   }
    104:   return 0;
    105: }

    107: /*@C
    108:    DMDAGetInfo - Gets information about a given distributed array.

    110:    Not Collective

    112:    Input Parameter:
    113: .  da - the distributed array

    115:    Output Parameters:
    116: +  dim      - dimension of the distributed array (1, 2, or 3)
    117: .  M        - global dimension in first direction of the array
    118: .  N        - global dimension in second direction of the array
    119: .  P        - global dimension in third direction of the array
    120: .  m        - corresponding number of procs in first dimension
    121: .  n        - corresponding number of procs in second dimension
    122: .  p        - corresponding number of procs in third dimension
    123: .  dof      - number of degrees of freedom per node
    124: .  s        - stencil width
    125: .  bx       - type of ghost nodes at boundary in first dimension
    126: .  by       - type of ghost nodes at boundary in second dimension
    127: .  bz       - type of ghost nodes at boundary in third dimension
    128: -  st       - stencil type, either DMDA_STENCIL_STAR or DMDA_STENCIL_BOX

    130:    Level: beginner

    132:    Note:
    133:    Use NULL (NULL_INTEGER in Fortran) in place of any output parameter that is not of interest.

    135: .seealso: DMView(), DMDAGetCorners(), DMDAGetLocalInfo()
    136: @*/
    137: PetscErrorCode  DMDAGetInfo(DM da,PetscInt *dim,PetscInt *M,PetscInt *N,PetscInt *P,PetscInt *m,PetscInt *n,PetscInt *p,PetscInt *dof,PetscInt *s,DMBoundaryType *bx,DMBoundaryType *by,DMBoundaryType *bz,DMDAStencilType *st)
    138: {
    139:   DM_DA *dd = (DM_DA*)da->data;

    142:   if (dim) *dim = da->dim;
    143:   if (M) {
    144:     if (dd->Mo < 0) *M = dd->M;
    145:     else *M = dd->Mo;
    146:   }
    147:   if (N) {
    148:     if (dd->No < 0) *N = dd->N;
    149:     else *N = dd->No;
    150:   }
    151:   if (P) {
    152:     if (dd->Po < 0) *P = dd->P;
    153:     else *P = dd->Po;
    154:   }
    155:   if (m) *m = dd->m;
    156:   if (n) *n = dd->n;
    157:   if (p) *p = dd->p;
    158:   if (dof) *dof = dd->w;
    159:   if (s) *s = dd->s;
    160:   if (bx) *bx = dd->bx;
    161:   if (by) *by = dd->by;
    162:   if (bz) *bz = dd->bz;
    163:   if (st) *st = dd->stencil_type;
    164:   return 0;
    165: }

    167: /*@C
    168:    DMDAGetLocalInfo - Gets information about a given distributed array and this processors location in it

    170:    Not Collective

    172:    Input Parameter:
    173: .  da - the distributed array

    175:    Output Parameters:
    176: .  dainfo - structure containing the information

    178:    Level: beginner

    180:    Notes:
    181:     See DMDALocalInfo for the information that is returned

    183: .seealso: DMDAGetInfo(), DMDAGetCorners(), DMDALocalInfo
    184: @*/
    185: PetscErrorCode  DMDAGetLocalInfo(DM da,DMDALocalInfo *info)
    186: {
    187:   PetscInt w;
    188:   DM_DA    *dd = (DM_DA*)da->data;

    192:   info->da  = da;
    193:   info->dim = da->dim;
    194:   if (dd->Mo < 0) info->mx = dd->M;
    195:   else info->mx = dd->Mo;
    196:   if (dd->No < 0) info->my = dd->N;
    197:   else info->my = dd->No;
    198:   if (dd->Po < 0) info->mz = dd->P;
    199:   else info->mz = dd->Po;
    200:   info->dof = dd->w;
    201:   info->sw  = dd->s;
    202:   info->bx  = dd->bx;
    203:   info->by  = dd->by;
    204:   info->bz  = dd->bz;
    205:   info->st  = dd->stencil_type;

    207:   /* since the xs, xe ... have all been multiplied by the number of degrees
    208:      of freedom per cell, w = dd->w, we divide that out before returning.*/
    209:   w        = dd->w;
    210:   info->xs = dd->xs/w + dd->xo;
    211:   info->xm = (dd->xe - dd->xs)/w;
    212:   /* the y and z have NOT been multiplied by w */
    213:   info->ys = dd->ys + dd->yo;
    214:   info->ym = (dd->ye - dd->ys);
    215:   info->zs = dd->zs + dd->zo;
    216:   info->zm = (dd->ze - dd->zs);

    218:   info->gxs = dd->Xs/w + dd->xo;
    219:   info->gxm = (dd->Xe - dd->Xs)/w;
    220:   /* the y and z have NOT been multiplied by w */
    221:   info->gys = dd->Ys + dd->yo;
    222:   info->gym = (dd->Ye - dd->Ys);
    223:   info->gzs = dd->Zs + dd->zo;
    224:   info->gzm = (dd->Ze - dd->Zs);
    225:   return 0;
    226: }