Large-Linear-System/VectorArithmetics.hpp at main · audictheprogrammer/Large-Linear-System · GitHub

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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#ifndef VECTOR_ARITHMETICS_HPP
#define VECTOR_ARITHMETICS_HPP

#include <iostream>
#include <vector>
#include <cassert>
#include <cmath>

using namespace std;


// OPERATORS.
vector<double> operator+(const vector<double>& u, const vector<double>& v){
    /* Addition operator between two vectors. */
    assert(u.size() == v.size());
    vector<double> res(u.size(), 0.);

    for (size_t i = 0; i < u.size(); i++){
        res[i] = u[i] + v[i];
    }

    return res;
}

vector<double> operator-(const vector<double>& u, const vector<double>& v){
    /* Substraction operator between two vectors. */
    assert(u.size() == v.size());
    vector<double> res(u.size(), 0.);

    for (size_t i = 0; i < u.size(); i++){
        res[i] = u[i] - v[i];
    }

    return res;
}

void operator+=(vector<double>& u, const vector<double>& v){
    /* Self-Add operator between two vectors. */
    assert(u.size() == v.size());
    for (size_t i = 0; i < u.size(); i++){
        u[i] += v[i];
    }
}

vector<double> operator*(const vector<double>& u, const double& l){
    /* Multiplication operator between a vector and a scalar. */
    vector<double> res(u.size(), 0.);
    for (size_t i = 0; i < u.size(); i++){
        res[i] = u[i] * l;
    }

    return res;
}

vector<double> operator*(const double& l, const vector<double>& u){
    /* Multiplication operator between a vector and a scalar. */
    vector<double> res(u.size(), 0.);
    for (size_t i = 0; i < u.size(); i++){
        res[i] = u[i] * l;
    }

    return res;
}

vector<double> operator/(const vector<double>& u, const double& l){
    /* Division operator between a vector and a scalar. */
    vector<double> res(u.size(), 0.);
    for (size_t i = 0; i < u.size(); i++){
        res[i] = u[i] / l;
    }

    return res;
}

ostream& operator<<(ostream& o, const vector<double>& u){
    /* Stream operator. */
    o << "[";
    for (size_t i = 0; i < u.size(); i++){
        o << u[i];
        if (i != u.size() - 1){
            o << "\t";
        }
    }
    o << "]" << endl;

    return o;
}


// FUNCTIONS.
double scalarProduct(const vector<double>& u, const vector<double>& v){
    /* Compute the scalar product between two vectors. */
    assert(u.size() == v.size());
    double res = 0;

    for (size_t i = 0; i < u.size(); i++){
        res += u[i] * v[i];
    }

    return res;
}

double Norm2(const vector<double>& u){
    /* Compute the Norm 2 of a vector. */
    double res = 0;
    for (size_t i = 0; i < u.size(); i++){
        res += u[i] * u[i];
    }
    return sqrt(res);

}


#endif