Line data    Source code

       1           1 : // Distributed under the MIT License.
       2             : // See LICENSE.txt for details.
       3             : 
       4             : /// \file
       5             : /// `blaze::DynamicMatrix` is a general-purpose dynamically sized matrix type.
       6             : /// This file implements interoperability of `blaze::DynamicMatrix` with our
       7             : /// data structures.
       8             : 
       9             : #pragma once
      10             : 
      11             : #include <blaze/math/DynamicMatrix.h>
      12             : #include <cstddef>
      13             : #include <pup.h>
      14             : #include <type_traits>
      15             : #include <vector>
      16             : 
      17             : #include "Options/Options.hpp"
      18             : #include "Options/ParseError.hpp"
      19             : #include "Utilities/Gsl.hpp"
      20             : 
      21             : namespace PUP {
      22             : /// @{
      23             : /// Serialization of blaze::DynamicMatrix
      24             : template <typename Type, bool SO, typename Alloc, typename Tag>
      25             : void pup(er& p, blaze::DynamicMatrix<Type, SO, Alloc, Tag>& t) {
      26             :   size_t rows = t.rows();
      27             :   size_t columns = t.columns();
      28             :   p | rows;
      29             :   p | columns;
      30             :   if (p.isUnpacking()) {
      31             :     t.resize(rows, columns);
      32             :   }
      33             :   const size_t spacing = t.spacing();
      34             :   const size_t data_size = spacing * (SO == blaze::rowMajor ? rows : columns);
      35             :   if (std::is_fundamental_v<Type>) {
      36             :     PUParray(p, t.data(), data_size);
      37             :   } else {
      38             :     for (size_t i = 0; i < data_size; ++i) {
      39             :       p | t.data()[i];
      40             :     }
      41             :   }
      42             : }
      43             : template <typename Type, bool SO, typename Alloc, typename Tag>
      44             : void operator|(er& p, blaze::DynamicMatrix<Type, SO, Alloc, Tag>& t) {
      45             :   pup(p, t);
      46             : }
      47             : /// @}
      48             : }  // namespace PUP
      49             : 
      50             : namespace DynamicMatrix_detail {
      51             : template <typename Type>
      52             : std::vector<std::vector<Type>> parse_to_vectors(const Options::Option& options);
      53             : }  // namespace DynamicMatrix_detail
      54             : 
      55             : template <typename Type, bool SO, typename Alloc, typename Tag>
      56           0 : struct Options::create_from_yaml<blaze::DynamicMatrix<Type, SO, Alloc, Tag>> {
      57             :   template <typename Metavariables>
      58           0 :   static blaze::DynamicMatrix<Type, SO, Alloc, Tag> create(
      59             :       const Options::Option& options) {
      60             :     const auto data = DynamicMatrix_detail::parse_to_vectors<Type>(options);
      61             :     const size_t num_rows = data.size();
      62             :     size_t num_cols = 0;
      63             :     if (num_rows > 0) {
      64             :       num_cols = data[0].size();
      65             :     }
      66             :     blaze::DynamicMatrix<Type, SO, Alloc, Tag> result(num_rows, num_cols);
      67             :     for (size_t i = 0; i < num_rows; i++) {
      68             :       const auto& row = gsl::at(data, i);
      69             :       if (row.size() != num_cols) {
      70             :         PARSE_ERROR(options.context(),
      71             :                     "All matrix rows must have the same size.");
      72             :       }
      73             :       std::copy(row.begin(), row.end(), blaze::row(result, i).begin());
      74             :     }
      75             :     return result;
      76             :   }
      77             : };