forked from andeplane/SimpleVisualizer
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mycontroller.cpp
107 lines (88 loc) · 2.84 KB
/
mycontroller.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
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
#include "mycontroller.h"
#include "simulator.h"
#include "billboards2d.h"
#include "points2d.h"
#include "controller.h"
#include <cmath>
#include <iostream>
#include <functional>
using std::cout; using std::endl; using std::function; using std::vector;
class MySimulator : public CompPhys::Simulator {
public:
MySimulator();
virtual void step();
std::vector<QVector2D> m_positions;
std::vector<QVector2D> m_velocities;
double m_springConstant;
virtual void synchronize(CompPhys::Controller *controller);
};
MyController::MyController()
{
CompPhys::Billboards2D *billboards = new CompPhys::Billboards2D(
[&](CompPhys::Billboards2D *obj) {
obj->setPositions(m_positions);
}
, ":/football.png");
billboards->setScale(0.1);
billboards->setColor(QColor("white"));
CompPhys::Points2D *points = new CompPhys::Points2D([&](CompPhys::Points2D *obj) {
obj->setData(m_positions, QVector3D(1.0, 0.0, 0.0));
});
points->setPointSize(10.0);
addRenderableObject(billboards);
addRenderableObject(points);
}
CompPhys::Simulator *MyController::createSimulator() const
{
return new MySimulator();
}
double MyController::springConstant() const
{
return m_springConstant;
}
void MyController::setSpringConstant(double arg)
{
if (m_springConstant == arg)
return;
m_springConstant = arg;
emit springConstantChanged(arg);
}
MySimulator::MySimulator() :
m_springConstant(1.0)
{
unsigned int numberOfBalls = 100;
for(unsigned int i=0; i<numberOfBalls; i++) {
float x = 2.0*(rand() / double(RAND_MAX)) - 1.0;
float y = 2.0*(rand() / double(RAND_MAX)) - 1.0;
float vx = 2.0*(rand() / double(RAND_MAX)) - 1.0;
float vy = 2.0*(rand() / double(RAND_MAX)) - 1.0;
x *= 0.5;
y *= 0.5;
vx *= 0.3;
vy *= 0.3;
m_positions.push_back(QVector2D(x,y));
m_velocities.push_back(QVector2D(vx,vy));
}
}
void MySimulator::step() {
qDebug() << QThread::currentThreadId() << " is MySimulator step";
float mass = 1.0;
float k = m_springConstant;
float dt = 0.01;
for(unsigned int i=0; i<m_positions.size(); i++) {
QVector2D force;
force.setX(-k*m_positions[i][0]);
force.setY(-k*m_positions[i][1]);
m_velocities[i][0] += force[0]/mass*dt;
m_velocities[i][1] += force[1]/mass*dt;
m_positions[i][0] += m_velocities[i][0]*dt;
m_positions[i][1] += m_velocities[i][1]*dt;
}
}
void MySimulator::synchronize(CompPhys::Controller *controller) {
qDebug() << QThread::currentThreadId() << " is MySimulator sync";
MyController* myController = static_cast<MyController*>(controller);
m_springConstant = myController->m_springConstant;
myController->m_positions = m_positions;
myController->m_velocities = m_velocities;
}