blob: 024e7b5dc78995d094b418acc4d712e48c517fba (
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
|
/*
* VisitorTest.cc
*
* Created on: Apr 3, 2011
* Author: jakob
*/
#include <iostream>
#include <vector>
#include "Element.h"
#include "CompositeElement.h"
#include "StraightElement.h"
#include "CurvedElement.h"
#include "Quadrupole.h"
#include "Dipole.h"
#include "FODO.h"
#include "ElementVisitor.h"
#include "Vector3D.h"
using namespace std;
using namespace vhc;
class PrintVisitor: public ElementVisitor {
public:
virtual void visit(const StraightElement* straight) const {
cout << "A straight element." << endl;
}
virtual void visit(const Quadrupole* quadrupole) const {
cout << "A quadrupole." << endl;
}
virtual void visit(const Dipole* dipole) const {
cout << "A dipole." << endl;
}
};
int main() {
PrintVisitor pv;
Dipole* dp = new Dipole (Vector3D::Null, Vector3D::i, 0.2, 0.5);
Quadrupole* qp = new Quadrupole(Vector3D::Null, Vector3D::i, 0.2, 2);
FODO* fodo = new FODO(Vector3D::Null, Vector3D::i, 0.2, 0.25, 2.0);
vector<Element*> elements;
elements.push_back(dp);
elements.push_back(qp);
elements.push_back(fodo);
for (int i = 0; i < elements.size(); ++i) {
elements[i]->accept(pv);
}
for (int i = 0; i < elements.size(); ++i) {
delete elements[i];
elements[i] = NULL;
}
return 0;
}
|
