| [1963] | 1 | /* | 
|---|
|  | 2 | Bullet Continuous Collision Detection and Physics Library | 
|---|
|  | 3 | Copyright (c) 2003-2006 Erwin Coumans  http://continuousphysics.com/Bullet/ | 
|---|
|  | 4 |  | 
|---|
|  | 5 | This software is provided 'as-is', without any express or implied warranty. | 
|---|
|  | 6 | In no event will the authors be held liable for any damages arising from the use of this software. | 
|---|
|  | 7 | Permission is granted to anyone to use this software for any purpose, | 
|---|
|  | 8 | including commercial applications, and to alter it and redistribute it freely, | 
|---|
|  | 9 | subject to the following restrictions: | 
|---|
|  | 10 |  | 
|---|
|  | 11 | 1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required. | 
|---|
|  | 12 | 2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software. | 
|---|
|  | 13 | 3. This notice may not be removed or altered from any source distribution. | 
|---|
|  | 14 | */ | 
|---|
|  | 15 |  | 
|---|
|  | 16 | #include "btUnionFind.h" | 
|---|
|  | 17 |  | 
|---|
|  | 18 |  | 
|---|
|  | 19 |  | 
|---|
|  | 20 | btUnionFind::~btUnionFind() | 
|---|
|  | 21 | { | 
|---|
|  | 22 | Free(); | 
|---|
|  | 23 |  | 
|---|
|  | 24 | } | 
|---|
|  | 25 |  | 
|---|
|  | 26 | btUnionFind::btUnionFind() | 
|---|
|  | 27 | { | 
|---|
|  | 28 |  | 
|---|
|  | 29 | } | 
|---|
|  | 30 |  | 
|---|
|  | 31 | void    btUnionFind::allocate(int N) | 
|---|
|  | 32 | { | 
|---|
|  | 33 | m_elements.resize(N); | 
|---|
|  | 34 | } | 
|---|
|  | 35 | void    btUnionFind::Free() | 
|---|
|  | 36 | { | 
|---|
|  | 37 | m_elements.clear(); | 
|---|
|  | 38 | } | 
|---|
|  | 39 |  | 
|---|
|  | 40 |  | 
|---|
|  | 41 | void    btUnionFind::reset(int N) | 
|---|
|  | 42 | { | 
|---|
|  | 43 | allocate(N); | 
|---|
|  | 44 |  | 
|---|
|  | 45 | for (int i = 0; i < N; i++) | 
|---|
|  | 46 | { | 
|---|
|  | 47 | m_elements[i].m_id = i; m_elements[i].m_sz = 1; | 
|---|
|  | 48 | } | 
|---|
|  | 49 | } | 
|---|
|  | 50 |  | 
|---|
|  | 51 |  | 
|---|
|  | 52 | class btUnionFindElementSortPredicate | 
|---|
|  | 53 | { | 
|---|
|  | 54 | public: | 
|---|
|  | 55 |  | 
|---|
|  | 56 | bool operator() ( const btElement& lhs, const btElement& rhs ) | 
|---|
|  | 57 | { | 
|---|
|  | 58 | return lhs.m_id < rhs.m_id; | 
|---|
|  | 59 | } | 
|---|
|  | 60 | }; | 
|---|
|  | 61 |  | 
|---|
|  | 62 | ///this is a special operation, destroying the content of btUnionFind. | 
|---|
|  | 63 | ///it sorts the elements, based on island id, in order to make it easy to iterate over islands | 
|---|
|  | 64 | void    btUnionFind::sortIslands() | 
|---|
|  | 65 | { | 
|---|
|  | 66 |  | 
|---|
|  | 67 | //first store the original body index, and islandId | 
|---|
|  | 68 | int numElements = m_elements.size(); | 
|---|
|  | 69 |  | 
|---|
|  | 70 | for (int i=0;i<numElements;i++) | 
|---|
|  | 71 | { | 
|---|
|  | 72 | m_elements[i].m_id = find(i); | 
|---|
|  | 73 | m_elements[i].m_sz = i; | 
|---|
|  | 74 | } | 
|---|
|  | 75 |  | 
|---|
|  | 76 | // Sort the vector using predicate and std::sort | 
|---|
|  | 77 | //std::sort(m_elements.begin(), m_elements.end(), btUnionFindElementSortPredicate); | 
|---|
|  | 78 | m_elements.quickSort(btUnionFindElementSortPredicate()); | 
|---|
|  | 79 |  | 
|---|
|  | 80 | } | 
|---|
|  | 81 |  | 
|---|