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