Function InferImageFormat

python/mod_cvcuda/nvcv/Image.cpp:276–357

}

nvcv::ImageFormat InferImageFormat(const std::vector<nvcv::DataType> &planePixTypes)
{
    if (planePixTypes.empty())
    {
        return nvcv::FMT_NONE;
    }

    static_assert(NVCV_PACKING_0 == 0, "Invalid 0 packing value");
    NVCV_ASSERT(planePixTypes.size() <= 4);

    nvcv::Packing packing[4] = {nvcv::Packing::NONE};

    int numChannels = 0;

    for (size_t p = 0; p < planePixTypes.size(); ++p)
    {
        packing[p] = planePixTypes[p].packing();
        numChannels += planePixTypes[p].numChannels();

        if (planePixTypes[p].dataKind() != planePixTypes[0].dataKind())
        {
            throw std::invalid_argument("Planes must all have the same data type");
        }
    }

    nvcv::DataKind dataKind = planePixTypes[0].dataKind();

    int numPlanes = planePixTypes.size();

    // Planar or packed?
    if (numPlanes == 1 || numChannels == numPlanes)
    {
        static const nvcv::ImageFormat baseFormatList[4]
            = {nvcv::FMT_U8, nvcv::FMT_2F32, nvcv::FMT_RGB8, nvcv::FMT_RGBA8};

        // Validate array index to prevent buffer overrun
        if (numChannels < 1 || numChannels > 4)
        {
            throw std::invalid_argument(
                util::FormatString("Invalid number of channels %d, must be between 1 and 4", numChannels));
        }

        nvcv::ImageFormat baseFormat = baseFormatList[numChannels - 1];

        nvcv::ColorModel model = baseFormat.colorModel();
        switch (model)
        {
        case nvcv::ColorModel::YCbCr:
            return nvcv::ImageFormat(baseFormat.colorSpec(), baseFormat.chromaSubsampling(), baseFormat.memLayout(),
                                     dataKind, baseFormat.swizzle(), packing[0], packing[1], packing[2], packing[3]);

        case nvcv::ColorModel::UNDEFINED:
            return nvcv::ImageFormat(baseFormat.memLayout(), dataKind, baseFormat.swizzle(), packing[0], packing[1],
                                     packing[2], packing[3]);
        case nvcv::ColorModel::RAW:
            return nvcv::ImageFormat(baseFormat.rawPattern(), baseFormat.memLayout(), dataKind, baseFormat.swizzle(),
                                     packing[0], packing[1], packing[2], packing[3]);
        default: