Android Renderscript установить соседний пиксель прозрачным

Question

У меня есть скрипт, который очищает пиксели определенного цвета.

uchar red = 100;
uchar green = 100;
uchar blue = 100;
float treshold = 100;

uchar4 __attribute__((kernel)) saturation(uchar4 in,uint32_t x, uint32_t y)
{
    float ddd = ((in.r - red)*(in.r - red) + (in.g - green)*(in.g - green) + (in.b - blue)*(in.b - blue));
    float dif = sqrt( ddd );
    if (dif <= treshold){
        in.a = 0;
        in.r = 0;
        in.g = 0;
        in.b = 0;
    }
    return in;
}

Это я запускаю в Java Lile:

        mScript.set_red((short)r);
        mScript.set_blue((short)b);
        mScript.set_green((short)g);
        mScript.set_treshold(treshold);
        mScript.forEach_saturation(mInAllocation, mOutAllocations);

Это работает, но мне нужен четкий сосед с определенным цветом в RenderScript? В насыщенности мы обрабатываем каждый пиксель, и я не знаю, как получить доступ ко всем пикселям.

Answer 1

Используйте глобальную переменную rs_allocation, а затем используйте функцию rsGetElementAt_uchar4 для выборки изображения в других местах:

#pragma rs_fp_relaxed
rs_allocation image;
int width_minus_one;
void RS_KERNEL root(uchar4 in, uint32_t x, uint32_t y) {
    int newX = min(x + 1, width_minus_one);
    uchar4 pixel = rsGetElementAt_uchar4(image, newX, y);
}

Джава:

mScript.set_image(mInAllocation);
mScript.set_width_minus_one(mInAllocation.getType().getX() - 1);