tsgUtils.hpp

/*
 * Copyright (c) 2017, Miroslav Stoyanov
 *
 * This file is part of
 * Toolkit for Adaptive Stochastic Modeling And Non-Intrusive ApproximatioN: TASMANIAN
 *
 * Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions
 *    and the following disclaimer in the documentation and/or other materials provided with the distribution.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse
 *    or promote products derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * UT-BATTELLE, LLC AND THE UNITED STATES GOVERNMENT MAKE NO REPRESENTATIONS AND DISCLAIM ALL WARRANTIES, BOTH EXPRESSED AND IMPLIED.
 * THERE ARE NO EXPRESS OR IMPLIED WARRANTIES OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, OR THAT THE USE OF THE SOFTWARE WILL NOT INFRINGE ANY PATENT,
 * COPYRIGHT, TRADEMARK, OR OTHER PROPRIETARY RIGHTS, OR THAT THE SOFTWARE WILL ACCOMPLISH THE INTENDED RESULTS OR THAT THE SOFTWARE OR ITS USE WILL NOT RESULT IN INJURY OR DAMAGE.
 * THE USER ASSUMES RESPONSIBILITY FOR ALL LIABILITIES, PENALTIES, FINES, CLAIMS, CAUSES OF ACTION, AND COSTS AND EXPENSES, CAUSED BY, RESULTING FROM OR ARISING OUT OF,
 * IN WHOLE OR IN PART THE USE, STORAGE OR DISPOSAL OF THE SOFTWARE.
 */
 
#ifndef __TASMANIAN_SPARSE_GRID_UTILS_HPP
#define __TASMANIAN_SPARSE_GRID_UTILS_HPP

 
#include "tsgMathUtils.hpp"

 
namespace TasGrid{

namespace Utils{

template<typename scalar_type>
void transpose(long long M, long long N, scalar_type const A[], scalar_type B[]);
template<typename scalar_type>
std::vector<scalar_type> transpose(long long M, long long N, scalar_type const A[]){
    std::vector<scalar_type> B(M * N);
    transpose(M, N, A, B.data());
    return B;
}

template<typename IntA, typename IntB>
inline size_t size_mult(IntA a, IntB b){ return static_cast<size_t>(a) * static_cast<size_t>(b); }

template<typename T, typename I>
std::vector<typename std::remove_const<T>::type> copyArray(T* x, I size){
    return (x == nullptr) ? std::vector<typename std::remove_const<T>::type>() :
        std::vector<typename std::remove_const<T>::type>(x, x + static_cast<size_t>(size));
}

template<typename T>
std::vector<T> mergeVectors(std::vector<std::vector<T>> const &vec){
    size_t total_size = 0;
    for(auto const &v : vec) total_size += v.size();
    std::vector<T> result;
    result.reserve(total_size);
    for(auto const &v : vec) result.insert(result.end(), v.begin(), v.end());
    return result;
}

template<typename T>
class Wrapper2D{
public:
    Wrapper2D(int stride_size, T *raw_data) : stride(static_cast<size_t>(stride_size)), data(raw_data){}
    ~Wrapper2D(){}

    T* getStrip(int i){ return &(data[size_mult(i, stride)]); }
 
private:
    size_t stride;
    T *data;
};

template<bool condition>
using use_if = typename std::enable_if<condition, void>::type;

template<typename T, typename U> T exchange(T& x, U new_x){
    T old_x = x;
    x = static_cast<T>(new_x);
    return old_x;
}

template<typename T, typename... Args>
std::unique_ptr<T> make_unique(Args&&... args){
    return std::unique_ptr<T>(new T(std::forward<Args>(args)...));
}
 
}
 
}
 
#endif