Function GetOutputFormat

python/mod_cvcuda/CvtColorUtil.cpp:244–276

};

nvcv::ImageFormat GetOutputFormat(nvcv::DataType in, NVCVColorConversionCode code)
{
    auto outFormatIt = kOutputFormat.find(code);
    if (outFormatIt == kOutputFormat.end())
    {
        throw std::runtime_error("Invalid color conversion code");
    }
    nvcv::ImageFormat outFormat{outFormatIt->second};

    auto inPackingParams = nvcv::GetParams(in.packing());
    int  inNumBits       = 0;
    for (const auto &numBits : inPackingParams.bits)
    {
        if (numBits > 0)
        {
            inNumBits = (inNumBits == 0) ? numBits : inNumBits;
            if (numBits != inNumBits)
            {
                throw std::runtime_error("Invalid input format, all channels must have the same bit-depth");
            }
        }
    }
    auto outPackingParams = nvcv::GetParams(outFormat.planePacking(0));
    for (auto &numBits : outPackingParams.bits)
    {
        numBits = (numBits > 0) ? inNumBits : numBits;
    }
    auto outPacking = nvcv::MakePacking(outPackingParams);
    outFormat = outFormat.swizzleAndPacking(outFormat.swizzle(), outPacking, nvcv::Packing::NONE, nvcv::Packing::NONE,
                                            nvcv::Packing::NONE);

    return outFormat;
}

int64_t GetOutputHeight(int64_t height, NVCVColorConversionCode code)
{