blob: 4ec8279794f8eaf120c9c72f51c3c1acb9de8168 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#ifndef _MATH_VECTOR_H_
#define _MATH_VECTOR_H_
#include <vector>
#include <iostream>
#include <cassert>
class Vec
{
private:
std::vector<double> _v;
public:
Vec(const size_t n = 0, const double val = 0) { _v.resize(n, val); }
Vec(const std::vector<double> & v) : _v(v) {}
const std::vector<double> & STLVec() const { return _v; }
std::vector<double> & STLVec() { return _v; }
size_t Size() const { return _v.size(); }
double & operator[](int i) { return _v[i]; }
const double & operator[](int i) const { return _v[i]; }
Vec & operator+=(const Vec & b) {
assert(b.Size() == _v.size());
for (size_t i = 0; i < _v.size(); i++) {
_v[i] += b[i];
}
return *this;
}
Vec & operator*=(const double c) {
for (size_t i = 0; i < _v.size(); i++) {
_v[i] *= c;
}
return *this;
}
void Project(const Vec & y) {
for (size_t i = 0; i < _v.size(); i++) {
// if (sign(_v[i]) != sign(y[i])) _v[i] = 0;
if (_v[i] * y[i] <=0) _v[i] = 0;
}
}
};
inline double dot_product(const Vec & a, const Vec & b)
{
double sum = 0;
for (size_t i = 0; i < a.Size(); i++) {
sum += a[i] * b[i];
}
return sum;
}
inline std::ostream & operator<<(std::ostream & s, const Vec & a)
{
s << "(";
for (size_t i = 0; i < a.Size(); i++) {
if (i != 0) s << ", ";
s << a[i];
}
s << ")";
return s;
}
inline const Vec operator+(const Vec & a, const Vec & b)
{
Vec v(a.Size());
assert(a.Size() == b.Size());
for (size_t i = 0; i < a.Size(); i++) {
v[i] = a[i] + b[i];
}
return v;
}
inline const Vec operator-(const Vec & a, const Vec & b)
{
Vec v(a.Size());
assert(a.Size() == b.Size());
for (size_t i = 0; i < a.Size(); i++) {
v[i] = a[i] - b[i];
}
return v;
}
inline const Vec operator*(const Vec & a, const double c)
{
Vec v(a.Size());
for (size_t i = 0; i < a.Size(); i++) {
v[i] = a[i] * c;
}
return v;
}
inline const Vec operator*(const double c, const Vec & a)
{
return a * c;
}
#endif
|
