| #include "rs_core.rsh" |
| #include "rs_structs.h" |
| |
| |
| // 565 Conversion bits taken from SkBitmap |
| #define SK_R16_BITS 5 |
| #define SK_G16_BITS 6 |
| #define SK_B16_BITS 5 |
| |
| #define SK_R16_SHIFT (SK_B16_BITS + SK_G16_BITS) |
| #define SK_G16_SHIFT (SK_B16_BITS) |
| #define SK_B16_SHIFT 0 |
| |
| #define SK_R16_MASK ((1 << SK_R16_BITS) - 1) |
| #define SK_G16_MASK ((1 << SK_G16_BITS) - 1) |
| #define SK_B16_MASK ((1 << SK_B16_BITS) - 1) |
| |
| #define SkGetPackedR16(color) (((unsigned)(color) >> SK_R16_SHIFT) & SK_R16_MASK) |
| #define SkGetPackedG16(color) (((unsigned)(color) >> SK_G16_SHIFT) & SK_G16_MASK) |
| #define SkGetPackedB16(color) (((unsigned)(color) >> SK_B16_SHIFT) & SK_B16_MASK) |
| |
| static inline unsigned SkR16ToR32(unsigned r) { |
| return (r << (8 - SK_R16_BITS)) | (r >> (2 * SK_R16_BITS - 8)); |
| } |
| |
| static inline unsigned SkG16ToG32(unsigned g) { |
| return (g << (8 - SK_G16_BITS)) | (g >> (2 * SK_G16_BITS - 8)); |
| } |
| |
| static inline unsigned SkB16ToB32(unsigned b) { |
| return (b << (8 - SK_B16_BITS)) | (b >> (2 * SK_B16_BITS - 8)); |
| } |
| |
| #define SkPacked16ToR32(c) SkR16ToR32(SkGetPackedR16(c)) |
| #define SkPacked16ToG32(c) SkG16ToG32(SkGetPackedG16(c)) |
| #define SkPacked16ToB32(c) SkB16ToB32(SkGetPackedB16(c)) |
| |
| static float3 getFrom565(uint16_t color) { |
| float3 result; |
| result.x = (float)SkPacked16ToR32(color); |
| result.y = (float)SkPacked16ToG32(color); |
| result.z = (float)SkPacked16ToB32(color); |
| return result; |
| } |
| |
| /** |
| * Allocation sampling |
| */ |
| static inline float __attribute__((overloadable)) |
| getElementAt1(const uint8_t *p, int32_t x) { |
| float r = p[x]; |
| return r; |
| } |
| |
| static inline float2 __attribute__((overloadable)) |
| getElementAt2(const uint8_t *p, int32_t x) { |
| x *= 2; |
| float2 r = {p[x], p[x+1]}; |
| return r; |
| } |
| |
| static inline float3 __attribute__((overloadable)) |
| getElementAt3(const uint8_t *p, int32_t x) { |
| x *= 4; |
| float3 r = {p[x], p[x+1], p[x+2]}; |
| return r; |
| } |
| |
| static inline float4 __attribute__((overloadable)) |
| getElementAt4(const uint8_t *p, int32_t x) { |
| x *= 4; |
| const uchar4 *p2 = (const uchar4 *)&p[x]; |
| return convert_float4(p2[0]); |
| } |
| |
| static inline float3 __attribute__((overloadable)) |
| getElementAt565(const uint8_t *p, int32_t x) { |
| x *= 2; |
| float3 r = getFrom565(((const uint16_t *)p)[0]); |
| return r; |
| } |
| |
| static inline float __attribute__((overloadable)) |
| getElementAt1(const uint8_t *p, size_t stride, int32_t x, int32_t y) { |
| p += y * stride; |
| float r = p[x]; |
| return r; |
| } |
| |
| static inline float2 __attribute__((overloadable)) |
| getElementAt2(const uint8_t *p, size_t stride, int32_t x, int32_t y) { |
| p += y * stride; |
| x *= 2; |
| float2 r = {p[x], p[x+1]}; |
| return r; |
| } |
| |
| static inline float3 __attribute__((overloadable)) |
| getElementAt3(const uint8_t *p, size_t stride, int32_t x, int32_t y) { |
| p += y * stride; |
| x *= 4; |
| float3 r = {p[x], p[x+1], p[x+2]}; |
| return r; |
| } |
| |
| static inline float4 __attribute__((overloadable)) |
| getElementAt4(const uint8_t *p, size_t stride, int32_t x, int32_t y) { |
| p += y * stride; |
| x *= 4; |
| float4 r = {p[x], p[x+1], p[x+2], p[x+3]}; |
| return r; |
| } |
| |
| static inline float3 __attribute__((overloadable)) |
| getElementAt565(const uint8_t *p, size_t stride, int32_t x, int32_t y) { |
| p += y * stride; |
| x *= 2; |
| float3 r = getFrom565(((const uint16_t *)p)[0]); |
| return r; |
| } |
| |
| |
| |
| |
| |
| static float4 __attribute__((overloadable)) |
| getSample_A(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float p0 = getElementAt1(p, iPixel); |
| float p1 = getElementAt1(p, next); |
| float r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| float4 ret = {0.f, 0.f, 0.f, r}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_L(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float p0 = getElementAt1(p, iPixel); |
| float p1 = getElementAt1(p, next); |
| float r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| float4 ret = {r, r, r, 1.f}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_LA(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float2 p0 = getElementAt2(p, iPixel); |
| float2 p1 = getElementAt2(p, next); |
| float2 r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| float4 ret = {r.x, r.x, r.x, r.y}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_RGB(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float3 p0 = getElementAt3(p, iPixel); |
| float3 p1 = getElementAt3(p, next); |
| float3 r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| float4 ret = {r.x, r.x, r.z, 1.f}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_565(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float3 p0 = getElementAt565(p, iPixel); |
| float3 p1 = getElementAt565(p, next); |
| float3 r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| float4 ret = {r.x, r.x, r.z, 1.f}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_RGBA(const uint8_t *p, int32_t iPixel, |
| int32_t next, float w0, float w1) { |
| float4 p0 = getElementAt4(p, iPixel); |
| float4 p1 = getElementAt4(p, next); |
| float4 r = p0 * w0 + p1 * w1; |
| r *= (1.f / 255.f); |
| return r; |
| } |
| |
| |
| static float4 __attribute__((overloadable)) |
| getSample_A(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float p0 = getElementAt1(p, stride, locX, locY); |
| float p1 = getElementAt1(p, stride, nextX, locY); |
| float p2 = getElementAt1(p, stride, locX, nextY); |
| float p3 = getElementAt1(p, stride, nextX, nextY); |
| float r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| float4 ret = {0.f, 0.f, 0.f, r}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_L(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float p0 = getElementAt1(p, stride, locX, locY); |
| float p1 = getElementAt1(p, stride, nextX, locY); |
| float p2 = getElementAt1(p, stride, locX, nextY); |
| float p3 = getElementAt1(p, stride, nextX, nextY); |
| float r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| float4 ret = {r, r, r, 1.f}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_LA(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float2 p0 = getElementAt2(p, stride, locX, locY); |
| float2 p1 = getElementAt2(p, stride, nextX, locY); |
| float2 p2 = getElementAt2(p, stride, locX, nextY); |
| float2 p3 = getElementAt2(p, stride, nextX, nextY); |
| float2 r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| float4 ret = {r.x, r.x, r.x, r.y}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_RGB(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float4 p0 = getElementAt4(p, stride, locX, locY); |
| float4 p1 = getElementAt4(p, stride, nextX, locY); |
| float4 p2 = getElementAt4(p, stride, locX, nextY); |
| float4 p3 = getElementAt4(p, stride, nextX, nextY); |
| float4 r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| float4 ret = {r.x, r.y, r.z, 1.f}; |
| return ret; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_RGBA(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float4 p0 = getElementAt4(p, stride, locX, locY); |
| float4 p1 = getElementAt4(p, stride, nextX, locY); |
| float4 p2 = getElementAt4(p, stride, locX, nextY); |
| float4 p3 = getElementAt4(p, stride, nextX, nextY); |
| float4 r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| return r; |
| } |
| static float4 __attribute__((overloadable)) |
| getSample_565(const uint8_t *p, size_t stride, |
| int locX, int locY, int nextX, int nextY, |
| float w0, float w1, float w2, float w3) { |
| float3 p0 = getElementAt565(p, stride, locX, locY); |
| float3 p1 = getElementAt565(p, stride, nextX, locY); |
| float3 p2 = getElementAt565(p, stride, locX, nextY); |
| float3 p3 = getElementAt565(p, stride, nextX, nextY); |
| float3 r = p0 * w0 + p1 * w1 + p2 * w2 + p3 * w3; |
| r *= (1.f / 255.f); |
| float4 ret; |
| ret.rgb = r; |
| ret.w = 1.f; |
| return ret; |
| } |
| |
| static float4 __attribute__((overloadable)) |
| getBilinearSample1D(const Allocation_t *alloc, float2 weights, |
| uint32_t iPixel, uint32_t next, |
| rs_data_kind dk, rs_data_type dt, uint32_t lod) { |
| |
| const uint8_t *p = (const uint8_t *)alloc->mHal.drvState.lod[lod].mallocPtr; |
| |
| switch(dk) { |
| case RS_KIND_PIXEL_RGBA: |
| return getSample_RGBA(p, iPixel, next, weights.x, weights.y); |
| case RS_KIND_PIXEL_A: |
| return getSample_A(p, iPixel, next, weights.x, weights.y); |
| case RS_KIND_PIXEL_RGB: |
| if (dt == RS_TYPE_UNSIGNED_5_6_5) { |
| return getSample_565(p, iPixel, next, weights.x, weights.y); |
| } |
| return getSample_RGB(p, iPixel, next, weights.x, weights.y); |
| case RS_KIND_PIXEL_L: |
| return getSample_L(p, iPixel, next, weights.x, weights.y); |
| case RS_KIND_PIXEL_LA: |
| return getSample_LA(p, iPixel, next, weights.x, weights.y); |
| |
| default: |
| //__builtin_unreachable(); |
| break; |
| } |
| |
| //__builtin_unreachable(); |
| return 0.f; |
| } |
| |
| static uint32_t wrapI(rs_sampler_value wrap, int32_t coord, int32_t size) { |
| if (wrap == RS_SAMPLER_WRAP) { |
| coord = coord % size; |
| if (coord < 0) { |
| coord += size; |
| } |
| } |
| if (wrap == RS_SAMPLER_MIRRORED_REPEAT) { |
| coord = coord % (size * 2); |
| if (coord < 0) { |
| coord = (size * 2) + coord; |
| } |
| if (coord >= size) { |
| coord = (size * 2 - 1) - coord; |
| } |
| } |
| return (uint32_t)max(0, min(coord, size - 1)); |
| } |
| |
| static float4 __attribute__((overloadable)) |
| getBilinearSample2D(const Allocation_t *alloc, float w0, float w1, float w2, float w3, |
| int lx, int ly, int nx, int ny, |
| rs_data_kind dk, rs_data_type dt, uint32_t lod) { |
| |
| const uint8_t *p = (const uint8_t *)alloc->mHal.drvState.lod[lod].mallocPtr; |
| size_t stride = alloc->mHal.drvState.lod[lod].stride; |
| |
| switch(dk) { |
| case RS_KIND_PIXEL_RGBA: |
| return getSample_RGBA(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| case RS_KIND_PIXEL_A: |
| return getSample_A(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| case RS_KIND_PIXEL_LA: |
| return getSample_LA(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| case RS_KIND_PIXEL_RGB: |
| if (dt == RS_TYPE_UNSIGNED_5_6_5) { |
| return getSample_565(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| } |
| return getSample_RGB(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| case RS_KIND_PIXEL_L: |
| return getSample_L(p, stride, lx, ly, nx, ny, w0, w1, w2, w3); |
| |
| default: |
| break; |
| } |
| |
| return 0.f; |
| } |
| |
| static float4 __attribute__((overloadable)) |
| getNearestSample(const Allocation_t *alloc, uint32_t iPixel, rs_data_kind dk, |
| rs_data_type dt, uint32_t lod) { |
| |
| const uint8_t *p = (const uint8_t *)alloc->mHal.drvState.lod[lod].mallocPtr; |
| |
| float4 result = {0.f, 0.f, 0.f, 255.f}; |
| |
| switch(dk) { |
| case RS_KIND_PIXEL_RGBA: |
| result = getElementAt4(p, iPixel); |
| break; |
| case RS_KIND_PIXEL_A: |
| result.w = getElementAt1(p, iPixel); |
| break; |
| case RS_KIND_PIXEL_LA: |
| result.zw = getElementAt2(p, iPixel); |
| result.xy = result.z; |
| break; |
| case RS_KIND_PIXEL_RGB: |
| if (dt == RS_TYPE_UNSIGNED_5_6_5) { |
| result.xyz = getElementAt565(p, iPixel); |
| } else { |
| result.xyz = getElementAt3(p, iPixel); |
| } |
| break; |
| case RS_KIND_PIXEL_L: |
| result.xyz = getElementAt1(p, iPixel); |
| |
| default: |
| //__builtin_unreachable(); |
| break; |
| } |
| |
| return result * 0.003921569f; |
| } |
| |
| static float4 __attribute__((overloadable)) |
| getNearestSample(const Allocation_t *alloc, uint2 iPixel, rs_data_kind dk, |
| rs_data_type dt, uint32_t lod) { |
| |
| const uint8_t *p = (const uint8_t *)alloc->mHal.drvState.lod[lod].mallocPtr; |
| size_t stride = alloc->mHal.drvState.lod[lod].stride; |
| |
| float4 result = {0.f, 0.f, 0.f, 255.f}; |
| |
| switch(dk) { |
| case RS_KIND_PIXEL_RGBA: |
| result = getElementAt4(p, stride, iPixel.x, iPixel.y); |
| break; |
| case RS_KIND_PIXEL_A: |
| result.w = getElementAt1(p, stride, iPixel.x, iPixel.y); |
| break; |
| case RS_KIND_PIXEL_LA: |
| result.zw = getElementAt2(p, stride, iPixel.x, iPixel.y); |
| result.xy = result.z; |
| break; |
| case RS_KIND_PIXEL_RGB: |
| if (dt == RS_TYPE_UNSIGNED_5_6_5) { |
| result.xyz = getElementAt565(p, stride, iPixel.x, iPixel.y); |
| } else { |
| result.xyz = getElementAt3(p, stride, iPixel.x, iPixel.y); |
| } |
| break; |
| |
| default: |
| //__builtin_unreachable(); |
| break; |
| } |
| |
| return result * 0.003921569f; |
| } |
| |
| static float4 __attribute__((overloadable)) |
| sample_LOD_LinearPixel(const Allocation_t *alloc, |
| rs_data_kind dk, rs_data_type dt, |
| rs_sampler_value wrapS, |
| float uv, uint32_t lod) { |
| |
| int32_t sourceW = alloc->mHal.drvState.lod[lod].dimX; |
| float pixelUV = uv * (float)(sourceW); |
| int32_t iPixel = floor(pixelUV); |
| float frac = pixelUV - (float)iPixel; |
| |
| if (frac < 0.5f) { |
| iPixel -= 1; |
| frac += 0.5f; |
| } else { |
| frac -= 0.5f; |
| } |
| |
| float oneMinusFrac = 1.0f - frac; |
| |
| float2 weights; |
| weights.x = oneMinusFrac; |
| weights.y = frac; |
| |
| uint32_t next = wrapI(wrapS, iPixel + 1, sourceW); |
| uint32_t location = wrapI(wrapS, iPixel, sourceW); |
| |
| return getBilinearSample1D(alloc, weights, location, next, dk, dt, lod); |
| } |
| |
| static float4 __attribute__((overloadable)) |
| sample_LOD_NearestPixel(const Allocation_t *alloc, |
| rs_data_kind dk, rs_data_type dt, |
| rs_sampler_value wrapS, |
| float uv, uint32_t lod) { |
| |
| int32_t sourceW = alloc->mHal.drvState.lod[lod].dimX; |
| int32_t iPixel = floor(uv * (float)(sourceW)); |
| uint32_t location = wrapI(wrapS, iPixel, sourceW); |
| |
| return getNearestSample(alloc, location, dk, dt, lod); |
| } |
| |
| static float4 __attribute__((overloadable)) |
| sample_LOD_LinearPixel(const Allocation_t *alloc, |
| rs_data_kind dk, rs_data_type dt, |
| rs_sampler_value wrapS, |
| rs_sampler_value wrapT, |
| float2 uv, uint32_t lod) { |
| |
| int sourceW = alloc->mHal.drvState.lod[lod].dimX; |
| int sourceH = alloc->mHal.drvState.lod[lod].dimY; |
| |
| float pixelU = uv.x * sourceW; |
| float pixelV = uv.y * sourceH; |
| int iPixelU = floor(pixelU); |
| int iPixelV = floor(pixelV); |
| float fracU = pixelU - iPixelU; |
| float fracV = pixelV - iPixelV; |
| |
| if (fracU < 0.5f) { |
| iPixelU -= 1; |
| fracU += 0.5f; |
| } else { |
| fracU -= 0.5f; |
| } |
| if (fracV < 0.5f) { |
| iPixelV -= 1; |
| fracV += 0.5f; |
| } else { |
| fracV -= 0.5f; |
| } |
| float oneMinusFracU = 1.0f - fracU; |
| float oneMinusFracV = 1.0f - fracV; |
| |
| float w0 = oneMinusFracU * oneMinusFracV; |
| float w1 = fracU * oneMinusFracV; |
| float w2 = oneMinusFracU * fracV; |
| float w3 = fracU * fracV; |
| |
| int nx = wrapI(wrapS, iPixelU + 1, sourceW); |
| int ny = wrapI(wrapT, iPixelV + 1, sourceH); |
| int lx = wrapI(wrapS, iPixelU, sourceW); |
| int ly = wrapI(wrapT, iPixelV, sourceH); |
| |
| return getBilinearSample2D(alloc, w0, w1, w2, w3, lx, ly, nx, ny, dk, dt, lod); |
| |
| } |
| |
| static float4 __attribute__((overloadable)) |
| sample_LOD_NearestPixel(const Allocation_t *alloc, |
| rs_data_kind dk, rs_data_type dt, |
| rs_sampler_value wrapS, |
| rs_sampler_value wrapT, |
| float2 uv, uint32_t lod) { |
| int sourceW = alloc->mHal.drvState.lod[lod].dimX; |
| int sourceH = alloc->mHal.drvState.lod[lod].dimY; |
| |
| float2 dimF; |
| dimF.x = (float)(sourceW); |
| dimF.y = (float)(sourceH); |
| int2 iPixel = convert_int2(floor(uv * dimF)); |
| |
| uint2 location; |
| location.x = wrapI(wrapS, iPixel.x, sourceW); |
| location.y = wrapI(wrapT, iPixel.y, sourceH); |
| return getNearestSample(alloc, location, dk, dt, lod); |
| } |
| |
| extern float4 __attribute__((overloadable)) |
| rsSample(rs_allocation a, rs_sampler s, float uv, float lod) { |
| |
| const Allocation_t *alloc = (const Allocation_t *)a.p; |
| const Sampler_t *prog = (Sampler_t *)s.p; |
| const Type_t *type = (Type_t *)alloc->mHal.state.type; |
| const Element_t *elem = type->mHal.state.element; |
| rs_data_kind dk = elem->mHal.state.dataKind; |
| rs_data_type dt = elem->mHal.state.dataType; |
| rs_sampler_value sampleMin = prog->mHal.state.minFilter; |
| rs_sampler_value sampleMag = prog->mHal.state.magFilter; |
| rs_sampler_value wrapS = prog->mHal.state.wrapS; |
| |
| if (!(alloc->mHal.state.usageFlags & RS_ALLOCATION_USAGE_GRAPHICS_TEXTURE)) { |
| return 0.f; |
| } |
| |
| if (lod <= 0.0f) { |
| if (sampleMag == RS_SAMPLER_NEAREST) { |
| return sample_LOD_NearestPixel(alloc, dk, dt, wrapS, uv, 0); |
| } |
| return sample_LOD_LinearPixel(alloc, dk, dt, wrapS, uv, 0); |
| } |
| |
| if (sampleMin == RS_SAMPLER_LINEAR_MIP_NEAREST) { |
| uint32_t maxLOD = type->mHal.state.lodCount - 1; |
| lod = min(lod, (float)maxLOD); |
| uint32_t nearestLOD = (uint32_t)round(lod); |
| return sample_LOD_LinearPixel(alloc, dk, dt, wrapS, uv, nearestLOD); |
| } |
| |
| if (sampleMin == RS_SAMPLER_LINEAR_MIP_LINEAR) { |
| uint32_t lod0 = (uint32_t)floor(lod); |
| uint32_t lod1 = (uint32_t)ceil(lod); |
| uint32_t maxLOD = type->mHal.state.lodCount - 1; |
| lod0 = min(lod0, maxLOD); |
| lod1 = min(lod1, maxLOD); |
| float4 sample0 = sample_LOD_LinearPixel(alloc, dk, dt, wrapS, uv, lod0); |
| float4 sample1 = sample_LOD_LinearPixel(alloc, dk, dt, wrapS, uv, lod1); |
| float frac = lod - (float)lod0; |
| return sample0 * (1.0f - frac) + sample1 * frac; |
| } |
| |
| return sample_LOD_NearestPixel(alloc, dk, dt, wrapS, uv, 0); |
| } |
| |
| extern float4 __attribute__((overloadable)) |
| rsSample(rs_allocation a, rs_sampler s, float location) { |
| return rsSample(a, s, location, 0); |
| } |
| |
| |
| extern float4 __attribute__((overloadable)) |
| rsSample(rs_allocation a, rs_sampler s, float2 uv, float lod) { |
| |
| const Allocation_t *alloc = (const Allocation_t *)a.p; |
| const Sampler_t *prog = (Sampler_t *)s.p; |
| const Type_t *type = (Type_t *)alloc->mHal.state.type; |
| const Element_t *elem = type->mHal.state.element; |
| rs_data_kind dk = elem->mHal.state.dataKind; |
| rs_data_type dt = elem->mHal.state.dataType; |
| rs_sampler_value sampleMin = prog->mHal.state.minFilter; |
| rs_sampler_value sampleMag = prog->mHal.state.magFilter; |
| rs_sampler_value wrapS = prog->mHal.state.wrapS; |
| rs_sampler_value wrapT = prog->mHal.state.wrapT; |
| |
| if (!(alloc->mHal.state.usageFlags & RS_ALLOCATION_USAGE_GRAPHICS_TEXTURE)) { |
| return 0.f; |
| } |
| |
| if (lod <= 0.0f) { |
| if (sampleMag == RS_SAMPLER_NEAREST) { |
| return sample_LOD_NearestPixel(alloc, dk, dt, wrapS, wrapT, uv, 0); |
| } |
| return sample_LOD_LinearPixel(alloc, dk, dt, wrapS, wrapT, uv, 0); |
| } |
| |
| if (sampleMin == RS_SAMPLER_LINEAR_MIP_NEAREST) { |
| uint32_t maxLOD = type->mHal.state.lodCount - 1; |
| lod = min(lod, (float)maxLOD); |
| uint32_t nearestLOD = (uint32_t)round(lod); |
| return sample_LOD_LinearPixel(alloc, dk, dt, wrapS, wrapT, uv, nearestLOD); |
| } |
| |
| if (sampleMin == RS_SAMPLER_LINEAR_MIP_LINEAR) { |
| uint32_t lod0 = (uint32_t)floor(lod); |
| uint32_t lod1 = (uint32_t)ceil(lod); |
| uint32_t maxLOD = type->mHal.state.lodCount - 1; |
| lod0 = min(lod0, maxLOD); |
| lod1 = min(lod1, maxLOD); |
| float4 sample0 = sample_LOD_LinearPixel(alloc, dk, dt, wrapS, wrapT, uv, lod0); |
| float4 sample1 = sample_LOD_LinearPixel(alloc, dk, dt, wrapS, wrapT, uv, lod1); |
| float frac = lod - (float)lod0; |
| return sample0 * (1.0f - frac) + sample1 * frac; |
| } |
| |
| return sample_LOD_NearestPixel(alloc, dk, dt, wrapS, wrapT, uv, 0); |
| } |
| |
| extern float4 __attribute__((overloadable)) |
| rsSample(rs_allocation a, rs_sampler s, float2 uv) { |
| |
| const Allocation_t *alloc = (const Allocation_t *)a.p; |
| const Sampler_t *prog = (Sampler_t *)s.p; |
| const Type_t *type = (Type_t *)alloc->mHal.state.type; |
| const Element_t *elem = type->mHal.state.element; |
| rs_data_kind dk = elem->mHal.state.dataKind; |
| rs_data_type dt = elem->mHal.state.dataType; |
| rs_sampler_value wrapS = prog->mHal.state.wrapS; |
| rs_sampler_value wrapT = prog->mHal.state.wrapT; |
| |
| if (!(alloc->mHal.state.usageFlags & RS_ALLOCATION_USAGE_GRAPHICS_TEXTURE)) { |
| return 0.f; |
| } |
| |
| if (prog->mHal.state.magFilter == RS_SAMPLER_NEAREST) { |
| return sample_LOD_NearestPixel(alloc, dk, dt, wrapS, wrapT, uv, 0); |
| } |
| return sample_LOD_LinearPixel(alloc, dk, dt, wrapS, wrapT, uv, 0); |
| } |
