-
Notifications
You must be signed in to change notification settings - Fork 345
/
Copy pathquiver3_2.cpp
73 lines (64 loc) · 3.13 KB
/
quiver3_2.cpp
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
#include <cmath>
#include <matplot/matplot.h>
#include <vector>
std::vector<double> get_x();
std::vector<double> get_y();
std::vector<double> get_z();
int main() {
using namespace matplot;
// Positions
vector_1d x = get_x();
vector_1d y = get_y();
vector_1d z = get_z();
// Vectors
vector_1d u, v, w, m;
for (size_t i = 0; i < x.size(); ++i) {
double t = std::exp(-std::pow(x[i], 2) - std::pow(y[i], 2) -
std::pow(z[i], 2));
u.emplace_back(x[i] * t);
v.emplace_back(y[i] * t);
w.emplace_back(z[i] * t);
m.emplace_back(std::sqrt(x[i] * x[i] + y[i] * y[i] + z[i] * z[i]));
}
quiver3(x, y, z, u, v, w, m, 0.3)->normalize(true).line_width(2);
show();
return 0;
}
std::vector<double> get_x() {
return {-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
-1, -1, -1, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5,
-0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5,
-0.5, -0.5, -0.5, -0.5, -0.5, -0.5, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0.5, 0.5,
0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5,
0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5,
0.5, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 1, 1};
}
std::vector<double> get_y() {
return {-1, -1, -1, -1, -1, -0.5, -0.5, -0.5, -0.5, -0.5, 0, 0,
0, 0, 0, 0.5, 0.5, 0.5, 0.5, 0.5, 1, 1, 1, 1,
1, -1, -1, -1, -1, -1, -0.5, -0.5, -0.5, -0.5, -0.5, 0,
0, 0, 0, 0, 0.5, 0.5, 0.5, 0.5, 0.5, 1, 1, 1,
1, 1, -1, -1, -1, -1, -1, -0.5, -0.5, -0.5, -0.5, -0.5,
0, 0, 0, 0, 0, 0.5, 0.5, 0.5, 0.5, 0.5, 1, 1,
1, 1, 1, -1, -1, -1, -1, -1, -0.5, -0.5, -0.5, -0.5,
-0.5, 0, 0, 0, 0, 0, 0.5, 0.5, 0.5, 0.5, 0.5, 1,
1, 1, 1, 1, -1, -1, -1, -1, -1, -0.5, -0.5, -0.5,
-0.5, -0.5, 0, 0, 0, 0, 0, 0.5, 0.5, 0.5, 0.5, 0.5,
1, 1, 1, 1, 1};
}
std::vector<double> get_z() {
return {-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1, -1, -0.5, 0, 0.5, 1,
-1, -0.5, 0, 0.5, 1};
}