Schnelles Suchen/Ersetzen von übereinstimmenden einzelnen Bytes in einem 8-Bit-Array, auf ARM

Question

Ich entwickle Bildverarbeitungsalgorithmen (mit GCC, Targeting ARMv7 (Raspberry Pi 2B)).

Insbesondere verwende ich einen einfachen Algorithmus, der Index in einer Maske ändert:

void ChangeIndex(uint8_t * mask, size_t size, uint8_t oldIndex, uint8_t newIndex) 
{ 
    for(size_t i = 0; i < size; ++i) 
    { 
     if(mask[i] == oldIndex) 
      mask[i] = newIndex; 
    } 
} 

Leider für die Zielplattform hat schlechte Leistung.

Gibt es eine Möglichkeit, es zu optimieren?

Answer 1

Die ARMv7-Plattform unterstützt SIMD-Anweisungen mit der Bezeichnung NEON. Mit Gebrauch von ihnen können Sie schneller Code:

#include <arm_neon.h> 

void ChangeIndex(uint8_t * mask, size_t size, uint8_t oldIndex, uint8_t newIndex) 
{ 
    size_t alignedSize = size/16*16, i = 0; 

    uint8x16_t _oldIndex = vdupq_n_u8(oldIndex); 
    uint8x16_t _newIndex = vdupq_n_u8(newIndex); 

    for(; i < alignedSize; i += 16) 
    { 
     uint8x16_t oldMask = vld1q_u8(mask + i); // loading of 128-bit vector 
     uint8x16_t condition = vceqq_u8(oldMask, _oldIndex); // compare two 128-bit vectors 
     uint8x16_t newMask = vbslq_u8(condition, _newIndex, oldMask); // selective copying of 128-bit vector 
     vst1q_u8(mask + i, newMask); // saving of 128-bit vector 
    } 

    for(; i < size; ++i) 
    { 
     if(mask[i] == oldIndex) 
      mask[i] = newIndex; 
    } 
}