1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
#ifndef __PETSCCTABLE_H
#define __PETSCCTABLE_H
#include <petscsys.h>
struct _n_PetscTable {
PetscInt *keytable;
PetscInt *table;
PetscInt count;
PetscInt tablesize;
PetscInt head;
PetscInt maxkey; /* largest key allowed */
};
typedef struct _n_PetscTable* PetscTable;
typedef PetscInt* PetscTablePosition;
PETSC_STATIC_INLINE unsigned long PetscHash(PetscTable ta,unsigned long x)
{
return(x%(unsigned long)ta->tablesize);
}
PETSC_STATIC_INLINE unsigned long PetscHashStep(PetscTable ta,unsigned long x)
{
return(1+(x%(unsigned long)(ta->tablesize-1)));
}
PETSC_EXTERN PetscErrorCode PetscTableCreate(const PetscInt,PetscInt,PetscTable*);
PETSC_EXTERN PetscErrorCode PetscTableCreateCopy(const PetscTable,PetscTable*);
PETSC_EXTERN PetscErrorCode PetscTableDestroy(PetscTable*);
PETSC_EXTERN PetscErrorCode PetscTableGetCount(const PetscTable,PetscInt*);
PETSC_EXTERN PetscErrorCode PetscTableIsEmpty(const PetscTable,PetscInt*);
PETSC_EXTERN PetscErrorCode PetscTableAddExpand(PetscTable,PetscInt,PetscInt,InsertMode);
PETSC_EXTERN PetscErrorCode PetscTableAddCountExpand(PetscTable,PetscInt);
PETSC_EXTERN PetscErrorCode PetscTableGetHeadPosition(PetscTable,PetscTablePosition*);
PETSC_EXTERN PetscErrorCode PetscTableGetNext(PetscTable,PetscTablePosition*,PetscInt*,PetscInt*);
PETSC_EXTERN PetscErrorCode PetscTableRemoveAll(PetscTable);
PETSC_STATIC_INLINE PetscErrorCode PetscTableAdd(PetscTable ta,PetscInt key,PetscInt data,InsertMode imode)
{
PetscErrorCode ierr;
PetscInt i,hash = (PetscInt)PetscHash(ta,(unsigned long)key);
PetscInt hashstep = (PetscInt)PetscHashStep(ta,(unsigned long)key);
PetscFunctionBegin;
if (key <= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"key <= 0");
if (key > ta->maxkey) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"key %D is greater than largest key allowed %D",key,ta->maxkey);
if (!data) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Null data");
for (i=0; i<ta->tablesize; i++) {
if (ta->keytable[hash] == key) {
switch (imode) {
case INSERT_VALUES:
ta->table[hash] = data; /* over write */
break;
case ADD_VALUES:
ta->table[hash] += data;
break;
case MAX_VALUES:
ta->table[hash] = PetscMax(ta->table[hash],data);
break;
case NOT_SET_VALUES:
case INSERT_ALL_VALUES:
case ADD_ALL_VALUES:
case INSERT_BC_VALUES:
case ADD_BC_VALUES:
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported InsertMode");
}
PetscFunctionReturn(0);
} else if (!ta->keytable[hash]) {
if (ta->count < 5*(ta->tablesize/6) - 1) {
ta->count++; /* add */
ta->keytable[hash] = key;
ta->table[hash] = data;
} else {
ierr = PetscTableAddExpand(ta,key,data,imode);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
hash = (hash + hashstep)%ta->tablesize;
}
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Full table");
/* PetscFunctionReturn(0); */
}
PETSC_STATIC_INLINE PetscErrorCode PetscTableAddCount(PetscTable ta,PetscInt key)
{
PetscErrorCode ierr;
PetscInt i,hash = (PetscInt)PetscHash(ta,(unsigned long)key);
PetscInt hashstep = (PetscInt)PetscHashStep(ta,(unsigned long)key);
PetscFunctionBegin;
if (key <= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"key <= 0");
if (key > ta->maxkey) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"key %D is greater than largest key allowed %D",key,ta->maxkey);
for (i=0; i<ta->tablesize; i++) {
if (ta->keytable[hash] == key) {
PetscFunctionReturn(0);
} else if (!ta->keytable[hash]) {
if (ta->count < 5*(ta->tablesize/6) - 1) {
ta->count++; /* add */
ta->keytable[hash] = key;
ta->table[hash] = ta->count;
} else {
ierr = PetscTableAddCountExpand(ta,key);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
hash = (hash + hashstep)%ta->tablesize;
}
SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Full table");
/* PetscFunctionReturn(0); */
}
/*
PetscTableFind - checks if a key is in the table
If data==0, then no table entry exists.
*/
PETSC_STATIC_INLINE PetscErrorCode PetscTableFind(PetscTable ta,PetscInt key,PetscInt *data)
{
PetscInt ii = 0;
PetscInt hash = (PetscInt)PetscHash(ta,(unsigned long)key);
PetscInt hashstep = (PetscInt)PetscHashStep(ta,(unsigned long)key);
PetscFunctionBegin;
*data = 0;
if (key <= 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Key <= 0");
if (key > ta->maxkey) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"key %D is greater than largest key allowed %D",key,ta->maxkey);
while (ii++ < ta->tablesize) {
if (!ta->keytable[hash]) break;
else if (ta->keytable[hash] == key) {
*data = ta->table[hash];
break;
}
hash = (hash + hashstep)%ta->tablesize;
}
PetscFunctionReturn(0);
}
#endif
|
