1 static char help[] = "Test PETSc integer hash map.\n\n"; 2 3 #include <petsc/private/hashmapi.h> 4 #include <petsc/private/hashmapiv.h> 5 #include <petscsys.h> 6 7 /* Unused, keep it for testing purposes */ 8 PETSC_HASH_MAP(HMapIP, PetscInt, void*, PetscHashInt, PetscHashEqual, NULL) 9 10 /* Unused, keep it for testing purposes */ 11 typedef struct { double x; double y; double z; } Point; 12 static Point origin = {0.0, 0.0, 0.0}; 13 PETSC_HASH_MAP(HMapIS, PetscInt, Point, PetscHashInt, PetscHashEqual, origin) 14 15 #define PetscAssert(expr) do { \ 16 if (PetscUnlikely(!(expr))) \ 17 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB, \ 18 "Assertion: `%s' failed.", \ 19 PetscStringize(expr)); \ 20 } while (0) 21 22 23 int main(int argc,char **argv) 24 { 25 PetscHMapI ht = NULL, hd; 26 PetscHMapIV htv; 27 PetscInt n, v, koff, keys[4], voff, vals[4],na,nb,i,size,*karray,off; 28 PetscScalar *varray,*vwork; 29 PetscBool has, flag; 30 PetscErrorCode ierr; 31 32 ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; 33 34 ierr = PetscHMapICreate(&ht);CHKERRQ(ierr); 35 PetscAssert(ht != NULL); 36 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 37 PetscAssert(n == 0); 38 39 ierr = PetscHMapIResize(ht,0);CHKERRQ(ierr); 40 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 41 PetscAssert(n == 0); 42 43 ierr = PetscHMapIHas(ht,123,&has);CHKERRQ(ierr); 44 PetscAssert(has == PETSC_FALSE); 45 ierr = PetscHMapIGet(ht,123,&v);CHKERRQ(ierr); 46 PetscAssert(v == -1); 47 48 ierr = PetscHMapISet(ht,123,42);CHKERRQ(ierr); 49 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 50 PetscAssert(n == 1); 51 ierr = PetscHMapIHas(ht,123,&has);CHKERRQ(ierr); 52 PetscAssert(has == PETSC_TRUE); 53 ierr = PetscHMapIGet(ht,123,&v);CHKERRQ(ierr); 54 PetscAssert(v == 42); 55 56 ierr = PetscHMapIDel(ht,123);CHKERRQ(ierr); 57 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 58 PetscAssert(n == 0); 59 ierr = PetscHMapIHas(ht,123,&has);CHKERRQ(ierr); 60 PetscAssert(has == PETSC_FALSE); 61 ierr = PetscHMapIGet(ht,123,&v);CHKERRQ(ierr); 62 PetscAssert(v == -1); 63 64 ierr = PetscHMapIQuerySet(ht,123,1,&flag);CHKERRQ(ierr); 65 PetscAssert(flag == PETSC_TRUE); 66 ierr = PetscHMapIQuerySet(ht,123,1,&flag);CHKERRQ(ierr); 67 PetscAssert(flag == PETSC_FALSE); 68 ierr = PetscHMapIQueryDel(ht,123,&flag);CHKERRQ(ierr); 69 PetscAssert(flag == PETSC_TRUE); 70 ierr = PetscHMapIQueryDel(ht,123,&flag);CHKERRQ(ierr); 71 PetscAssert(flag == PETSC_FALSE); 72 73 ierr = PetscHMapIResize(ht,13);CHKERRQ(ierr); 74 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 75 PetscAssert(n == 0); 76 77 ierr = PetscHMapIClear(ht);CHKERRQ(ierr); 78 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 79 PetscAssert(n == 0); 80 81 ierr = PetscHMapISet(ht,321,24);CHKERRQ(ierr); 82 ierr = PetscHMapISet(ht,123,42);CHKERRQ(ierr); 83 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 84 PetscAssert(n == 2); 85 86 koff = 0; keys[0] = keys[1] = 0;CHKERRQ(ierr); 87 ierr = PetscHMapIGetKeys(ht,&koff,keys);CHKERRQ(ierr); 88 ierr = PetscSortInt(koff,keys);CHKERRQ(ierr); 89 PetscAssert(koff == 2); 90 PetscAssert(keys[0] == 123); 91 PetscAssert(keys[1] == 321); 92 93 voff = 0; vals[0] = vals[1] = 0; 94 ierr = PetscHMapIGetVals(ht,&voff,vals);CHKERRQ(ierr); 95 ierr = PetscSortInt(voff,vals);CHKERRQ(ierr); 96 PetscAssert(voff == 2); 97 PetscAssert(vals[0] == 24); 98 PetscAssert(vals[1] == 42); 99 100 koff = 0; keys[0] = keys[1] = 0; 101 voff = 0; vals[0] = vals[1] = 0; 102 ierr = PetscHMapIDuplicate(ht,&hd);CHKERRQ(ierr); 103 ierr = PetscHMapIGetKeys(ht,&koff,keys);CHKERRQ(ierr); 104 ierr = PetscHMapIGetVals(ht,&voff,vals);CHKERRQ(ierr); 105 ierr = PetscSortInt(koff,keys);CHKERRQ(ierr); 106 ierr = PetscSortInt(voff,vals);CHKERRQ(ierr); 107 PetscAssert(koff == 2); 108 PetscAssert(voff == 2); 109 PetscAssert(keys[0] == 123); 110 PetscAssert(keys[1] == 321); 111 PetscAssert(vals[0] == 24); 112 PetscAssert(vals[1] == 42); 113 ierr = PetscHMapIDestroy(&hd);CHKERRQ(ierr); 114 115 ierr = PetscHMapISet(ht,0,0);CHKERRQ(ierr); 116 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 117 PetscAssert(n != 0); 118 ierr = PetscHMapIReset(ht);CHKERRQ(ierr); 119 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 120 PetscAssert(n == 0); 121 ierr = PetscHMapIReset(ht);CHKERRQ(ierr); 122 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 123 PetscAssert(n == 0); 124 ierr = PetscHMapISet(ht,0,0);CHKERRQ(ierr); 125 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 126 PetscAssert(n != 0); 127 128 ierr = PetscHMapIDestroy(&ht);CHKERRQ(ierr); 129 PetscAssert(ht == NULL); 130 131 ierr = PetscHMapICreate(&ht);CHKERRQ(ierr); 132 ierr = PetscHMapIReset(ht);CHKERRQ(ierr); 133 ierr = PetscHMapIGetSize(ht,&n);CHKERRQ(ierr); 134 PetscAssert(n == 0); 135 ierr = PetscHMapIDestroy(&ht);CHKERRQ(ierr); 136 137 ierr = PetscHMapIVCreate(&htv);CHKERRQ(ierr); 138 n = 10; 139 ierr = PetscHMapIVResize(htv,n);CHKERRQ(ierr); 140 ierr = PetscHMapIVGetCapacity(htv,&na);CHKERRQ(ierr); 141 PetscAssert(na>=n); 142 for (i=0; i<n; i++) { 143 ierr = PetscHMapIVSet(htv,i+100,10.);CHKERRQ(ierr); 144 } 145 ierr = PetscHMapIVGetCapacity(htv,&nb);CHKERRQ(ierr); 146 PetscAssert(nb>=na); 147 for (i=0; i<(2*n); i++) { 148 ierr = PetscHMapIVAddValue(htv,i+100,5.);CHKERRQ(ierr); 149 } 150 ierr = PetscHMapIVGetSize(htv,&size);CHKERRQ(ierr); 151 PetscAssert(size==(2*n)); 152 ierr = PetscMalloc3(size,&karray,size,&varray,size,&vwork);CHKERRQ(ierr); 153 off = 0; 154 ierr = PetscHMapIVGetPairs(htv,&off,karray,varray);CHKERRQ(ierr); 155 PetscAssert(off==(2*n)); 156 ierr = PetscSortIntWithDataArray(off,karray,varray,sizeof(PetscScalar),vwork);CHKERRQ(ierr); 157 for (i=0; i<n; i++) { 158 PetscAssert(karray[i]==(i+100)); 159 PetscAssert(karray[n+i]==(n+i+100)); 160 PetscAssert(varray[i]==15.); 161 PetscAssert(varray[n+i]==5.); 162 } 163 ierr = PetscFree3(karray,varray,vwork);CHKERRQ(ierr); 164 ierr = PetscHMapIVDestroy(&htv);CHKERRQ(ierr); 165 166 ierr = PetscFinalize(); 167 return ierr; 168 } 169 170 171 /*TEST 172 173 test: 174 175 TEST*/ 176