147c6ae99SBarry Smith 247c6ae99SBarry Smith /* 347c6ae99SBarry Smith Code for manipulating distributed regular arrays in parallel. 447c6ae99SBarry Smith */ 547c6ae99SBarry Smith 6c6db04a5SJed Brown #include <private/daimpl.h> /*I "petscdmda.h" I*/ 747c6ae99SBarry Smith 847c6ae99SBarry Smith /* Logging support */ 97087cfbeSBarry Smith PetscClassId DM_CLASSID; 107087cfbeSBarry Smith PetscClassId ADDA_CLASSID; 1167a56275SMatthew G Knepley PetscLogEvent DMDA_LocalADFunction; 1247c6ae99SBarry Smith 1347c6ae99SBarry Smith #undef __FUNCT__ 1447c6ae99SBarry Smith #define __FUNCT__ "DMDestroy_Private" 1547c6ae99SBarry Smith /* 1647c6ae99SBarry Smith DMDestroy_Private - handles the work vectors created by DMGetGlobalVector() and DMGetLocalVector() 1747c6ae99SBarry Smith 1847c6ae99SBarry Smith */ 197087cfbeSBarry Smith PetscErrorCode DMDestroy_Private(DM dm,PetscBool *done) 2047c6ae99SBarry Smith { 2147c6ae99SBarry Smith PetscErrorCode ierr; 2247c6ae99SBarry Smith PetscErrorCode i,cnt = 0; 2347c6ae99SBarry Smith 2447c6ae99SBarry Smith PetscFunctionBegin; 2547c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 2647c6ae99SBarry Smith *done = PETSC_FALSE; 2747c6ae99SBarry Smith 2847c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 2947c6ae99SBarry Smith if (dm->localin[i]) {cnt++;} 3047c6ae99SBarry Smith if (dm->globalin[i]) {cnt++;} 3147c6ae99SBarry Smith } 3247c6ae99SBarry Smith 3347c6ae99SBarry Smith if (--((PetscObject)dm)->refct - cnt > 0) PetscFunctionReturn(0); 3447c6ae99SBarry Smith 3547c6ae99SBarry Smith /* 3647c6ae99SBarry Smith Need this test because the dm references the vectors that 3747c6ae99SBarry Smith reference the dm, so destroying the dm calls destroy on the 3847c6ae99SBarry Smith vectors that cause another destroy on the dm 3947c6ae99SBarry Smith */ 4047c6ae99SBarry Smith if (((PetscObject)dm)->refct < 0) PetscFunctionReturn(0); 4147c6ae99SBarry Smith ((PetscObject)dm)->refct = 0; 4247c6ae99SBarry Smith 4347c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 4447c6ae99SBarry Smith if (dm->localout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a local vector obtained with DMGetLocalVector()"); 45*fcfd50ebSBarry Smith if (dm->localin[i]) {ierr = VecDestroy(&dm->localin[i]);CHKERRQ(ierr);} 4647c6ae99SBarry Smith if (dm->globalout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a global vector obtained with DMGetGlobalVector()"); 47*fcfd50ebSBarry Smith if (dm->globalin[i]) {ierr = VecDestroy(&dm->globalin[i]);CHKERRQ(ierr);} 4847c6ae99SBarry Smith } 49*fcfd50ebSBarry Smith if (dm->ltogmap) {ierr = ISLocalToGlobalMappingDestroy(&dm->ltogmap);CHKERRQ(ierr);} 50*fcfd50ebSBarry Smith if (dm->ltogmapb) {ierr = ISLocalToGlobalMappingDestroy(&dm->ltogmapb);CHKERRQ(ierr);} 511411c6eeSJed Brown 5247c6ae99SBarry Smith *done = PETSC_TRUE; 5347c6ae99SBarry Smith PetscFunctionReturn(0); 5447c6ae99SBarry Smith } 5547c6ae99SBarry Smith 5647c6ae99SBarry Smith #undef __FUNCT__ 579a42bb27SBarry Smith #define __FUNCT__ "DMDestroy_DA" 58*fcfd50ebSBarry Smith PetscErrorCode DMDestroy_DA(DM *da) 5947c6ae99SBarry Smith { 6047c6ae99SBarry Smith PetscErrorCode ierr; 6147c6ae99SBarry Smith PetscErrorCode i; 6247c6ae99SBarry Smith PetscBool done; 63*fcfd50ebSBarry Smith DM_DA *dd = (DM_DA*)(*da)->data; 6447c6ae99SBarry Smith 6547c6ae99SBarry Smith PetscFunctionBegin; 6647c6ae99SBarry Smith PetscValidHeaderSpecific(da,DM_CLASSID,1); 6747c6ae99SBarry Smith 6847c6ae99SBarry Smith ierr = DMDestroy_Private((DM)da,&done);CHKERRQ(ierr); 6947c6ae99SBarry Smith if (!done) PetscFunctionReturn(0); 700c010503SBarry Smith 7147c6ae99SBarry Smith /* destroy the external/common part */ 72aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 7347c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedout[i]);CHKERRQ(ierr); 7447c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedin[i]);CHKERRQ(ierr); 7547c6ae99SBarry Smith ierr = PetscFree(dd->adstartout[i]);CHKERRQ(ierr); 7647c6ae99SBarry Smith ierr = PetscFree(dd->adstartin[i]);CHKERRQ(ierr); 7747c6ae99SBarry Smith } 78aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 7947c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedout[i]);CHKERRQ(ierr); 8047c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedin[i]);CHKERRQ(ierr); 8147c6ae99SBarry Smith ierr = PetscFree(dd->admfstartout[i]);CHKERRQ(ierr); 8247c6ae99SBarry Smith ierr = PetscFree(dd->admfstartin[i]);CHKERRQ(ierr); 8347c6ae99SBarry Smith } 84aa219208SBarry Smith for (i=0; i<DMDA_MAX_WORK_ARRAYS; i++) { 8547c6ae99SBarry Smith ierr = PetscFree(dd->startghostedout[i]);CHKERRQ(ierr); 8647c6ae99SBarry Smith ierr = PetscFree(dd->startghostedin[i]);CHKERRQ(ierr); 8747c6ae99SBarry Smith ierr = PetscFree(dd->startout[i]);CHKERRQ(ierr); 8847c6ae99SBarry Smith ierr = PetscFree(dd->startin[i]);CHKERRQ(ierr); 8947c6ae99SBarry Smith } 9047c6ae99SBarry Smith 9147c6ae99SBarry Smith /* if memory was published with AMS then destroy it */ 9247c6ae99SBarry Smith ierr = PetscObjectDepublish(da);CHKERRQ(ierr); 9347c6ae99SBarry Smith 94*fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->ltog);CHKERRQ(ierr); 95*fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->gtol);CHKERRQ(ierr); 96*fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->ltol);CHKERRQ(ierr); 97*fcfd50ebSBarry Smith ierr = VecDestroy(&dd->natural);CHKERRQ(ierr); 98*fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->gton);CHKERRQ(ierr); 99*fcfd50ebSBarry Smith ierr = AODestroy(&dd->ao);CHKERRQ(ierr); 10047c6ae99SBarry Smith 101db87c5ecSEthan Coon ierr = PetscFree(dd->idx);CHKERRQ(ierr); 10247c6ae99SBarry Smith ierr = PetscFree(dd->lx);CHKERRQ(ierr); 10347c6ae99SBarry Smith ierr = PetscFree(dd->ly);CHKERRQ(ierr); 10447c6ae99SBarry Smith ierr = PetscFree(dd->lz);CHKERRQ(ierr); 10547c6ae99SBarry Smith ierr = PetscFree(da->vectype);CHKERRQ(ierr); 10647c6ae99SBarry Smith 10747c6ae99SBarry Smith if (dd->fieldname) { 10847c6ae99SBarry Smith for (i=0; i<dd->w; i++) { 10947c6ae99SBarry Smith ierr = PetscFree(dd->fieldname[i]);CHKERRQ(ierr); 11047c6ae99SBarry Smith } 11147c6ae99SBarry Smith ierr = PetscFree(dd->fieldname);CHKERRQ(ierr); 11247c6ae99SBarry Smith } 113*fcfd50ebSBarry Smith ierr = ISColoringDestroy(&dd->localcoloring);CHKERRQ(ierr); 114*fcfd50ebSBarry Smith ierr = ISColoringDestroy(&dd->ghostedcoloring);CHKERRQ(ierr); 11547c6ae99SBarry Smith 116*fcfd50ebSBarry Smith ierr = VecDestroy(&dd->coordinates);CHKERRQ(ierr);} 117*fcfd50ebSBarry Smith ierr = VecDestroy(&dd->ghosted_coordinates);CHKERRQ(ierr);} 118*fcfd50ebSBarry Smith if (dd->da_coordinates && da != dd->da_coordinates) {ierr = DMDestroy(&dd->da_coordinates);CHKERRQ(ierr);} 11947c6ae99SBarry Smith 12047c6ae99SBarry Smith ierr = PetscFree(dd->neighbors);CHKERRQ(ierr); 12147c6ae99SBarry Smith ierr = PetscFree(dd->dfill);CHKERRQ(ierr); 12247c6ae99SBarry Smith ierr = PetscFree(dd->ofill);CHKERRQ(ierr); 12347c6ae99SBarry Smith ierr = PetscFree(dd->e);CHKERRQ(ierr); 12447c6ae99SBarry Smith 125895459faSBarry Smith ierr = PetscFree(dd);CHKERRQ(ierr); 12647c6ae99SBarry Smith ierr = PetscHeaderDestroy(da);CHKERRQ(ierr); 12747c6ae99SBarry Smith PetscFunctionReturn(0); 12847c6ae99SBarry Smith } 129