void
GusdPrimWrapper::loadPrimvars(
UsdTimeCode time,
const GT_RefineParms* rparms,
int minUniform,
int minPoint,
int minVertex,
const string& primPath,
GT_AttributeListHandle* vertex,
GT_AttributeListHandle* point,
GT_AttributeListHandle* primitive,
GT_AttributeListHandle* constant,
const GT_DataArrayHandle& remapIndicies ) const
{
// Primvars will be loaded if they match a provided pattern.
// By default, set the pattern to match only "Cd". Then write
// over this pattern if there is one provided in rparms.
const char* Cd = "Cd";
UT_String primvarPattern(Cd);
if (rparms) {
rparms->import("usd:primvarPattern", primvarPattern);
}
std::vector<UsdGeomPrimvar> authoredPrimvars;
bool hasCdPrimvar = false;
{
UsdGeomImageable prim = getUsdPrimForRead();
UsdGeomPrimvar colorPrimvar = prim.GetPrimvar(GusdTokens->Cd);
if (colorPrimvar && colorPrimvar.GetAttr().HasAuthoredValueOpinion()) {
hasCdPrimvar = true;
}
// It's common for "Cd" to be the only primvar to load.
// In this case, avoid getting all other authored primvars.
if (primvarPattern == Cd) {
if (hasCdPrimvar) {
authoredPrimvars.push_back(colorPrimvar);
} else {
// There is no authored "Cd" primvar.
// Try to find "displayColor" instead.
colorPrimvar = prim.GetPrimvar(UsdGeomTokens->primvarsDisplayColor);
if (colorPrimvar &&
colorPrimvar.GetAttr().HasAuthoredValueOpinion()) {
authoredPrimvars.push_back(colorPrimvar);
}
}
} else if (primvarPattern != "") {
authoredPrimvars = prim.GetAuthoredPrimvars();
}
}
// Is it better to sort the attributes and build the attributes all at once.
for( const UsdGeomPrimvar &primvar : authoredPrimvars )
{
DBG(cerr << "loadPrimvar " << primvar.GetPrimvarName() << "\t" << primvar.GetTypeName() << "\t" << primvar.GetInterpolation() << endl);
UT_String name(primvar.GetPrimvarName());
// One special case we always handle here is to change
// the name of the USD "displayColor" primvar to "Cd",
// as long as there is not already a "Cd" primvar.
if (!hasCdPrimvar &&
primvar.GetName() == UsdGeomTokens->primvarsDisplayColor) {
name = Cd;
}
// If the name of this primvar doesn't
// match the primvarPattern, skip it.
if (!name.multiMatch(primvarPattern, 1, " ")) {
continue;
}
GT_DataArrayHandle gtData = convertPrimvarData( primvar, time );
if( !gtData )
{
TF_WARN( "Failed to convert primvar %s:%s %s.",
primPath.c_str(),
primvar.GetPrimvarName().GetText(),
primvar.GetTypeName().GetAsToken().GetText() );
continue;
}
// usd vertex primvars are assigned to points
if( primvar.GetInterpolation() == UsdGeomTokens->vertex )
{
if( gtData->entries() < minPoint ) {
TF_WARN( "Not enough values found for primvar: %s:%s. "
"%zd values given for %d points.",
primPath.c_str(),
primvar.GetPrimvarName().GetText(),
gtData->entries(), minPoint );
}
else {
if (remapIndicies) {
gtData = new GT_DAIndirect( remapIndicies, gtData );
}
if( point ) {
*point = (*point)->addAttribute( name.c_str(), gtData, true );
}
}
}
else if( primvar.GetInterpolation() == UsdGeomTokens->faceVarying )
{
if( gtData->entries() < minVertex ) {
TF_WARN( "Not enough values found for primvar: %s:%s. "
"%zd values given for %d verticies.",
primPath.c_str(),
primvar.GetPrimvarName().GetText(),
gtData->entries(), minVertex );
}
else if( vertex ) {
*vertex = (*vertex)->addAttribute( name.c_str(), gtData, true );
}
}
else if( primvar.GetInterpolation() == UsdGeomTokens->uniform )
{
if( gtData->entries() < minUniform ) {
TF_WARN( "Not enough values found for primvar: %s:%s. "
"%zd values given for %d faces.",
primPath.c_str(),
primvar.GetPrimvarName().GetText(),
gtData->entries(), minUniform );
}
else if( primitive ) {
*primitive = (*primitive)->addAttribute( name.c_str(), gtData, true );
}
}
else if( primvar.GetInterpolation() == UsdGeomTokens->constant )
{
if( constant ) {
*constant = (*constant)->addAttribute( name.c_str(), gtData, true );
}
}
}
}
/* static */
GU_PrimPacked*
GusdGU_PackedUSD::Build(
GU_Detail& detail,
const UT_StringHolder& fileName,
const SdfPath& primPath,
UsdTimeCode frame,
const char* lod,
GusdPurposeSet purposes,
const UsdPrim& prim,
const UT_Matrix4D* xform )
{
auto packedPrim = GU_PrimPacked::build( detail, k_typeName );
auto impl = UTverify_cast<GusdGU_PackedUSD *>(packedPrim->implementation());
impl->m_fileName = fileName;
impl->m_primPath = primPath;
impl->m_frame = frame;
if( prim && !prim.IsA<UsdGeomBoundable>() )
{
UsdGeomImageable geom = UsdGeomImageable(prim);
std::vector<UsdGeomPrimvar> authoredPrimvars = geom.GetAuthoredPrimvars();
GT_DataArrayHandle buffer;
for( const UsdGeomPrimvar &primvar : authoredPrimvars ) {
// XXX This is temporary code, we need to factor the usd read code into GT_Utils.cpp
// to avoid duplicates and read for types GfHalf,double,int,string ...
GT_DataArrayHandle gtData = GusdPrimWrapper::convertPrimvarData( primvar, frame );
if (!gtData)
continue;
const UT_String name(primvar.GetPrimvarName());
const GT_Storage gtStorage = gtData->getStorage();
const GT_Size gtTupleSize = gtData->getTupleSize();
GA_Attribute *anAttr = detail.addTuple(GT_Util::getGAStorage(gtStorage), GA_ATTRIB_PRIMITIVE, name,
gtTupleSize);
if( !anAttr ) {
// addTuple could fail for various reasons, like if there's a
// non-alphanumeric character in the primvar name.
continue;
}
if( const GA_AIFTuple *aIFTuple = anAttr->getAIFTuple()) {
const float* flatArray = gtData->getF32Array( buffer );
aIFTuple->set( anAttr, packedPrim->getMapOffset(), flatArray, gtTupleSize );
} else {
//TF_WARN( "Unsupported primvar type: %s, %s, tupleSize = %zd",
// GT_String( name ), GTstorage( gtStorage ), gtTupleSize );
}
}
}
if( lod )
{
#if SYS_VERSION_FULL_INT < 0x10050000
impl->intrinsicSetViewportLOD( lod );
#else
impl->intrinsicSetViewportLOD( packedPrim, lod );
#endif
}
impl->setPurposes( purposes );
// It seems that Houdini may reuse memory for packed implementations with
// out calling the constructor to initialize data.
impl->resetCaches();
// If a UsdPrim was passed in, make sure it is used.
impl->m_usdPrim = prim;
if (xform) {
impl->setTransform(*xform);
} else {
impl->updateTransform();
}
return packedPrim;
}
