svn merge ^/trunk/blender -r47201:47210
[blender.git] / source / blender / blenlib / BLI_kdopbvh.h
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2006 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Daniel Genrich, Andre Pinto
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28
29 #ifndef __BLI_KDOPBVH_H__
30 #define __BLI_KDOPBVH_H__
31
32 /** \file BLI_kdopbvh.h
33  *  \ingroup bli
34  *  \author Daniel Genrich
35  *  \author Andre Pinto
36  */
37
38 #ifdef __cplusplus
39 extern "C" { 
40 #endif
41
42 #include <float.h>
43
44 struct BVHTree;
45 typedef struct BVHTree BVHTree;
46
47 typedef struct BVHTreeOverlap {
48         int indexA;
49         int indexB;
50 } BVHTreeOverlap;
51
52 typedef struct BVHTreeNearest {
53         int index;          /* the index of the nearest found (untouched if none is found within a dist radius from the given coordinates) */
54         float co[3];        /* nearest coordinates (untouched it none is found within a dist radius from the given coordinates) */
55         float no[3];        /* normal at nearest coordinates (untouched it none is found within a dist radius from the given coordinates) */
56         float dist;         /* squared distance to search arround */
57 } BVHTreeNearest;
58
59 typedef struct BVHTreeRay {
60         float origin[3];    /* ray origin */
61         float direction[3]; /* ray direction */
62         float radius;       /* radius around ray */
63 } BVHTreeRay;
64
65 typedef struct BVHTreeRayHit {
66         int index;          /* index of the tree node (untouched if no hit is found) */
67         float co[3];        /* coordinates of the hit point */
68         float no[3];        /* normal on hit point */
69         float dist;         /* distance to the hit point */
70 } BVHTreeRayHit;
71
72 /* callback must update nearest in case it finds a nearest result */
73 typedef void (*BVHTree_NearestPointCallback)(void *userdata, int index, const float *co, BVHTreeNearest *nearest);
74
75 /* callback must update hit in case it finds a nearest successful hit */
76 typedef void (*BVHTree_RayCastCallback)(void *userdata, int index, const BVHTreeRay *ray, BVHTreeRayHit *hit);
77
78 /* callback to range search query */
79 typedef void (*BVHTree_RangeQuery)(void *userdata, int index, float squared_dist);
80
81 BVHTree *BLI_bvhtree_new(int maxsize, float epsilon, char tree_type, char axis);
82 void BLI_bvhtree_free(BVHTree *tree);
83
84 /* construct: first insert points, then call balance */
85 int BLI_bvhtree_insert(BVHTree *tree, int index, const float *co, int numpoints);
86 void BLI_bvhtree_balance(BVHTree *tree);
87
88 /* update: first update points/nodes, then call update_tree to refit the bounding volumes */
89 int BLI_bvhtree_update_node(BVHTree *tree, int index, const float *co, const float *co_moving, int numpoints);
90 void BLI_bvhtree_update_tree(BVHTree *tree);
91
92 /* collision/overlap: check two trees if they overlap, alloc's *overlap with length of the int return value */
93 BVHTreeOverlap *BLI_bvhtree_overlap(BVHTree *tree1, BVHTree *tree2, unsigned int *result);
94
95 float BLI_bvhtree_getepsilon(BVHTree *tree);
96
97 /* find nearest node to the given coordinates (if nearest is given it will only search nodes where square distance is smaller than nearest->dist) */
98 int BLI_bvhtree_find_nearest(BVHTree *tree, const float co[3], BVHTreeNearest *nearest, BVHTree_NearestPointCallback callback, void *userdata);
99
100 int BLI_bvhtree_ray_cast(BVHTree *tree, const float co[3], const float *dir, float radius, BVHTreeRayHit *hit, BVHTree_RayCastCallback callback, void *userdata);
101
102 float BLI_bvhtree_bb_raycast(float *bv, const float light_start[3], const float light_end[3], float pos[3]);
103
104 /* range query */
105 int BLI_bvhtree_range_query(BVHTree *tree, const float co[3], float radius, BVHTree_RangeQuery callback, void *userdata);
106
107 #ifdef __cplusplus
108 }
109 #endif
110
111 #endif // __BLI_KDOPBVH_H__
112