MObject createSubD(double iFrame, SubDAndColors & iNode,
MObject & iParent)
{
Alembic::AbcGeom::ISubDSchema schema = iNode.mMesh.getSchema();
Alembic::AbcCoreAbstract::index_t index, ceilIndex;
getWeightAndIndex(iFrame, schema.getTimeSampling(),
schema.getNumSamples(), index, ceilIndex);
Alembic::AbcGeom::ISubDSchema::Sample samp;
schema.get(samp, Alembic::Abc::ISampleSelector(index));
MString name(iNode.mMesh.getName().c_str());
MFnMesh fnMesh;
MFloatPointArray pointArray;
Alembic::Abc::P3fArraySamplePtr emptyPtr;
fillPoints(pointArray, samp.getPositions(), emptyPtr, 0.0);
fillTopology(fnMesh, iParent, pointArray, samp.getFaceIndices(),
samp.getFaceCounts());
fnMesh.setName(iNode.mMesh.getName().c_str());
setInitialShadingGroup(fnMesh.partialPathName());
MObject obj = fnMesh.object();
setUVs(iFrame, fnMesh, schema.getUVsParam());
setColors(iFrame, fnMesh, iNode.mC3s, iNode.mC4s, true);
// add the mFn-specific attributes to fnMesh node
MFnNumericAttribute numAttr;
MString attrName("SubDivisionMesh");
MObject attrObj = numAttr.create(attrName, attrName,
MFnNumericData::kBoolean, 1);
numAttr.setKeyable(true);
numAttr.setHidden(false);
fnMesh.addAttribute(attrObj, MFnDependencyNode::kLocalDynamicAttr);
if (samp.getInterpolateBoundary() > 0)
{
attrName = MString("interpolateBoundary");
attrObj = numAttr.create(attrName, attrName, MFnNumericData::kBoolean,
samp.getInterpolateBoundary());
numAttr.setKeyable(true);
numAttr.setHidden(false);
fnMesh.addAttribute(attrObj, MFnDependencyNode::kLocalDynamicAttr);
}
if (samp.getFaceVaryingInterpolateBoundary() > 0)
{
attrName = MString("faceVaryingInterpolateBoundary");
attrObj = numAttr.create(attrName, attrName, MFnNumericData::kBoolean,
samp.getFaceVaryingInterpolateBoundary());
numAttr.setKeyable(true);
numAttr.setHidden(false);
fnMesh.addAttribute(attrObj, MFnDependencyNode::kLocalDynamicAttr);
}
if (samp.getFaceVaryingPropagateCorners() > 0)
{
attrName = MString("faceVaryingPropagateCorners");
attrObj = numAttr.create(attrName, attrName, MFnNumericData::kBoolean,
samp.getFaceVaryingPropagateCorners());
numAttr.setKeyable(true);
numAttr.setHidden(false);
fnMesh.addAttribute(attrObj, MFnDependencyNode::kLocalDynamicAttr);
}
#if MAYA_API_VERSION >= 201100
Alembic::Abc::Int32ArraySamplePtr holes = samp.getHoles();
if (holes && !holes->size() == 0)
{
unsigned int numHoles = (unsigned int)holes->size();
MUintArray holeData(numHoles);
for (unsigned int i = 0; i < numHoles; ++i)
{
holeData[i] = (*holes)[i];
}
if (fnMesh.setInvisibleFaces(holeData) != MS::kSuccess)
{
MString warn = "Failed to set holes on: ";
warn += iNode.mMesh.getName().c_str();
printWarning(warn);
}
}
#endif
Alembic::Abc::FloatArraySamplePtr creases = samp.getCreaseSharpnesses();
if (creases && !creases->size() == 0)
{
Alembic::Abc::Int32ArraySamplePtr indices = samp.getCreaseIndices();
Alembic::Abc::Int32ArraySamplePtr lengths = samp.getCreaseLengths();
std::size_t numLengths = lengths->size();
MUintArray edgeIds;
MDoubleArray creaseData;
std::size_t curIndex = 0;
// curIndex incremented here to move on to the next crease length
for (std::size_t i = 0; i < numLengths; ++i, ++curIndex)
{
std::size_t len = (*lengths)[i] - 1;
float creaseSharpness = (*creases)[i];
// curIndex incremented here to go between all the edges that make
// up a given length
for (std::size_t j = 0; j < len; ++j, ++curIndex)
{
Alembic::Util::int32_t vertA = (*indices)[curIndex];
Alembic::Util::int32_t vertB = (*indices)[curIndex+1];
MItMeshVertex itv(obj);
int prev;
itv.setIndex(vertA, prev);
MIntArray edges;
itv.getConnectedEdges(edges);
std::size_t numEdges = edges.length();
for (unsigned int k = 0; k < numEdges; ++k)
{
int oppVert = -1;
itv.getOppositeVertex(oppVert, edges[k]);
if (oppVert == vertB)
{
creaseData.append(creaseSharpness);
edgeIds.append(edges[k]);
break;
}
}
}
}
if (fnMesh.setCreaseEdges(edgeIds, creaseData) != MS::kSuccess)
{
MString warn = "Failed to set creases on: ";
warn += iNode.mMesh.getName().c_str();
printWarning(warn);
}
}
Alembic::Abc::FloatArraySamplePtr corners = samp.getCornerSharpnesses();
if (corners && !corners->size() == 0)
{
Alembic::Abc::Int32ArraySamplePtr cornerVerts = samp.getCornerIndices();
unsigned int numCorners = static_cast<unsigned int>(corners->size());
MUintArray vertIds(numCorners);
MDoubleArray cornerData(numCorners);
for (unsigned int i = 0; i < numCorners; ++i)
{
cornerData[i] = (*corners)[i];
vertIds[i] = (*cornerVerts)[i];
}
if (fnMesh.setCreaseVertices(vertIds, cornerData) != MS::kSuccess)
{
MString warn = "Failed to set corners on: ";
warn += iNode.mMesh.getName().c_str();
printWarning(warn);
}
}
return obj;
}
void uiDrawManagerDrawOverride::addUIDrawables(
const MDagPath& objPath,
MHWRender::MUIDrawManager& drawManager,
const MHWRender::MFrameContext& frameContext,
const MUserData* data)
{
const uiDrawManagerData* thisdata = dynamic_cast<const uiDrawManagerData*>(data);
if (!thisdata) {
return;
}
switch (thisdata->fUIType)
{
case uiDrawManager::kText:
{
// Draw a text "uiDrawManager"
// All drawing operations must take place between calls to beginDrawable()
// and endDrawable().
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setFontSize(thisdata->fTextFontSize);
drawManager.setFontIncline(thisdata->fTextIncline);
drawManager.setFontWeight(thisdata->fTextWeight);
drawManager.setFontStretch(thisdata->fTextStretch);
drawManager.setFontLine(thisdata->fTextLine);
MString faceName = uiDrawManagerData::fFontList[thisdata->fFontFaceIndex];
drawManager.setFontName(faceName);
int boxSize[] = { thisdata->fTextBoxWidth, thisdata->fTextBoxHeight };
if (thisdata->fDraw2D) {
// uiDrawManagerData::fPosition gives a screen space position
// where 2D UI item is located.
drawManager.text2d(thisdata->fPosition, thisdata->fText, thisdata->fTextAlignment,
boxSize[0]+boxSize[1] == 0 ? NULL : boxSize, &thisdata->fTextBoxColor, false);
}
else {
// for 3D items, place it at the origin of the world space.
drawManager.text(thisdata->fPosition, thisdata->fText, thisdata->fTextAlignment,
boxSize[0]+boxSize[1] == 0 ? NULL : boxSize, &thisdata->fTextBoxColor, false);
}
drawManager.endDrawable();
}
break;
case uiDrawManager::kLine:
case uiDrawManager::kLineList:
case uiDrawManager::kLineStrip:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
if (thisdata->fUIType == uiDrawManager::kLineStrip)
drawManager.lineStrip(thisdata->fLines, thisdata->fDraw2D);
else
drawManager.lineList(thisdata->fLines, thisdata->fDraw2D);
drawManager.endDrawable();
}
break;
case uiDrawManager::kPoint:
case uiDrawManager::kPointList:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setPointSize(thisdata->fPointSize);
drawManager.points(thisdata->fPoints, thisdata->fDraw2D);
drawManager.endDrawable();
}
break;
case uiDrawManager::kRect:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
if (thisdata->fDraw2D) {
// For 2d rectangle, an up vector in screen space is used to determine its X
// and Y directions. In addition, "fRectScaleX" and "fRectScaleY"(in pixels)
// specify the half-lengths of the 2d rectangle.
drawManager.rect2d(thisdata->fPosition, thisdata->fRectUp,
thisdata->fRectScaleX, thisdata->fRectScaleY, thisdata->fIsFilled);
}
else {
// For 3d rectangle, the up vector should not be parallel with the normal vector.
drawManager.rect(thisdata->fPosition, thisdata->fRectUp, thisdata->fRectNormal,
thisdata->fRectScaleX, thisdata->fRectScaleY, thisdata->fIsFilled);
}
drawManager.endDrawable();
}
break;
case uiDrawManager::kQuad:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
// prepare primitive type
MUIDrawManager::Primitive mode =
thisdata->fIsFilled ? MUIDrawManager::kTriStrip : MUIDrawManager::kClosedLine;
// prepare position list
MPointArray position;
for (int i = 0; i < 4; ++i) {
position.append(thisdata->fQuadVertex[i]);
}
// prepare index
MUintArray index;
index.append(0);
index.append(1);
index.append(3);
index.append(2);
// draw mesh
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
if (thisdata->fDraw2D) {
drawManager.mesh2d(mode, position, NULL, thisdata->fIsFilled ? &index : NULL);
}
else {
drawManager.mesh(mode, position, NULL, NULL, thisdata->fIsFilled ? &index : NULL);
}
drawManager.endDrawable();
}
break;
case uiDrawManager::kSphere:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
drawManager.sphere(thisdata->fPosition, thisdata->fRadius, thisdata->fIsFilled);
drawManager.endDrawable();
}
break;
case uiDrawManager::kCircle:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
if (thisdata->fDraw2D) {
// The radius in specified as pixel unit for 2d items.
drawManager.circle2d(thisdata->fPosition, thisdata->fRadius,
thisdata->fIsFilled);
}
else {
drawManager.circle(thisdata->fPosition, thisdata->fCircleNormal, thisdata->fRadius,
thisdata->fIsFilled);
}
drawManager.endDrawable();
}
break;
case uiDrawManager::kArc:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
if (thisdata->fDraw2D) {
// If 2d, the range of the arc is defined by the start and end vectors
// specified in screen space.
drawManager.arc2d(thisdata->fPosition, thisdata->fArcStart, thisdata->fArcEnd,
thisdata->fRadius, thisdata->fIsFilled);
}
else {
// For 3d arc, the projections of the start and end vectors onto the arc plane(
// determined by the normal vector) determine the range of the arc.
drawManager.arc(thisdata->fPosition, thisdata->fArcStart, thisdata->fArcEnd,
thisdata->fArcNormal, thisdata->fRadius, thisdata->fIsFilled);
}
drawManager.endDrawable();
}
break;
case uiDrawManager::kIcon:
{
MString iconName = uiDrawManagerData::fIconList[thisdata->fIconIndex];
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.icon(thisdata->fPosition, iconName, thisdata->fIconScale);
drawManager.endDrawable();
}
break;
case uiDrawManager::kCone:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
drawManager.cone(thisdata->fPosition,
thisdata->fConeDirection,
thisdata->fRadius,
thisdata->fConeHeight,
thisdata->fIsFilled);
drawManager.endDrawable();
}
break;
case uiDrawManager::kBox:
{
drawManager.beginDrawable();
drawManager.setColor(thisdata->fColor);
drawManager.setLineWidth(thisdata->fLineWidth);
drawManager.setLineStyle(thisdata->fLineStyle);
drawManager.setPaintStyle(thisdata->fShaded ?
MUIDrawManager::kShaded : MUIDrawManager::kFlat);
drawManager.box(thisdata->fPosition,
thisdata->fBoxUp,
thisdata->fBoxRight,
thisdata->fBoxScale[0],
thisdata->fBoxScale[1],
thisdata->fBoxScale[2],
thisdata->fIsFilled);
drawManager.endDrawable();
}
break;
default:
perror("unhandled ui types.");
break;
}
}