1*c4762a1bSJed Brown static char help[] = "Tests PetscHeapCreate()\n\n"; 2*c4762a1bSJed Brown 3*c4762a1bSJed Brown #include <petscsys.h> 4*c4762a1bSJed Brown #include <petscviewer.h> 5*c4762a1bSJed Brown 6*c4762a1bSJed Brown int main(int argc,char **args) 7*c4762a1bSJed Brown { 8*c4762a1bSJed Brown PetscErrorCode ierr; 9*c4762a1bSJed Brown PetscHeap h; 10*c4762a1bSJed Brown PetscInt id,val,cnt,*values; 11*c4762a1bSJed Brown 12*c4762a1bSJed Brown ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 13*c4762a1bSJed Brown ierr = PetscHeapCreate(9,&h);CHKERRQ(ierr); 14*c4762a1bSJed Brown ierr = PetscHeapAdd(h,0,100);CHKERRQ(ierr); 15*c4762a1bSJed Brown ierr = PetscHeapAdd(h,1,19);CHKERRQ(ierr); 16*c4762a1bSJed Brown ierr = PetscHeapAdd(h,2,36);CHKERRQ(ierr); 17*c4762a1bSJed Brown ierr = PetscHeapAdd(h,3,17);CHKERRQ(ierr); 18*c4762a1bSJed Brown ierr = PetscHeapAdd(h,4,3);CHKERRQ(ierr); 19*c4762a1bSJed Brown ierr = PetscHeapAdd(h,5,25);CHKERRQ(ierr); 20*c4762a1bSJed Brown ierr = PetscHeapAdd(h,6,1);CHKERRQ(ierr); 21*c4762a1bSJed Brown ierr = PetscHeapAdd(h,8,2);CHKERRQ(ierr); 22*c4762a1bSJed Brown ierr = PetscHeapAdd(h,9,7);CHKERRQ(ierr); 23*c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_SELF,"Initial heap:\n");CHKERRQ(ierr); 24*c4762a1bSJed Brown ierr = PetscHeapView(h,NULL);CHKERRQ(ierr); 25*c4762a1bSJed Brown 26*c4762a1bSJed Brown ierr = PetscHeapPop(h,&id,&val);CHKERRQ(ierr); 27*c4762a1bSJed Brown ierr = PetscHeapStash(h,id,val+10);CHKERRQ(ierr); 28*c4762a1bSJed Brown ierr = PetscHeapPop(h,&id,&val);CHKERRQ(ierr); 29*c4762a1bSJed Brown ierr = PetscHeapStash(h,id,val+10);CHKERRQ(ierr); 30*c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_SELF,"Pop two items, increment, and place in stash:\n");CHKERRQ(ierr); 31*c4762a1bSJed Brown ierr = PetscHeapView(h,NULL);CHKERRQ(ierr); 32*c4762a1bSJed Brown 33*c4762a1bSJed Brown ierr = PetscHeapUnstash(h);CHKERRQ(ierr); 34*c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_SELF,"After unpacking the stash:\n");CHKERRQ(ierr); 35*c4762a1bSJed Brown ierr = PetscHeapView(h,NULL);CHKERRQ(ierr); 36*c4762a1bSJed Brown 37*c4762a1bSJed Brown ierr = PetscMalloc1(9,&values);CHKERRQ(ierr); 38*c4762a1bSJed Brown ierr = PetscHeapPop(h,&id,&val);CHKERRQ(ierr); 39*c4762a1bSJed Brown cnt = 0; 40*c4762a1bSJed Brown while (id >= 0) { 41*c4762a1bSJed Brown values[cnt++] = val; 42*c4762a1bSJed Brown ierr = PetscHeapPop(h,&id,&val);CHKERRQ(ierr); 43*c4762a1bSJed Brown } 44*c4762a1bSJed Brown ierr = PetscPrintf(PETSC_COMM_SELF,"Sorted values:\n");CHKERRQ(ierr); 45*c4762a1bSJed Brown ierr = PetscIntView(cnt,values,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); 46*c4762a1bSJed Brown ierr = PetscFree(values);CHKERRQ(ierr); 47*c4762a1bSJed Brown ierr = PetscHeapDestroy(&h);CHKERRQ(ierr); 48*c4762a1bSJed Brown ierr = PetscFinalize(); 49*c4762a1bSJed Brown return ierr; 50*c4762a1bSJed Brown } 51*c4762a1bSJed Brown 52*c4762a1bSJed Brown 53*c4762a1bSJed Brown 54*c4762a1bSJed Brown 55*c4762a1bSJed Brown /*TEST 56*c4762a1bSJed Brown 57*c4762a1bSJed Brown test: 58*c4762a1bSJed Brown 59*c4762a1bSJed Brown TEST*/ 60