ufo-filters-data 0.14.1+dfsg1-1.1 / usr / share / ufo / correlate.cl

/*
 * Copyright (C) 2017 Karlsruhe Institute of Technology
 *
 * This file is part of Ufo.
 *
 * This library is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 */

kernel void
diff (global float *refs, global float *input, global float *result, unsigned height)
{
    const size_t x = get_global_id (0);
    const size_t y = get_global_id (1);
    const size_t width = get_global_size (0);
    const size_t idx = y * width + x;
    const size_t in_y = y % height;
    const float diff = refs[idx] - input[in_y * width + x];
    result[idx] = diff * diff;
}

kernel void
sum (global float *result, global float *matrix, unsigned width, unsigned height, unsigned row)
{
    const unsigned x = get_global_id (0);
    const unsigned offset = x * width * height;
    const unsigned total = width * height;
    float sum = 0.0f;

    for (unsigned i = 0; i < total; i++) {
        sum += result[offset + i];
    }

    matrix[get_global_size (0) * row + x] = sum;
}