basicArraySupport.h

Go to the documentation of this file.

// Xerus - A General Purpose Tensor Library
// Copyright (C) 2014-2017 Benjamin Huber and Sebastian Wolf. 
// 
// Xerus is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License,
// or (at your option) any later version.
// 
// Xerus 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 Affero General Public License for more details.
// 
// You should have received a copy of the GNU Affero General Public License
// along with Xerus. If not, see <http://www.gnu.org/licenses/>.
//
// For further information on Xerus visit https://libXerus.org 
// or contact us at contact@libXerus.org.

#pragma once

#include <type_traits>
#include <memory>
#include <cstring>

#include "standard.h"

namespace xerus {
    namespace misc {
        template<class T>
        std::unique_ptr<T[]> make_unique_array(T* _ptr) {
            return std::unique_ptr<T[]>(_ptr);
        }
        
        template <typename T, typename std::enable_if<std::is_trivial<T>::value, int>::type = 0>
        inline void set_zero(T* const __restrict _x, const size_t _n) noexcept {
            memset(_x, 0, _n*sizeof(T));
        }
        
        
        template <typename T, typename std::enable_if<std::is_trivial<T>::value, int>::type = 0>
        inline void copy(T* const __restrict _dest, const T* const __restrict _src, const size_t _n) noexcept {
            memcpy(_dest, _src, _n*sizeof(T));
        }
        
        
        template <typename T, typename std::enable_if<std::is_trivial<T>::value, int>::type = 0>
        inline void copy_inplace(T* const _x, const T* const _y, const size_t _n) noexcept {
            memmove(_x, _y, _n*sizeof(T));
        }
        
        
        template <typename T>
        inline void copy_scaled(T* const __restrict _x, const T _alpha, const T* const _y, const size_t _n) noexcept {
            for(size_t i = 0; i < _n; ++i) {
                _x[i] = _alpha*_y[i];
            }
        }
        
        
        template <typename T>
        inline void scale(T* const __restrict _x, const T _alpha, const size_t _n) noexcept {
            for(size_t i = 0; i < _n; i++) {
                _x[i] *= _alpha;
            }
        }
        
        
        template <typename T>
        inline void add(T* const __restrict _x, const T* const __restrict _y, const size_t _n) noexcept {
            for(size_t i = 0; i < _n; i++) {
                _x[i] += _y[i];
            }
        }
        
        
        template <typename T>
        inline void add_scaled(T* const __restrict _x, const T _alpha, const T* const __restrict _y, const size_t _n) noexcept {
            for(size_t i = 0; i < _n; i++) {
                _x[i] += _alpha*_y[i];
            }
        }
    }
}