1 2 #if !defined(__PETSCBT_H) 3 #define __PETSCBT_H 4 PETSC_EXTERN_CXX_BEGIN 5 6 /*S 7 PetscBT - PETSc bitarrays 8 9 Level: advanced 10 11 PetscBTCreate(m,&bt) - creates a bit array with enough room to hold m values 12 PetscBTDestroy(&bt) - destroys the bit array 13 PetscBTMemzero(m,bt) - zeros the entire bit array (sets all values to false) 14 PetscBTSet(bt,index) - sets a particular entry as true 15 PetscBTClear(bt,index) - sets a particular entry as false 16 PetscBTLookup(bt,index) - returns the value 17 PetscBTLookupSet(bt,index) - returns the value and then sets it true 18 PetscBTLength(m) - returns number of bytes in array with m bits 19 PetscBTView(m,bt,viewer) - prints all the entries in a bit array 20 21 We do not currently check error flags on PetscBTSet(), PetscBTClear(), PetscBTLookup(), 22 PetcBTLookupSet(), PetscBTLength() cause error checking would cost hundreds more cycles then 23 the operation. 24 25 S*/ 26 typedef char* PetscBT; 27 28 extern char _BT_mask; 29 extern char _BT_c; 30 extern PetscInt _BT_idx; 31 32 PETSC_STATIC_INLINE PetscInt PetscBTLength(PetscInt m) 33 { 34 return ((m)/PETSC_BITS_PER_BYTE+1); 35 } 36 37 PETSC_STATIC_INLINE PetscErrorCode PetscBTMemzero(PetscInt m,PetscBT array) 38 { 39 return PetscMemzero(array,sizeof(char)*((m)/PETSC_BITS_PER_BYTE+1)); 40 } 41 42 PETSC_STATIC_INLINE PetscErrorCode PetscBTDestroy(PetscBT *array) 43 { 44 return PetscFree(*array); 45 } 46 47 PETSC_STATIC_INLINE PetscInt PetscBTLookup(PetscBT array,PetscInt index) 48 { 49 return (_BT_idx = (index)/PETSC_BITS_PER_BYTE, 50 _BT_c = array[_BT_idx], 51 _BT_mask = (char)1 << ((index)%PETSC_BITS_PER_BYTE), 52 (_BT_c & _BT_mask) != 0); 53 } 54 55 PETSC_STATIC_INLINE PetscErrorCode PetscBTView(PetscInt m,const PetscBT bt,PetscViewer viewer) 56 { 57 PetscInt i; 58 PetscErrorCode ierr; 59 60 if (!viewer) viewer = PETSC_VIEWER_STDOUT_SELF; 61 ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 62 for (i=0; i<m; i++) { 63 ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%D %d\n",i,PetscBTLookup(bt,i));CHKERRQ(ierr); 64 } ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 65 ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 66 return 0; 67 } 68 69 PETSC_STATIC_INLINE PetscErrorCode PetscBTCreate(PetscInt m,PetscBT *array) 70 { 71 return (PetscMalloc(((m)/PETSC_BITS_PER_BYTE+1)*sizeof(char),array) || PetscBTMemzero(m,*array)); 72 } 73 74 PETSC_STATIC_INLINE char PetscBTLookupSet(PetscBT array,PetscInt index) 75 { 76 return (_BT_idx = (index)/PETSC_BITS_PER_BYTE, 77 _BT_c = array[_BT_idx], 78 _BT_mask = (char)1 << ((index)%PETSC_BITS_PER_BYTE), 79 array[_BT_idx] = _BT_c | _BT_mask, 80 _BT_c & _BT_mask); 81 } 82 83 PETSC_STATIC_INLINE PetscErrorCode PetscBTSet(PetscBT array,PetscInt index) 84 { 85 return (_BT_idx = (index)/PETSC_BITS_PER_BYTE, 86 _BT_c = array[_BT_idx], 87 _BT_mask = (char)1 << ((index)%PETSC_BITS_PER_BYTE), 88 array[_BT_idx] = _BT_c | _BT_mask,0); 89 } 90 91 PETSC_STATIC_INLINE PetscErrorCode PetscBTClear(PetscBT array,PetscInt index) 92 { 93 return (_BT_idx = (index)/PETSC_BITS_PER_BYTE, 94 _BT_c = array[_BT_idx], 95 _BT_mask = (char)1 << ((index)%PETSC_BITS_PER_BYTE), 96 array[_BT_idx] = _BT_c & (~_BT_mask),0); 97 } 98 99 100 PETSC_EXTERN_CXX_END 101 #endif 102