2012-04-02 74 views
3

此问题与以前回答的问题有关: Fast 24-bit array -> 32-bit array conversion? 在一个答案中,interjay寄出了用于转换RGB24-> RGB32的SSE3代码,但是我也需要反向转换(RGB32-> RGB24) 。我给它一个镜头(见下文),我的代码可以工作,但它比interjay的代码更复杂,而且显着更慢。我看不出如何完全颠倒指令:_mm_alignr_epi8在这种情况下似乎没有帮助,但我对SSE3并不熟悉。不对称是不可避免的,还是有更快的替代和换位?快速的32位数组 - > SSE3中的24位数组转换? (RGB32 - > RGB24)

RGB32 - > RGB24:

__m128i *src = ... 
__m128i *dst = ... 
__m128i mask = _mm_setr_epi8(0,1,2,4, 5,6,8,9, 10,12,13,14, -1,-1,-1,-1); 
for (UINT i = 0; i < Pixels; i += 16) { 
    __m128i sa = _mm_shuffle_epi8(_mm_load_si128(src), mask); 
    __m128i sb = _mm_shuffle_epi8(_mm_load_si128(src + 1), mask); 
    __m128i sc = _mm_shuffle_epi8(_mm_load_si128(src + 2), mask); 
    __m128i sd = _mm_shuffle_epi8(_mm_load_si128(src + 3), mask); 
    _mm_store_si128(dst, _mm_or_si128(sa, _mm_slli_si128(sb, 12))); 
    _mm_store_si128(dst + 1, _mm_or_si128(_mm_srli_si128(sb, 4), _mm_slli_si128(sc, 8))); 
    _mm_store_si128(dst + 2, _mm_or_si128(_mm_srli_si128(sc, 8), _mm_slli_si128(sd, 4))); 
    src += 4; 
    dst += 3; 
} 

RGB24 - > RGB32(礼貌interjay):

__m128i *src = ... 
__m128i *dst = ... 
__m128i mask = _mm_setr_epi8(0,1,2,-1, 3,4,5,-1, 6,7,8,-1, 9,10,11,-1); 
for (UINT i = 0; i < Pixels; i += 16) { 
    __m128i sa = _mm_load_si128(src); 
    __m128i sb = _mm_load_si128(src + 1); 
    __m128i sc = _mm_load_si128(src + 2); 
    __m128i val = _mm_shuffle_epi8(sa, mask); 
    _mm_store_si128(dst, val); 
    val = _mm_shuffle_epi8(_mm_alignr_epi8(sb, sa, 12), mask); 
    _mm_store_si128(dst + 1, val); 
    val = _mm_shuffle_epi8(_mm_alignr_epi8(sc, sb, 8), mask); 
    _mm_store_si128(dst + 2, val); 
    val = _mm_shuffle_epi8(_mm_alignr_epi8(sc, sc, 4), mask); 
    _mm_store_si128(dst + 3, val); 
    src += 3; 
    dst += 4; 
} 
+0

那么,没有SSE4.1允许? – harold 2012-04-02 22:05:16

+0

您只需在4个输入寄存器上使用6个掩码将它们转换为3个输出寄存器。你不能绕过三个'或',因为'pshufb'设置一个字节为0或者由掩码索引的值。 – hirschhornsalz 2012-04-02 22:14:48

回答

0

您可以this answer和更改洗牌面具从RGB32去RGB24。

最大的区别是直接计算洗牌,并使用按位操作来避免转换。另外,使用对齐的流式写入而不是对齐的写入不会污染缓存。

0

老问题,但我试图解决同样的问题,所以...

您可以使用palignr如果右对齐它的第二个操作数,即置零的低字节。您需要第二个,第三个和第四个单词的左对齐版本,以及第一个,第二个和第三个单词的右对齐版本。

对于第二个和第三个单词,如果我使用shift来计算左对齐的右对齐版本,GCC会稍微高兴一些。如果我使用两个不同的pshufb,它会产生3个不必要的移动。

这是代码。它正好使用8个寄存器;如果您处于64位模式,则可以尝试将其展开两次。

__m128i mask_right = _mm_set_epi8(14, 13, 12, 10, 9, 8, 6, 5, 4, 2, 1, 0, 0x80, 0x80, 0x80, 0x80); 
    __m128i mask = _mm_set_epi8(0x80, 0x80, 0x80, 0x80, 14, 13, 12, 10, 9, 8, 6, 5, 4, 2, 1, 0); 

    for (; n; n -= 16, d += 48, s += 64) { 
      __m128i v0 = _mm_load_si128((__m128i *) &s[0]); 
      __m128i v1 = _mm_load_si128((__m128i *) &s[16]); 
      __m128i v2 = _mm_load_si128((__m128i *) &s[32]); 
      __m128i v3 = _mm_load_si128((__m128i *) &s[48]); 

      v0 = _mm_shuffle_epi8(v0, mask_right); 
      v1 = _mm_shuffle_epi8(v1, mask); 
      v2 = _mm_shuffle_epi8(v2, mask); 
      v3 = _mm_shuffle_epi8(v3, mask); 

      v0 = _mm_alignr_epi8(v1, v0, 4); 
      v1 = _mm_slli_si128(v1, 4);  // mask -> mask_right 
      v1 = _mm_alignr_epi8(v2, v1, 8); 
      v2 = _mm_slli_si128(v2, 4);  // mask -> mask_right 
      v2 = _mm_alignr_epi8(v3, v2, 12); 

      _mm_store_si128((__m128i *) &d[0], v0); 
      _mm_store_si128((__m128i *) &d[16], v1); 
      _mm_store_si128((__m128i *) &d[32], v2); 
    } 

中央部分也可能是这样写的。编译器只产生一个指令,看起来它有更多的并行性,但需要基准来给出正确答案:

  v0 = _mm_shuffle_epi8(v0, mask_right); 
      v1 = _mm_shuffle_epi8(v1, mask); 
      v2 = _mm_shuffle_epi8(v2, mask_right); 
      v3 = _mm_shuffle_epi8(v3, mask); 

      __m128i v2l = v2; 
      v0 = _mm_alignr_epi8(v1, v0, 4); 
      v1 = _mm_slli_si128(v1, 4);    // mask -> mask_right 
      v2 = _mm_alignr_epi8(v3, v2, 12); 
      v2l = _mm_srli_si128(v2l, 4);   // mask_right -> mask 
      v1 = _mm_alignr_epi8(v2l, v1, 8);