| /* |
| * Copyright (C) 2012 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| |
| #include "rsdCore.h" |
| #include "rsdIntrinsics.h" |
| #include "rsdAllocation.h" |
| |
| #include "rsdIntrinsicInlines.h" |
| |
| using namespace android; |
| using namespace android::renderscript; |
| |
| struct YuvParams { |
| ObjectBaseRef<Allocation> alloc; |
| }; |
| |
| static void YuvToRGB_Bind(const Context *dc, const Script *script, |
| void * intrinsicData, uint32_t slot, Allocation *data) { |
| YuvParams *cp = (YuvParams *)intrinsicData; |
| rsAssert(slot == 0); |
| cp->alloc.set(data); |
| } |
| |
| |
| |
| static uchar4 rsYuvToRGBA_uchar4(uchar y, uchar u, uchar v) { |
| short Y = ((short)y) - 16; |
| short U = ((short)u) - 128; |
| short V = ((short)v) - 128; |
| |
| short4 p; |
| p.r = (Y * 298 + V * 409 + 128) >> 8; |
| p.g = (Y * 298 - U * 100 - V * 208 + 128) >> 8; |
| p.b = (Y * 298 + U * 516 + 128) >> 8; |
| p.a = 255; |
| if(p.r < 0) { |
| p.r = 0; |
| } |
| if(p.r > 255) { |
| p.r = 255; |
| } |
| if(p.g < 0) { |
| p.g = 0; |
| } |
| if(p.g > 255) { |
| p.g = 255; |
| } |
| if(p.b < 0) { |
| p.b = 0; |
| } |
| if(p.b > 255) { |
| p.b = 255; |
| } |
| |
| return (uchar4){p.r, p.g, p.b, p.a}; |
| } |
| |
| //extern "C" void rsdIntrinsicYuvToRGB_K(void *dst, const uchar *Y, const uchar *uv, uint32_t count); |
| |
| static void YuvToRGB_uchar4(const RsForEachStubParamStruct *p, |
| uint32_t xstart, uint32_t xend, |
| uint32_t instep, uint32_t outstep) { |
| YuvParams *cp = (YuvParams *)p->usr; |
| DrvAllocation *din = (DrvAllocation *)cp->alloc->mHal.drv; |
| const uchar *pin = (const uchar *)din->lod[0].mallocPtr; |
| |
| const uchar *Y = pin + (p->y * p->dimX); |
| const uchar *uv = pin + (p->dimX * p->dimY); |
| uv += (p->y>>1) * p->dimX; |
| |
| uchar4 *out = (uchar4 *)p->out; |
| uint32_t x1 = xstart; |
| uint32_t x2 = xend; |
| |
| if(x2 > x1) { |
| #if 0//defined(ARCH_ARM_HAVE_NEON) |
| int32_t len = (x2 - x1 - 1) >> 1; |
| if(len > 0) { |
| rsdIntrinsicYuvToRGB_K(out, Y, uv, len); |
| x1 += len << 1; |
| out += len << 1; |
| } |
| #endif |
| |
| // ALOGE("y %i %i %i", p->y, x1, x2); |
| while(x1 < x2) { |
| uchar u = uv[(x1 & 0xffffe) + 1]; |
| uchar v = uv[(x1 & 0xffffe) + 0]; |
| *out = rsYuvToRGBA_uchar4(Y[x1], u, v); |
| out++; |
| x1++; |
| *out = rsYuvToRGBA_uchar4(Y[x1], u, v); |
| out++; |
| x1++; |
| } |
| } |
| } |
| |
| void * rsdIntrinsic_InitYuvToRGB(const android::renderscript::Context *dc, |
| android::renderscript::Script *script, |
| RsdIntriniscFuncs_t *funcs) { |
| |
| script->mHal.info.exportedVariableCount = 1; |
| funcs->bind = YuvToRGB_Bind; |
| funcs->root = YuvToRGB_uchar4; |
| YuvParams *cp = (YuvParams *)calloc(1, sizeof(YuvParams)); |
| return cp; |
| } |
| |
| |