historical/toontown-classic.git/panda/include/eggUtilities.I
2024-01-16 11:20:27 -06:00

120 lines
3.8 KiB
Text

/**
* PANDA 3D SOFTWARE
* Copyright (c) Carnegie Mellon University. All rights reserved.
*
* All use of this software is subject to the terms of the revised BSD
* license. You should have received a copy of this license along
* with this source code in a file named "LICENSE."
*
* @file eggUtilities.I
* @author drose
* @date 1999-02-10
*/
#include "eggGroup.h"
#include "eggPrimitive.h"
#include "eggVertexPool.h"
#include <algorithm>
/**
* Splits a vertex into two or more vertices, each an exact copy of the
* original and in the same vertex pool.
*
* The splitting is based on some arbitrary property of the primitives that
* own the vertex. In the extreme, each primitive may get a different copy of
* the vertex, although it is also possible for some primitives to still share
* vertices.
*
* This decision is made based on the function object 'sequence'. This object
* must define the following function:
*
* int operator () (const EggPrimitive *prim) const;
*
* This function returns a sequence number, which determines which primitives
* will share which vertices. The sequence number 0 refers to the original
* vertex pointer; other sequence numbers indicate new vertices. Other than
* that, the sequence number is totally arbitrary. Primitives for which the
* sequence number is the same will end up sharing the same copy of the
* vertex.
*/
template<class FunctionObject>
void
split_vertex(EggVertex *vert, const FunctionObject &sequence) {
// Did we start in a happy world?
vert->test_pref_integrity();
vert->test_gref_integrity();
EggVertexPool *pool = vert->get_pool();
// Define a map of ints to vert pointers, to indicate which sequence numbers
// we have already created vertices for.
typedef pmap<int, EggVertex *> Sequences;
Sequences _sequences;
// Get a copy of the list of primitives that reference this vertex. We must
// have a copy because we will be modifying the list as we traverse it.
typedef pvector<EggPrimitive *> Prims;
Prims prims;
prims.reserve(vert->pref_size());
std::copy(vert->pref_begin(), vert->pref_end(), std::back_inserter(prims));
// Now walk through the list of primitives that reference this vertex.
Prims::const_iterator pri;
for (pri = prims.begin(); pri != prims.end(); ++pri) {
EggPrimitive *prim = *pri;
prim->test_ref_count_integrity();
int seq = sequence(prim);
if (seq != 0) {
// Here's a new sequence number! Have we already defined it?
EggVertex *new_vert = nullptr;
Sequences::const_iterator si = _sequences.find(seq);
if (si != _sequences.end()) {
// Yes, we've seen this sequence number before. Use the same vertex.
new_vert = (*si).second;
} else {
// No, this is the first time we've encountered this sequence. Split
// the vertex.
new_vert = new EggVertex(*vert);
pool->add_vertex(new_vert);
_sequences[seq] = new_vert;
// The new vertex gets all the same group memberships as the old one.
EggVertex::GroupRef::const_iterator gri;
for (gri = vert->gref_begin(); gri != vert->gref_end(); ++gri) {
EggGroup *group = *gri;
group->ref_vertex(new_vert, group->get_vertex_membership(vert));
}
}
// Now replace the vertex in the primitive.
EggPrimitive::iterator pi;
for (pi = prim->begin(); pi != prim->end(); ++pi) {
if (*pi == vert) {
prim->replace(pi, new_vert);
}
}
}
}
#ifndef NDEBUG
// Now verify everything is still happy.
vert->test_pref_integrity();
vert->test_gref_integrity();
Sequences::const_iterator si;
for (si = _sequences.begin();
si != _sequences.end();
++si) {
EggVertex *new_vert = (*si).second;
new_vert->test_gref_integrity();
new_vert->test_pref_integrity();
}
#endif // NDEBUG
}