Function DecompressTextureSetWithGraphicsAPI

tools/cli/GraphicsPasses.cpp:185–310

}

bool DecompressTextureSetWithGraphicsAPI(
    nvrhi::IDevice* device,
    nvrhi::ICommandList* commandList,
    nvrhi::ITimerQuery* timerQuery,
    GraphicsDecompressionPass& gdp,
    GDeflateFeatures* gdeflateFeatures,
    ntc::IContext* context,
    ntc::ITextureSetMetadata* metadata,
    ntc::IStream* inputFile,
    int mipLevels,
    bool enableDithering,
    GraphicsResourcesForTextureSet const& graphicsResources)
{
    // In some cases, this function is called without a file - which means we reuse the previously uploaded data.
    if (inputFile)
    {
        if (!gdp.SetLatentDataFromTextureSet(commandList, context, gdeflateFeatures, inputFile, metadata))
        {
            fprintf(stderr, "GraphicsDecompressionPass::SetInputData failed.\n");
            return false;
        }
    }

    int const numTextures = int(graphicsResources.perTexture.size());
    
    // Write UAV descriptors for all necessary mip levels into the descriptor table
    for (int mipLevel = 0; mipLevel < mipLevels; ++mipLevel)
    {
        for (int index = 0; index < numTextures; ++index)
        {
            const auto bindingSetItem = nvrhi::BindingSetItem::Texture_UAV(
                mipLevel * numTextures + index,
                graphicsResources.perTexture[index].color,
                nvrhi::Format::UNKNOWN,
                nvrhi::TextureSubresourceSet(mipLevel, 1, 0, 1));

            gdp.WriteDescriptor(bindingSetItem);
        }
    }

    ntc::InferenceWeightType weightType = metadata->GetBestSupportedWeightType();
    if (weightType == ntc::InferenceWeightType::Unknown)
    {
        fprintf(stderr, "The texture set does not provide any weights compatible with the current device.\n");
        return false;
    }

    commandList->open();

    if (!gdp.SetWeightsFromTextureSet(commandList, metadata, weightType))
    {
        fprintf(stderr, "GraphicsDecompressionPass::SetWeightsFromTextureSet failed.\n");
        commandList->close();
        return false;
    }

    commandList->beginTimerQuery(timerQuery);