vc_aos_types.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 "algebra/impl/vc_aos_approximately_equal.hpp"
#include "algebra/impl/vc_aos_concepts.hpp"
#include "algebra/impl/vc_aos_getter.hpp"
#include "algebra/impl/vc_aos_transform3.hpp"
#include "detray/algebra/common/matrix.hpp"
#include "detray/algebra/common/vector.hpp"
#include "detray/algebra/concepts.hpp"
#include "detray/algebra/type_traits.hpp"
 
// System include(s).
#include <array>
#include <cstddef>
 
// Vc include(s).
#ifdef _MSC_VER
#pragma warning(push, 0)
#endif  // MSVC
#include <Vc/Vc>
#ifdef _MSC_VER
#pragma warning(pop)
#endif  // MSVC
 
namespace detray {
 
namespace algebra::vc_aos {

using index_type = std::size_t;
template <concepts::value T>
using value_type = T;
template <concepts::value T>
using scalar_type = T;
template <concepts::value T, index_type N>
using storage_type = Vc::SimdArray<T, N>;
template <concepts::value T>
using value_type = T;
template <concepts::value T>
using scalar_type = T;
template <concepts::value T, std::size_t N>
using vector_type = algebra::storage::vector<N, T, storage_type>;
template <concepts::value T, index_type ROWS, index_type COLS>
using matrix_type = algebra::storage::matrix<storage_type, T, ROWS, COLS>;

template <concepts::value T>
using vector2 = vector_type<T, 2>;
template <concepts::value T>
using point2 = vector2<T>;
template <concepts::value T>
using vector3 = vector_type<T, 3>;
template <concepts::value T>
using point3 = vector3<T>;
template <concepts::value T>
using vector6 = vector_type<T, 6>;
template <concepts::value T>
using vector8 = vector_type<T, 8>;

template <concepts::value T>
using transform3 = vc_aos::math::transform3<vc_aos::storage_type, T>;

using element_getter = algebra::storage::element_getter;
using block_getter = algebra::storage::block_getter;
 
}  // namespace algebra::vc_aos
 
DETRAY_ALGEBRA_DEFINE_TYPE_TRAITS(algebra::vc_aos)
 
namespace traits {

template <typename T, auto N>
struct index<algebra::vc_aos::storage_type<T, N>> {
  using type = algebra::vc_aos::index_type;
};

template <typename T, auto N>
struct index<Vc_1::Vector<T, Vc_1::simd_abi::fixed_size<N>>> {
  using type = algebra::vc_aos::index_type;
};

template <typename T, auto N>
struct value<algebra::vc_aos::storage_type<T, N>> {
  using type = T;
};

template <typename T, auto N>
struct value<Vc_1::Vector<T, Vc_1::simd_abi::fixed_size<N>>> {
  using type = T;
};
 
// Vector and storage types are different
template <typename T, auto N>
struct dimensions<algebra::vc_aos::storage_type<T, N>> {
  using index_type = algebra::vc_aos::index_type;

  static constexpr index_type _dim{1};
  static constexpr index_type _rows{N};
  static constexpr index_type _columns{1};
};
 
// Vector and storage types are different
template <typename T, auto N>
struct dimensions<Vc_1::Vector<T, Vc_1::simd_abi::fixed_size<N>>> {
  using index_type = algebra::vc_aos::index_type;

  static constexpr index_type _dim{1};
  static constexpr index_type _rows{N};
  static constexpr index_type _columns{1};
};
 
}  // namespace traits
 
}  // namespace detray