147c6ae99SBarry Smith 247c6ae99SBarry Smith /* 347c6ae99SBarry Smith Code for manipulating distributed regular arrays in parallel. 447c6ae99SBarry Smith */ 547c6ae99SBarry Smith 6*b45d2f2cSJed Brown #include <petsc-private/daimpl.h> /*I "petscdmda.h" I*/ 747c6ae99SBarry Smith 847c6ae99SBarry Smith /* Logging support */ 97087cfbeSBarry Smith PetscClassId ADDA_CLASSID; 1067a56275SMatthew G Knepley PetscLogEvent DMDA_LocalADFunction; 1147c6ae99SBarry Smith 1247c6ae99SBarry Smith #undef __FUNCT__ 1347c6ae99SBarry Smith #define __FUNCT__ "DMDestroy_Private" 1447c6ae99SBarry Smith /* 1547c6ae99SBarry Smith DMDestroy_Private - handles the work vectors created by DMGetGlobalVector() and DMGetLocalVector() 1647c6ae99SBarry Smith 1747c6ae99SBarry Smith */ 187087cfbeSBarry Smith PetscErrorCode DMDestroy_Private(DM dm,PetscBool *done) 1947c6ae99SBarry Smith { 2047c6ae99SBarry Smith PetscErrorCode ierr; 2147c6ae99SBarry Smith PetscErrorCode i,cnt = 0; 2247c6ae99SBarry Smith 2347c6ae99SBarry Smith PetscFunctionBegin; 2447c6ae99SBarry Smith PetscValidHeaderSpecific(dm,DM_CLASSID,1); 2547c6ae99SBarry Smith *done = PETSC_FALSE; 2647c6ae99SBarry Smith 2747c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 2847c6ae99SBarry Smith if (dm->localin[i]) {cnt++;} 2947c6ae99SBarry Smith if (dm->globalin[i]) {cnt++;} 3047c6ae99SBarry Smith } 3147c6ae99SBarry Smith 3247c6ae99SBarry Smith if (--((PetscObject)dm)->refct - cnt > 0) PetscFunctionReturn(0); 3347c6ae99SBarry Smith 3447c6ae99SBarry Smith /* 3547c6ae99SBarry Smith Need this test because the dm references the vectors that 3647c6ae99SBarry Smith reference the dm, so destroying the dm calls destroy on the 3747c6ae99SBarry Smith vectors that cause another destroy on the dm 3847c6ae99SBarry Smith */ 3947c6ae99SBarry Smith if (((PetscObject)dm)->refct < 0) PetscFunctionReturn(0); 4047c6ae99SBarry Smith ((PetscObject)dm)->refct = 0; 4147c6ae99SBarry Smith 4247c6ae99SBarry Smith for (i=0; i<DM_MAX_WORK_VECTORS; i++) { 4347c6ae99SBarry Smith if (dm->localout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a local vector obtained with DMGetLocalVector()"); 4496e147daSBarry Smith ierr = VecDestroy(&dm->localin[i]);CHKERRQ(ierr); 4547c6ae99SBarry Smith if (dm->globalout[i]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Destroying a DM that has a global vector obtained with DMGetGlobalVector()"); 4696e147daSBarry Smith ierr = VecDestroy(&dm->globalin[i]);CHKERRQ(ierr); 4747c6ae99SBarry Smith } 4896e147daSBarry Smith ierr = ISLocalToGlobalMappingDestroy(&dm->ltogmap);CHKERRQ(ierr); 4996e147daSBarry Smith ierr = ISLocalToGlobalMappingDestroy(&dm->ltogmapb);CHKERRQ(ierr); 501411c6eeSJed Brown 5147c6ae99SBarry Smith *done = PETSC_TRUE; 5247c6ae99SBarry Smith PetscFunctionReturn(0); 5347c6ae99SBarry Smith } 5447c6ae99SBarry Smith 5547c6ae99SBarry Smith #undef __FUNCT__ 569a42bb27SBarry Smith #define __FUNCT__ "DMDestroy_DA" 576bf464f9SBarry Smith PetscErrorCode DMDestroy_DA(DM da) 5847c6ae99SBarry Smith { 5947c6ae99SBarry Smith PetscErrorCode ierr; 6047c6ae99SBarry Smith PetscErrorCode i; 616bf464f9SBarry Smith DM_DA *dd = (DM_DA*)da->data; 6247c6ae99SBarry Smith 6347c6ae99SBarry Smith PetscFunctionBegin; 6447c6ae99SBarry Smith /* destroy the external/common part */ 65aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 6647c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedout[i]);CHKERRQ(ierr); 6747c6ae99SBarry Smith ierr = PetscFree(dd->adstartghostedin[i]);CHKERRQ(ierr); 6847c6ae99SBarry Smith ierr = PetscFree(dd->adstartout[i]);CHKERRQ(ierr); 6947c6ae99SBarry Smith ierr = PetscFree(dd->adstartin[i]);CHKERRQ(ierr); 7047c6ae99SBarry Smith } 71aa219208SBarry Smith for (i=0; i<DMDA_MAX_AD_ARRAYS; i++) { 7247c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedout[i]);CHKERRQ(ierr); 7347c6ae99SBarry Smith ierr = PetscFree(dd->admfstartghostedin[i]);CHKERRQ(ierr); 7447c6ae99SBarry Smith ierr = PetscFree(dd->admfstartout[i]);CHKERRQ(ierr); 7547c6ae99SBarry Smith ierr = PetscFree(dd->admfstartin[i]);CHKERRQ(ierr); 7647c6ae99SBarry Smith } 77aa219208SBarry Smith for (i=0; i<DMDA_MAX_WORK_ARRAYS; i++) { 7847c6ae99SBarry Smith ierr = PetscFree(dd->startghostedout[i]);CHKERRQ(ierr); 7947c6ae99SBarry Smith ierr = PetscFree(dd->startghostedin[i]);CHKERRQ(ierr); 8047c6ae99SBarry Smith ierr = PetscFree(dd->startout[i]);CHKERRQ(ierr); 8147c6ae99SBarry Smith ierr = PetscFree(dd->startin[i]);CHKERRQ(ierr); 8247c6ae99SBarry Smith } 8347c6ae99SBarry Smith 8447c6ae99SBarry Smith /* if memory was published with AMS then destroy it */ 8547c6ae99SBarry Smith ierr = PetscObjectDepublish(da);CHKERRQ(ierr); 8647c6ae99SBarry Smith 87fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->ltog);CHKERRQ(ierr); 88fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->gtol);CHKERRQ(ierr); 89fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->ltol);CHKERRQ(ierr); 90fcfd50ebSBarry Smith ierr = VecDestroy(&dd->natural);CHKERRQ(ierr); 91fcfd50ebSBarry Smith ierr = VecScatterDestroy(&dd->gton);CHKERRQ(ierr); 92fcfd50ebSBarry Smith ierr = AODestroy(&dd->ao);CHKERRQ(ierr); 9347c6ae99SBarry Smith 94db87c5ecSEthan Coon ierr = PetscFree(dd->idx);CHKERRQ(ierr); 9547c6ae99SBarry Smith ierr = PetscFree(dd->lx);CHKERRQ(ierr); 9647c6ae99SBarry Smith ierr = PetscFree(dd->ly);CHKERRQ(ierr); 9747c6ae99SBarry Smith ierr = PetscFree(dd->lz);CHKERRQ(ierr); 9847c6ae99SBarry Smith 9947c6ae99SBarry Smith if (dd->fieldname) { 10047c6ae99SBarry Smith for (i=0; i<dd->w; i++) { 10147c6ae99SBarry Smith ierr = PetscFree(dd->fieldname[i]);CHKERRQ(ierr); 10247c6ae99SBarry Smith } 10347c6ae99SBarry Smith ierr = PetscFree(dd->fieldname);CHKERRQ(ierr); 10447c6ae99SBarry Smith } 105fcfd50ebSBarry Smith ierr = ISColoringDestroy(&dd->localcoloring);CHKERRQ(ierr); 106fcfd50ebSBarry Smith ierr = ISColoringDestroy(&dd->ghostedcoloring);CHKERRQ(ierr); 10747c6ae99SBarry Smith 1086bf464f9SBarry Smith ierr = VecDestroy(&dd->coordinates);CHKERRQ(ierr); 1096bf464f9SBarry Smith ierr = VecDestroy(&dd->ghosted_coordinates);CHKERRQ(ierr); 1106bf464f9SBarry Smith ierr = DMDestroy(&dd->da_coordinates);CHKERRQ(ierr); 11147c6ae99SBarry Smith 11247c6ae99SBarry Smith ierr = PetscFree(dd->neighbors);CHKERRQ(ierr); 11347c6ae99SBarry Smith ierr = PetscFree(dd->dfill);CHKERRQ(ierr); 11447c6ae99SBarry Smith ierr = PetscFree(dd->ofill);CHKERRQ(ierr); 11547c6ae99SBarry Smith ierr = PetscFree(dd->e);CHKERRQ(ierr); 11647c6ae99SBarry Smith PetscFunctionReturn(0); 11747c6ae99SBarry Smith } 118