/*
* Copyright (C) 2013 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 "rsCpuIntrinsic.h"
#include "rsCpuIntrinsicInlines.h"
#ifdef RS_COMPATIBILITY_LIB
#include "rsCompatibilityLib.h"
#endif
#ifndef RS_COMPATIBILITY_LIB
#include "hardware/gralloc.h"
#endif
using namespace android;
using namespace android::renderscript;
namespace android {
namespace renderscript {
class RsdCpuScriptIntrinsicYuvToRGB : public RsdCpuScriptIntrinsic {
public:
virtual void populateScript(Script *);
virtual void invokeFreeChildren();
virtual void setGlobalObj(uint32_t slot, ObjectBase *data);
virtual ~RsdCpuScriptIntrinsicYuvToRGB();
RsdCpuScriptIntrinsicYuvToRGB(RsdCpuReferenceImpl *ctx, const Script *s, const Element *e);
protected:
ObjectBaseRef<Allocation> alloc;
static void kernel(const RsForEachStubParamStruct *p,
uint32_t xstart, uint32_t xend,
uint32_t instep, uint32_t outstep);
};
}
}
void RsdCpuScriptIntrinsicYuvToRGB::setGlobalObj(uint32_t slot, ObjectBase *data) {
rsAssert(slot == 0);
alloc.set(static_cast<Allocation *>(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.x = (Y * 298 + V * 409 + 128) >> 8;
p.y = (Y * 298 - U * 100 - V * 208 + 128) >> 8;
p.z = (Y * 298 + U * 516 + 128) >> 8;
p.w = 255;
if(p.x < 0) {
p.x = 0;
}
if(p.x > 255) {
p.x = 255;
}
if(p.y < 0) {
p.y = 0;
}
if(p.y > 255) {
p.y = 255;
}
if(p.z < 0) {
p.z = 0;
}
if(p.z > 255) {
p.z = 255;
}
return (uchar4){p.x, p.y, p.z, p.w};
}
static short YuvCoeff[] = {
298, 409, -100, 516, -208, 255, 0, 0,
16, 16, 16, 16, 16, 16, 16, 16,
128, 128, 128, 128, 128, 128, 128, 128,
298, 298, 298, 298, 298, 298, 298, 298,
255, 255, 255, 255, 255, 255, 255, 255
};
extern "C" void rsdIntrinsicYuv_K(void *dst, const uchar *Y, const uchar *uv, uint32_t count, const short *param);
extern "C" void rsdIntrinsicYuvR_K(void *dst, const uchar *Y, const uchar *uv, uint32_t count, const short *param);
extern "C" void rsdIntrinsicYuv2_K(void *dst, const uchar *Y, const uchar *u, const uchar *v, uint32_t count, const short *param);
void RsdCpuScriptIntrinsicYuvToRGB::kernel(const RsForEachStubParamStruct *p,
uint32_t xstart, uint32_t xend,
uint32_t instep, uint32_t outstep) {
RsdCpuScriptIntrinsicYuvToRGB *cp = (RsdCpuScriptIntrinsicYuvToRGB *)p->usr;
if (!cp->alloc.get()) {
ALOGE("YuvToRGB executed without input, skipping");
return;
}
const uchar *pinY = (const uchar *)cp->alloc->mHal.drvState.lod[0].mallocPtr;
if (pinY == NULL) {
ALOGE("YuvToRGB executed without data, skipping");
return;
}
size_t strideY = cp->alloc->mHal.drvState.lod[0].stride;
// calculate correct stride in legacy case
if (cp->alloc->mHal.drvState.lod[0].dimY == 0) {
strideY = p->dimX;
}
const uchar *Y = pinY + (p->y * strideY);
uchar4 *out = (uchar4 *)p->out;
uint32_t x1 = xstart;
uint32_t x2 = xend;
size_t cstep = cp->alloc->mHal.drvState.yuv.step;
const uchar *pinU = (const uchar *)cp->alloc->mHal.drvState.lod[1].mallocPtr;
const size_t strideU = cp->alloc->mHal.drvState.lod[1].stride;
const uchar *u = pinU + ((p->y >> 1) * strideU);
const uchar *pinV = (const uchar *)cp->alloc->mHal.drvState.lod[2].mallocPtr;
const size_t strideV = cp->alloc->mHal.drvState.lod[2].stride;
const uchar *v = pinV + ((p->y >> 1) * strideV);
//ALOGE("pinY, %p, Y, %p, p->y, %d, strideY, %d", pinY, Y, p->y, strideY);
//ALOGE("pinU, %p, U, %p, p->y, %d, strideU, %d", pinU, u, p->y, strideU);
//ALOGE("pinV, %p, V, %p, p->y, %d, strideV, %d", pinV, v, p->y, strideV);
//ALOGE("dimX, %d, dimY, %d", cp->alloc->mHal.drvState.lod[0].dimX, cp->alloc->mHal.drvState.lod[0].dimY);
//ALOGE("p->dimX, %d, p->dimY, %d", p->dimX, p->dimY);
if (pinU == NULL) {
// Legacy yuv support didn't fill in uv
v = ((uint8_t *)cp->alloc->mHal.drvState.lod[0].mallocPtr) +
(strideY * p->dimY) +
((p->y >> 1) * strideY);
u = v + 1;
cstep = 2;
}
#if defined(ARCH_ARM_HAVE_VFP)
if((x2 > x1) && gArchUseSIMD) {
// The neon paths may over-read by up to 8 bytes
int32_t len = (x2 - x1 - 8) >> 3;
if(len > 0) {
if (cstep == 1) {
rsdIntrinsicYuv2_K(out, Y, u, v, len, YuvCoeff);
x1 += len << 3;
out += len << 3;
} else if (cstep == 2) {
// Check for proper interleave
intptr_t ipu = (intptr_t)u;
intptr_t ipv = (intptr_t)v;
if (ipu == (ipv + 1)) {
rsdIntrinsicYuv_K(out, Y, v, len, YuvCoeff);
x1 += len << 3;
out += len << 3;
} else if (ipu == (ipv - 1)) {
rsdIntrinsicYuvR_K(out, Y, u, len, YuvCoeff);
x1 += len << 3;
out += len << 3;
}
}
}
}
#endif
if(x2 > x1) {
// ALOGE("y %i %i %i", p->y, x1, x2);
while(x1 < x2) {
int cx = (x1 >> 1) * cstep;
*out = rsYuvToRGBA_uchar4(Y[x1], u[cx], v[cx]);
out++;
x1++;
*out = rsYuvToRGBA_uchar4(Y[x1], u[cx], v[cx]);
out++;
x1++;
}
}
}
RsdCpuScriptIntrinsicYuvToRGB::RsdCpuScriptIntrinsicYuvToRGB(
RsdCpuReferenceImpl *ctx, const Script *s, const Element *e)
: RsdCpuScriptIntrinsic(ctx, s, e, RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB) {
mRootPtr = &kernel;
}
RsdCpuScriptIntrinsicYuvToRGB::~RsdCpuScriptIntrinsicYuvToRGB() {
}
void RsdCpuScriptIntrinsicYuvToRGB::populateScript(Script *s) {
s->mHal.info.exportedVariableCount = 1;
}
void RsdCpuScriptIntrinsicYuvToRGB::invokeFreeChildren() {
alloc.clear();
}
RsdCpuScriptImpl * rsdIntrinsic_YuvToRGB(RsdCpuReferenceImpl *ctx,
const Script *s, const Element *e) {
return new RsdCpuScriptIntrinsicYuvToRGB(ctx, s, e);
}