ops_vectors.h

Go to the documentation of this file.

/* +------------------------------------------------------------------------+
   |                     Mobile Robot Programming Toolkit (MRPT)            |
   |                          http://www.mrpt.org/                          |
   |                                                                        |
   | Copyright (c) 2005-2018, Individual contributors, see AUTHORS file     |
   | See: http://www.mrpt.org/Authors - All rights reserved.                |
   | Released under BSD License. See details in http://www.mrpt.org/License |
   +------------------------------------------------------------------------+ */
#pragma once

#include <mrpt/math/CMatrixTemplateNumeric.h>
#include <iomanip>  // for setprecision(), etc.
#include <iterator>  // std::ostream_iterator
#include <mrpt/serialization/CArchive.h>

// Many of the functions originally in this file are now in ops_containers.h
#include <mrpt/math/ops_containers.h>

namespace mrpt::math
{
// Frwd. decl.
template <typename T, std::size_t N>
class CArrayNumeric;

/** \addtogroup container_ops_grp
 *  @{ */

/** \name Generic std::vector element-wise operations
 * @{
 */

/** a*=b (element-wise multiplication) */
template <typename T1, typename T2>
inline std::vector<T1>& operator*=(std::vector<T1>& a, const std::vector<T2>& b)
{
    ASSERT_EQUAL_(a.size(), b.size());
    const size_t N = a.size();
    for (size_t i = 0; i < N; i++) a[i] *= b[i];
    return a;
}

/** a*=k (multiplication by a constant) */
template <typename T1>
inline std::vector<T1>& operator*=(std::vector<T1>& a, const T1 b)
{
    const size_t N = a.size();
    for (size_t i = 0; i < N; i++) a[i] *= b;
    return a;
}

/** a*b (element-wise multiplication) */
template <typename T1, typename T2>
inline std::vector<T1> operator*(
    const std::vector<T1>& a, const std::vector<T2>& b)
{
    ASSERT_EQUAL_(a.size(), b.size());
    const size_t N = a.size();
    std::vector<T1> ret(N);
    for (size_t i = 0; i < N; i++) ret[i] = a[i] * b[i];
    return ret;
}

/** a+=b (element-wise sum) */
template <typename T1, typename T2>
inline std::vector<T1>& operator+=(std::vector<T1>& a, const std::vector<T2>& b)
{
    ASSERT_EQUAL_(a.size(), b.size());
    const size_t N = a.size();
    for (size_t i = 0; i < N; i++) a[i] += b[i];
    return a;
}

/** a+=b (sum a constant) */
template <typename T1>
inline std::vector<T1>& operator+=(std::vector<T1>& a, const T1 b)
{
    const size_t N = a.size();
    for (size_t i = 0; i < N; i++) a[i] += b;
    return a;
}

/** a+b (element-wise sum) */
template <typename T1, typename T2>
inline std::vector<T1> operator+(
    const std::vector<T1>& a, const std::vector<T2>& b)
{
    ASSERT_EQUAL_(a.size(), b.size());
    const size_t N = a.size();
    std::vector<T1> ret(N);
    for (size_t i = 0; i < N; i++) ret[i] = a[i] + b[i];
    return ret;
}

template <typename T1, typename T2>
inline std::vector<T1> operator-(
    const std::vector<T1>& v1, const std::vector<T2>& v2)
{
    ASSERT_EQUAL_(v1.size(), v2.size());
    std::vector<T1> res(v1.size());
    for (size_t i = 0; i < v1.size(); i++) res[i] = v1[i] - v2[i];
    return res;
}

/** @} */

/** A template function for printing out the contents of a std::vector variable.
 */
template <class T>
std::ostream& operator<<(std::ostream& out, const std::vector<T>& d)
{
    const std::streamsize old_pre = out.precision();
    const std::ios_base::fmtflags old_flags = out.flags();
    out << "[" << std::fixed << std::setprecision(4);
    std::copy(d.begin(), d.end(), std::ostream_iterator<T>(out, " "));
    out << "]";
    out.flags(old_flags);
    out.precision(old_pre);
    return out;
}

/** A template function for printing out the contents of a std::vector variable.
 */
template <class T>
std::ostream& operator<<(std::ostream& out, std::vector<T>* d)
{
    const std::streamsize old_pre = out.precision();
    const std::ios_base::fmtflags old_flags = out.flags();
    out << "[" << std::fixed << std::setprecision(4);
    copy(d->begin(), d->end(), std::ostream_iterator<T>(out, " "));
    out << "]";
    out.flags(old_flags);
    out.precision(old_pre);
    return out;
}

/** Binary dump of a CArrayNumeric<T,N> to a stream. */
template <typename T, size_t N>
mrpt::serialization::CArchive& operator<<(
    mrpt::serialization::CArchive& ostrm, const CArrayNumeric<T, N>& a)
{
    ostrm << mrpt::typemeta::TTypeName<CArrayNumeric<T, N>>::get();
    if (N) ostrm.WriteBufferFixEndianness<T>(&a[0], N);
    return ostrm;
}

/** Binary read of a CArrayNumeric<T,N> from a stream. */
template <typename T, size_t N>
mrpt::serialization::CArchive& operator>>(
    mrpt::serialization::CArchive& istrm, CArrayNumeric<T, N>& a)
{
    static const std::string namExpect =
        mrpt::typemeta::TTypeName<CArrayNumeric<T, N>>::get();
    std::string nam;
    istrm >> nam;
    ASSERTMSG_(
        nam == namExpect, format(
                              "Error deserializing: expected '%s', got '%s'",
                              namExpect.c_str(), nam.c_str()));
    if (N) istrm.ReadBufferFixEndianness<T>(&a[0], N);
    return istrm;
}

/**  @} */  // end of grouping

}