1// Copyright (C) 2011 The Android Open Source Project 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14 15#include "ip.rsh" 16 17uint32_t gMaxIteration = 500; 18uint32_t gDimX = 1024; 19uint32_t gDimY = 1024; 20 21float lowerBoundX = -2.f; 22float lowerBoundY = -2.f; 23float scaleFactor = 4.f; 24 25uchar4 RS_KERNEL root(uint32_t x, uint32_t y) { 26 float2 p; 27 p.x = lowerBoundX + ((float)x / gDimX) * scaleFactor; 28 p.y = lowerBoundY + ((float)y / gDimY) * scaleFactor; 29 30 float2 t = 0; 31 float2 t2 = t * t; 32 int iter = 0; 33 while((t2.x + t2.y < 4.f) && (iter < gMaxIteration)) { 34 float xtemp = t2.x - t2.y + p.x; 35 t.y = 2 * t.x * t.y + p.y; 36 t.x = xtemp; 37 iter++; 38 t2 = t * t; 39 } 40 41 if(iter >= gMaxIteration) { 42 // write a non-transparent black pixel 43 return (uchar4){0, 0, 0, 0xff}; 44 } else { 45 float mi3 = gMaxIteration / 3.f; 46 if (iter <= (gMaxIteration / 3)) 47 return (uchar4){0xff * (iter / mi3), 0, 0, 0xff}; 48 else if (iter <= (((gMaxIteration / 3) * 2))) 49 return (uchar4){0xff - (0xff * ((iter - mi3) / mi3)), 50 (0xff * ((iter - mi3) / mi3)), 0, 0xff}; 51 else 52 return (uchar4){0, 0xff - (0xff * ((iter - (mi3 * 2)) / mi3)), 53 (0xff * ((iter - (mi3 * 2)) / mi3)), 0xff}; 54 } 55} 56 57uchar4 RS_KERNEL rootD(uint32_t x, uint32_t y) { 58 double2 p; 59 p.x = lowerBoundX + ((float)x / gDimX) * scaleFactor; 60 p.y = lowerBoundY + ((float)y / gDimY) * scaleFactor; 61 62 double2 t = 0; 63 double2 t2 = t * t; 64 int iter = 0; 65 while((t2.x + t2.y < 4.f) && (iter < gMaxIteration)) { 66 double xtemp = t2.x - t2.y + p.x; 67 t.y = 2 * t.x * t.y + p.y; 68 t.x = xtemp; 69 iter++; 70 t2 = t * t; 71 } 72 73 if(iter >= gMaxIteration) { 74 // write a non-transparent black pixel 75 return (uchar4){0, 0, 0, 0xff}; 76 } else { 77 double mi3 = gMaxIteration / 3.f; 78 if (iter <= (gMaxIteration / 3)) 79 return (uchar4){0xff * (iter / mi3), 0, 0, 0xff}; 80 else if (iter <= (((gMaxIteration / 3) * 2))) 81 return (uchar4){0xff - (0xff * ((iter - mi3) / mi3)), 82 (0xff * ((iter - mi3) / mi3)), 0, 0xff}; 83 else 84 return (uchar4){0, 0xff - (0xff * ((iter - (mi3 * 2)) / mi3)), 85 (0xff * ((iter - (mi3 * 2)) / mi3)), 0xff}; 86 } 87} 88 89