Line data    Source code

       1           0 : // Distributed under the MIT License.
       2             : // See LICENSE.txt for details.
       3             : 
       4             : #pragma once
       5             : 
       6             : #include <complex>
       7             : #include <cstddef>
       8             : #include <stdexcept>
       9             : #include <string>
      10             : 
      11             : #include "DataStructures/DataVector.hpp"
      12             : 
      13             : namespace py_bindings::detail {
      14             : // The bindings pass spin-weighted complex data as real, interleaved
      15             : // [re, im, re, im, ...] arrays so they can be constructed directly from numpy.
      16             : 
      17             : template <typename ComplexVector>
      18             : ComplexVector interleaved_to_complex(const DataVector& interleaved) {
      19             :   if (interleaved.size() % 2 != 0) {
      20             :     throw std::invalid_argument(
      21             :         "Interleaved [re, im, ...] array must have an even number of "
      22             :         "entries, not " +
      23             :         std::to_string(interleaved.size()) + ".");
      24             :   }
      25             :   ComplexVector result{interleaved.size() / 2};
      26             :   for (size_t i = 0; i < result.size(); ++i) {
      27             :     result[i] =
      28             :         std::complex<double>(interleaved[2 * i], interleaved[2 * i + 1]);
      29             :   }
      30             :   return result;
      31             : }
      32             : 
      33             : template <typename ComplexVector>
      34             : DataVector complex_to_interleaved(const ComplexVector& complex_values) {
      35             :   DataVector result{2 * complex_values.size()};
      36             :   for (size_t i = 0; i < complex_values.size(); ++i) {
      37             :     result[2 * i] = complex_values[i].real();
      38             :     result[2 * i + 1] = complex_values[i].imag();
      39             :   }
      40             :   return result;
      41             : }
      42             : }  // namespace py_bindings::detail