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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/*
* Main.cc
*
* Created on: Mar 24, 2011
* Author: jakob
*/
#include <iostream>
#include <QApplication>
#include <QWidget>
#include <cmath>
#include "Stage.h"
#include "Accelerator.h"
#include "StraightElement.h"
#include "Dipole.h"
#include "Particle.h"
#include "FODO.h"
#include "Vector3D.h"
#include "constants.h"
#include <vector>
using namespace vhc;
std::vector< Particle > createParticles(const Vector3D& position, int n, double mass = constants::ELECTRON_MASS, double charge = constants::E, double energy = 0, Vector3D direction = Vector3D::i) {
std::vector< Particle > v;
double r = 0.1;
for (int i = 0; i < n; ++i) {
double x = (rand() % 1000) / 1000.0 * r;
double y = (rand() % 1000) / 1000.0 * sqrt(r * r - x * x);
double z = (rand() % 1000) / 1000.0 * sqrt(r * r - y * y - x * x);
v.push_back(Particle(position + Vector3D(x, y, z), mass, charge, energy, direction));
}
return v;
}
Accelerator* standardAccelerator() {
FODO e1 = FODO(Vector3D(3, 2, 0), Vector3D(3, -2, 0), 0.2, 1.0, 5.0);
Dipole e2 = Dipole(e1.getExitPosition(), Vector3D(2, -3, 0), 0.2, 1);
FODO e3 = FODO(e2.getExitPosition(), Vector3D(-2, -3, 0), 0.2, 1, 5.0);
Dipole e4 = Dipole(e3.getExitPosition(), Vector3D(-3, -2, 0), 0.2, 1);
FODO e5 = FODO(e4.getExitPosition(), Vector3D(-3, 2, 0), 0.2, 1.0, 5.0);
Dipole e6 = Dipole(e5.getExitPosition(), Vector3D(-2, 3, 0), 0.2, 1);
FODO e7 = FODO(e6.getExitPosition(), Vector3D(2, 3, 0), 0.2, 1.0, 5.0);
Dipole e8 = Dipole(e7.getExitPosition(), e1.getEntryPosition(), 0.2, 1);
Accelerator* acc = new Accelerator();
acc->add(e1);
acc->add(e2);
acc->add(e3);
acc->add(e4);
acc->add(e5);
acc->add(e6);
acc->add(e7);
acc->add(e8);
std::vector< Particle > ps = createParticles(e1.getEntryPosition(), 1000);
for (int i = 0; i < ps.size(); ++i) {
acc->add(ps[i]);
}
return acc;
}
Accelerator* linear() {
FODO element = FODO(Vector3D(0, 0, 0), Vector3D(4, 0, 0), 0.2, 0.2, 120);
Accelerator* acc = new Accelerator();
Element* celement = acc->add(element);
Particle e(Vector3D(0, 0.15, 0.01), constants::ELECTRON_MASS, constants::E, 14E9 * constants::E, Vector3D::i);
Particle* ce = acc->add(e);
ce->setElement(celement);
return acc;
}
Accelerator* singleDipole() {
Vector3D entry = Vector3D(0, 2, 0);
Vector3D exit = Vector3D(2, 0, 0);
double sectionRadius = 0.2;
double curvature = 0.5;
Vector3D direction = entry.cross(Vector3D::k);
double mass = constants::ELECTRON_MASS;
double charge = constants::E;
double energy = 1 * 1E9 * constants::E;
Particle particle = Particle(entry, mass, charge, energy, direction);
double Bz = particle.getGamma() * particle.getMass() * curvature * particle.getVelocity().norm() / particle.getCharge();
std::cout << "|B:|" << Bz << std::endl;
Dipole element = Dipole(entry, exit, sectionRadius, curvature, Vector3D::k * Bz);
Accelerator* acc = new Accelerator();
Element* celement = acc->add(element);
std::vector< Particle > ps = createParticles(element.getEntryPosition(), 10, constants::ELECTRON_MASS, constants::E, energy, direction);
for (int i = 0; i < ps.size(); ++i) {
Particle* cparticle = acc->add(ps[i]);
cparticle->setElement(celement);
}
Particle* cparticle = acc->add(particle);
cparticle->setElement(celement);
return acc;
}
int main(int argc, char *argv[])
{
QApplication app(argc, argv);
vhc::Stage window;
Accelerator* acc = singleDipole();
window.accelerator = acc;
window.showFullScreen();
//window.resize(QSize(500, 500));
window.setWindowTitle("Virtual Hadron Collider");
window.show();
return app.exec();
delete acc; acc = NULL;
}
|
