1d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams/* 2d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * Copyright (C) 2012 The Android Open Source Project 3d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * 4d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * you may not use this file except in compliance with the License. 6d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * You may obtain a copy of the License at 7d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * 8d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * http://www.apache.org/licenses/LICENSE-2.0 9d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * 10d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * Unless required by applicable law or agreed to in writing, software 11d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * See the License for the specific language governing permissions and 14d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams * limitations under the License. 15d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams */ 16d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 17d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 18d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams#include "rsdCore.h" 19d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams#include "rsdIntrinsics.h" 20d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams#include "rsdAllocation.h" 21d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 22d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams#include "rsdIntrinsicInlines.h" 23d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 24d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsusing namespace android; 25d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsusing namespace android::renderscript; 26d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 27d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsstruct ConvolveParams { 28d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams float fp[28]; 29d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams short ip[28]; 30d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams ObjectBaseRef<Allocation> alloc; 31d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams}; 32d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 33d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsstatic void Convolve5x5_Bind(const Context *dc, const Script *script, 34d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams void * intrinsicData, uint32_t slot, Allocation *data) { 35d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams ConvolveParams *cp = (ConvolveParams *)intrinsicData; 36d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams rsAssert(slot == 1); 37d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams cp->alloc.set(data); 38d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams} 39d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 40d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsstatic void Convolve5x5_SetVar(const Context *dc, const Script *script, void * intrinsicData, 41d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t slot, void *data, size_t dataLength) { 42d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams ConvolveParams *cp = (ConvolveParams *)intrinsicData; 43d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 44d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams rsAssert(slot == 0); 45d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams memcpy (cp->fp, data, dataLength); 46d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams for(int ct=0; ct < 25; ct++) { 47d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams cp->ip[ct] = (short)(cp->fp[ct] * 255.f + 0.5f); 48d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams } 49d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams} 50d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 51d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 52d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsstatic void One(const RsForEachStubParamStruct *p, uint32_t x, uchar4 *out, 53d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py0, const uchar4 *py1, const uchar4 *py2, const uchar4 *py3, const uchar4 *py4, 54d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const float* coeff) { 55d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 56d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x0 = rsMax((int32_t)x-2, 0); 57d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x1 = rsMax((int32_t)x-1, 0); 58d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x2 = x; 59d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x3 = rsMin((int32_t)x+1, (int32_t)(p->dimX-1)); 60d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x4 = rsMin((int32_t)x+2, (int32_t)(p->dimX-1)); 61d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 62d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams float4 px = convert_float4(py0[x0]) * coeff[0] + 63d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py0[x1]) * coeff[1] + 64d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py0[x2]) * coeff[2] + 65d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py0[x3]) * coeff[3] + 66d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py0[x4]) * coeff[4] + 67d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 68d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py1[x0]) * coeff[5] + 69d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py1[x1]) * coeff[6] + 70d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py1[x2]) * coeff[7] + 71d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py1[x3]) * coeff[8] + 72d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py1[x4]) * coeff[9] + 73d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 74d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py2[x0]) * coeff[10] + 75d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py2[x1]) * coeff[11] + 76d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py2[x2]) * coeff[12] + 77d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py2[x3]) * coeff[13] + 78d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py2[x4]) * coeff[14] + 79d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 80d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py3[x0]) * coeff[15] + 81d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py3[x1]) * coeff[16] + 82d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py3[x2]) * coeff[17] + 83d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py3[x3]) * coeff[18] + 84d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py3[x4]) * coeff[19] + 85d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 86d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py4[x0]) * coeff[20] + 87d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py4[x1]) * coeff[21] + 88d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py4[x2]) * coeff[22] + 89d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py4[x3]) * coeff[23] + 90d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams convert_float4(py4[x4]) * coeff[24]; 91d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 92d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams px = clamp(px, 0.f, 255.f); 93d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uchar4 o = {(uchar)px.x, (uchar)px.y, (uchar)px.z, (uchar)px.w}; 94d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams *out = o; 95d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams} 96d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 97a1b08e2cacf3891fcd6895422c6124887b75975eJason Samsextern "C" void rsdIntrinsicConvolve5x5_K(void *dst, const void *y0, const void *y1, 98a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams const void *y2, const void *y3, const void *y4, 99a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams const short *coef, uint32_t count); 100a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams 101d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsstatic void Convolve5x5_uchar4(const RsForEachStubParamStruct *p, 102d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t xstart, uint32_t xend, 103d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t instep, uint32_t outstep) { 104d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams ConvolveParams *cp = (ConvolveParams *)p->usr; 105b801b949e286275b5d19a33135235ba68d3a19a9Jason Sams if (!cp->alloc.get()) { 106b801b949e286275b5d19a33135235ba68d3a19a9Jason Sams ALOGE("Convolve5x5 executed without input, skipping"); 107b801b949e286275b5d19a33135235ba68d3a19a9Jason Sams return; 108b801b949e286275b5d19a33135235ba68d3a19a9Jason Sams } 109d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams DrvAllocation *din = (DrvAllocation *)cp->alloc->mHal.drv; 110d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar *pin = (const uchar *)din->lod[0].mallocPtr; 111d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 112d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t y0 = rsMax((int32_t)p->y-2, 0); 113d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t y1 = rsMax((int32_t)p->y-1, 0); 114d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t y2 = p->y; 115d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t y3 = rsMin((int32_t)p->y+1, (int32_t)(p->dimY-1)); 116d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t y4 = rsMin((int32_t)p->y+2, (int32_t)(p->dimY-1)); 117d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 118d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py0 = (const uchar4 *)(pin + din->lod[0].stride * y0); 119d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py1 = (const uchar4 *)(pin + din->lod[0].stride * y1); 120d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py2 = (const uchar4 *)(pin + din->lod[0].stride * y2); 121d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py3 = (const uchar4 *)(pin + din->lod[0].stride * y3); 122d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams const uchar4 *py4 = (const uchar4 *)(pin + din->lod[0].stride * y4); 123d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 124d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uchar4 *out = (uchar4 *)p->out; 125d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x1 = xstart; 126d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams uint32_t x2 = xend; 127d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 128a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams while((x1 < x2) && (x1 < 2)) { 129a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams One(p, x1, out, py0, py1, py2, py3, py4, cp->fp); 130a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams out++; 131a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams x1++; 132a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams } 133a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams 134a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams#if defined(ARCH_ARM_HAVE_NEON) 135a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams if((x1 + 3) < x2) { 136a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams uint32_t len = (x2 - x1 - 3) >> 1; 137a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams rsdIntrinsicConvolve5x5_K(out, py0, py1, py2, py3, py4, cp->ip, len); 138a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams out += len << 1; 139a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams x1 += len << 1; 140a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams } 141a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams#endif 142a1b08e2cacf3891fcd6895422c6124887b75975eJason Sams 143d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams while(x1 < x2) { 144d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams One(p, x1, out, py0, py1, py2, py3, py4, cp->fp); 145d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams out++; 146d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams x1++; 147d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams } 148d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams} 149d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 150d85e283087ecd00760a0d8d0c9d8482cda845efcJason Samsvoid * rsdIntrinsic_InitConvolve5x5(const android::renderscript::Context *dc, 151d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams android::renderscript::Script *script, 152d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams RsdIntriniscFuncs_t *funcs) { 153d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 154d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams script->mHal.info.exportedVariableCount = 2; 15578b050ea1e13108110ce3b8ead63252e5ebe2468Jason Sams funcs->setVarObj = Convolve5x5_Bind; 156d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams funcs->setVar = Convolve5x5_SetVar; 157d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams funcs->root = Convolve5x5_uchar4; 158d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 159d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams ConvolveParams *cp = (ConvolveParams *)calloc(1, sizeof(ConvolveParams)); 160d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams for(int ct=0; ct < 25; ct++) { 161d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams cp->fp[ct] = 1.f / 25.f; 162d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams cp->ip[ct] = (short)(cp->fp[ct] * 255.f + 0.5f); 163d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams } 164d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams return cp; 165d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams} 166d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 167d85e283087ecd00760a0d8d0c9d8482cda845efcJason Sams 168