Can halide express coordinate extraction operations

[cl87]

hi，can halide express following coordinate extraction operation，thanks

int GetMask(uint8_t* dst,uint8_t* src,int width,int height,int stride){
int count = 0;
for(int h = 0; h < height; h++){
for(int w = 0; w < width; w++){
if (src[h * stride+w] == 1) {
dst[count++] = w;
dst[count++] = h;
};
}
}
return count >> 1;
}