array_operators.hpp

// This file is part of the ACTS project.
//
// Copyright (C) 2016 CERN for the benefit of the ACTS project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
#pragma once
 
// Project include(s)
#include "detray/algebra/concepts.hpp"
#include "detray/definitions/detail/qualifiers.hpp"
 
// System include(s).
#include <cstddef>
#include <type_traits>
 
namespace detray::algebra::storage {

template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator*(
    const array_t<scalar_t, N> &a, scalar_t s) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = a[i] * s;
  }
 
  return result;
}
 
template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator*(
    scalar_t s, const array_t<scalar_t, N> &a) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = s * a[i];
  }
 
  return result;
}
 
template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator-(
    const array_t<scalar_t, N> &a, const array_t<scalar_t, N> &b) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = a[i] - b[i];
  }
 
  return result;
}
 
template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator+(
    const array_t<scalar_t, N> &a, const array_t<scalar_t, N> &b) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = a[i] + b[i];
  }
 
  return result;
}
 
template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator*(
    const array_t<scalar_t, N> &a, const array_t<scalar_t, N> &b) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = a[i] * b[i];
  }
 
  return result;
}
 
template <template <typename, std::size_t> class array_t,
          concepts::scalar scalar_t, std::size_t N>
DETRAY_HOST_DEVICE constexpr array_t<scalar_t, N> operator/(
    const array_t<scalar_t, N> &a, const array_t<scalar_t, N> &b) noexcept {
  array_t<scalar_t, N> result;
  for (std::size_t i = 0u; i < N; ++i) {
    result[i] = a[i] / b[i];
  }
 
  return result;
}

 
}  // namespace detray::algebra::storage