-
Notifications
You must be signed in to change notification settings - Fork 494
Expand file tree
/
Copy pathCTFCoder.cxx
More file actions
76 lines (68 loc) · 2.78 KB
/
CTFCoder.cxx
File metadata and controls
76 lines (68 loc) · 2.78 KB
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
// Copyright CERN and copyright holders of ALICE O2. This software is
// distributed under the terms of the GNU General Public License v3 (GPL
// Version 3), copied verbatim in the file "COPYING".
//
// See http://alice-o2.web.cern.ch/license for full licensing information.
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
/// \file CTFCoder.cxx
/// \author ruben.shahoyan@cern.ch
/// \brief class for entropy encoding/decoding of PHOS data
#include "PHOSReconstruction/CTFCoder.h"
#include "CommonUtils/StringUtils.h"
#include <TTree.h>
using namespace o2::phos;
///___________________________________________________________________________________
// Register encoded data in the tree (Fill is not called, will be done by caller)
void CTFCoder::appendToTree(TTree& tree, CTF& ec)
{
ec.appendToTree(tree, mDet.getName());
}
///___________________________________________________________________________________
// extract and decode data from the tree
void CTFCoder::readFromTree(TTree& tree, int entry, std::vector<TriggerRecord>& trigVec, std::vector<Cell>& cellVec)
{
assert(entry >= 0 && entry < tree.GetEntries());
CTF ec;
ec.readFromTree(tree, mDet.getName(), entry);
decode(ec, trigVec, cellVec);
}
///________________________________
void CTFCoder::createCoders(const std::string& dictPath, o2::ctf::CTFCoderBase::OpType op)
{
bool mayFail = true; // RS FIXME if the dictionary file is not there, do not produce exception
auto buff = readDictionaryFromFile<CTF>(dictPath, mayFail);
if (!buff.size()) {
if (mayFail) {
return;
}
throw std::runtime_error("Failed to create CTF dictionaty");
}
const auto* ctf = CTF::get(buff.data());
auto getFreq = [ctf](CTF::Slots slot) -> o2::rans::FrequencyTable {
o2::rans::FrequencyTable ft;
auto bl = ctf->getBlock(slot);
auto md = ctf->getMetadata(slot);
ft.addFrequencies(bl.getDict(), bl.getDict() + bl.getNDict(), md.min, md.max);
return std::move(ft);
};
auto getProbBits = [ctf](CTF::Slots slot) -> int {
return ctf->getMetadata(slot).probabilityBits;
};
// just to get types
uint16_t bcInc = 0, entries = 0, cellTime = 0, energy = 0, packedid = 0;
uint32_t orbitInc = 0;
uint8_t status = 0;
#define MAKECODER(part, slot) createCoder<decltype(part)>(op, getFreq(slot), getProbBits(slot), int(slot))
// clang-format off
MAKECODER(bcInc, CTF::BLC_bcIncTrig);
MAKECODER(orbitInc, CTF::BLC_orbitIncTrig);
MAKECODER(entries, CTF::BLC_entriesTrig);
MAKECODER(packedid, CTF::BLC_packedID);
MAKECODER(cellTime, CTF::BLC_time);
MAKECODER(energy, CTF::BLC_energy);
MAKECODER(status, CTF::BLC_status);
// clang-format on
}