guess_list_1d.hpp

Go to the documentation of this file.

#ifndef GUESS_LIST_1D_HPP
#define GUESS_LIST_1D_HPP

#include "my_mirror_view.hpp"

template<class Device>
struct GuessList1D{
    View<int**,CLayout,Device> list;

    GuessList1D(){}

    GuessList1D(const UniformRange& range, const View<double*, Device>& view)
     : list("guess_list_1D", 2, range.n)
    {
        // Run a binary search for each value of
        // the uniform grid --> upper and lower
        // index on the non-uniform mesh
        for(int i=0; i<range.n; i++){
            double val = range.get_val(i);
            int down=1;
            int up=view.size();
            int middle=down + (up-down)/2;
            while(up-down > 1){
                if (val >= view(middle-1)){
                    down = middle;
                }else{
                    up = middle;
                }
                middle = down + (up-down)/2;
            }

            list(0,i)=down;
            list(1,i)=up;

            if (val < view(down-1) || val > view(up-1)){
                list(0,i) = -1;
                list(1,i) = -1;
            }
        }

        for(int i=0; i<(range.n-1); i++){
            // Correct the upper bound to make sure that
            // upper and lower bound enclose the interval
            // range.val_min+i*range.dval to range.val_min+(i+1)*range.dval
            list(1, i) = list(1, i+1);
        }
    }

    // Create a mirror with a different device type
    template<class Device2>
    inline GuessList1D<Device2> mirror() const{
        GuessList1D<Device2> m;

        m.list = my_mirror_view(list, Device2());
        mirror_copy(m.list, list);

        return m;
    }
};

#endif