Research1/vector.cpp at master · hydrogen602/Research1 · 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
#include <cstdio>
#include "vector.h"
#include "state.h"
#include "data/vector3.h"

Vector::Vector() {
}

Vector::Vector(int n) {
    data.resize(n, 0.0);
}

Vector::Vector(const Vector& src) {
    data.resize(src.data.size(), 0);
    for (unsigned int i = 0; i < src.data.size(); ++i) {
        data[i] = src.data[i];
    }
}

// Vector::Vector(Vector&& src): data{src.data} {
//     // move should destroy original

//     // how to ?
// }

Vector& Vector::operator=(const Vector& other) {
    data.resize(other.data.size(), 0);
    for (unsigned int i = 0; i < other.data.size(); ++i) {
        data[i] = other.data[i];
    }
    //printf("Copy assign\n");
    return *this;
}

Vector& Vector::operator*=(double h) {
    for (unsigned int i = 0; i < data.size(); ++i) {
        data[i] *= h;
    }
    return *this;
}

Vector& Vector::operator+=(const Vector& other) {
    if (data.size() != other.data.size()) {
        throw ERR_VECTOR_SIZE_MISMATCH;
    }
    for (unsigned int i = 0; i < data.size(); ++i) {
        data[i] += other.data[i];
    }
    return *this;
}

double& Vector::operator[](unsigned int i) {
    if (i >= data.size()) {
        throw ERR_VECTOR_OUT_OF_BOUNDS;
    }
    return data[i];
}

const double& Vector::operator[](unsigned int i) const {
    if (i >= data.size()) {
        throw ERR_VECTOR_OUT_OF_BOUNDS;
    }
    return data[i];
}

vector3 Vector::getPos(unsigned int i) const {
    return vector3(data[i * 6 + 0], data[i * 6 + 1], data[i * 6 + 2]);
}

vector3 Vector::getVel(unsigned int i) const {
    return vector3(data[i * 6 + 3], data[i * 6 + 4], data[i * 6 + 5]);
}

Vector Vector::operator/(double m) const {
    Vector v(*this);
    for (unsigned int i = 0; i < data.size(); ++i) {
        v[i] = data[i] / m;
    }
    return v;
}

void Vector::resize(int n, double d) {
    data.resize(n, d);
}

void Vector::addBody(double x, double y, double z, double vx, double vy, double vz) {
    data.push_back(x);
    data.push_back(y);
    data.push_back(z);
    data.push_back(vx);
    data.push_back(vy);
    data.push_back(vz);
}

unsigned int Vector::size() const {
    return data.size();
}

void Vector::printOut(unsigned int i) const {
    printf("  <%e, %e, %e, %e, %e, %e>\n", data[i], data[i+1], data[i+2], data[i+3], data[i+4], data[i+5]);
}

void Vector::printOut() const {
    printf("[\n");
    for (unsigned int i = 0; i < data.size(); i += 6) {
        printf("  <%e, %e, %e, %e, %e, %e>\n", data[i], data[i+1], data[i+2], data[i+3], data[i+4], data[i+5]);
    }
    printf("]\n");
}