Lines Matching refs:bucket_indices
32 PetscCount *bucket_indices; member
53 PetscCall(PetscFree((*tree)->bucket_indices)); in PetscKDTreeDestroy()
133 PetscSegBuffer stems, leaves, bucket_indices, bucket_coords; member
148 PetscCount *bucket_indices; in PetscKDTreeBuildStemAndLeaves() local
156 PetscCall(PetscSegBufferGetSize(kd_build->bucket_indices, &leaf->indices_handle)); in PetscKDTreeBuildStemAndLeaves()
157 PetscCall(PetscSegBufferGet(kd_build->bucket_indices, leaf->count, &bucket_indices)); in PetscKDTreeBuildStemAndLeaves()
158 PetscCall(PetscArraycpy(bucket_indices, &sorted_indices[start], leaf->count)); in PetscKDTreeBuildStemAndLeaves()
165 … = 0; d < dim; d++) bucket_coords[d * leaf->count + i] = tree->coords[bucket_indices[i] * dim + d]; in PetscKDTreeBuildStemAndLeaves()
290 PetscCall(PetscSegBufferCreate(sizeof(PetscCount), num_coords, &kd_build->bucket_indices)); in PetscKDTreeCreate()
297 PetscCall(PetscSegBufferGetSize(kd_build->bucket_indices, &tree->num_bucket_indices)); in PetscKDTreeCreate()
300 PetscCall(PetscSegBufferExtractAlloc(kd_build->bucket_indices, &tree->bucket_indices)); in PetscKDTreeCreate()
309 PetscCall(PetscSegBufferDestroy(&kd_build->bucket_indices)); in PetscKDTreeCreate()
332 PetscCount point_index = tree->bucket_indices[leaf.indices_handle + i]; in PetscKDTreeQueryLeaf()
355 *index = tree->bucket_indices[leaf.indices_handle + i]; in PetscKDTreeQueryLeaf_CopyCoords()
493 PetscCount bucket_index = tree->bucket_indices[leaf.indices_handle + j]; in PetscKDTreeView()