ATLAS Offline Software
Classes | Macros | Typedefs | Functions
test-hdf5-writer.cxx File Reference
#include "HDF5Utils/Writer.h"
Include dependency graph for test-hdf5-writer.cxx:

Go to the source code of this file.

Classes

struct  out_t
 

Macros

#define ADD(NAME)   consumers.add(#NAME, [](const out_t& o){ return o.NAME;}, 0)
 

Typedefs

using consumer_t = H5Utils::Consumers< const out_t & >
 

Functions

consumer_t getConsumers ()
 
std::vector< out_tgetOutputs (int offset, size_t length, float factor)
 
template<size_t N>
auto nestOutputs (int offset, size_t length)
 
template<>
auto nestOutputs< 1 > (int offset, size_t length)
 
void fill (H5::Group &out_file, size_t iterations)
 
int main (int nargs, char *argv[])
 

Macro Definition Documentation

◆ ADD

#define ADD (   NAME)    consumers.add(#NAME, [](const out_t& o){ return o.NAME;}, 0)

Typedef Documentation

◆ consumer_t

Definition at line 25 of file test-hdf5-writer.cxx.

Function Documentation

◆ fill()

void fill ( H5::Group &  out_file,
size_t  iterations 
)

Definition at line 95 of file test-hdf5-writer.cxx.

95  {
96 
97  const int deflate = 7;
98 
99  // scalar output
100  using scalar_writer_t = H5Utils::Writer<0, consumer_t::input_type>;
101  scalar_writer_t::configuration_type scalar_config;
102  scalar_config.name = "scalar";
103  scalar_config.deflate = deflate;
104  consumer_t consumers = getConsumers();
105  scalar_writer_t scalar(out_file, consumers, scalar_config);
106  for (size_t n = 0; n < iterations; n++) {
107  scalar.fill(getOutputs(1 + n, 1, 0.5).at(0));
108  }
109 
110  // 1d output
112  d1_t::configuration_type d1_config;
113  d1_config.name = "1d";
114  d1_config.extent = {10};
115  d1_config.chunks = {5};
116  d1_config.deflate = deflate;
117  d1_t d1(out_file, consumers, d1_config);
118  for (size_t n = 0; n < iterations; n++) {
119  d1.fill(getOutputs(n, 10, 0.5));
120  }
121 
122  // 4d output
124  d4_t::configuration_type d4_config;
125  d4_config.name = "4d";
126  d4_config.extent = {2,3,4,5};
127  d4_config.chunks = {1,2,1,2};
128  d4_config.deflate = deflate;
129  d4_t d4(out_file, consumers, d4_config);
130  for (size_t n = 0; n < iterations; n++) {
131  auto vals = nestOutputs<4>(n, 2);
132  // store some specific value to make sure we have the indexing
133  // right
134  vals.at(1).at(2).at(3).at(4).stype = 86;
135  d4.fill(std::move(vals));
136  }
137 }

◆ getConsumers()

consumer_t getConsumers ( )

Definition at line 27 of file test-hdf5-writer.cxx.

27  {
28  consumer_t consumers;
29 #define ADD(NAME) consumers.add(#NAME, [](const out_t& o){ return o.NAME;}, 0)
30  ADD(ctype);
31  ADD(stype);
32  ADD(itype);
33  ADD(ltype);
34  ADD(lltype);
35  ADD(uctype);
36  ADD(ustype);
37  ADD(uitype);
38  ADD(ultype);
39  ADD(ulltype);
40  ADD(ftype);
41  ADD(dtype);
42  ADD(btype);
43 #undef ADD
45  consumers.add("half" , [](const out_t& o) { return o.ftype; }, 0, half);
46  consumers.add("dhalf", [](const out_t& o) { return o.dtype; }, 0, half);
47  return consumers;
48 }

◆ getOutputs()

std::vector<out_t> getOutputs ( int  offset,
size_t  length,
float  factor 
)

Definition at line 53 of file test-hdf5-writer.cxx.

53  {
54  std::vector<out_t> outvec;
55  for (size_t n = 0; n < length; n++) {
56  out_t out;
57  long long int shifted = n + offset;
58  double factored = shifted*factor;
59  out.dtype = factored;
60  out.ftype = factored;
61  out.ctype = shifted;
62  out.stype = shifted;
63  out.itype = shifted;
64  out.ltype = shifted;
65  out.lltype = shifted;
66  out.uctype = shifted;
67  out.ustype = shifted;
68  out.uitype = shifted;
69  out.ultype = shifted;
70  out.ulltype = shifted;
71  out.btype = n % 2;
72  outvec.push_back(out);
73  }
74  return outvec;
75 }

◆ main()

int main ( int  nargs,
char *  argv[] 
)

Definition at line 139 of file test-hdf5-writer.cxx.

139  {
140  H5::H5File out_file("output.h5", H5F_ACC_TRUNC);
141  size_t iterations = 1;
142  if (nargs > 2) {
143  return 1;
144  }
145  if (nargs > 1) iterations = std::atoi(argv[1]);
146  fill(out_file, iterations);
147  return 0;
148 }

◆ nestOutputs()

template<size_t N>
auto nestOutputs ( int  offset,
size_t  length 
)

Definition at line 78 of file test-hdf5-writer.cxx.

78  {
79  using ret_t = decltype(
80  nestOutputs<N-1>(std::declval<int>(),std::declval<size_t>()));
81  std::vector<ret_t> ret;
82  for (size_t n = 0; n < length; n++) {
83  ret.push_back(nestOutputs<N-1>(n + offset, length + 1));
84  }
85  return ret;
86 }

◆ nestOutputs< 1 >()

template<>
auto nestOutputs< 1 > ( int  offset,
size_t  length 
)

Definition at line 88 of file test-hdf5-writer.cxx.

88  {
89  return getOutputs(offset, length, 0.5);
90 }
H5Utils::Consumers::add
void add(const std::string &name, const std::function< T(I)> &, const T &default_value=T(), Compression=Compression::STANDARD)
This should be the only method you need in this class.
Definition: Writer.h:164
getOutputs
std::vector< out_t > getOutputs(int offset, size_t length, float factor)
Definition: test-hdf5-writer.cxx:53
CaloLCW_tf.nargs
nargs
Definition: CaloLCW_tf.py:26
PlotCalibFromCool.dtype
dtype
Definition: PlotCalibFromCool.py:495
python.AthDsoLogger.out
out
Definition: AthDsoLogger.py:71
dq_defect_virtual_defect_validation.d1
d1
Definition: dq_defect_virtual_defect_validation.py:79
getConsumers
consumer_t getConsumers()
Definition: test-hdf5-writer.cxx:27
H5Utils::Compression::HALF_PRECISION
@ HALF_PRECISION
H5Utils::Writer
Writer.
Definition: Writer.h:349
out_t::dtype
double dtype
Definition: test-hdf5-writer.cxx:11
LArCellNtuple.argv
argv
Definition: LArCellNtuple.py:152
beamspotman.n
n
Definition: beamspotman.py:731
out_t
Definition: test-hdf5-writer.cxx:10
fill
void fill(H5::Group &out_file, size_t iterations)
Definition: test-hdf5-writer.cxx:95
H5Utils::Consumers
Definition: Writer.h:129
convertTimingResiduals.offset
offset
Definition: convertTimingResiduals.py:71
out_t::ftype
float ftype
Definition: test-hdf5-writer.cxx:12
CxxUtils::atoi
int atoi(std::string_view str)
Helper functions to unpack numbers decoded in string into integers and doubles The strings are requir...
Definition: Control/CxxUtils/Root/StringUtils.cxx:85
length
double length(const pvec &v)
Definition: FPGATrackSimLLPDoubletHoughTransformTool.cxx:26
H5Utils::defaults::deflate
const int deflate
Definition: defaults.h:10
ADD
#define ADD(NAME)
PlotCalibFromCool.vals
vals
Definition: PlotCalibFromCool.py:474