void ModelWriter::visitFaces(SUEntitiesRef ents, VisitorFunc func, bool recursive, Transform transform)
{
size_t count =0;
SUEntitiesGetNumFaces(ents,&count);
vector<SUFaceRef> faces(count);
SUEntitiesGetFaces(ents,count,faces.data(),&count);
for(int i=0; i < count; i++){
(*this.*func)(faces[i], transform);
}
SUEntitiesGetNumGroups(ents,&count);
vector<SUGroupRef> groups(count);
SUEntitiesGetGroups(ents,count,groups.data(),&count);
for(int i=0; i < count; i++){
SUEntitiesRef ents2 = SU_INVALID;
SUGroupGetEntities(groups[i],&ents2);
Transform t;
SUGroupGetTransform(groups[i], (SUTransformation*)&t);
visitFaces(ents2,func,true, transform * t);
}
if(recursive) {
SUEntitiesGetNumInstances(ents,&count);
vector<SUComponentInstanceRef> instances(count);
SUEntitiesGetInstances(ents,count,instances.data(),&count);
for(int i=0; i < count; i++){
Transform t;
SUComponentDefinitionRef def = SU_INVALID;
SUEntitiesRef ents2 = SU_INVALID;
SUComponentInstanceGetTransform(instances[i],(SUTransformation*)&t);
SUComponentInstanceGetDefinition(instances[i],&def);
SUComponentDefinitionGetEntities(def,&ents2);
visitFaces(ents2,func,true, transform * t);
}
}
}
int SkpModel::LoadEntitiesRecursive(SUEntitiesRef entities){
int out = 0;
//Get Entities from Instances
size_t instanceCount;
SUResult res = SUEntitiesGetNumInstances(entities, &instanceCount);
if (res != SU_ERROR_NONE)
return 1;
if (instanceCount > 0) {
std::vector<SUComponentInstanceRef> instances(instanceCount);
std::vector<SUEntitiesRef> entities2(instanceCount);
SUEntitiesGetInstances(entities, instanceCount,
&instances[0], &instanceCount);
out += Instances2Entities(&instances, &entities2);
for (size_t i = 0; i < instanceCount; i++) {
entities_.push_back(entities2[i]);
LoadEntitiesRecursive(entities2[i]);
}
}
//Get Entities from groups
size_t groupCount = 0;
res = SUEntitiesGetNumGroups(entities, &groupCount);
if (res != SU_ERROR_NONE)
return 1;
if (groupCount > 0){
std::vector<SUGroupRef> groups(groupCount);
std::vector<SUEntitiesRef> entities2(groupCount);
SUEntitiesGetGroups(entities, groupCount, &groups[0], &groupCount);
out += Groups2Entities(&groups, &entities2);
for (size_t i=0; i < groupCount; i++){
entities_.push_back(entities2[i]);
LoadEntitiesRecursive(entities2[i]);
}
}
return out;
}
