|
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
|
#include "DataFormats/Luminosity/interface/BeamCurrentInfo.h"
#include "DataFormats/Math/interface/libminifloat.h"
#include <iomanip>
#include <ostream>
#include <iostream>
const float BeamCurrentInfo::scaleFactor = 1e10;
float BeamCurrentInfo::getBeam1IntensityBX(int bx) const { return beam1IntensitiesUnpacked_.at(bx); }
const std::vector<float>& BeamCurrentInfo::getBeam1Intensities() const { return beam1IntensitiesUnpacked_; }
float BeamCurrentInfo::getBeam2IntensityBX(int bx) const { return beam2IntensitiesUnpacked_.at(bx); }
const std::vector<float>& BeamCurrentInfo::getBeam2Intensities() const { return beam2IntensitiesUnpacked_; }
bool BeamCurrentInfo::isProductEqual(BeamCurrentInfo const& next) const {
return (beam1IntensitiesPacked_ == next.beam1IntensitiesPacked_ &&
beam2IntensitiesPacked_ == next.beam2IntensitiesPacked_);
}
void BeamCurrentInfo::fillBeamIntensities(const std::vector<float>& beam1Intensities,
const std::vector<float>& beam2Intensities) {
beam1IntensitiesUnpacked_.assign(beam1Intensities.begin(), beam1Intensities.end());
beam2IntensitiesUnpacked_.assign(beam2Intensities.begin(), beam2Intensities.end());
packData();
}
void BeamCurrentInfo::fill(const std::vector<float>& beam1Intensities, const std::vector<float>& beam2Intensities) {
fillBeamIntensities(beam1Intensities, beam2Intensities);
}
// Convert unpacked data to packed data (when it is filled).
void BeamCurrentInfo::packData() {
beam1IntensitiesPacked_.resize(beam1IntensitiesUnpacked_.size());
beam2IntensitiesPacked_.resize(beam2IntensitiesUnpacked_.size());
for (unsigned int i = 0; i < beam1IntensitiesUnpacked_.size(); i++) {
beam1IntensitiesPacked_[i] = MiniFloatConverter::float32to16(beam1IntensitiesUnpacked_[i] / scaleFactor);
}
for (unsigned int i = 0; i < beam2IntensitiesUnpacked_.size(); i++) {
beam2IntensitiesPacked_[i] = MiniFloatConverter::float32to16(beam2IntensitiesUnpacked_[i] / scaleFactor);
}
// Re-unpack the data so that it matches the packed data.
unpackData();
}
// Convert packed data to unpacked data when reading back data
void BeamCurrentInfo::unpackData(const std::vector<uint16_t>& packed, std::vector<float>& unpacked) {
unpacked.resize(packed.size());
for (unsigned int i = 0; i < packed.size(); i++) {
unpacked[i] = MiniFloatConverter::float16to32(packed[i]) * scaleFactor;
}
}
void BeamCurrentInfo::unpackData() {
unpackData(beam1IntensitiesPacked_, beam1IntensitiesUnpacked_);
unpackData(beam2IntensitiesPacked_, beam2IntensitiesUnpacked_);
}
std::ostream& operator<<(std::ostream& s, const BeamCurrentInfo& beamInfo) {
s << std::endl << "Dumping BeamCurrentInfo..." << std::endl;
s << " beam1Intensities = ";
const std::vector<float>& b1int = beamInfo.getBeam1Intensities();
const std::vector<uint16_t>& b1intPacked = beamInfo.getBeam1IntensitiesPacked();
for (unsigned int i = 0; i < 10 && i < b1int.size(); ++i) {
s << b1int[i] << " ";
}
s << "..." << std::endl << " (packed: ";
for (unsigned int i = 0; i < 10 && i < b1intPacked.size(); ++i) {
s << b1intPacked[i] << " ";
}
s << "...)" << std::endl;
s << " beam2Intensities = ";
const std::vector<float>& b2int = beamInfo.getBeam2Intensities();
for (unsigned int i = 0; i < 10 && i < b2int.size(); ++i) {
s << b2int[i] << " ";
}
s << " ..." << std::endl;
return s;
}
|