147c6ae99SBarry Smith #define PETSCDM_DLL 247c6ae99SBarry Smith 347c6ae99SBarry Smith /* 447c6ae99SBarry Smith Code for manipulating distributed regular arrays in parallel. 547c6ae99SBarry Smith */ 647c6ae99SBarry Smith 7e1589f56SBarry Smith #include "private/daimpl.h" /*I "petscdm.h" I*/ 847c6ae99SBarry Smith 947c6ae99SBarry Smith /* Logging support */ 107087cfbeSBarry Smith PetscClassId DM_CLASSID; 117087cfbeSBarry Smith PetscClassId ADDA_CLASSID; 12aa219208SBarry Smith PetscLogEvent DMDA_GlobalToLocal, DMDA_LocalToGlobal, DMDA_LocalADFunction; 1347c6ae99SBarry Smith 1447c6ae99SBarry Smith #undef __FUNCT__ 1547c6ae99SBarry Smith #define __FUNCT__ "DMDestroy_Private" 1647c6ae99SBarry Smith /* 1747c6ae99SBarry Smith DMDestroy_Private - handles the work vectors created by DMGetGlobalVector() and DMGetLocalVector() 1847c6ae99SBarry Smith 1947c6ae99SBarry Smith */ 207087cfbeSBarry Smith PetscErrorCode DMDestroy_Private(DM dm,PetscBool *done) 2147c6ae99SBarry Smith { 2247c6ae99SBarry Smith PetscErrorCode ierr; 2347c6ae99SBarry Smith PetscErrorCode i,cnt = 0; 2447c6ae99SBarry Smith 2547c6ae99SBarry Smith PetscFunctionBegin; 2647c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 2747c6ae99SBarry Smith *done = PETSC_FALSE; 2847c6ae99SBarry Smith 2947c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 3047c6ae99SBarry Smith if (dm->localin[i]) {cnt++;} 3147c6ae99SBarry Smith if (dm->globalin[i]) {cnt++;} 3247c6ae99SBarry Smith } 3347c6ae99SBarry Smith 3447c6ae99SBarry Smith if (--((PetscObject)dm)->refct - cnt > 0) PetscFunctionReturn(0); 3547c6ae99SBarry Smith 3647c6ae99SBarry Smith /* 3747c6ae99SBarry Smith Need this test because the dm references the vectors that 3847c6ae99SBarry Smith reference the dm, so destroying the dm calls destroy on the 3947c6ae99SBarry Smith vectors that cause another destroy on the dm 4047c6ae99SBarry Smith */ 4147c6ae99SBarry Smith if (((PetscObject)dm)->refct < 0) PetscFunctionReturn(0); 4247c6ae99SBarry Smith ((PetscObject)dm)->refct = 0; 4347c6ae99SBarry Smith 4447c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 4547c6ae99SBarry Smith if (dm->localout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a local vector obtained with DMGetLocalVector()"); 4647c6ae99SBarry Smith if (dm->localin[i]) {ierr = VecDestroy(dm->localin[i]);CHKERRQ(ierr);} 4747c6ae99SBarry Smith if (dm->globalout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a global vector obtained with DMGetGlobalVector()"); 4847c6ae99SBarry Smith if (dm->globalin[i]) {ierr = VecDestroy(dm->globalin[i]);CHKERRQ(ierr);} 4947c6ae99SBarry Smith } 501411c6eeSJed Brown if (dm->ltogmap) {ierr = ISLocalToGlobalMappingDestroy(dm->ltogmap);CHKERRQ(ierr);} 511411c6eeSJed Brown if (dm->ltogmapb) {ierr = ISLocalToGlobalMappingDestroy(dm->ltogmapb);CHKERRQ(ierr);} 521411c6eeSJed Brown 5347c6ae99SBarry Smith *done = PETSC_TRUE; 5447c6ae99SBarry Smith PetscFunctionReturn(0); 5547c6ae99SBarry Smith } 5647c6ae99SBarry Smith 5747c6ae99SBarry Smith #undef __FUNCT__ 589a42bb27SBarry Smith #define __FUNCT__ "DMDestroy_DA" 597087cfbeSBarry Smith PetscErrorCode DMDestroy_DA(DM da) 6047c6ae99SBarry Smith { 6147c6ae99SBarry Smith PetscErrorCode ierr; 6247c6ae99SBarry Smith PetscErrorCode i; 6347c6ae99SBarry Smith PetscBool done; 6447c6ae99SBarry Smith DM_DA *dd = (DM_DA*)da->data; 6547c6ae99SBarry Smith 6647c6ae99SBarry Smith PetscFunctionBegin; 6747c6ae99SBarry Smith PetscValidHeaderSpecific(da,DM_CLASSID,1); 6847c6ae99SBarry Smith 6947c6ae99SBarry Smith ierr = DMDestroy_Private((DM)da,&done);CHKERRQ(ierr); 7047c6ae99SBarry Smith if (!done) PetscFunctionReturn(0); 710c010503SBarry Smith 7247c6ae99SBarry Smith /* destroy the external/common part */ 73aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 7447c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedout[i]);CHKERRQ(ierr); 7547c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedin[i]);CHKERRQ(ierr); 7647c6ae99SBarry Smith ierr = PetscFree(dd->adstartout[i]);CHKERRQ(ierr); 7747c6ae99SBarry Smith ierr = PetscFree(dd->adstartin[i]);CHKERRQ(ierr); 7847c6ae99SBarry Smith } 79aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 8047c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedout[i]);CHKERRQ(ierr); 8147c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedin[i]);CHKERRQ(ierr); 8247c6ae99SBarry Smith ierr = PetscFree(dd->admfstartout[i]);CHKERRQ(ierr); 8347c6ae99SBarry Smith ierr = PetscFree(dd->admfstartin[i]);CHKERRQ(ierr); 8447c6ae99SBarry Smith } 85aa219208SBarry Smith for (i=0; i<DMDA_MAX_WORK_ARRAYS; i++) { 8647c6ae99SBarry Smith ierr = PetscFree(dd->startghostedout[i]);CHKERRQ(ierr); 8747c6ae99SBarry Smith ierr = PetscFree(dd->startghostedin[i]);CHKERRQ(ierr); 8847c6ae99SBarry Smith ierr = PetscFree(dd->startout[i]);CHKERRQ(ierr); 8947c6ae99SBarry Smith ierr = PetscFree(dd->startin[i]);CHKERRQ(ierr); 9047c6ae99SBarry Smith } 9147c6ae99SBarry Smith 9247c6ae99SBarry Smith /* if memory was published with AMS then destroy it */ 9347c6ae99SBarry Smith ierr = PetscObjectDepublish(da);CHKERRQ(ierr); 9447c6ae99SBarry Smith 9547c6ae99SBarry Smith if (dd->ltog) {ierr = VecScatterDestroy(dd->ltog);CHKERRQ(ierr);} 9647c6ae99SBarry Smith if (dd->gtol) {ierr = VecScatterDestroy(dd->gtol);CHKERRQ(ierr);} 9747c6ae99SBarry Smith if (dd->ltol) {ierr = VecScatterDestroy(dd->ltol);CHKERRQ(ierr);} 9847c6ae99SBarry Smith if (dd->natural){ 9947c6ae99SBarry Smith ierr = VecDestroy(dd->natural);CHKERRQ(ierr); 10047c6ae99SBarry Smith } 10147c6ae99SBarry Smith if (dd->gton) { 10247c6ae99SBarry Smith ierr = VecScatterDestroy(dd->gton);CHKERRQ(ierr); 10347c6ae99SBarry Smith } 10447c6ae99SBarry Smith 10547c6ae99SBarry Smith if (dd->ao) { 10647c6ae99SBarry Smith ierr = AODestroy(dd->ao);CHKERRQ(ierr); 10747c6ae99SBarry Smith } 10847c6ae99SBarry Smith 109*db87c5ecSEthan Coon ierr = PetscFree(dd->idx);CHKERRQ(ierr); 11047c6ae99SBarry Smith ierr = PetscFree(dd->lx);CHKERRQ(ierr); 11147c6ae99SBarry Smith ierr = PetscFree(dd->ly);CHKERRQ(ierr); 11247c6ae99SBarry Smith ierr = PetscFree(dd->lz);CHKERRQ(ierr); 11347c6ae99SBarry Smith ierr = PetscFree(da->vectype);CHKERRQ(ierr); 11447c6ae99SBarry Smith 11547c6ae99SBarry Smith if (dd->fieldname) { 11647c6ae99SBarry Smith for (i=0; i<dd->w; i++) { 11747c6ae99SBarry Smith ierr = PetscFree(dd->fieldname[i]);CHKERRQ(ierr); 11847c6ae99SBarry Smith } 11947c6ae99SBarry Smith ierr = PetscFree(dd->fieldname);CHKERRQ(ierr); 12047c6ae99SBarry Smith } 12147c6ae99SBarry Smith 12247c6ae99SBarry Smith if (dd->localcoloring) { 12347c6ae99SBarry Smith ierr = ISColoringDestroy(dd->localcoloring);CHKERRQ(ierr); 12447c6ae99SBarry Smith } 12547c6ae99SBarry Smith if (dd->ghostedcoloring) { 12647c6ae99SBarry Smith ierr = ISColoringDestroy(dd->ghostedcoloring);CHKERRQ(ierr); 12747c6ae99SBarry Smith } 12847c6ae99SBarry Smith 12947c6ae99SBarry Smith if (dd->coordinates) {ierr = VecDestroy(dd->coordinates);CHKERRQ(ierr);} 13047c6ae99SBarry Smith if (dd->ghosted_coordinates) {ierr = VecDestroy(dd->ghosted_coordinates);CHKERRQ(ierr);} 1319a42bb27SBarry Smith if (dd->da_coordinates && da != dd->da_coordinates) {ierr = DMDestroy(dd->da_coordinates);CHKERRQ(ierr);} 13247c6ae99SBarry Smith 13347c6ae99SBarry Smith ierr = PetscFree(dd->neighbors);CHKERRQ(ierr); 13447c6ae99SBarry Smith ierr = PetscFree(dd->dfill);CHKERRQ(ierr); 13547c6ae99SBarry Smith ierr = PetscFree(dd->ofill);CHKERRQ(ierr); 13647c6ae99SBarry Smith ierr = PetscFree(dd->e);CHKERRQ(ierr); 13747c6ae99SBarry Smith 138895459faSBarry Smith ierr = PetscFree(dd);CHKERRQ(ierr); 13947c6ae99SBarry Smith ierr = PetscHeaderDestroy(da);CHKERRQ(ierr); 14047c6ae99SBarry Smith PetscFunctionReturn(0); 14147c6ae99SBarry Smith } 142