tests/tests/hardware/src/android/hardware/camera2/cts/means_yuvx_444_2d_to_1d.rs

/*
 * 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;
}