在ARM NEON中将向量加载到数组的边界上

Question

我尝试使用NEON内在函数为ARM优化一些图像处理算法。 对于某些过滤器，需要在该点附近加载元素。 例如，要过滤像素p[x]的图像，我需要加载p[x - 1] ， p[x]和p[x + 1] 。 如果x=0 ，那么我加载p[0] ， p[0]和p[1] 。 如果x=width-1 ，那么我加载p[width-2] ， p[width-1]和p[width-1] 。

因此，如果我有一个向量：

uint8x16_t a = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};

如何从以下向量中得到它：

uint8x16_t b = {0, 0, 1, 2, 3, 4, 5, 6, 7,  8,  9, 10, 11, 12, 13, 14};
uint8x16_t c = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 15};

Answer 1

我认为以下功能将对您的情况有用：

template <size_t count> inline uint8x16_t LoadBeforeFirst(uint8x16_t first)
{
    return vextq_u8(vextq_u8(first, first, count), first, 16 - count);
}

template <size_t count> inline uint8x16_t LoadAfterLast(uint8x16_t last)
{
    return vextq_u8(last, vextq_u8(last, last, 16 - count), count);
}