| /* |
| * Copyright 2014 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "common.rs" |
| |
| // #define LOG_DEBUG 1 |
| |
| // Must be YUV 420 888 (flexible YUV) |
| rs_allocation mInput; |
| |
| uint32_t width; |
| float inv_width; // 1/w |
| uint32_t src_x; // x-offset from mInput |
| uint32_t src_y; // y-offset from mInput |
| |
| // Average a 2D array -> 1D array (by each row) |
| // Input: mInput must be yuvf_420 |
| yuvx_444 RS_KERNEL means_yuvf_420(uint32_t x) { |
| |
| LOGD("x", x); |
| |
| uint3 sum = { 0, 0, 0 }; |
| |
| for (uint32_t i = 0; i < width; ++i) { |
| uchar py = rsGetElementAtYuv_uchar_Y(mInput, src_x + i, src_y + x); |
| uchar pu = rsGetElementAtYuv_uchar_U(mInput, src_x + i, src_y + x); |
| uchar pv = rsGetElementAtYuv_uchar_V(mInput, src_x + i, src_y + x); |
| |
| yuvx_444 elem = { py, pu, pv }; |
| |
| LOGD("elem", elem); |
| |
| sum.x += elem.x; |
| sum.y += elem.y; |
| sum.z += elem.z; |
| } |
| |
| sum.x *= inv_width; // multiply by 1/w |
| sum.y *= inv_width; // multiply by 1/w |
| sum.z *= inv_width; // multiply by 1/w |
| |
| yuvx_444 avg = convert_yuvx_444(sum); |
| |
| return avg; |
| } |
| |
| // Average a 2D array -> 1D array (by each row) |
| // Input: mInput must be yuvx_444 |
| yuvx_444 RS_KERNEL means_yuvx_444(uint32_t x) { |
| |
| LOGD("x", x); |
| |
| uint3 sum = { 0, 0, 0 }; |
| |
| for (uint32_t i = 0; i < width; ++i) { |
| yuvx_444 elem = rsGetElementAt_yuvx_444(mInput, src_x + i, src_y + x); |
| |
| LOGD("elem", elem); |
| |
| sum.x += elem.x; |
| sum.y += elem.y; |
| sum.z += elem.z; |
| } |
| |
| sum.x *= inv_width; // multiply by 1/w |
| sum.y *= inv_width; // multiply by 1/w |
| sum.z *= inv_width; // multiply by 1/w |
| |
| yuvx_444 avg = convert_yuvx_444(sum); |
| |
| return avg; |
| } |
| |