150b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma/* 250b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * Copyright (C) 2012 The Android Open Source Project 350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * 450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * Licensed under the Apache License, Version 2.0 (the "License"); 550b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * you may not use this file except in compliance with the License. 650b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * You may obtain a copy of the License at 750b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * 850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * http://www.apache.org/licenses/LICENSE-2.0 950b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * 1050b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * Unless required by applicable law or agreed to in writing, software 1150b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * distributed under the License is distributed on an "AS IS" BASIS, 1250b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * See the License for the specific language governing permissions and 1450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma * limitations under the License. 1550b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma */ 1650b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 1750b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharmars_allocation in_alloc; 1850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharmars_sampler sampler; 1950b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 2050b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharmastatic float2 center, neg_center, inv_dimensions, axis_scale; 2150b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharmastatic float alpha, radius2, factor; 2250b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 2350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharmavoid init_filter(uint32_t dim_x, uint32_t dim_y, float center_x, float center_y, float k) { 2450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma center.x = center_x; 2550b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma center.y = center_y; 2650b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma neg_center = -center; 2750b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma inv_dimensions.x = 1.f / (float)dim_x; 2850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma inv_dimensions.y = 1.f / (float)dim_y; 29d3d6570df4f39af697ba090fa8069aef6acfbcf4Stephen Hines alpha = k * 2.0f + 0.75f; 3050b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 3150b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma axis_scale = (float2)1.f; 3250b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma if (dim_x > dim_y) 3350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma axis_scale.y = (float)dim_y / (float)dim_x; 3450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma else 3550b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma axis_scale.x = (float)dim_x / (float)dim_y; 3650b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 37d3d6570df4f39af697ba090fa8069aef6acfbcf4Stephen Hines const float bound2 = 0.25f * (axis_scale.x*axis_scale.x + axis_scale.y*axis_scale.y); 3850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float bound = sqrt(bound2); 39d3d6570df4f39af697ba090fa8069aef6acfbcf4Stephen Hines const float radius = 1.15f * bound; 4050b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma radius2 = radius*radius; 4150b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float max_radian = M_PI_2 - atan(alpha / bound * sqrt(radius2 - bound2)); 4250b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma factor = bound / max_radian; 4350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma} 4450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 45d3d6570df4f39af697ba090fa8069aef6acfbcf4Stephen Hinesuchar4 __attribute__((kernel)) root(uint32_t x, uint32_t y) { 4650b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma // Convert x and y to floating point coordinates with center as origin 4750b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float2 inCoord = {(float)x, (float)y}; 4850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float2 coord = mad(inCoord, inv_dimensions, neg_center); 4950b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float2 scaledCoord = axis_scale * coord; 5050b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float dist2 = scaledCoord.x*scaledCoord.x + scaledCoord.y*scaledCoord.y; 5136fa5a42ba1e09d1aff2087ad352e6d63f36f652Jason Sams const float inv_dist = half_rsqrt(dist2); 5236fa5a42ba1e09d1aff2087ad352e6d63f36f652Jason Sams const float radian = M_PI_2 - atan((alpha * half_sqrt(radius2 - dist2)) * inv_dist); 5350b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float scalar = radian * factor * inv_dist; 5450b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float2 new_coord = mad(coord, scalar, center); 5550b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma const float4 fout = rsSample(in_alloc, sampler, new_coord); 56d3d6570df4f39af697ba090fa8069aef6acfbcf4Stephen Hines return rsPackColorTo8888(fout); 5750b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma} 5850b1ba0083115c4fb578f13e0b94c6d80b5b1841Rajeev Sharma 59