array_upsampler3.h

Go to the documentation of this file.

/*
**  array_upsampler3.h
**
**  This is part of Shiokaze, a research-oriented fluid solver for computer graphics.
**  Created by Ryoichi Ando <rand@nii.ac.jp> on Sep 16, 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_UPSAMPLER3_H
#define SHKZ_ARRAY_UPSAMPLER3_H
//
#include <shiokaze/array/array_interpolator3.h>
//
SHKZ_BEGIN_NAMESPACE
//
namespace array_upsampler3 {
    template<class T> void upsample_to_double_cell( const array3<T> &array, double dx, array3<T> &doubled_array ) {
        //
        assert(doubled_array.shape() == 2*array.shape());
        //
        array.const_serial_actives([&]( int i, int j, int k, auto &it ) {
            for( int ii=0; ii<2; ++ii ) for( int jj=0; jj<2; ++jj ) for( int kk=0; kk<2; ++kk ) {
                vec3i pi (2*i+ii,2*j+jj,2*k+kk);
                if( ! doubled_array.shape().out_of_bounds(pi)) {
                    doubled_array.set(pi,0.0);
                }
            }
        });
        //
        doubled_array.parallel_actives([&]( int i, int j, int k, auto &it, int tn) {
            vec3d p = 0.5*vec3i(i,j,k).cell()-vec3d(0.5,0.5,0.5);
            it.set(array_interpolator3::interpolate<T>(array,p));
        });
        //
        if( array.is_levelset()) {
            doubled_array.set_as_levelset(0.5*array.get_background_value());
            doubled_array.flood_fill();
        }
    }
    //
    template<class T> void upsample_to_double_nodal( const array3<T> &array, double dx, array3<T> &doubled_array ) {
        //
        assert(doubled_array.shape() == 2*array.shape()-shape3(1,1,1));
        //
        array.const_serial_actives([&]( int i, int j, int k, auto &it ) {
            for( int ii=0; ii<2; ++ii ) for( int jj=0; jj<2; ++jj ) for( int kk=0; kk<2; ++kk ) {
                vec3i pi (2*i+ii,2*j+jj,2*k+kk);
                if( ! doubled_array.shape().out_of_bounds(pi)) {
                    doubled_array.set(pi,0.0);
                }
            }
        });
        //
        doubled_array.parallel_actives([&]( int i, int j, int k, auto &it, int tn) {
            vec3d p = 0.5*vec3i(i,j,k).nodal();
            it.set(array_interpolator3::interpolate<T>(array,p));
        });
        //
        if( array.is_levelset()) {
            doubled_array.set_as_levelset(0.5*array.get_background_value());
            doubled_array.flood_fill();
        }
    }
};
//
SHKZ_END_NAMESPACE
//
#endif