void MayaMeshWriter::writePoly(
const Alembic::AbcGeom::OV2fGeomParam::Sample & iUVs)
{
MStatus status = MS::kSuccess;
MFnMesh lMesh( mDagPath, &status );
if ( !status )
{
MGlobal::displayError( "MFnMesh() failed for MayaMeshWriter" );
}
std::vector<float> points;
std::vector<Alembic::Util::int32_t> facePoints;
std::vector<Alembic::Util::int32_t> pointCounts;
fillTopology(points, facePoints, pointCounts);
Alembic::AbcGeom::ON3fGeomParam::Sample normalsSamp;
std::vector<float> normals;
getPolyNormals(normals);
if (!normals.empty())
{
normalsSamp.setScope( Alembic::AbcGeom::kFacevaryingScope );
normalsSamp.setVals(Alembic::AbcGeom::N3fArraySample(
(const Imath::V3f *) &normals.front(), normals.size() / 3));
}
Alembic::AbcGeom::OPolyMeshSchema::Sample samp(
Alembic::Abc::V3fArraySample((const Imath::V3f *)&points.front(),
points.size() / 3),
Alembic::Abc::Int32ArraySample(facePoints),
Alembic::Abc::Int32ArraySample(pointCounts), iUVs, normalsSamp);
// if this mesh is animated, write out the animated geometry
if (mIsGeometryAnimated)
{
mPolySchema.set(samp);
}
else
{
mPolySchema.set(samp);
}
writeColor();
}
static void export_alembic_xform_by_buffer(AlembicArchive &archive, const RenderedBuffer & renderedBuffer, int renderedBufferIndex)
{
Alembic::AbcGeom::OObject topObj(*archive.archive, Alembic::AbcGeom::kTop);
Alembic::AbcGeom::OXform xform;
if (archive.xform_map.find(renderedBufferIndex) != archive.xform_map.end())
{
xform = archive.xform_map[renderedBufferIndex];
}
else
{
xform = Alembic::AbcGeom::OXform(topObj, "xform_" + umbase::UMStringUtil::number_to_string(renderedBufferIndex), archive.timesampling);
archive.xform_map[renderedBufferIndex] = xform;
}
bool isFirstMesh = false;
Alembic::AbcGeom::OPolyMesh polyMesh;
if (archive.mesh_map.find(renderedBufferIndex) != archive.mesh_map.end())
{
polyMesh = archive.mesh_map[renderedBufferIndex];
}
else
{
polyMesh = Alembic::AbcGeom::OPolyMesh(xform, "mesh_" + to_string(renderedBufferIndex), archive.timesampling);
archive.mesh_map[renderedBufferIndex] = polyMesh;
isFirstMesh = true;
Alembic::AbcGeom::OPolyMeshSchema &meshSchema = polyMesh.getSchema();
archive.schema_map[renderedBufferIndex] = meshSchema;
}
Alembic::AbcGeom::OPolyMeshSchema &meshSchema = archive.schema_map[renderedBufferIndex];
meshSchema.setTimeSampling(archive.timesampling);
std::vector<Alembic::Util::int32_t> faceList;
std::vector<Alembic::Util::int32_t> faceCountList;
const RenderedBuffer::UVList &uvList = renderedBuffer.uvs;
const RenderedBuffer::VertexList &vertexList = renderedBuffer.vertecies;
const RenderedBuffer::NormalList &normalList = renderedBuffer.normals;
RenderedBuffer::UVList& temporary_uv = archive.temporary_uv_list;
temporary_uv.resize(uvList.size());
RenderedBuffer::NormalList& temporary_normal = archive.temporary_normal_list;
temporary_normal.resize(normalList.size());
RenderedBuffer::VertexList& temporary_vertex = archive.temporary_vertex_list;
temporary_vertex.resize(vertexList.size());
const int materialSize = static_cast<int>(renderedBuffer.materials.size());
int totalFaceCount = 0;
for (int k = 0; k < materialSize; ++k)
{
RenderedMaterial* material = renderedBuffer.materials.at(k);
totalFaceCount += material->surface.faces.size();
}
if (archive.surface_size_map.find(renderedBufferIndex) == archive.surface_size_map.end())
{
archive.surface_size_map[renderedBufferIndex] = 0;
}
int& preSurfaceSize = archive.surface_size_map[renderedBufferIndex];
bool isFirstSurface = totalFaceCount != preSurfaceSize;
if (!isFirstMesh && isFirstSurface)
{
return;
}
preSurfaceSize = totalFaceCount;
faceCountList.resize(totalFaceCount);
faceList.resize(totalFaceCount * 3);
int faceCounter = 0;
for (int k = 0; k < materialSize; ++k)
{
RenderedMaterial* material = renderedBuffer.materials.at(k);
const int faceSize = material->surface.faces.size();
for (int n = 0; n < faceSize; ++n)
{
UMVec3i face = material->surface.faces[n];
faceList[faceCounter * 3 + 0] = (face.x - 1);
faceList[faceCounter * 3 + 1] = (face.y - 1);
faceList[faceCounter * 3 + 2] = (face.z - 1);
faceCountList[faceCounter] = 3;
++faceCounter;
}
}
Alembic::AbcGeom::OPolyMeshSchema::Sample sample;
// vertex
for (int n = 0, nsize = vertexList.size(); n < nsize; ++n)
{
temporary_vertex[n].z = -vertexList[n].z;
}
Alembic::AbcGeom::P3fArraySample positions( (const Imath::V3f *) &temporary_vertex.front(), temporary_vertex.size());
sample.setPositions(positions);
// face index
if (isFirstMesh)
{
Alembic::Abc::Int32ArraySample faceIndices(faceList);
Alembic::Abc::Int32ArraySample faceCounts(faceCountList);
sample.setFaceIndices(faceIndices);
sample.setFaceCounts(faceCounts);
}
// UVs
if (!uvList.empty() && archive.is_export_uvs)
{
for (int n = 0, nsize = uvList.size(); n < nsize; ++n)
{
temporary_uv[n].y = 1.0f - uvList[n].y;
}
Alembic::AbcGeom::OV2fGeomParam::Sample uvSample;
uvSample.setScope(Alembic::AbcGeom::kVertexScope );
uvSample.setVals(Alembic::AbcGeom::V2fArraySample( ( const Imath::V2f *) &temporary_uv.front(), temporary_uv.size()));
sample.setUVs(uvSample);
}
// Normals
if (!normalList.empty() && archive.is_export_normals)
{
for (int n = 0, nsize = normalList.size(); n < nsize; ++n)
{
temporary_normal[n].z = -normalList[n].z;
}
Alembic::AbcGeom::ON3fGeomParam::Sample normalSample;
normalSample.setScope(Alembic::AbcGeom::kVertexScope );
normalSample.setVals(Alembic::AbcGeom::N3fArraySample( (const Alembic::AbcGeom::N3f *) &temporary_normal.front(), temporary_normal.size()));
sample.setNormals(normalSample);
}
meshSchema.set(sample);
}
static void export_alembic_xform_by_material_direct(AlembicArchive &archive, const RenderedBuffer & renderedBuffer, int renderedBufferIndex)
{
Alembic::AbcGeom::OObject topObj(*archive.archive, Alembic::AbcGeom::kTop);
for (int k = 0, ksize = static_cast<int>(renderedBuffer.materials.size()); k < ksize; ++k)
{
Alembic::AbcGeom::OPolyMesh polyMesh;
const int key = renderedBufferIndex * 10000 + k;
Alembic::AbcGeom::OXform xform;
if (archive.xform_map.find(key) != archive.xform_map.end())
{
xform = archive.xform_map[key];
}
else
{
xform = Alembic::AbcGeom::OXform(topObj, "xform_" + to_string(renderedBufferIndex) + "_material_" + to_string(k) , archive.timesampling);
archive.xform_map[key] = xform;
}
bool isFirstMesh = false;
if (archive.mesh_map.find(key) != archive.mesh_map.end())
{
polyMesh = archive.mesh_map[key];
}
else
{
polyMesh = Alembic::AbcGeom::OPolyMesh(xform, "mesh_" + to_string(renderedBufferIndex) + "_material_" + to_string(k), archive.timesampling);
archive.mesh_map[key] = polyMesh;
isFirstMesh = true;
Alembic::AbcGeom::OPolyMeshSchema &meshSchema = polyMesh.getSchema();
archive.schema_map[key] = meshSchema;
}
if (archive.surface_size_map.find(key) == archive.surface_size_map.end())
{
archive.surface_size_map[key] = 0;
}
Alembic::AbcGeom::OPolyMeshSchema &meshSchema = archive.schema_map[key];
meshSchema.setTimeSampling(archive.timesampling);
Alembic::AbcGeom::OPolyMeshSchema::Sample empty;
std::vector<Alembic::Util::int32_t> faceList;
std::vector<Alembic::Util::int32_t> faceCountList;
const RenderedBuffer::UVList &uvList = renderedBuffer.uvs;
const RenderedBuffer::VertexList &vertexList = renderedBuffer.vertecies;
const RenderedBuffer::NormalList &normalList = renderedBuffer.normals;
RenderedBuffer::VertexList vertexListByMaterial;
RenderedBuffer::UVList uvListByMaterial;
RenderedBuffer::NormalList normalListByMaterial;
RenderedMaterial* material = renderedBuffer.materials.at(k);
const int materialSurfaceSize = static_cast<int>(material->surface.faces.size());
vertexListByMaterial.resize(materialSurfaceSize * 3);
faceList.resize(materialSurfaceSize * 3);
faceCountList.resize(materialSurfaceSize);
if (!uvList.empty())
{
uvListByMaterial.resize(materialSurfaceSize * 3);
}
if (!normalList.empty())
{
normalListByMaterial.resize(materialSurfaceSize * 3);
}
int& preSurfaceSize = archive.surface_size_map[key];
bool isFirstSurface = material->surface.faces.size() != preSurfaceSize;
if (!isFirstMesh && isFirstSurface)
{
continue;
}
// re assign par material
int lastIndex = 0;
for (int n = 0; n < materialSurfaceSize; ++n)
{
UMVec3i face = material->surface.faces[n];
const int f1 = face.x - 1;
const int f2 = face.y - 1;
const int f3 = face.z - 1;
int vi1 = n * 3 + 0;
int vi2 = n * 3 + 1;
int vi3 = n * 3 + 2;
vertexListByMaterial[vi1] = vertexList.at(f1);
vertexListByMaterial[vi2] = vertexList.at(f2);
vertexListByMaterial[vi3] = vertexList.at(f3);
if (!uvList.empty() && archive.is_export_uvs)
{
uvListByMaterial[vi1] = uvList.at(f1);
uvListByMaterial[vi2] = uvList.at(f2);
uvListByMaterial[vi3] = uvList.at(f3);
}
if (!normalList.empty() && archive.is_export_normals)
{
normalListByMaterial[vi1] = normalList.at(f1);
normalListByMaterial[vi2] = normalList.at(f2);
normalListByMaterial[vi3] = normalList.at(f3);
}
faceList[vi1] = vi1;
faceList[vi2] = vi2;
faceList[vi3] = vi3;
faceCountList[n] = 3;
}
preSurfaceSize = material->surface.faces.size();
for (int n = 0, nsize = vertexListByMaterial.size(); n < nsize; ++n)
{
vertexListByMaterial[n].z = -vertexListByMaterial[n].z;
}
Alembic::AbcGeom::OPolyMeshSchema::Sample sample;
// vertex
Alembic::AbcGeom::P3fArraySample positions( (const Imath::V3f *) &vertexListByMaterial.front(), vertexListByMaterial.size());
sample.setPositions(positions);
// face index
if (isFirstMesh)
{
Alembic::Abc::Int32ArraySample faceIndices(faceList);
Alembic::Abc::Int32ArraySample faceCounts(faceCountList);
sample.setFaceIndices(faceIndices);
sample.setFaceCounts(faceCounts);
}
// UVs
if (!uvListByMaterial.empty() && archive.is_export_uvs)
{
for (int n = 0, nsize = uvListByMaterial.size(); n < nsize; ++n)
{
uvListByMaterial[n].y = 1.0f - uvListByMaterial[n].y;
}
Alembic::AbcGeom::OV2fGeomParam::Sample uvSample;
uvSample.setScope(Alembic::AbcGeom::kVertexScope );
uvSample.setVals(Alembic::AbcGeom::V2fArraySample( ( const Imath::V2f *) &uvListByMaterial.front(), uvListByMaterial.size()));
sample.setUVs(uvSample);
}
// Normals
if (!normalListByMaterial.empty() && archive.is_export_normals)
{
for (int n = 0, nsize = normalListByMaterial.size(); n < nsize; ++n)
{
normalListByMaterial[n].z = -normalListByMaterial[n].z;
}
Alembic::AbcGeom::ON3fGeomParam::Sample normalSample;
normalSample.setScope(Alembic::AbcGeom::kVertexScope );
normalSample.setVals(Alembic::AbcGeom::N3fArraySample( (const Alembic::AbcGeom::N3f *) &normalListByMaterial.front(), normalListByMaterial.size()));
sample.setNormals(normalSample);
}
meshSchema.set(sample);
}
}
