|
Blender V4.3
|
#include "MEM_guardedalloc.h"#include "BLI_kdtree_impl.h"#include "BLI_math_base.h"#include "BLI_utildefines.h"#include <string.h>#include "BLI_strict_flags.h"Go to the source code of this file.
Classes | |
| struct | KDTreeNode_head |
| struct | KDTreeNode |
| struct | KDTree |
| struct | DeDuplicateParams |
Macros | |
| #define | _BLI_KDTREE_CONCAT_AUX(MACRO_ARG1, MACRO_ARG2) MACRO_ARG1##MACRO_ARG2 |
| #define | _BLI_KDTREE_CONCAT(MACRO_ARG1, MACRO_ARG2) _BLI_KDTREE_CONCAT_AUX(MACRO_ARG1, MACRO_ARG2) |
| #define | BLI_kdtree_nd_(id) _BLI_KDTREE_CONCAT(KDTREE_PREFIX_ID, _##id) |
| #define | KD_STACK_INIT 100 /* initial size for array (on the stack) */ |
| #define | KD_NEAR_ALLOC_INC 100 /* alloc increment for collecting nearest */ |
| #define | KD_FOUND_ALLOC_INC 50 /* alloc increment for collecting nearest */ |
| #define | KD_NODE_UNSET ((uint)-1) |
| #define | KD_NODE_ROOT_IS_INIT ((uint)-2) |
| #define | NODE_TEST_NEAREST(node) |
Typedefs | |
| typedef struct KDTreeNode_head | KDTreeNode_head |
| typedef struct KDTreeNode | KDTreeNode |
Functions | |
| KDTree *BLI_kdtree_nd_ | new (uint nodes_len_capacity) |
| void BLI_kdtree_nd_ | free (KDTree *tree) |
| void BLI_kdtree_nd_ | insert (KDTree *tree, int index, const float co[KD_DIMS]) |
| static uint | kdtree_balance (KDTreeNode *nodes, uint nodes_len, uint axis, const uint ofs) |
| void BLI_kdtree_nd_ | balance (KDTree *tree) |
| static uint * | realloc_nodes (uint *stack, uint *stack_len_capacity, const bool is_alloc) |
| int BLI_kdtree_nd_ | find_nearest (const KDTree *tree, const float co[KD_DIMS], KDTreeNearest *r_nearest) |
| int BLI_kdtree_nd_ | find_nearest_cb (const KDTree *tree, const float co[KD_DIMS], int(*filter_cb)(void *user_data, int index, const float co[KD_DIMS], float dist_sq), void *user_data, KDTreeNearest *r_nearest) |
| static void | nearest_ordered_insert (KDTreeNearest *nearest, uint *nearest_len, const uint nearest_len_capacity, const int index, const float dist, const float co[KD_DIMS]) |
| int BLI_kdtree_nd_ | find_nearest_n_with_len_squared_cb (const KDTree *tree, const float co[KD_DIMS], KDTreeNearest r_nearest[], const uint nearest_len_capacity, float(*len_sq_fn)(const float co_search[KD_DIMS], const float co_test[KD_DIMS], const void *user_data), const void *user_data) |
| int BLI_kdtree_nd_ | find_nearest_n (const KDTree *tree, const float co[KD_DIMS], KDTreeNearest r_nearest[], uint nearest_len_capacity) |
| static int | nearest_cmp_dist (const void *a, const void *b) |
| static void | nearest_add_in_range (KDTreeNearest **r_nearest, uint nearest_index, uint *nearest_len_capacity, const int index, const float dist, const float co[KD_DIMS]) |
| int BLI_kdtree_nd_ | range_search_with_len_squared_cb (const KDTree *tree, const float co[KD_DIMS], KDTreeNearest **r_nearest, const float range, float(*len_sq_fn)(const float co_search[KD_DIMS], const float co_test[KD_DIMS], const void *user_data), const void *user_data) |
| int BLI_kdtree_nd_ | range_search (const KDTree *tree, const float co[KD_DIMS], KDTreeNearest **r_nearest, float range) |
| void BLI_kdtree_nd_ | range_search_cb (const KDTree *tree, const float co[KD_DIMS], float range, bool(*search_cb)(void *user_data, int index, const float co[KD_DIMS], float dist_sq), void *user_data) |
| static int * | kdtree_order (const KDTree *tree) |
Local Math API | |
| static void | copy_vn_vn (float v0[KD_DIMS], const float v1[KD_DIMS]) |
| static float | len_squared_vnvn (const float v0[KD_DIMS], const float v1[KD_DIMS]) |
| static float | len_squared_vnvn_cb (const float co_kdtree[KD_DIMS], const float co_search[KD_DIMS], const void *UNUSED(user_data)) |
BLI_kdtree_3d_calc_duplicates_fast | |
| static void | deduplicate_recursive (const struct DeDuplicateParams *p, uint i) |
| int BLI_kdtree_nd_ | calc_duplicates_fast (const KDTree *tree, const float range, bool use_index_order, int *duplicates) |
BLI_kdtree_3d_deduplicate | |
| static int | kdtree_cmp_bool (const bool a, const bool b) |
| static int | kdtree_node_cmp_deduplicate (const void *n0_p, const void *n1_p) |
| int BLI_kdtree_nd_ | deduplicate (KDTree *tree) |
| #define _BLI_KDTREE_CONCAT | ( | MACRO_ARG1, | |
| MACRO_ARG2 ) _BLI_KDTREE_CONCAT_AUX(MACRO_ARG1, MACRO_ARG2) |
Definition at line 20 of file kdtree_impl.h.
| #define _BLI_KDTREE_CONCAT_AUX | ( | MACRO_ARG1, | |
| MACRO_ARG2 ) MACRO_ARG1##MACRO_ARG2 |
Definition at line 19 of file kdtree_impl.h.
| #define BLI_kdtree_nd_ | ( | id | ) | _BLI_KDTREE_CONCAT(KDTREE_PREFIX_ID, _##id) |
Definition at line 21 of file kdtree_impl.h.
Referenced by find_nearest_n(), and range_search().
| #define KD_FOUND_ALLOC_INC 50 /* alloc increment for collecting nearest */ |
Definition at line 49 of file kdtree_impl.h.
Referenced by nearest_add_in_range().
| #define KD_NEAR_ALLOC_INC 100 /* alloc increment for collecting nearest */ |
Definition at line 48 of file kdtree_impl.h.
Referenced by realloc_nodes().
When set we know all values are unbalanced, otherwise clear them when re-balancing: see #62210.
Definition at line 57 of file kdtree_impl.h.
Definition at line 51 of file kdtree_impl.h.
Referenced by balance(), deduplicate_recursive(), find_nearest(), find_nearest_cb(), find_nearest_n_with_len_squared_cb(), insert(), kdtree_balance(), range_search_cb(), and range_search_with_len_squared_cb().
Definition at line 47 of file kdtree_impl.h.
Referenced by find_nearest(), find_nearest_cb(), find_nearest_n_with_len_squared_cb(), range_search_cb(), and range_search_with_len_squared_cb().
| #define NODE_TEST_NEAREST | ( | node | ) |
Referenced by find_nearest_cb().
| typedef struct KDTreeNode KDTreeNode |
| typedef struct KDTreeNode_head KDTreeNode_head |
| void BLI_kdtree_nd_ balance | ( | KDTree * | tree | ) |
Definition at line 198 of file kdtree_impl.h.
References KD_NODE_ROOT_IS_INIT, KD_NODE_UNSET, kdtree_balance(), and tree.
| int BLI_kdtree_nd_ calc_duplicates_fast | ( | const KDTree * | tree, |
| const float | range, | ||
| bool | use_index_order, | ||
| int * | duplicates ) |
Find duplicate points in range. Favors speed over quality since it doesn't find the best target vertex for merging. Nodes are looped over, duplicates are added when found. Nevertheless results are predictable.
| range | Coordinates in this range are candidates to be merged. |
| use_index_order | Loop over the coordinates ordered by KDTreeNode.index At the expense of some performance, this ensures the layout of the tree doesn't influence the iteration order. |
| duplicates | An array of int's the length of KDTree.nodes_len Values initialized to -1 are candidates to me merged. Setting the index to its own position in the array prevents it from being touched, although it can still be used as a target. |
Definition at line 883 of file kdtree_impl.h.
References copy_vn_vn(), deduplicate_recursive(), DeDuplicateParams::duplicates, ELEM, KDTreeNode::index, kdtree_order(), MEM_freeN(), DeDuplicateParams::nodes, range, DeDuplicateParams::search, DeDuplicateParams::search_co, square_f(), and tree.
Definition at line 63 of file kdtree_impl.h.
References KD_DIMS.
Referenced by calc_duplicates_fast(), find_nearest(), find_nearest_cb(), insert(), nearest_add_in_range(), and nearest_ordered_insert().
| int BLI_kdtree_nd_ deduplicate | ( | KDTree * | tree | ) |
Remove exact duplicates (run before balancing).
Keep the first element added when duplicates are found.
Definition at line 980 of file kdtree_impl.h.
References KD_DIMS, kdtree_node_cmp_deduplicate(), and tree.
|
static |
Definition at line 836 of file kdtree_impl.h.
References deduplicate_recursive(), DeDuplicateParams::duplicates, DeDuplicateParams::duplicates_found, int, KD_NODE_UNSET, len_squared_vnvn(), DeDuplicateParams::nodes, DeDuplicateParams::range, DeDuplicateParams::range_sq, DeDuplicateParams::search, and DeDuplicateParams::search_co.
Referenced by calc_duplicates_fast(), and deduplicate_recursive().
| int BLI_kdtree_nd_ find_nearest | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| KDTreeNearest * | r_nearest ) |
Find nearest returns index, and -1 if no node is found.
Definition at line 230 of file kdtree_impl.h.
References BLI_assert, KDTreeNode::co, copy_vn_vn(), KDTreeNode::d, KDTreeNode::index, KD_DIMS, KD_NODE_UNSET, KD_STACK_INIT, KDTreeNode::left, len_squared_vnvn(), MEM_freeN(), node, realloc_nodes(), KDTreeNode::right, sqrtf, tree, and UNLIKELY.
| int BLI_kdtree_nd_ find_nearest_cb | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| int(* | filter_cb )(void *user_data, int index, const float co[KD_DIMS], float dist_sq), | ||
| void * | user_data, | ||
| KDTreeNearest * | r_nearest ) |
A version of #BLI_kdtree_3d_find_nearest which runs a callback to filter out values.
| filter_cb | Filter find results, Return codes: (1: accept, 0: skip, -1: immediate exit). |
Definition at line 336 of file kdtree_impl.h.
References ARRAY_SIZE, BLI_assert, KDTreeNearest::co, KDTreeNode::co, copy_vn_vn(), KDTreeNearest::dist, FLT_MAX, KDTreeNearest::index, KDTreeNode::index, KD_DIMS, KD_NODE_UNSET, KD_STACK_INIT, MEM_freeN(), NODE_TEST_NEAREST, NULL, realloc_nodes(), sqrtf, tree, and UNLIKELY.
| int BLI_kdtree_nd_ find_nearest_n | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| KDTreeNearest | r_nearest[], | ||
| uint | nearest_len_capacity ) |
Definition at line 588 of file kdtree_impl.h.
References BLI_kdtree_nd_, find_nearest_n_with_len_squared_cb(), NULL, and tree.
| int BLI_kdtree_nd_ find_nearest_n_with_len_squared_cb | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| KDTreeNearest | r_nearest[], | ||
| const uint | nearest_len_capacity, | ||
| float(* | len_sq_fn )(const float co_search[KD_DIMS], const float co_test[KD_DIMS], const void *user_data), | ||
| const void * | user_data ) |
Find nearest_len_capacity nearest returns number of points found, with results in nearest.
| r_nearest | An array of nearest, sized at least nearest_len_capacity. |
Definition at line 474 of file kdtree_impl.h.
References ARRAY_SIZE, BLI_assert, KDTreeNode::co, KDTreeNode::d, KDTreeNearest::dist, KDTreeNearest::index, KDTreeNode::index, KD_DIMS, KD_NODE_UNSET, KD_STACK_INIT, KDTreeNode::left, len_squared_vnvn_cb(), MEM_freeN(), nearest_ordered_insert(), NULL, realloc_nodes(), KDTreeNode::right, sqrtf, tree, and UNLIKELY.
Referenced by find_nearest_n().
| void BLI_kdtree_nd_ free | ( | KDTree * | tree | ) |
Definition at line 109 of file kdtree_impl.h.
References MEM_freeN(), and tree.
Referenced by blender::ResourceScope::add(), iTaSC::Cache::addCacheItem(), iTaSC::Armature::addConstraint(), aligned_free(), libmv::aligned_free(), arg_handle_addons_set(), Freestyle::BezierII(), BKE_blender_atexit(), BKE_blender_atexit_unregister(), BKE_gpencil_stroke_editcurve_generate(), BKE_rigidbody_free_world(), BLI_dynstr_clear(), BLI_exists(), BLI_freelist(), BLI_getenv(), BLI_rng_shuffle_array(), BLI_system_backtrace(), btFreeDefault(), button_matches_search_filter(), callback_main_atexit(), ccl_try_align(), iTaSC::CacheChannel::clear(), blender::gpu::MTLUniformBuf::clear_to_zero(), iTaSC::Cache::clearCacheFrom(), createGPUShader(), curve_draw_exec(), blender::ed::curves::curves_draw_exec(), CustomData_external_write(), data_device_handle_drop(), blender::RawAllocator::deallocate(), GuardedAllocator< T >::deallocate(), MemoryAllocator< N >::destroy(), device_oneapi_capabilities(), ED_region_draw_cb_remove_by_type(), event_to_buf(), Freestyle::FitCurveWrapper::FitCubic(), blender::bke::pbvh::free_tree(), GPUDevice::generic_alloc(), blender::ed::geometry::geometry_attribute_add_invoke(), blender::gpu::MTLContext::get_null_buffer(), GHOST_SystemX11::getClipboard(), GHOST_SystemX11::getClipboard_xcout(), GHOST_SystemWayland::getClipboardImage(), GHOST_DropTargetX11::getGhostData(), GHOST_DropTargetX11::GHOST_HandleClientMessage(), GHOST_URL_decode_alloc(), GHOST_WindowWin32::GHOST_WindowWin32(), GHOST_WindowX11::GHOST_WindowX11(), gim_free(), GIZMO_GT_snap_3d(), Freestyle::gts_vertex_principal_directions(), gwl_simple_buffer_free_data(), gwl_simple_buffer_set_from_string(), gwl_window_cursor_custom_free(), GHOST_SystemCocoa::handleDraggingEvent(), IFileStream::IFileStream(), imb_free_dds_buffer(), imb_savewebp(), insert_key_menu_invoke(), MEM_guarded_printmemlist_stats(), MEM_lockfree_freeN(), memfd_create_sealed(), menu_item_enum_opname_menu_active(), blender::ed::space_node::node_space_subtype_item_extend(), blender::ed::object::object_remove_parent_deform_modifiers(), blender::ed::object::ocean_bake_exec(), OFileStream::OFileStream(), osx_user_locale(), blender::ed::greasepencil::polyline_detect_corners(), blender::ed::greasepencil::polyline_fit_curve(), processEvent(), GHOST_SystemX11::putClipboard(), pyrna_enum_as_string(), pyrna_prop_to_enum_bitfield(), RB_shape_delete(), read_file_as_buffer(), rect_bevel_smooth(), recursive_operation_impl(), rem_memblock(), RNA_enum_is_equal(), RNA_property_as_string(), RNA_property_enum_bitflag_identifiers(), RNA_property_enum_identifier(), RNA_property_enum_item_from_value(), RNA_property_enum_items_gettexted_all(), RNA_property_enum_name(), RNA_property_enum_step(), RNA_property_enum_value(), GHOST_WindowWin32::setTitle(), GHOST_SystemWin32::showMessageBox(), GHOST_SystemX11::showMessageBox(), SKY_arhosekskymodelstate_free(), space_type_set_or_cycle_exec(), split(), strbuf_free(), thumb_data_vertical_flip(), u_free_getenv(), UI_block_set_active_operator(), ui_but_event_property_operator_string(), ui_def_but_rna(), ui_def_but_rna__menu(), ui_icon_view_menu_cb(), ui_item_enum_expand_exec(), ui_item_rna_size(), ui_menu_enumpropname(), uiItemEnumO_string(), uiItemEnumR_string_prop(), uiItemsEnumR(), uiItemsFullEnumO(), util_aligned_free(), where_am_i(), WM_clipboard_image_get(), wm_clipboard_text_get_ex(), WM_jobs_customdata_set(), WM_paint_cursor_remove_by_type(), blender::io::grease_pencil::PDFExporter::write_to_file(), blender::io::grease_pencil::SVGExporter::write_to_file(), wWinMain(), iTaSC::CacheEntry::~CacheEntry(), GHOST_ContextWGL::~GHOST_ContextWGL(), GHOST_EventDragnDrop::~GHOST_EventDragnDrop(), GHOST_EventString::~GHOST_EventString(), and iTaSC::Armature::JointConstraint_struct::~JointConstraint_struct().
| void BLI_kdtree_nd_ insert | ( | KDTree * | tree, |
| int | index, | ||
| const float | co[KD_DIMS] ) |
Construction: first insert points, then call balance. Normal is optional.
Definition at line 120 of file kdtree_impl.h.
References BLI_assert, copy_vn_vn(), KD_NODE_UNSET, MAX2, and tree.
|
static |
Definition at line 142 of file kdtree_impl.h.
References KDTreeNode::co, KD_DIMS, KD_NODE_UNSET, kdtree_balance(), left, node, and SWAP.
Referenced by balance(), and kdtree_balance().
|
static |
Definition at line 943 of file kdtree_impl.h.
References b.
Referenced by kdtree_node_cmp_deduplicate().
|
static |
Definition at line 951 of file kdtree_impl.h.
References KDTreeNode::co, KDTreeNode::d, KD_DIMS, and kdtree_cmp_bool().
Referenced by deduplicate().
Use when we want to loop over nodes ordered by index. Requires indices to be aligned with nodes.
Definition at line 807 of file kdtree_impl.h.
References int, MEM_mallocN, and tree.
Referenced by calc_duplicates_fast().
Definition at line 70 of file kdtree_impl.h.
References KD_DIMS, and square_f().
Referenced by deduplicate_recursive(), find_nearest(), len_squared_vnvn_cb(), and range_search_cb().
|
static |
Definition at line 79 of file kdtree_impl.h.
References len_squared_vnvn().
Referenced by find_nearest_n_with_len_squared_cb(), and range_search_with_len_squared_cb().
|
static |
Definition at line 612 of file kdtree_impl.h.
References copy_vn_vn(), KD_FOUND_ALLOC_INC, MEM_reallocN_id, sqrtf, and UNLIKELY.
Referenced by range_search_with_len_squared_cb().
|
static |
Definition at line 597 of file kdtree_impl.h.
References b, and KDTreeNearest::dist.
Referenced by range_search_with_len_squared_cb().
|
static |
Definition at line 442 of file kdtree_impl.h.
References copy_vn_vn().
Referenced by find_nearest_n_with_len_squared_cb().
| KDTree *BLI_kdtree_nd_ new | ( | uint | nodes_len_capacity | ) |
Creates or free a kdtree
Definition at line 91 of file kdtree_impl.h.
References KD_NODE_ROOT_IS_INIT, MEM_mallocN, and tree.
| int BLI_kdtree_nd_ range_search | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| KDTreeNearest ** | r_nearest, | ||
| float | range ) |
Definition at line 720 of file kdtree_impl.h.
References BLI_kdtree_nd_, NULL, range, range_search_with_len_squared_cb(), and tree.
| void BLI_kdtree_nd_ range_search_cb | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| float | range, | ||
| bool(* | search_cb )(void *user_data, int index, const float co[KD_DIMS], float dist_sq), | ||
| void * | user_data ) |
A version of #BLI_kdtree_3d_range_search which runs a callback instead of allocating an array.
| search_cb | Called for every node found in range, false return value performs an early exit. |
Definition at line 737 of file kdtree_impl.h.
References ARRAY_SIZE, BLI_assert, KD_DIMS, KD_NODE_UNSET, KD_STACK_INIT, len_squared_vnvn(), MEM_freeN(), range, realloc_nodes(), tree, and UNLIKELY.
| int BLI_kdtree_nd_ range_search_with_len_squared_cb | ( | const KDTree * | tree, |
| const float | co[KD_DIMS], | ||
| KDTreeNearest ** | r_nearest, | ||
| const float | range, | ||
| float(* | len_sq_fn )(const float co_search[KD_DIMS], const float co_test[KD_DIMS], const void *user_data), | ||
| const void * | user_data ) |
Range search returns number of points nearest_len, with results in nearest
| r_nearest | Allocated array of nearest nearest_len (caller is responsible for freeing). |
Definition at line 638 of file kdtree_impl.h.
References ARRAY_SIZE, BLI_assert, KD_DIMS, KD_NODE_UNSET, KD_STACK_INIT, len_squared_vnvn_cb(), MEM_freeN(), nearest_add_in_range(), nearest_cmp_dist(), NULL, range, realloc_nodes(), tree, and UNLIKELY.
Referenced by range_search().
Definition at line 214 of file kdtree_impl.h.
References KD_NEAR_ALLOC_INC, MEM_freeN(), and MEM_mallocN.
Referenced by find_nearest(), find_nearest_cb(), find_nearest_n_with_len_squared_cb(), range_search_cb(), and range_search_with_len_squared_cb().