array_extrapolator3.h

Go to the documentation of this file.

/*
**  array_extrapolator3.h
**
**  This is part of Shiokaze, a research-oriented fluid solver for computer graphics.
**  Created by Ryoichi Ando <rand@nii.ac.jp> on Feb 14, 2018.
**
**  Permission is hereby granted, free of charge, to any person obtaining a copy of
**  this software and associated documentation files (the "Software"), to deal in
**  the Software without restriction, including without limitation the rights to use,
**  copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
**  Software, and to permit persons to whom the Software is furnished to do so,
**  subject to the following conditions:
**
**  The above copyright notice and this permission notice shall be included in all copies
**  or substantial portions of the Software.
**
**  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
**  INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
**  PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
**  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
**  CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
**  OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
//
#ifndef SHKZ_ARRAY_EXTRAPOLATOR3_H
#define SHKZ_ARRAY_EXTRAPOLATOR3_H
//
#include "array3.h"
#include "shared_array3.h"
//
SHKZ_BEGIN_NAMESPACE
//
namespace array_extrapolator3 {
    template<class T> unsigned extrapolate_if ( array3<T> &array, std::function<bool(int i, int j, int k, int thread_index )> func, int count=1 ) {
        //
        std::vector<unsigned> counters(array.get_thread_num());
        const shape3 shape(array.shape());
        //
        array.dilate([&](int i, int j, int k, auto &it, int tn ) {
            if( func(i,j,k,tn)) {
                T sum (0.0);
                int weight = 0;
                vec3i query[] = {vec3i(i+1,j,k),vec3i(i-1,j,k),vec3i(i,j+1,k),
                                 vec3i(i,j-1,k),vec3i(i,j,k-1),vec3i(i,j,k+1)};
                for( int nq=0; nq<6; nq++ ) {
                    vec3i qi (query[nq]);
                    if( ! shape.out_of_bounds(qi) ) {
                        if( array.active(qi)) {
                            sum += array(qi);
                            weight ++;
                        }
                    }
                }
                if( weight ) {
                    counters[tn] ++;
                    it.set(sum / weight);
                }
            }
        },count);
        //
        unsigned sum (0);
        for( const auto &e : counters ) sum += e;
        return sum;
    }
    template<class T> unsigned extrapolate ( array3<T> &array, int count=1 ) {
        return extrapolate_if(array,[&](int i, int j, int k, int tn){ return true; },count);
    }
};
//
SHKZ_END_NAMESPACE
//
#endif
//