Newer
Older
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
#include "Event/BesEvtReader.hh"
#include "Event/EventList.hh"
#include "Event/Event.hh"
#include "ErrLogger/ErrLogger.hh"
BesEvtReader::BesEvtReader()
{}
BesEvtReader::BesEvtReader(const std::vector<std::string>& files, int particles, int skip):
numParticles(particles),
linesToSkip(skip)
{
if (0 == files.size()) {
Alert << "empty list of event files" << endmsg;
exit(1);
}
std::vector<std::string>::const_iterator iter = files.begin();
for (; iter != files.end(); ++iter)
fileNames.push_back(*iter);
currentFile = fileNames.begin();
}
BesEvtReader::~BesEvtReader()
{}
bool BesEvtReader::fillAll(EventList& evtList)
{
while (currentFile != fileNames.end()) {
currentStream.open(currentFile->c_str());
if (!currentStream) {
Alert << "can not open " << *currentFile << endmsg;
exit(1);
}
while (!currentStream.eof()) {
double e,px,py,pz;
Event* newEvent = new Event();
int parts;
for (parts = 0; parts < numParticles; parts++) {
currentStream >> px >> py >> pz >> e;
newEvent->addParticle(e,px,py,pz);
}
if (!currentStream.fail()) {
evtList.add(newEvent);
for (parts = 0; parts < linesToSkip; parts++)
currentStream >> px >> py >> pz >> e;
}
}
currentStream.close();
++currentFile;
}
evtList.rewind();
return true; // success
}