Function PadAndStack

tests/cvcuda/system/TestOpPadAndStack.cpp:32–96

namespace test = nvcv::test;

static void PadAndStack(std::vector<uint8_t> &hDst, const std::vector<std::vector<uint8_t>> &hBatchSrc,
                        const nvcv::TensorDataAccessStridedImagePlanar &dDstData, const int srcWidth,
                        const int srcHeight, const int srcRowStride, const int srcPixPitch,
                        const std::vector<int> &topVec, const std::vector<int> &leftVec,
                        const NVCVBorderType borderType, const float borderValue)
{
    int dstPixPitch  = dDstData.numChannels();
    int dstRowStride = dDstData.rowStride() / sizeof(uint8_t);
    int dstImgPitch  = dDstData.sampleStride() / sizeof(uint8_t);

    int2 coord, size{srcWidth, srcHeight};

    for (int db = 0; db < dDstData.numSamples(); db++)
    {
        for (int di = 0; di < dDstData.numRows(); di++)
        {
            coord.y = di - topVec[db];

            for (int dj = 0; dj < dDstData.numCols(); dj++)
            {
                coord.x = dj - leftVec[db];

                for (int dk = 0; dk < dDstData.numChannels(); dk++)
                {
                    uint8_t out = 0;

                    if (coord.x >= 0 && coord.x < size.x && coord.y >= 0 && coord.y < size.y)
                    {
                        out = hBatchSrc[db][coord.y * srcRowStride + coord.x * srcPixPitch + dk];
                    }
                    else
                    {
                        if (borderType == NVCV_BORDER_CONSTANT)
                        {
                            out = static_cast<uint8_t>(borderValue);
                        }
                        else
                        {
                            if (borderType == NVCV_BORDER_REPLICATE)
                            {
                                test::ReplicateBorderIndex(coord, size);
                            }
                            else if (borderType == NVCV_BORDER_WRAP)
                            {
                                test::WrapBorderIndex(coord, size);
                            }
                            else if (borderType == NVCV_BORDER_REFLECT)
                            {
                                test::ReflectBorderIndex(coord, size);
                            }
                            else if (borderType == NVCV_BORDER_REFLECT101)
                            {
                                test::Reflect101BorderIndex(coord, size);
                            }

                            out = hBatchSrc[db][coord.y * srcRowStride + coord.x * srcPixPitch + dk];
                        }
                    }