1/* 2 * Copyright (C) 2012 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include "ip.rsh" 18#pragma rs_fp_relaxed 19 20uchar RS_KERNEL genRand() { 21 return (uchar)rsRand(0xff); 22} 23 24/* 25 * Convolution matrix of distance 2 with fixed point of 'kShiftBits' bits 26 * shifted. Thus the sum of this matrix should be 'kShiftValue'. Entries of 27 * small values are not calculated to gain efficiency. 28 * The order ot pixels represented in this matrix is: 29 * 1 2 3 30 * 4 0 5 31 * 6 7 8 32 * and the matrix should be: {230, 56, 114, 56, 114, 114, 56, 114, 56}. 33 * However, since most of the valus are identical, we only use the first three 34 * entries and the entries corresponding to the pixels is: 35 * 1 2 1 36 * 2 0 2 37 * 1 2 1 38 */ 39 40int32_t gWMask; 41int32_t gHMask; 42 43rs_allocation gBlendSource; 44uchar RS_KERNEL blend9(uint32_t x, uint32_t y) { 45 uint32_t x1 = (x-1) & gWMask; 46 uint32_t x2 = (x+1) & gWMask; 47 uint32_t y1 = (y-1) & gHMask; 48 uint32_t y2 = (y+1) & gHMask; 49 50 uint p00 = 56 * rsGetElementAt_uchar(gBlendSource, x1, y1); 51 uint p01 = 114 * rsGetElementAt_uchar(gBlendSource, x, y1); 52 uint p02 = 56 * rsGetElementAt_uchar(gBlendSource, x2, y1); 53 uint p10 = 114 * rsGetElementAt_uchar(gBlendSource, x1, y); 54 uint p11 = 230 * rsGetElementAt_uchar(gBlendSource, x, y); 55 uint p12 = 114 * rsGetElementAt_uchar(gBlendSource, x2, y); 56 uint p20 = 56 * rsGetElementAt_uchar(gBlendSource, x1, y2); 57 uint p21 = 114 * rsGetElementAt_uchar(gBlendSource, x, y2); 58 uint p22 = 56 * rsGetElementAt_uchar(gBlendSource, x2, y2); 59 60 p00 += p01; 61 p02 += p10; 62 p11 += p12; 63 p20 += p21; 64 65 p22 += p00; 66 p02 += p11; 67 68 p20 += p22; 69 p20 += p02; 70 71 p20 = min(p20 >> 10, (uint)255); 72 return (uchar)p20; 73} 74 75float gNoiseStrength; 76 77rs_allocation gNoise; 78uchar4 RS_KERNEL root(uchar4 in, uint32_t x, uint32_t y) { 79 float4 ip = convert_float4(in); 80 float pnoise = (float) rsGetElementAt_uchar(gNoise, x & gWMask, y & gHMask); 81 82 float energy_level = ip.r + ip.g + ip.b; 83 float energy_mask = (28.f - sqrt(energy_level)) * 0.03571f; 84 pnoise = (pnoise - 128.f) * energy_mask; 85 86 ip += pnoise * gNoiseStrength; 87 ip = clamp(ip, 0.f, 255.f); 88 89 uchar4 p = convert_uchar4(ip); 90 p.a = 0xff; 91 return p; 92} 93