Array_complex_numbers.h

Go to the documentation of this file.

#ifndef __Array_complex_numbers_H__
#define __Array_complex_numbers_H__

/*
    Copyright (C) 2004- 2011, Hammersmith Imanet Ltd
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/
#include "stir/Array.h"
#include <complex>

START_NAMESPACE_STIR


template <int num_dimensions, typename elemT>
Array<num_dimensions, std::complex<elemT>>&
operator*=(Array<num_dimensions, std::complex<elemT>>& lhs, const elemT& rhs)
{
  typename Array<num_dimensions, std::complex<elemT>>::iterator iter1 = lhs.begin();
  while (iter1 != lhs.end())
    *iter1++ *= rhs;
  return lhs;
}

template <int num_dimensions, typename elemT>
Array<num_dimensions, std::complex<elemT>>&
operator/=(Array<num_dimensions, std::complex<elemT>>& lhs, const elemT& rhs)
{
  typename Array<num_dimensions, std::complex<elemT>>::iterator iter1 = lhs.begin();
  while (iter1 != lhs.end())
    *iter1++ /= rhs;
  return lhs;
}

template <int num_dimensions, typename elemT>
Array<num_dimensions, std::complex<elemT>>&
operator+=(Array<num_dimensions, std::complex<elemT>>& lhs, const elemT& rhs)
{
  typename Array<num_dimensions, std::complex<elemT>>::iterator iter1 = lhs.begin();
  while (iter1 != lhs.end())
    *iter1++ += rhs;
  return lhs;
}

template <int num_dimensions, typename elemT>
Array<num_dimensions, std::complex<elemT>>&
operator-=(Array<num_dimensions, std::complex<elemT>>& lhs, const elemT& rhs)
{
  typename Array<num_dimensions, std::complex<elemT>>::iterator iter1 = lhs.begin();
  while (iter1 != lhs.end())
    *iter1++ -= rhs;
  return lhs;
}

// a few common cases given explictly here such that we don't get conversion warnings all the time.
inline void
assign(std::complex<double>& x, const int y)
{
  x = static_cast<std::complex<double>>(y);
}

inline void
assign(std::complex<float>& x, const int y)
{
  x = static_cast<float>(y);
}
END_NAMESPACE_STIR

#endif