1 #pragma once
2
3 #include <petsc/private/hashtable.h>
4 #include <petsc/private/pcbddcstructsimpl.h>
5
PCBDDCGraphNodeHash(const PCBDDCGraphNode * node)6 static inline PetscHash_t PCBDDCGraphNodeHash(const PCBDDCGraphNode *node)
7 {
8 PetscHash_t hash;
9 hash = PetscHashCombine(PetscHashInt(node->count), PetscHashInt(node->which_dof));
10 hash = PetscHashCombine(hash, PetscHashInt(node->special_dof));
11 for (PetscInt i = 0; i < node->count; i++) hash = PetscHashCombine(hash, PetscHashInt(node->neighbours_set[i]));
12 hash = PetscHashCombine(hash, PetscHashInt(node->local_groups_count));
13 if (!node->shared) {
14 for (PetscInt i = 0; i < node->local_groups_count; i++) hash = PetscHashCombine(hash, PetscHashInt(node->local_groups[i]));
15 }
16 return hash;
17 }
18
PCBDDCGraphNodeEqual(const PCBDDCGraphNode * a,const PCBDDCGraphNode * b)19 static inline int PCBDDCGraphNodeEqual(const PCBDDCGraphNode *a, const PCBDDCGraphNode *b)
20 {
21 if (a->count != b->count) return 0;
22 if (a->which_dof != b->which_dof) return 0;
23 if (a->special_dof != b->special_dof) return 0;
24 /* check only for same local groups if not shared
25 shared dofs at the process boundaries will be handled differently */
26 PetscBool mpi_shared = a->shared;
27 PetscBool same_set;
28 PetscCallContinue(PetscArraycmp(a->neighbours_set, b->neighbours_set, a->count, &same_set));
29 if (same_set && !mpi_shared) {
30 if (a->local_groups_count != b->local_groups_count) same_set = PETSC_FALSE;
31 else PetscCallContinue(PetscArraycmp(a->local_groups, b->local_groups, a->local_groups_count, &same_set));
32 }
33 return same_set ? 1 : 0;
34 }
35
36 PETSC_HASH_MAP(HMapPCBDDCGraphNode, PCBDDCGraphNode *, PetscInt, PCBDDCGraphNodeHash, PCBDDCGraphNodeEqual, -1)
37