vectorop.h

Go to the documentation of this file.

// -*- C++ -*-
#ifndef __STL_ADDON_VECTOROP_H__
#define __STL_ADDON_VECTOROP_H__

#include <assert.h>
#include <vector>

namespace lemga {
namespace op {

using std::vector;

template <typename R, typename N>
vector<R>& operator+= (vector<R>& u, const vector<N>& v) {
    assert(u.size() == v.size());
    typename vector<R>::iterator x = u.begin();
    typename vector<N>::const_iterator y = v.begin();
    for (; x != u.end(); ++x, ++y) *x += *y;
    return u;
}

template <typename R>
vector<R> operator- (const vector<R>& v) {
    vector<R> s(v);
    for (typename vector<R>::iterator x = s.begin(); x != s.end(); ++x)
        *x = -*x;
    return s;
}

template <typename R, typename N>
inline R inner_product (const R& u, const N& v) {
    return u * v;
}
template <typename RET, typename R, typename N>
RET inner_product (const vector<R>& u, const vector<N>& v) {
    assert(u.size() == v.size());
    RET s(0);
    typename vector<R>::const_iterator x = u.begin();
    typename vector<N>::const_iterator y = v.begin();
    for (; x != u.end(); ++x, ++y) s += inner_product<RET>(*x, *y);
    return s;
}

template <typename R, typename N>
vector<R>& operator*= (vector<R>& v, const N& r) {
    for (typename vector<R>::iterator x = v.begin(); x != v.end(); ++x)
        *x *= r;
    return v;
}

}} // namespace lemga::op

#ifdef  __VECTOROP_H__
#warning "This header file may conflict with another `vectorop.h' file."
#endif
#define __VECTOROP_H__
#endif

---