Materials_Sample()

static void HoudiniEngine::Materials_Sample ( )

static

Definition at line 895 of file HoudiniTestSource.h.

    {
        const char* hdaFile = "";
 
        HAPI_Session session;
        
        HAPI_ThriftServerOptions serverOptions{ 0 };
        serverOptions.autoClose = true;
        serverOptions.timeoutMs = 3000.0f;
 
        HE_CHECK(HAPI_StartThriftNamedPipeServer(&serverOptions, "hapi", nullptr, nullptr))
 
        HAPI_SessionInfo sessionInfo = HAPI_SessionInfo_Create();
        HE_CHECK(HAPI_CreateThriftNamedPipeSession(&session, "hapi", &sessionInfo))
 
        HAPI_CookOptions cookOptions = HAPI_CookOptions_Create();
        HE_CHECK(HAPI_Initialize(&session, &cookOptions, true, -1, nullptr, nullptr, nullptr, nullptr, nullptr))
 
        HAPI_AssetLibraryId assetLibId;
        HE_CHECK(HAPI_LoadAssetLibraryFromFile(&session, hdaFile, true, &assetLibId));
 
        int assetCount;
        HE_CHECK(HAPI_GetAvailableAssetCount(&session, assetLibId, &assetCount))
 
        if (assetCount > 1)
        {
            std::cout << "Should only be loading 1 asset here" << std::endl;
        }
 
        HAPI_StringHandle assetSh;
        HE_CHECK(HAPI_GetAvailableAssets(&session, assetLibId, &assetSh, assetCount))
 
        std::string assetName = GetString(session, assetSh);
 
        HAPI_NodeId nodeId;
        HE_CHECK(HAPI_CreateNode(&session, -1, assetName.c_str(), "BrandonTest", false, &nodeId))
 
        HE_CHECK(HAPI_CookNode(&session, nodeId, &cookOptions))
 
        int cookStatus;
        HAPI_Result cookResult;
 
        do
        {
            cookResult = HAPI_GetStatus(&session, HAPI_STATUS_COOK_STATE, &cookStatus);
        }
        while (cookStatus > HAPI_STATE_MAX_READY_STATE && cookResult == HAPI_RESULT_SUCCESS);
 
        HE_CHECK(cookResult);
        HE_CHECK_COOK(cookStatus);
 
        HAPI_GeoInfo geoInfo;
        HE_CHECK(HAPI_GetDisplayGeoInfo(&session, nodeId, &geoInfo))
 
        HAPI_PartInfo partInfo;
        HE_CHECK(HAPI_GetPartInfo(&session, geoInfo.nodeId, 0, &partInfo))
 
        bool areAllTheSame = false;
        std::vector<HAPI_NodeId> materialIds(partInfo.faceCount);
        HE_CHECK(HAPI_GetMaterialNodeIdsOnFaces(&session, geoInfo.nodeId, partInfo.id, &areAllTheSame, &materialIds.front(), 0, partInfo.faceCount))
 
        if(!areAllTheSame)
        {
            std::cout << "All materials should be the same." << std::endl;
        }
 
        for (int i = 0; i < partInfo.faceCount; ++i)
        {
            if (materialIds[i] != materialIds[0])
            {
                std::cout << "All material ids should be the same." << std::endl;
            }
        }
 
        HAPI_MaterialInfo materialInfo;
        HE_CHECK(HAPI_GetMaterialInfo(&session, materialIds[0], &materialInfo))
 
        if (materialInfo.nodeId != materialIds[0] ||
            materialInfo.nodeId < 0 ||
            materialInfo.exists != true ||
            materialInfo.hasChanged != true)
        {
            std::cout << "Did not successfully extract the first material" << std::endl;
        }
 
        HAPI_NodeInfo materialNodeInfo;
        HE_CHECK(HAPI_GetNodeInfo(&session, materialInfo.nodeId, &materialNodeInfo))
 
        std::cout << GetString(session, materialNodeInfo.nameSH) << std::endl;
 
        std::vector<HAPI_ParmInfo> parmInfos(materialNodeInfo.parmCount);
        HE_CHECK(HAPI_GetParameters(&session, materialNodeInfo.id, parmInfos.data(), 0, materialNodeInfo.parmCount))
 
        int baseColorMapIndex = -1;
        for (int i = 0; i < materialNodeInfo.parmCount; ++i)
        {
            if (GetString(session, parmInfos[i].nameSH) == "baseColorMap")
            {
                baseColorMapIndex = i;
                break;
            }
        }
 
        if (baseColorMapIndex < 0)
        {
            std::cout << "Could not find the base color map parameter" << std::endl;
        }
 
        HAPI_StringHandle basePath;
        HE_CHECK(HAPI_GetParmStringValue(&session, materialNodeInfo.id, "baseColorMap", 0, true, &basePath))
 
        std::cout << "Base Color Map Path: " << GetString(session, basePath) << std::endl;
 
        HE_CHECK(HAPI_RenderTextureToImage(&session, materialNodeInfo.id, baseColorMapIndex))
 
        HAPI_ImageInfo imgInfo;
        HE_CHECK(HAPI_GetImageInfo(&session, materialNodeInfo.id, &imgInfo))
 
        std::cout << "Image Width = " << imgInfo.xRes << std::endl
            << "Image Height = " << imgInfo.yRes << std::endl
            << "Image Format = " << GetString(session, imgInfo.imageFileFormatNameSH) << std::endl;
 
        HE_CHECK(HAPI_SetImageInfo(&session, materialNodeInfo.id, &imgInfo))
 
        int imagePlaneCount;
        HE_CHECK(HAPI_GetImagePlaneCount(&session, materialNodeInfo.id, &imagePlaneCount))
 
        std::vector<HAPI_StringHandle> imagePlanes(imagePlaneCount);
        HE_CHECK(HAPI_GetImagePlanes(&session, materialNodeInfo.id, imagePlanes.data(), imagePlaneCount))
 
        for (int j = 0; j < imagePlaneCount; ++j)
        {
            std::string imagePlaneName = GetString(session, imagePlanes[j]);
            std::cout << "Image Plane [ " << j << " ] = " << imagePlaneName << std::endl;
 
            int destinationFilePath;
            HE_CHECK(HAPI_ExtractImageToFile(&session, materialNodeInfo.id, nullptr, imagePlaneName.c_str(), "./examples/", nullptr, &destinationFilePath))
        }
 
        HE_CHECK(HAPI_Cleanup(&session));
    }