1572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams/* 2572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Copyright (C) 2012 The Android Open Source Project 3572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 4572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * you may not use this file except in compliance with the License. 6572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * You may obtain a copy of the License at 7572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 8572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * http://www.apache.org/licenses/LICENSE-2.0 9572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 10572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Unless required by applicable law or agreed to in writing, software 11572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * See the License for the specific language governing permissions and 14572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * limitations under the License. 15572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams */ 16572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 17572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams#include "ip.rsh" 186a804605b99cab4ffa3cc55c691338fd4a5396eaJason Sams#pragma rs_fp_relaxed 19572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 20025b5f82971c431eb22df3c9d0f00b3cbe426bdbChris Wailesuchar RS_KERNEL genRand() { 21572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams return (uchar)rsRand(0xff); 22572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 23572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 24572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams/* 25572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * Convolution matrix of distance 2 with fixed point of 'kShiftBits' bits 26572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * shifted. Thus the sum of this matrix should be 'kShiftValue'. Entries of 27572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * small values are not calculated to gain efficiency. 28572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * The order ot pixels represented in this matrix is: 29572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 1 2 3 30572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 4 0 5 31572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 6 7 8 32572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * and the matrix should be: {230, 56, 114, 56, 114, 114, 56, 114, 56}. 33572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * However, since most of the valus are identical, we only use the first three 34572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * entries and the entries corresponding to the pixels is: 35572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 1 2 1 36572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 2 0 2 37572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams * 1 2 1 38572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams */ 39572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 40572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsint32_t gWMask; 41572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsint32_t gHMask; 42572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 43572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsrs_allocation gBlendSource; 44025b5f82971c431eb22df3c9d0f00b3cbe426bdbChris Wailesuchar RS_KERNEL blend9(uint32_t x, uint32_t y) { 45572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint32_t x1 = (x-1) & gWMask; 46572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint32_t x2 = (x+1) & gWMask; 47572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint32_t y1 = (y-1) & gHMask; 48572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint32_t y2 = (y+1) & gHMask; 49572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 50572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p00 = 56 * rsGetElementAt_uchar(gBlendSource, x1, y1); 51572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p01 = 114 * rsGetElementAt_uchar(gBlendSource, x, y1); 52572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p02 = 56 * rsGetElementAt_uchar(gBlendSource, x2, y1); 53572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p10 = 114 * rsGetElementAt_uchar(gBlendSource, x1, y); 54572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p11 = 230 * rsGetElementAt_uchar(gBlendSource, x, y); 55572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p12 = 114 * rsGetElementAt_uchar(gBlendSource, x2, y); 56572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p20 = 56 * rsGetElementAt_uchar(gBlendSource, x1, y2); 57572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p21 = 114 * rsGetElementAt_uchar(gBlendSource, x, y2); 58572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uint p22 = 56 * rsGetElementAt_uchar(gBlendSource, x2, y2); 59572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 60572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p00 += p01; 61572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p02 += p10; 62572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p11 += p12; 63572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p20 += p21; 64572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 65572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p22 += p00; 66572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p02 += p11; 67572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 68572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p20 += p22; 69572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p20 += p02; 70572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 71572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p20 = min(p20 >> 10, (uint)255); 72572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams return (uchar)p20; 73572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 74572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 75572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsfloat gNoiseStrength; 76572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 77572a5031a5d8602db0bec0b253428a034bd4dd59Jason Samsrs_allocation gNoise; 78025b5f82971c431eb22df3c9d0f00b3cbe426bdbChris Wailesuchar4 RS_KERNEL root(uchar4 in, uint32_t x, uint32_t y) { 79572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams float4 ip = convert_float4(in); 80572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams float pnoise = (float) rsGetElementAt_uchar(gNoise, x & gWMask, y & gHMask); 81572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 82572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams float energy_level = ip.r + ip.g + ip.b; 83572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams float energy_mask = (28.f - sqrt(energy_level)) * 0.03571f; 84572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams pnoise = (pnoise - 128.f) * energy_mask; 85572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 86572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams ip += pnoise * gNoiseStrength; 87572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams ip = clamp(ip, 0.f, 255.f); 88572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams 89572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams uchar4 p = convert_uchar4(ip); 90572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams p.a = 0xff; 91572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams return p; 92572a5031a5d8602db0bec0b253428a034bd4dd59Jason Sams} 93