blob: f01368e6838521453bf118408d83700b4253d87a (
plain) (
blame)
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
131
|
/*
* Beam.cc
*
* Created on: May 6, 2011
* Author: jakob
*/
#include <math.h>
#include "Beam.h"
#include "Element.h"
using namespace std;
namespace vhc {
Beam::Beam(const Particle& referenceParticle, int quantity, int lambda): referenceParticle(referenceParticle), quantity(quantity), lambda(lambda) {
init(referenceParticle, quantity, lambda);
}
Beam::~Beam() {
for (list<Particle*>::iterator i = particles.begin(); i != particles.end(); ++i) {
delete *i;
*i = NULL;
}
particles.clear();
}
int Beam::getSize() const {
return particles.size();
}
int Beam::getLambda() const {
return lambda;
}
const Particle& Beam::getReferenceParticle() const {
return referenceParticle;
}
double Beam::getVR2() const {
double z2 = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
z2 += (**i).getPosition().getZ() * (**i).getPosition().getZ();
}
return sqrt(z2 / particles.size());
}
double Beam::getVV2() const {
double v2 = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
v2 += (**i).getVelocity().getZ() * (**i).getVelocity().getZ();
}
return v2 / particles.size();
}
double Beam::getVRV2() const {
double vz = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
vz += (**i).getPosition().getZ();
}
return (vz / particles.size()) * (vz / particles.size());
}
double Beam::getHR2() const {
double r2 = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
Vector3D u = ((**i).getElement())->getHorizontalAt((**i).getPosition());
r2 += (**i).getPosition().dot(u) * (**i).getPosition().dot(u);
}
return r2 / particles.size();
}
double Beam::getHV2() const {
double v2 = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
Vector3D u = ((**i).getElement())->getHorizontalAt((**i).getPosition());
v2 += (**i).getVelocity().dot(u) * (**i).getVelocity().dot(u);
}
return v2 / particles.size();
}
double Beam::getHRV2() const {
double vr = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
Vector3D u = ((**i).getElement())->getHorizontalAt((**i).getPosition());
vr += (**i).getPosition().dot(u) * (**i).getVelocity().dot(u);
}
return (vr / particles.size()) * (vr / particles.size());
}
double Beam::getAverageEnergy() const {
double e = 0;
for (list<Particle*>::const_iterator i = particles.begin(); i != particles.end(); ++i) {
e += (**i).getEnergy();
}
return e / particles.size();
}
double Beam::getVerticalEmittance() const {
return sqrt(getVR2() * getVV2() - getVRV2());
}
double Beam::getHorizontalEmittance() const {
return sqrt(getHR2() * getHV2() - getHRV2());
}
double Beam::getVerticalA11() const {
return getVR2() / getVerticalEmittance();
}
double Beam::getVerticalA12() const {
return -getVRV2() / getVerticalEmittance();
}
double Beam::getVerticalA22() const {
return getVV2() / getVerticalEmittance();
}
double Beam::getHorizontalA11() const {
return getHR2() / getHorizontalEmittance();
}
double Beam::getHorizontalA12() const {
return -getHRV2() / getVerticalEmittance();
}
double Beam::getHorizontalA22() const {
return getHV2() / getHorizontalEmittance();
}
}
|
