| /* | |
| * Copyright (C) 2008-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 <malloc.h> | |
| #include "RenderScript.h" | |
| #include "rsCppInternal.h" | |
| using namespace android; | |
| using namespace RSC; | |
| ScriptIntrinsic::ScriptIntrinsic(sp<RS> rs, int id, sp<const Element> e) | |
| : Script(nullptr, rs) { | |
| mID = createDispatch(rs, RS::dispatch->ScriptIntrinsicCreate(rs->getContext(), id, | |
| e != nullptr ? e->getID() : 0)); | |
| mElement = e; | |
| } | |
| ScriptIntrinsic::~ScriptIntrinsic() { | |
| } | |
| sp<ScriptIntrinsic3DLUT> ScriptIntrinsic3DLUT::create(sp<RS> rs, sp<const Element> e) { | |
| if (e->isCompatible(Element::U8_4(rs)) == false) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Element not supported for intrinsic"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsic3DLUT(rs, e); | |
| } | |
| ScriptIntrinsic3DLUT::ScriptIntrinsic3DLUT(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_3DLUT, e) { | |
| } | |
| void ScriptIntrinsic3DLUT::forEach(sp<Allocation> ain, sp<Allocation> aout) { | |
| if (ain->getType()->getElement()->isCompatible(mElement) == false || | |
| aout->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "3DLUT forEach element mismatch"); | |
| return; | |
| } | |
| Script::forEach(0, ain, aout, nullptr, 0); | |
| } | |
| void ScriptIntrinsic3DLUT::setLUT(sp<Allocation> lut) { | |
| sp<const Type> t = lut->getType(); | |
| if (!t->getElement()->isCompatible(mElement)) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "setLUT element does not match"); | |
| return; | |
| } | |
| if (t->getZ() == 0) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, "setLUT Allocation must be 3D"); | |
| return; | |
| } | |
| Script::setVar(0, lut); | |
| } | |
| sp<ScriptIntrinsicBlend> ScriptIntrinsicBlend::create(sp<RS> rs, sp<const Element> e) { | |
| if (e->isCompatible(Element::U8_4(rs)) == false) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Element not supported for intrinsic"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicBlend(rs, e); | |
| } | |
| ScriptIntrinsicBlend::ScriptIntrinsicBlend(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_BLEND, e) { | |
| } | |
| void ScriptIntrinsicBlend::forEachClear(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(0, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSrc(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(1, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachDst(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(2, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSrcOver(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(3, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachDstOver(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(4, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSrcIn(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(5, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachDstIn(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(6, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSrcOut(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(7, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachDstOut(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(8, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSrcAtop(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(9, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachDstAtop(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(10, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachXor(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(11, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachMultiply(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(14, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachAdd(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(34, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlend::forEachSubtract(sp<Allocation> in, sp<Allocation> out) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false || | |
| out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blend"); | |
| } | |
| Script::forEach(35, in, out, nullptr, 0); | |
| } | |
| sp<ScriptIntrinsicBlur> ScriptIntrinsicBlur::create(sp<RS> rs, sp<const Element> e) { | |
| if ((e->isCompatible(Element::U8_4(rs)) == false) && | |
| (e->isCompatible(Element::U8(rs)) == false)) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blur"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicBlur(rs, e); | |
| } | |
| ScriptIntrinsicBlur::ScriptIntrinsicBlur(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_BLUR, e) { | |
| } | |
| void ScriptIntrinsicBlur::setInput(sp<Allocation> in) { | |
| if (in->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blur input"); | |
| return; | |
| } | |
| Script::setVar(1, in); | |
| } | |
| void ScriptIntrinsicBlur::forEach(sp<Allocation> out) { | |
| if (out->getType()->getElement()->isCompatible(mElement) == false) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element in blur output"); | |
| return; | |
| } | |
| Script::forEach(0, nullptr, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicBlur::setRadius(float radius) { | |
| if (radius > 0.f && radius <= 25.f) { | |
| Script::setVar(0, &radius, sizeof(float)); | |
| } else { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Blur radius out of 0-25 pixel bound"); | |
| } | |
| } | |
| sp<ScriptIntrinsicColorMatrix> ScriptIntrinsicColorMatrix::create(sp<RS> rs) { | |
| return new ScriptIntrinsicColorMatrix(rs, Element::RGBA_8888(rs)); | |
| } | |
| ScriptIntrinsicColorMatrix::ScriptIntrinsicColorMatrix(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_COLOR_MATRIX, e) { | |
| float add[4] = {0.f, 0.f, 0.f, 0.f}; | |
| setAdd(add); | |
| } | |
| void ScriptIntrinsicColorMatrix::forEach(sp<Allocation> in, sp<Allocation> out) { | |
| if (!(in->getType()->getElement()->isCompatible(Element::U8(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::U8_2(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::U8_3(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::U8_4(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::F32(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::F32_2(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::F32_3(mRS))) && | |
| !(in->getType()->getElement()->isCompatible(Element::F32_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for ColorMatrix"); | |
| return; | |
| } | |
| if (!(out->getType()->getElement()->isCompatible(Element::U8(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U8_2(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U8_3(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U8_4(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::F32(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::F32_2(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::F32_3(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::F32_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for ColorMatrix"); | |
| return; | |
| } | |
| Script::forEach(0, in, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicColorMatrix::setAdd(float* add) { | |
| Script::setVar(1, (void*)add, sizeof(float) * 4); | |
| } | |
| void ScriptIntrinsicColorMatrix::setColorMatrix3(float* m) { | |
| float temp[16]; | |
| temp[0] = m[0]; | |
| temp[1] = m[1]; | |
| temp[2] = m[2]; | |
| temp[3] = 0.f; | |
| temp[4] = m[3]; | |
| temp[5] = m[4]; | |
| temp[6] = m[5]; | |
| temp[7] = 0.f; | |
| temp[8] = m[6]; | |
| temp[9] = m[7]; | |
| temp[10] = m[8]; | |
| temp[11] = 0.f; | |
| temp[12] = 0.f; | |
| temp[13] = 0.f; | |
| temp[14] = 0.f; | |
| temp[15] = 1.f; | |
| setColorMatrix4(temp); | |
| } | |
| void ScriptIntrinsicColorMatrix::setColorMatrix4(float* m) { | |
| Script::setVar(0, (void*)m, sizeof(float) * 16); | |
| } | |
| void ScriptIntrinsicColorMatrix::setGreyscale() { | |
| float matrix[] = {0.299f, 0.299f, 0.299f,0.587f,0.587f,0.587f,0.114f,0.114f, 0.114f}; | |
| setColorMatrix3(matrix); | |
| } | |
| void ScriptIntrinsicColorMatrix::setRGBtoYUV() { | |
| float matrix[] = { 0.299f, -0.14713f, 0.615f, 0.587f, -0.28886f, -0.51499f, 0.114f, 0.436f, -0.10001f}; | |
| setColorMatrix3(matrix); | |
| } | |
| void ScriptIntrinsicColorMatrix::setYUVtoRGB() { | |
| float matrix[] = {1.f, 1.f, 1.f, 0.f, -0.39465f, 2.03211f, 1.13983f, -0.5806f, 0.f}; | |
| setColorMatrix3(matrix); | |
| } | |
| sp<ScriptIntrinsicConvolve3x3> ScriptIntrinsicConvolve3x3::create(sp<RS> rs, sp<const Element> e) { | |
| if (!(e->isCompatible(Element::U8(rs))) && | |
| !(e->isCompatible(Element::U8_2(rs))) && | |
| !(e->isCompatible(Element::U8_3(rs))) && | |
| !(e->isCompatible(Element::U8_4(rs))) && | |
| !(e->isCompatible(Element::F32(rs))) && | |
| !(e->isCompatible(Element::F32_2(rs))) && | |
| !(e->isCompatible(Element::F32_3(rs))) && | |
| !(e->isCompatible(Element::F32_4(rs)))) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for Convolve3x3"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicConvolve3x3(rs, e); | |
| } | |
| ScriptIntrinsicConvolve3x3::ScriptIntrinsicConvolve3x3(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_CONVOLVE_3x3, e) { | |
| } | |
| void ScriptIntrinsicConvolve3x3::setInput(sp<Allocation> in) { | |
| if (!(in->getType()->getElement()->isCompatible(mElement))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Element mismatch in Convolve3x3"); | |
| return; | |
| } | |
| Script::setVar(1, in); | |
| } | |
| void ScriptIntrinsicConvolve3x3::forEach(sp<Allocation> out) { | |
| if (!(out->getType()->getElement()->isCompatible(mElement))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Element mismatch in Convolve3x3"); | |
| return; | |
| } | |
| Script::forEach(0, nullptr, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicConvolve3x3::setCoefficients(float* v) { | |
| Script::setVar(0, (void*)v, sizeof(float) * 9); | |
| } | |
| sp<ScriptIntrinsicConvolve5x5> ScriptIntrinsicConvolve5x5::create(sp<RS> rs, sp<const Element> e) { | |
| if (!(e->isCompatible(Element::U8(rs))) && | |
| !(e->isCompatible(Element::U8_2(rs))) && | |
| !(e->isCompatible(Element::U8_3(rs))) && | |
| !(e->isCompatible(Element::U8_4(rs))) && | |
| !(e->isCompatible(Element::F32(rs))) && | |
| !(e->isCompatible(Element::F32_2(rs))) && | |
| !(e->isCompatible(Element::F32_3(rs))) && | |
| !(e->isCompatible(Element::F32_4(rs)))) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for Convolve5x5"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicConvolve5x5(rs, e); | |
| } | |
| ScriptIntrinsicConvolve5x5::ScriptIntrinsicConvolve5x5(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_CONVOLVE_5x5, e) { | |
| } | |
| void ScriptIntrinsicConvolve5x5::setInput(sp<Allocation> in) { | |
| if (!(in->getType()->getElement()->isCompatible(mElement))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Element mismatch in Convolve5x5 input"); | |
| return; | |
| } | |
| Script::setVar(1, in); | |
| } | |
| void ScriptIntrinsicConvolve5x5::forEach(sp<Allocation> out) { | |
| if (!(out->getType()->getElement()->isCompatible(mElement))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Element mismatch in Convolve5x5 output"); | |
| return; | |
| } | |
| Script::forEach(0, nullptr, out, nullptr, 0); | |
| } | |
| void ScriptIntrinsicConvolve5x5::setCoefficients(float* v) { | |
| Script::setVar(0, (void*)v, sizeof(float) * 25); | |
| } | |
| sp<ScriptIntrinsicHistogram> ScriptIntrinsicHistogram::create(sp<RS> rs, sp<const Element> e) { | |
| return new ScriptIntrinsicHistogram(rs, e); | |
| } | |
| ScriptIntrinsicHistogram::ScriptIntrinsicHistogram(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_HISTOGRAM, e) { | |
| } | |
| void ScriptIntrinsicHistogram::setOutput(sp<Allocation> out) { | |
| if (!(out->getType()->getElement()->isCompatible(Element::U32(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U32_2(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U32_3(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::U32_4(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::I32(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::I32_2(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::I32_3(mRS))) && | |
| !(out->getType()->getElement()->isCompatible(Element::I32_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for Histogram output"); | |
| return; | |
| } | |
| if (out->getType()->getX() != 256 || | |
| out->getType()->getY() != 0 || | |
| out->getType()->hasMipmaps()) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Invalid Allocation type for Histogram output"); | |
| return; | |
| } | |
| mOut = out; | |
| Script::setVar(1, out); | |
| } | |
| void ScriptIntrinsicHistogram::setDotCoefficients(float r, float g, float b, float a) { | |
| if ((r < 0.f) || (g < 0.f) || (b < 0.f) || (a < 0.f)) { | |
| return; | |
| } | |
| if ((r + g + b + a) > 1.f) { | |
| return; | |
| } | |
| FieldPacker fp(16); | |
| fp.add(r); | |
| fp.add(g); | |
| fp.add(b); | |
| fp.add(a); | |
| Script::setVar(0, fp.getData(), fp.getLength()); | |
| } | |
| void ScriptIntrinsicHistogram::forEach(sp<Allocation> ain) { | |
| if (ain->getType()->getElement()->getVectorSize() < | |
| mOut->getType()->getElement()->getVectorSize()) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, | |
| "Input vector size must be >= output vector size"); | |
| return; | |
| } | |
| if (!(ain->getType()->getElement()->isCompatible(Element::U8(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::U8_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, | |
| "Input allocation to Histogram must be U8 or U8_4"); | |
| return; | |
| } | |
| Script::forEach(0, ain, nullptr, nullptr, 0); | |
| } | |
| void ScriptIntrinsicHistogram::forEach_dot(sp<Allocation> ain) { | |
| if (mOut->getType()->getElement()->getVectorSize() != 1) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, | |
| "Output Histogram allocation must have vector size of 1 " \ | |
| "when used with forEach_dot"); | |
| return; | |
| } | |
| if (!(ain->getType()->getElement()->isCompatible(Element::U8(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::U8_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, | |
| "Input allocation to Histogram must be U8 or U8_4"); | |
| return; | |
| } | |
| Script::forEach(1, ain, nullptr, nullptr, 0); | |
| } | |
| sp<ScriptIntrinsicLUT> ScriptIntrinsicLUT::create(sp<RS> rs, sp<const Element> e) { | |
| if (!(e->isCompatible(Element::U8_4(rs)))) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for LUT"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicLUT(rs, e); | |
| } | |
| ScriptIntrinsicLUT::ScriptIntrinsicLUT(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_LUT, e), mDirty(true) { | |
| LUT = Allocation::createSized(rs, Element::U8(rs), 1024); | |
| for (int i = 0; i < 256; i++) { | |
| mCache[i] = i; | |
| mCache[i+256] = i; | |
| mCache[i+512] = i; | |
| mCache[i+768] = i; | |
| } | |
| setVar(0, LUT); | |
| } | |
| void ScriptIntrinsicLUT::forEach(sp<Allocation> ain, sp<Allocation> aout) { | |
| if (mDirty) { | |
| LUT->copy1DFrom((void*)mCache); | |
| mDirty = false; | |
| } | |
| if (!(ain->getType()->getElement()->isCompatible(Element::U8_4(mRS))) || | |
| !(aout->getType()->getElement()->isCompatible(Element::U8_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for LUT"); | |
| return; | |
| } | |
| Script::forEach(0, ain, aout, nullptr, 0); | |
| } | |
| void ScriptIntrinsicLUT::setTable(unsigned int offset, unsigned char base, unsigned int length, unsigned char* lutValues) { | |
| if ((base + length) > 256 || length == 0) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, "LUT out of range"); | |
| return; | |
| } | |
| mDirty = true; | |
| for (unsigned int i = 0; i < length; i++) { | |
| mCache[offset + base + i] = lutValues[i]; | |
| } | |
| } | |
| void ScriptIntrinsicLUT::setRed(unsigned char base, unsigned int length, unsigned char* lutValues) { | |
| setTable(0, base, length, lutValues); | |
| } | |
| void ScriptIntrinsicLUT::setGreen(unsigned char base, unsigned int length, unsigned char* lutValues) { | |
| setTable(256, base, length, lutValues); | |
| } | |
| void ScriptIntrinsicLUT::setBlue(unsigned char base, unsigned int length, unsigned char* lutValues) { | |
| setTable(512, base, length, lutValues); | |
| } | |
| void ScriptIntrinsicLUT::setAlpha(unsigned char base, unsigned int length, unsigned char* lutValues) { | |
| setTable(768, base, length, lutValues); | |
| } | |
| ScriptIntrinsicLUT::~ScriptIntrinsicLUT() { | |
| } | |
| sp<ScriptIntrinsicResize> ScriptIntrinsicResize::create(sp<RS> rs) { | |
| return new ScriptIntrinsicResize(rs, nullptr); | |
| } | |
| ScriptIntrinsicResize::ScriptIntrinsicResize(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_RESIZE, e) { | |
| } | |
| void ScriptIntrinsicResize::forEach_bicubic(sp<Allocation> aout) { | |
| if (aout == mInput) { | |
| mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Resize Input and Ouput cannot be the same"); | |
| } | |
| if (!(mInput->getType()->getElement()->isCompatible(aout->getType()->getElement()))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Resize forEach element mismatch"); | |
| return; | |
| } | |
| Script::forEach(0, nullptr, aout, nullptr, 0); | |
| } | |
| void ScriptIntrinsicResize::setInput(sp<Allocation> ain) { | |
| if (!(ain->getType()->getElement()->isCompatible(Element::U8(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::U8_2(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::U8_3(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::U8_4(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::F32(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::F32_2(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::F32_3(mRS))) && | |
| !(ain->getType()->getElement()->isCompatible(Element::F32_4(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for Resize Input"); | |
| return; | |
| } | |
| mInput = ain; | |
| Script::setVar(0, ain); | |
| } | |
| sp<ScriptIntrinsicYuvToRGB> ScriptIntrinsicYuvToRGB::create(sp<RS> rs, sp<const Element> e) { | |
| if (!(e->isCompatible(Element::U8_4(rs)))) { | |
| rs->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for YuvToRGB"); | |
| return nullptr; | |
| } | |
| return new ScriptIntrinsicYuvToRGB(rs, e); | |
| } | |
| ScriptIntrinsicYuvToRGB::ScriptIntrinsicYuvToRGB(sp<RS> rs, sp<const Element> e) | |
| : ScriptIntrinsic(rs, RS_SCRIPT_INTRINSIC_ID_YUV_TO_RGB, e) { | |
| } | |
| void ScriptIntrinsicYuvToRGB::setInput(sp<Allocation> in) { | |
| if (!(in->getType()->getElement()->isCompatible(Element::YUV(mRS)))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for input in YuvToRGB"); | |
| return; | |
| } | |
| Script::setVar(0, in); | |
| } | |
| void ScriptIntrinsicYuvToRGB::forEach(sp<Allocation> out) { | |
| if (!(out->getType()->getElement()->isCompatible(mElement))) { | |
| mRS->throwError(RS_ERROR_INVALID_ELEMENT, "Invalid element for output in YuvToRGB"); | |
| return; | |
| } | |
| Script::forEach(0, nullptr, out, nullptr, 0); | |
| } |