Actual source code: gr1.c
petsc-3.7.1 2016-05-15
Report Typos and Errors

      2: /*
      3:    Plots vectors obtained with DMDACreate1d()
      4: */

      6: #include <petsc/private/dmdaimpl.h>      /*I  "petscdmda.h"   I*/

     10: /*@
     11:     DMDASetUniformCoordinates - Sets a DMDA coordinates to be a uniform grid

     13:   Collective on DMDA

     15:   Input Parameters:
     16: +  da - the distributed array object
     17: .  xmin,xmax - extremes in the x direction
     18: .  ymin,ymax - extremes in the y direction (value ignored for 1 dimensional problems)
     19: -  zmin,zmax - extremes in the z direction (value ignored for 1 or 2 dimensional problems)

     21:   Level: beginner

     23: .seealso: DMSetCoordinates(), DMGetCoordinates(), DMDACreate1d(), DMDACreate2d(), DMDACreate3d()

     25: @*/
     26: PetscErrorCode  DMDASetUniformCoordinates(DM da,PetscReal xmin,PetscReal xmax,PetscReal ymin,PetscReal ymax,PetscReal zmin,PetscReal zmax)
     27: {
     28:   MPI_Comm         comm;
     29:   PetscSection     section;
     30:   DM               cda;
     31:   DMBoundaryType   bx,by,bz;
     32:   Vec              xcoor;
     33:   PetscScalar      *coors;
     34:   PetscReal        hx,hy,hz_;
     35:   PetscInt         i,j,k,M,N,P,istart,isize,jstart,jsize,kstart,ksize,dim,cnt;
     36:   PetscErrorCode   ierr;

     40:   DMDAGetInfo(da,&dim,&M,&N,&P,0,0,0,0,0,&bx,&by,&bz,0);
     41:   if (xmax < xmin) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_INCOMP,"xmax must be larger than xmin %g %g",(double)xmin,(double)xmax);
     42:   if ((ymax < ymin) && (dim > 1)) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_INCOMP,"ymax must be larger than ymin %g %g",(double)ymin,(double)ymax);
     43:   if ((zmax < zmin) && (dim > 2)) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_INCOMP,"zmax must be larger than zmin %g %g",(double)zmin,(double)zmax);
     44:   PetscObjectGetComm((PetscObject)da,&comm);
     45:   DMGetDefaultSection(da,&section);
     46:   DMDAGetCorners(da,&istart,&jstart,&kstart,&isize,&jsize,&ksize);
     47:   DMGetCoordinateDM(da, &cda);
     48:   if (section) {
     49:     /* This would be better as a vector, but this is compatible */
     50:     PetscInt numComp[3]      = {1, 1, 1};
     51:     PetscInt numVertexDof[3] = {1, 1, 1};

     53:     DMDASetFieldName(cda, 0, "x");
     54:     if (dim > 1) {DMDASetFieldName(cda, 1, "y");}
     55:     if (dim > 2) {DMDASetFieldName(cda, 2, "z");}
     56:     DMDACreateSection(cda, numComp, numVertexDof, NULL, NULL);
     57:   }
     58:   DMCreateGlobalVector(cda, &xcoor);
     59:   if (section) {
     60:     PetscSection csection;
     61:     PetscInt     vStart, vEnd;

     63:     DMGetDefaultGlobalSection(cda,&csection);
     64:     VecGetArray(xcoor,&coors);
     65:     DMDAGetHeightStratum(da, dim, &vStart, &vEnd);
     66:     if (bx == DM_BOUNDARY_PERIODIC) hx  = (xmax-xmin)/(M+1);
     67:     else                              hx  = (xmax-xmin)/(M ? M : 1);
     68:     if (by == DM_BOUNDARY_PERIODIC) hy  = (ymax-ymin)/(N+1);
     69:     else                              hy  = (ymax-ymin)/(N ? N : 1);
     70:     if (bz == DM_BOUNDARY_PERIODIC) hz_ = (zmax-zmin)/(P+1);
     71:     else                              hz_ = (zmax-zmin)/(P ? P : 1);
     72:     switch (dim) {
     73:     case 1:
     74:       for (i = 0; i < isize+1; ++i) {
     75:         PetscInt v = i+vStart, dof, off;

     77:         PetscSectionGetDof(csection, v, &dof);
     78:         PetscSectionGetOffset(csection, v, &off);
     79:         if (off >= 0) {
     80:           coors[off] = xmin + hx*(i+istart);
     81:         }
     82:       }
     83:       break;
     84:     case 2:
     85:       for (j = 0; j < jsize+1; ++j) {
     86:         for (i = 0; i < isize+1; ++i) {
     87:           PetscInt v = j*(isize+1)+i+vStart, dof, off;

     89:           PetscSectionGetDof(csection, v, &dof);
     90:           PetscSectionGetOffset(csection, v, &off);
     91:           if (off >= 0) {
     92:             coors[off+0] = xmin + hx*(i+istart);
     93:             coors[off+1] = ymin + hy*(j+jstart);
     94:           }
     95:         }
     96:       }
     97:       break;
     98:     case 3:
     99:       for (k = 0; k < ksize+1; ++k) {
    100:         for (j = 0; j < jsize+1; ++j) {
    101:           for (i = 0; i < isize+1; ++i) {
    102:             PetscInt v = (k*(jsize+1)+j)*(isize+1)+i+vStart, dof, off;

    104:             PetscSectionGetDof(csection, v, &dof);
    105:             PetscSectionGetOffset(csection, v, &off);
    106:             if (off >= 0) {
    107:               coors[off+0] = xmin + hx*(i+istart);
    108:               coors[off+1] = ymin + hy*(j+jstart);
    109:               coors[off+2] = zmin + hz_*(k+kstart);
    110:             }
    111:           }
    112:         }
    113:       }
    114:       break;
    115:     default:
    116:       SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Cannot create uniform coordinates for this dimension %D\n",dim);
    117:     }
    118:     VecRestoreArray(xcoor,&coors);
    119:     DMSetCoordinates(da,xcoor);
    120:     PetscLogObjectParent((PetscObject)da,(PetscObject)xcoor);
    121:     VecDestroy(&xcoor);
    122:     return(0);
    123:   }
    124:   if (dim == 1) {
    125:     if (bx == DM_BOUNDARY_PERIODIC) hx = (xmax-xmin)/M;
    126:     else hx = (xmax-xmin)/(M-1);
    127:     VecGetArray(xcoor,&coors);
    128:     for (i=0; i<isize; i++) {
    129:       coors[i] = xmin + hx*(i+istart);
    130:     }
    131:     VecRestoreArray(xcoor,&coors);
    132:   } else if (dim == 2) {
    133:     if (bx == DM_BOUNDARY_PERIODIC) hx = (xmax-xmin)/(M);
    134:     else hx = (xmax-xmin)/(M-1);
    135:     if (by == DM_BOUNDARY_PERIODIC) hy = (ymax-ymin)/(N);
    136:     else hy = (ymax-ymin)/(N-1);
    137:     VecGetArray(xcoor,&coors);
    138:     cnt  = 0;
    139:     for (j=0; j<jsize; j++) {
    140:       for (i=0; i<isize; i++) {
    141:         coors[cnt++] = xmin + hx*(i+istart);
    142:         coors[cnt++] = ymin + hy*(j+jstart);
    143:       }
    144:     }
    145:     VecRestoreArray(xcoor,&coors);
    146:   } else if (dim == 3) {
    147:     if (bx == DM_BOUNDARY_PERIODIC) hx = (xmax-xmin)/(M);
    148:     else hx = (xmax-xmin)/(M-1);
    149:     if (by == DM_BOUNDARY_PERIODIC) hy = (ymax-ymin)/(N);
    150:     else hy = (ymax-ymin)/(N-1);
    151:     if (bz == DM_BOUNDARY_PERIODIC) hz_ = (zmax-zmin)/(P);
    152:     else hz_ = (zmax-zmin)/(P-1);
    153:     VecGetArray(xcoor,&coors);
    154:     cnt  = 0;
    155:     for (k=0; k<ksize; k++) {
    156:       for (j=0; j<jsize; j++) {
    157:         for (i=0; i<isize; i++) {
    158:           coors[cnt++] = xmin + hx*(i+istart);
    159:           coors[cnt++] = ymin + hy*(j+jstart);
    160:           coors[cnt++] = zmin + hz_*(k+kstart);
    161:         }
    162:       }
    163:     }
    164:     VecRestoreArray(xcoor,&coors);
    165:   } else SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Cannot create uniform coordinates for this dimension %D\n",dim);
    166:   DMSetCoordinates(da,xcoor);
    167:   PetscLogObjectParent((PetscObject)da,(PetscObject)xcoor);
    168:   VecDestroy(&xcoor);
    169:   return(0);
    170: }

    174: /*
    175:     Allows a user to select a subset of the fields to be drawn by VecView() when the vector comes from a DMDA
    176: */
    177: PetscErrorCode DMDASelectFields(DM da,PetscInt *outfields,PetscInt **fields)
    178: {
    180:   PetscInt       step,ndisplayfields,*displayfields,k,j;
    181:   PetscBool      flg;

    184:   DMDAGetInfo(da,0,0,0,0,0,0,0,&step,0,0,0,0,0);
    185:   PetscMalloc1(step,&displayfields);
    186:   for (k=0; k<step; k++) displayfields[k] = k;
    187:   ndisplayfields = step;
    188:   PetscOptionsGetIntArray(NULL,NULL,"-draw_fields",displayfields,&ndisplayfields,&flg);
    189:   if (!ndisplayfields) ndisplayfields = step;
    190:   if (!flg) {
    191:     char       **fields;
    192:     const char *fieldname;
    193:     PetscInt   nfields = step;
    194:     PetscMalloc1(step,&fields);
    195:     PetscOptionsGetStringArray(NULL,NULL,"-draw_fields_by_name",fields,&nfields,&flg);
    196:     if (flg) {
    197:       ndisplayfields = 0;
    198:       for (k=0; k<nfields;k++) {
    199:         for (j=0; j<step; j++) {
    200:           DMDAGetFieldName(da,j,&fieldname);
    201:           PetscStrcmp(fieldname,fields[k],&flg);
    202:           if (flg) {
    203:             goto found;
    204:           }
    205:         }
    206:         SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_USER,"Unknown fieldname %s",fields[k]);
    207: found:  displayfields[ndisplayfields++] = j;
    208:       }
    209:     }
    210:     for (k=0; k<nfields; k++) {
    211:       PetscFree(fields[k]);
    212:     }
    213:     PetscFree(fields);
    214:   }
    215:   *fields    = displayfields;
    216:   *outfields = ndisplayfields;
    217:   return(0);
    218: }

    220: #include <petscdraw.h>

    224: PetscErrorCode VecView_MPI_Draw_DA1d(Vec xin,PetscViewer v)
    225: {
    226:   DM                da;
    227:   PetscErrorCode    ierr;
    228:   PetscMPIInt       rank,size,tag;
    229:   PetscInt          i,n,N,dof,istart,isize,j,nbounds;
    230:   MPI_Status        status;
    231:   PetscReal         min,max,xmin = 0.0,xmax = 0.0,tmp = 0.0,xgtmp = 0.0;
    232:   const PetscScalar *array,*xg;
    233:   PetscDraw         draw;
    234:   PetscBool         isnull,useports = PETSC_FALSE,showmarkers = PETSC_FALSE;
    235:   MPI_Comm          comm;
    236:   PetscDrawAxis     axis;
    237:   Vec               xcoor;
    238:   DMBoundaryType    bx;
    239:   const char        *tlabel = NULL,*xlabel = NULL;
    240:   const PetscReal   *bounds;
    241:   PetscInt          *displayfields;
    242:   PetscInt          k,ndisplayfields;
    243:   PetscBool         hold;
    244:   PetscDrawViewPorts *ports = NULL;
    245:   PetscViewerFormat  format;

    248:   PetscViewerDrawGetDraw(v,0,&draw);
    249:   PetscDrawIsNull(draw,&isnull);
    250:   if (isnull) return(0);
    251:   PetscViewerDrawGetBounds(v,&nbounds,&bounds);

    253:   VecGetDM(xin,&da);
    254:   if (!da) SETERRQ(PetscObjectComm((PetscObject)xin),PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
    255:   PetscObjectGetComm((PetscObject)xin,&comm);
    256:   MPI_Comm_size(comm,&size);
    257:   MPI_Comm_rank(comm,&rank);

    259:   PetscOptionsGetBool(NULL,NULL,"-draw_vec_use_markers",&showmarkers,NULL);

    261:   DMDAGetInfo(da,NULL,&N,NULL,NULL,NULL,NULL,NULL,&dof,NULL,&bx,NULL,NULL,NULL);
    262:   DMDAGetCorners(da,&istart,NULL,NULL,&isize,NULL,NULL);
    263:   VecGetArrayRead(xin,&array);
    264:   VecGetLocalSize(xin,&n);
    265:   n    = n/dof;

    267:   /* Get coordinates of nodes */
    268:   DMGetCoordinates(da,&xcoor);
    269:   if (!xcoor) {
    270:     DMDASetUniformCoordinates(da,0.0,1.0,0.0,0.0,0.0,0.0);
    271:     DMGetCoordinates(da,&xcoor);
    272:   }
    273:   VecGetArrayRead(xcoor,&xg);
    274:   DMDAGetCoordinateName(da,0,&xlabel);

    276:   /* Determine the min and max coordinate in plot */
    277:   if (!rank) xmin = PetscRealPart(xg[0]);
    278:   if (rank == size-1) xmax = PetscRealPart(xg[n-1]);
    279:   MPI_Bcast(&xmin,1,MPIU_REAL,0,comm);
    280:   MPI_Bcast(&xmax,1,MPIU_REAL,size-1,comm);

    282:   DMDASelectFields(da,&ndisplayfields,&displayfields);
    283:   PetscViewerGetFormat(v,&format);
    284:   PetscOptionsGetBool(NULL,NULL,"-draw_ports",&useports,NULL);
    285:   if (format == PETSC_VIEWER_DRAW_PORTS) useports = PETSC_TRUE;
    286:   if (useports) {
    287:     PetscViewerDrawGetDraw(v,0,&draw);
    288:     PetscViewerDrawGetDrawAxis(v,0,&axis);
    289:     PetscDrawCheckResizedWindow(draw);
    290:     PetscDrawClear(draw);
    291:     PetscDrawViewPortsCreate(draw,ndisplayfields,&ports);
    292:   }

    294:   /* Loop over each field; drawing each in a different window */
    295:   for (k=0; k<ndisplayfields; k++) {
    296:     j = displayfields[k];

    298:     /* determine the min and max value in plot */
    299:     VecStrideMin(xin,j,NULL,&min);
    300:     VecStrideMax(xin,j,NULL,&max);
    301:     if (j < nbounds) {
    302:       min = PetscMin(min,bounds[2*j]);
    303:       max = PetscMax(max,bounds[2*j+1]);
    304:     }
    305:     if (min == max) {
    306:       min -= 1.e-5;
    307:       max += 1.e-5;
    308:     }

    310:     if (useports) {
    311:       PetscDrawViewPortsSet(ports,k);
    312:       DMDAGetFieldName(da,j,&tlabel);
    313:     } else {
    314:       const char *title;
    315:       PetscViewerDrawGetHold(v,&hold);
    316:       PetscViewerDrawGetDraw(v,k,&draw);
    317:       PetscViewerDrawGetDrawAxis(v,k,&axis);
    318:       DMDAGetFieldName(da,j,&title);
    319:       if (title) {PetscDrawSetTitle(draw,title);}
    320:       PetscDrawCheckResizedWindow(draw);
    321:       if (!hold) {PetscDrawClear(draw);}
    322:     }
    323:     PetscDrawAxisSetLabels(axis,tlabel,xlabel,NULL);
    324:     PetscDrawAxisSetLimits(axis,xmin,xmax,min,max);
    325:     PetscDrawAxisDraw(axis);

    327:     /* draw local part of vector */
    328:     PetscObjectGetNewTag((PetscObject)xin,&tag);
    329:     if (rank < size-1) { /*send value to right */
    330:       MPI_Send((void*)&xg[n-1],1,MPIU_REAL,rank+1,tag,comm);
    331:       MPI_Send((void*)&array[j+(n-1)*dof],1,MPIU_REAL,rank+1,tag,comm);
    332:     }
    333:     if (rank) { /* receive value from left */
    334:       MPI_Recv(&xgtmp,1,MPIU_REAL,rank-1,tag,comm,&status);
    335:       MPI_Recv(&tmp,1,MPIU_REAL,rank-1,tag,comm,&status);
    336:     }
    337:     PetscDrawCollectiveBegin(draw);
    338:     if (rank) {
    339:       PetscDrawLine(draw,xgtmp,tmp,PetscRealPart(xg[0]),PetscRealPart(array[j]),PETSC_DRAW_RED);
    340:       if (showmarkers) {PetscDrawPoint(draw,xgtmp,tmp,PETSC_DRAW_BLACK);}
    341:     }
    342:     for (i=1; i<n; i++) {
    343:       PetscDrawLine(draw,PetscRealPart(xg[i-1]),PetscRealPart(array[j+dof*(i-1)]),PetscRealPart(xg[i]),PetscRealPart(array[j+dof*i]),PETSC_DRAW_RED);
    344:       if (showmarkers) {PetscDrawMarker(draw,PetscRealPart(xg[i-1]),PetscRealPart(array[j+dof*(i-1)]),PETSC_DRAW_BLACK);}
    345:     }
    346:     if (rank == size-1) {
    347:       if (showmarkers) {PetscDrawMarker(draw,PetscRealPart(xg[n-1]),PetscRealPart(array[j+dof*(n-1)]),PETSC_DRAW_BLACK);}
    348:     }
    349:     PetscDrawCollectiveEnd(draw);
    350:     PetscDrawFlush(draw);
    351:     PetscDrawPause(draw);
    352:     if (!useports) {PetscDrawSave(draw);}
    353:   }
    354:   if (useports) {
    355:     PetscViewerDrawGetDraw(v,0,&draw);
    356:     PetscDrawSave(draw);
    357:   }

    359:   PetscDrawViewPortsDestroy(ports);
    360:   PetscFree(displayfields);
    361:   VecRestoreArrayRead(xcoor,&xg);
    362:   VecRestoreArrayRead(xin,&array);
    363:   return(0);
    364: }